Update python examples for openai v1

autoblocksai · Dec 1, 2023 · 61c49b0 · 61c49b0
1 parent 9438222
commit 61c49b0
Show file tree

Hide file tree

Showing 6 changed files with 17 additions and 15 deletions.
diff --git a/Python/openai-manual/main.py b/Python/openai-manual/main.py
@@ -5,6 +5,7 @@
 
 import dotenv
 from autoblocks.tracer import AutoblocksTracer
+from autoblocks.vendor.openai import serialize_completion
 from openai import OpenAI
 
 dotenv.load_dotenv(".env")
@@ -41,12 +42,12 @@ def main():
     tracer.send_event("ai.request", span_id=span_id, properties=params)
     try:
         start_time = time.time()
-        openai_response = client.chat.completions.create(**params)
+        completion = client.chat.completions.create(**params)
         tracer.send_event(
             "ai.response",
             span_id=span_id,
             properties=dict(
-                response=openai_response,
+                response=serialize_completion(completion),
                 latency=(time.time() - start_time) * 1000,
             ),
         )

diff --git a/Python/openai-manual/poetry.lock b/Python/openai-manual/poetry.lock
diff --git a/Python/openai-manual/pyproject.toml b/Python/openai-manual/pyproject.toml
@@ -9,5 +9,5 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = "^3.9"
 openai = "^1.0.0"
-autoblocksai = "^0.0.14"
+autoblocksai = "^0.0.17"
 python-dotenv = "^1.0.0"
diff --git a/Python/pytest-replays/main.py b/Python/pytest-replays/main.py
@@ -6,6 +6,7 @@
 
 import dotenv
 from autoblocks.tracer import AutoblocksTracer
+from autoblocks.vendor.openai import serialize_completion
 from openai import OpenAI
 
 dotenv.load_dotenv(".env")
@@ -48,16 +49,16 @@ def run(content: str, trace_id: Optional[str] = None):
 
     try:
         start_time = time.time()
-        response = client.chat.completions.create(**params)
+        completion = client.chat.completions.create(**params)
         tracer.send_event(
             "ai.response",
             span_id=span_id,
             properties=dict(
-                response=response,
+                response=serialize_completion(completion),
                 latency_ms=(time.time() - start_time) * 1000,
             ),
         )
-        return response.choices[0].message
+        return completion.choices[0].message
     except Exception as error:
         tracer.send_event(
             "ai.error",

diff --git a/Python/pytest-replays/poetry.lock b/Python/pytest-replays/poetry.lock
diff --git a/Python/pytest-replays/pyproject.toml b/Python/pytest-replays/pyproject.toml
@@ -9,7 +9,7 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = "^3.9"
 openai = "^1.0.0"
-autoblocksai = "^0.0.14"
+autoblocksai = "^0.0.17"
 python-dotenv = "^1.0.0"
 pytest = "^7.4.3"
 wrapt = "^1.15.0"