llama-index-llms-openai 0.3.8__tar.gz → 0.3.9__tar.gz

{llama_index_llms_openai-0.3.8 → llama_index_llms_openai-0.3.9}/PKG-INFO

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

{llama_index_llms_openai-0.3.8 → llama_index_llms_openai-0.3.9}/llama_index/llms/openai/base.py

@@ -59,7 +59,6 @@ from llama_index.core.llms.llm import ToolSelection
 from llama_index.core.llms.utils import parse_partial_json
 from llama_index.core.types import BaseOutputParser, Model, PydanticProgramMode
 from llama_index.llms.openai.utils import (
-    ALL_AVAILABLE_MODELS,
     O1_MODELS,
     OpenAIToolCall,
     create_retry_decorator,
@@ -218,10 +217,6 @@ class OpenAI(FunctionCallingLLM):
         default=False,
         description="Whether to use strict mode for invoking tools/using schemas.",
     )
-    supports_content_blocks: bool = Field(
-        default=False,
-        description="Whether the model supports content blocks in chat messages.",
-    )
 
     _client: Optional[SyncOpenAI] = PrivateAttr()
     _aclient: Optional[AsyncOpenAI] = PrivateAttr()
@@ -267,10 +262,6 @@ class OpenAI(FunctionCallingLLM):
         if model in O1_MODELS:
             temperature = 1.0
 
-        supports_content_blocks = kwargs.pop(
-            "supports_content_blocks", model in ALL_AVAILABLE_MODELS
-        )
-
         super().__init__(
             model=model,
             temperature=temperature,
@@ -290,7 +281,6 @@ class OpenAI(FunctionCallingLLM):
             pydantic_program_mode=pydantic_program_mode,
             output_parser=output_parser,
             strict=strict,
-            supports_content_blocks=supports_content_blocks,
             **kwargs,
         )
 
@@ -436,7 +426,6 @@ class OpenAI(FunctionCallingLLM):
         message_dicts = to_openai_message_dicts(
             messages,
             model=self.model,
-            supports_content_blocks=self.supports_content_blocks,
         )
 
         if self.reuse_client:
@@ -475,7 +464,6 @@ class OpenAI(FunctionCallingLLM):
         message_dicts = to_openai_message_dicts(
             messages,
             model=self.model,
-            supports_content_blocks=self.supports_content_blocks,
         )
 
         def gen() -> ChatResponseGen:
@@ -689,7 +677,6 @@ class OpenAI(FunctionCallingLLM):
         message_dicts = to_openai_message_dicts(
             messages,
             model=self.model,
-            supports_content_blocks=self.supports_content_blocks,
         )
 
         if self.reuse_client:
@@ -726,7 +713,6 @@ class OpenAI(FunctionCallingLLM):
         message_dicts = to_openai_message_dicts(
             messages,
             model=self.model,
-            supports_content_blocks=self.supports_content_blocks,
         )
 
         async def gen() -> ChatResponseAsyncGen:

{llama_index_llms_openai-0.3.8 → llama_index_llms_openai-0.3.9}/llama_index/llms/openai/utils.py

@@ -256,22 +256,14 @@ def to_openai_message_dict(
     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 = []
     content_txt = ""
     for block in message.blocks:
         if isinstance(block, TextBlock):
-            if (
-                message.role.value in ("assistant", "tool", "system")
-                or not supports_content_blocks
-            ):
-                # Despite the docs say otherwise, when role is ASSISTANT, SYSTEM
-                # or TOOL, 'content' cannot be a list and must be string instead.
-                content_txt += block.text
-            else:
-                content.append({"type": "text", "text": block.text})
+            content.append({"type": "text", "text": block.text})
+            content_txt += block.text
         elif isinstance(block, ImageBlock):
             if block.url:
                 content.append(
@@ -293,11 +285,15 @@ def to_openai_message_dict(
             msg = f"Unsupported content block type: {type(block).__name__}"
             raise ValueError(msg)
 
+    # NOTE: Despite what the openai docs say, if the role is ASSISTANT, SYSTEM
+    # or TOOL, 'content' cannot be a list and must be string instead.
+    # Furthermore, if all blocks are text blocks, we can use the content_txt
+    # as the content. This will avoid breaking openai-like APIs.
     message_dict = {
         "role": message.role.value,
         "content": content_txt
         if message.role.value in ("assistant", "tool", "system")
-        or not supports_content_blocks
+        or all(isinstance(block, TextBlock) for block in message.blocks)
         else content,
     }
 
@@ -324,7 +320,6 @@ 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 [
@@ -332,7 +327,6 @@ def to_openai_message_dicts(
             message,
             drop_none=drop_none,
             model=model,
-            supports_content_blocks=supports_content_blocks,
         )
         for message in messages
     ]

{llama_index_llms_openai-0.3.8 → llama_index_llms_openai-0.3.9}/pyproject.toml

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