adjust unwrap + add pydantic to test reqs

alizenhom · alizenhom · commit 1dacf8dbfaaa · 2024-10-21T11:16:48.000+03:00
diff --git a/instrumentation/opentelemetry-instrumentation-openai-v2/src/opentelemetry/instrumentation/openai_v2/__init__.py b/instrumentation/opentelemetry-instrumentation-openai-v2/src/opentelemetry/instrumentation/openai_v2/__init__.py
@@ -75,4 +75,6 @@ def _instrument(self, **kwargs):
         )
 
     def _uninstrument(self, **kwargs):
-        unwrap("openai.resources.chat.completions.Completions", "create")
+        import openai
+
+        unwrap(openai, "resources.chat.completions.Completions.create")
diff --git a/instrumentation/opentelemetry-instrumentation-openai-v2/test-requirements.txt b/instrumentation/opentelemetry-instrumentation-openai-v2/test-requirements.txt
@@ -1,4 +1,5 @@
 openai==1.0.0
+pydantic==2.8.2
 Deprecated==1.2.14
 importlib-metadata==6.11.0
 packaging==24.0

Original file line number	Diff line number	Diff line change
`@@ -75,4 +75,6 @@ def _instrument(self, **kwargs):`
`75`	`75`	`)`
`76`	`76`
`77`	`77`	`def _uninstrument(self, **kwargs):`
`78`		`- unwrap("openai.resources.chat.completions.Completions", "create")`
	`78`	`+ import openai`
	`79`	`+`
	`80`	`+ unwrap(openai, "resources.chat.completions.Completions.create")`