model_dump_json -> model_dump

autoblocksai · Mar 2, 2024 · 8e7b29f · 8e7b29f
1 parent 6da2a64
commit 8e7b29f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/Python/openai-tracing/src/main.py b/Python/openai-tracing/src/main.py
@@ -46,9 +46,9 @@ def main():
             "ai.response",
             span_id=span_id,
             properties=dict(
-                # OpenAI v1 returns pydantic models, which have a model_dump_json
-                # method for converting to JSON.
-                response=completion.model_dump_json(),
+                # OpenAI v1 returns pydantic models, which have a model_dump
+                # method for converting to a dictionary.
+                response=completion.model_dump(),
                 latency=(time.time() - start_time) * 1000,
             ),
         )

diff --git a/Python/prompt-sdk-headless/src/main.py b/Python/prompt-sdk-headless/src/main.py
@@ -56,9 +56,9 @@ def main():
             "ai.response",
             prompt_tracking=prompt.track(),
             properties=dict(
-                # OpenAI v1 returns pydantic models, which have a model_dump_json
+                # OpenAI v1 returns pydantic models, which have a model_dump
                 # method for converting to JSON.
-                response=response.model_dump_json(),
+                response=response.model_dump(),
             ),
         )