v2.0.38

phidatahq · Nov 10, 2023 · 04c55b7 · 04c55b7
1 parent eb23703
commit 04c55b7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/phi/llm/openai/chat.py b/phi/llm/openai/chat.py
@@ -1,3 +1,4 @@
+import json
 from typing import Optional, List, Iterator, Dict, Any, Union, Tuple
 
 from phi.llm.base import LLM
@@ -139,7 +140,6 @@ def invoke_model_stream(self, messages: List[Message]) -> Iterator[ChatCompletio
         if get_from_env("OPENAI_API_KEY") is None:
             logger.debug("--o-o-- Using phi-proxy")
             try:
-                import json
                 from phi.api.llm import openai_chat_stream
 
                 for chunk in openai_chat_stream(
@@ -157,7 +157,7 @@ def invoke_model_stream(self, messages: List[Message]) -> Iterator[ChatCompletio
                             chunks = "[" + chunk.replace("}{", "},{") + "]"
                             for completion_chunk in json.loads(chunks):
                                 try:
-                                    yield ChatCompletionChunk.model_validate_json(completion_chunk)
+                                    yield ChatCompletionChunk.model_validate(completion_chunk)
                                 except Exception as e:
                                     logger.warning(e)
                         else:

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "phidata"
-version = "2.0.37"
+version = "2.0.38"
 description = "AI Toolkit for Engineers"
 requires-python = ">=3.7"
 readme = "README.md"