Merge pull request #1041 from parea-ai/fix-stream-w-old-openai-version

type checking on old openai versions

Author: jalexanderII

parea/wrapper/openai/openai.py

@@ -276,13 +276,15 @@ def _kwargs_to_llm_configuration(kwargs):
 
     @staticmethod
     def _get_output(result: Any, model: Optional[str] = None) -> str:
+        PARSED_CHAT_COMPLETION_AVAILABLE = False
         try:
-            from openai.types.chat import ParsedChatCompletion, ParsedChatCompletionMessage
+            from openai.types.chat import ParsedChatCompletionMessage
+
+            PARSED_CHAT_COMPLETION_AVAILABLE = True
         except ImportError:
-            ParsedChatCompletion = None
             ParsedChatCompletionMessage = None
 
-        if not isinstance(result, (OpenAIObject, ParsedChatCompletion)) and isinstance(result, dict):
+        if isinstance(result, dict):
             result = convert_to_openai_object(
                 {
                     "choices": [
@@ -295,8 +297,8 @@ def _get_output(result: Any, model: Optional[str] = None) -> str:
                 }
             )
         response_message = result.choices[0].message
-        if isinstance(response_message, ParsedChatCompletionMessage):
-            completion = response_message.parsed.model_dump_json() if response_message.parsed else ""
+        if PARSED_CHAT_COMPLETION_AVAILABLE and isinstance(response_message, ParsedChatCompletionMessage):
+            completion = json_dumps(response_message.parsed) if response_message.parsed else ""
         elif not response_message.get("content", None) if is_old_openai else not response_message.content:
             completion = OpenAIWrapper._format_function_call(response_message)
         else:

pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "poetry.core.masonry.api"
 [tool.poetry]
 name = "parea-ai"
 packages = [{ include = "parea" }]
-version = "0.2.195"
+version = "0.2.196"
 description = "Parea python sdk"
 readme = "README.md"
 authors = ["joel-parea-ai <[email protected]>"]