opentelemetry-instrumentation-openai 0.21.1__tar.gz → 0.21.3__tar.gz

{opentelemetry_instrumentation_openai-0.21.1 → opentelemetry_instrumentation_openai-0.21.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: opentelemetry-instrumentation-openai
-Version: 0.21.1
+Version: 0.21.3
 Summary: OpenTelemetry OpenAI instrumentation
 Home-page: https://github.com/traceloop/openllmetry/tree/main/packages/opentelemetry-instrumentation-openai
 License: Apache-2.0
@@ -14,9 +14,9 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Provides-Extra: instruments
-Requires-Dist: opentelemetry-api (>=1.24.0,<2.0.0)
-Requires-Dist: opentelemetry-instrumentation (>=0.45b0,<0.46)
-Requires-Dist: opentelemetry-semantic-conventions (>=0.45b0,<0.46)
+Requires-Dist: opentelemetry-api (>=1.25.0,<2.0.0)
+Requires-Dist: opentelemetry-instrumentation (>=0.46b0,<0.47)
+Requires-Dist: opentelemetry-semantic-conventions (>=0.46b0,<0.47)
 Requires-Dist: opentelemetry-semantic-conventions-ai (==0.2.0)
 Requires-Dist: tiktoken (>=0.6.0,<1)
 Project-URL: Repository, https://github.com/traceloop/openllmetry/tree/main/packages/opentelemetry-instrumentation-openai

{opentelemetry_instrumentation_openai-0.21.1 → opentelemetry_instrumentation_openai-0.21.3}/opentelemetry/instrumentation/openai/shared/chat_wrappers.py

@@ -429,7 +429,9 @@ def _set_streaming_token_metrics(
                 **shared_attributes,
                 "llm.usage.token_type": "completion",
             }
-            token_counter.record(completion_usage, attributes=attributes_with_token_type)
+            token_counter.record(
+                completion_usage, attributes=attributes_with_token_type
+            )
 
 
 class ChatStream(ObjectProxy):
@@ -458,6 +460,8 @@ class ChatStream(ObjectProxy):
     ):
         super().__init__(response)
 
+        print("HEYY", response.__class__.__name__)
+
         self._span = span
         self._instance = instance
         self._token_counter = token_counter
@@ -476,6 +480,15 @@ class ChatStream(ObjectProxy):
     def __enter__(self):
         return self
 
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.__wrapped__.__exit__(exc_type, exc_val, exc_tb)
+
+    async def __aenter__(self):
+        return self
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        await self.__wrapped__.__aexit__(exc_type, exc_val, exc_tb)
+
     def __iter__(self):
         return self
 

opentelemetry_instrumentation_openai-0.21.3/opentelemetry/instrumentation/openai/version.py

@@ -0,0 +1 @@
+__version__ = "0.21.3"

{opentelemetry_instrumentation_openai-0.21.1 → opentelemetry_instrumentation_openai-0.21.3}/pyproject.toml

@@ -8,7 +8,7 @@ show_missing = true
 
 [tool.poetry]
 name = "opentelemetry-instrumentation-openai"
-version = "0.21.1"
+version = "0.21.3"
 description = "OpenTelemetry OpenAI instrumentation"
 authors = [
   "Gal Kleinman <gal@traceloop.com>",
@@ -24,9 +24,9 @@ include = "opentelemetry/instrumentation/openai"
 
 [tool.poetry.dependencies]
 python = ">=3.9,<4"
-opentelemetry-api = "^1.24.0"
-opentelemetry-instrumentation = "^0.45b0"
-opentelemetry-semantic-conventions = "^0.45b0"
+opentelemetry-api = "^1.25.0"
+opentelemetry-instrumentation = "^0.46b0"
+opentelemetry-semantic-conventions = "^0.46b0"
 opentelemetry-semantic-conventions-ai = "0.2.0"
 tiktoken = ">=0.6.0, <1"
 

opentelemetry_instrumentation_openai-0.21.1/opentelemetry/instrumentation/openai/version.py

@@ -1 +0,0 @@
-__version__ = "0.21.1"