From 69e054947d587ff2548b101ece690d21d3c38f74 Mon Sep 17 00:00:00 2001
From: Stainless Bot <107565488+stainless-bot@users.noreply.github.com>
Date: Fri, 24 Nov 2023 11:42:36 -0500
Subject: [PATCH] chore(internal): send more detailed x-stainless headers
 (#877)

---
 pyproject.toml                | 1 +
 src/openai/_client.py         | 4 +++-
 src/openai/_utils/__init__.py | 1 +
 src/openai/_utils/_utils.py   | 9 +++++++++
 4 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index f17def16b6..d2d85a5018 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -13,6 +13,7 @@ dependencies = [
     "typing-extensions>=4.5, <5",
     "anyio>=3.5.0, <4",
     "distro>=1.7.0, <2",
+    "sniffio",
     "tqdm > 4"
 ]
 requires-python = ">= 3.7.1"
diff --git a/src/openai/_client.py b/src/openai/_client.py
index aa00073281..202162070b 100644
--- a/src/openai/_client.py
+++ b/src/openai/_client.py
@@ -20,7 +20,7 @@
     ProxiesTypes,
     RequestOptions,
 )
-from ._utils import is_given, is_mapping
+from ._utils import is_given, is_mapping, get_async_library
 from ._version import __version__
 from ._streaming import Stream as Stream
 from ._streaming import AsyncStream as AsyncStream
@@ -147,6 +147,7 @@ def auth_headers(self) -> dict[str, str]:
     def default_headers(self) -> dict[str, str | Omit]:
         return {
             **super().default_headers,
+            "X-Stainless-Async": "false",
             "OpenAI-Organization": self.organization if self.organization is not None else Omit(),
             **self._custom_headers,
         }
@@ -356,6 +357,7 @@ def auth_headers(self) -> dict[str, str]:
     def default_headers(self) -> dict[str, str | Omit]:
         return {
             **super().default_headers,
+            "X-Stainless-Async": f"async:{get_async_library()}",
             "OpenAI-Organization": self.organization if self.organization is not None else Omit(),
             **self._custom_headers,
         }
diff --git a/src/openai/_utils/__init__.py b/src/openai/_utils/__init__.py
index d3397212de..400ca9b828 100644
--- a/src/openai/_utils/__init__.py
+++ b/src/openai/_utils/__init__.py
@@ -25,6 +25,7 @@
 from ._utils import deepcopy_minimal as deepcopy_minimal
 from ._utils import extract_type_arg as extract_type_arg
 from ._utils import is_required_type as is_required_type
+from ._utils import get_async_library as get_async_library
 from ._utils import is_annotated_type as is_annotated_type
 from ._utils import maybe_coerce_float as maybe_coerce_float
 from ._utils import get_required_header as get_required_header
diff --git a/src/openai/_utils/_utils.py b/src/openai/_utils/_utils.py
index 4b51dcb2e8..d2bfc91a70 100644
--- a/src/openai/_utils/_utils.py
+++ b/src/openai/_utils/_utils.py
@@ -18,6 +18,8 @@
 from pathlib import Path
 from typing_extensions import Required, Annotated, TypeGuard, get_args, get_origin
 
+import sniffio
+
 from .._types import Headers, NotGiven, FileTypes, NotGivenOr, HeadersLike
 from .._compat import is_union as _is_union
 from .._compat import parse_date as parse_date
@@ -406,3 +408,10 @@ def get_required_header(headers: HeadersLike, header: str) -> str:
             return value
 
     raise ValueError(f"Could not find {header} header")
+
+
+def get_async_library() -> str:
+    try:
+        return sniffio.current_async_library()
+    except Exception:
+        return "false"