llama-index-llms-openai 0.3.4__tar.gz → 0.3.5__tar.gz

{llama_index_llms_openai-0.3.4 → llama_index_llms_openai-0.3.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llama-index-llms-openai
-Version: 0.3.4
+Version: 0.3.5
 Summary: llama-index llms openai integration
 License: MIT
 Author: llama-index

{llama_index_llms_openai-0.3.4 → llama_index_llms_openai-0.3.5}/llama_index/llms/openai/base.py

@@ -217,6 +217,10 @@ class OpenAI(FunctionCallingLLM):
         default=False,
         description="Whether to use strict mode for invoking tools/using schemas.",
     )
+    supports_content_blocks: bool = Field(
+        default=True,
+        description="Whether the model supports content blocks in chat messages.",
+    )
 
     _client: Optional[SyncOpenAI] = PrivateAttr()
     _aclient: Optional[AsyncOpenAI] = PrivateAttr()
@@ -423,7 +427,11 @@ class OpenAI(FunctionCallingLLM):
     @llm_retry_decorator
     def _chat(self, messages: Sequence[ChatMessage], **kwargs: Any) -> ChatResponse:
         client = self._get_client()
-        message_dicts = to_openai_message_dicts(messages, model=self.model)
+        message_dicts = to_openai_message_dicts(
+            messages,
+            model=self.model,
+            supports_content_blocks=self.supports_content_blocks,
+        )
 
         if self.reuse_client:
             response = client.chat.completions.create(
@@ -458,7 +466,11 @@ class OpenAI(FunctionCallingLLM):
         self, messages: Sequence[ChatMessage], **kwargs: Any
     ) -> ChatResponseGen:
         client = self._get_client()
-        message_dicts = to_openai_message_dicts(messages, model=self.model)
+        message_dicts = to_openai_message_dicts(
+            messages,
+            model=self.model,
+            supports_content_blocks=self.supports_content_blocks,
+        )
 
         def gen() -> ChatResponseGen:
             content = ""
@@ -668,7 +680,11 @@ class OpenAI(FunctionCallingLLM):
         self, messages: Sequence[ChatMessage], **kwargs: Any
     ) -> ChatResponse:
         aclient = self._get_aclient()
-        message_dicts = to_openai_message_dicts(messages, model=self.model)
+        message_dicts = to_openai_message_dicts(
+            messages,
+            model=self.model,
+            supports_content_blocks=self.supports_content_blocks,
+        )
 
         if self.reuse_client:
             response = await aclient.chat.completions.create(
@@ -701,7 +717,11 @@ class OpenAI(FunctionCallingLLM):
         self, messages: Sequence[ChatMessage], **kwargs: Any
     ) -> ChatResponseAsyncGen:
         aclient = self._get_aclient()
-        message_dicts = to_openai_message_dicts(messages, model=self.model)
+        message_dicts = to_openai_message_dicts(
+            messages,
+            model=self.model,
+            supports_content_blocks=self.supports_content_blocks,
+        )
 
         async def gen() -> ChatResponseAsyncGen:
             content = ""

{llama_index_llms_openai-0.3.4 → llama_index_llms_openai-0.3.5}/llama_index/llms/openai/utils.py

@@ -253,7 +253,10 @@ def is_function_calling_model(model: str) -> bool:
 
 
 def to_openai_message_dict(
-    message: ChatMessage, drop_none: bool = False, model: Optional[str] = None
+    message: ChatMessage,
+    drop_none: bool = False,
+    model: Optional[str] = None,
+    supports_content_blocks: bool = False,
 ) -> ChatCompletionMessageParam:
     """Convert a ChatMessage to an OpenAI message dict."""
     content = []
@@ -291,6 +294,7 @@ def to_openai_message_dict(
         "role": message.role.value,
         "content": content_txt
         if message.role.value in ("assistant", "tool", "system")
+        or not supports_content_blocks
         else content,
     }
 
@@ -317,10 +321,16 @@ def to_openai_message_dicts(
     messages: Sequence[ChatMessage],
     drop_none: bool = False,
     model: Optional[str] = None,
+    supports_content_blocks: bool = False,
 ) -> List[ChatCompletionMessageParam]:
     """Convert generic messages to OpenAI message dicts."""
     return [
-        to_openai_message_dict(message, drop_none=drop_none, model=model)
+        to_openai_message_dict(
+            message,
+            drop_none=drop_none,
+            model=model,
+            supports_content_blocks=supports_content_blocks,
+        )
         for message in messages
     ]
 

{llama_index_llms_openai-0.3.4 → llama_index_llms_openai-0.3.5}/pyproject.toml

@@ -29,7 +29,7 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-llms-openai"
 readme = "README.md"
-version = "0.3.4"
+version = "0.3.5"
 
 [tool.poetry.dependencies]
 python = ">=3.9,<4.0"