llama-index-llms-openai 0.1.27__py3-none-any.whl → 0.1.29__py3-none-any.whl

llama_index/llms/openai/base.py

@@ -201,6 +201,10 @@ class OpenAI(FunctionCallingLLM):
     api_key: str = Field(default=None, description="The OpenAI API key.")
     api_base: str = Field(description="The base URL for OpenAI API.")
     api_version: str = Field(description="The API version for OpenAI API.")
+    strict: bool = Field(
+        default=False,
+        description="Whether to use strict mode for invoking tools/using schemas.",
+    )
 
     _client: Optional[SyncOpenAI] = PrivateAttr()
     _aclient: Optional[AsyncOpenAI] = PrivateAttr()
@@ -229,6 +233,7 @@ class OpenAI(FunctionCallingLLM):
         completion_to_prompt: Optional[Callable[[str], str]] = None,
         pydantic_program_mode: PydanticProgramMode = PydanticProgramMode.DEFAULT,
         output_parser: Optional[BaseOutputParser] = None,
+        strict: bool = False,
         **kwargs: Any,
     ) -> None:
         additional_kwargs = additional_kwargs or {}
@@ -257,6 +262,7 @@ class OpenAI(FunctionCallingLLM):
             completion_to_prompt=completion_to_prompt,
             pydantic_program_mode=pydantic_program_mode,
             output_parser=output_parser,
+            strict=strict,
             **kwargs,
         )
 
@@ -832,6 +838,7 @@ class OpenAI(FunctionCallingLLM):
         verbose: bool = False,
         allow_parallel_tool_calls: bool = False,
         tool_choice: Union[str, dict] = "auto",
+        strict: Optional[bool] = None,
         **kwargs: Any,
     ) -> Dict[str, Any]:
         """Predict and call the tool."""
@@ -840,6 +847,20 @@ class OpenAI(FunctionCallingLLM):
         # misralai uses the same openai tool format
         tool_specs = [tool.metadata.to_openai_tool() for tool in tools]
 
+        # if strict is passed in, use, else default to the class-level attribute, else default to True`
+        if strict is not None:
+            strict = strict
+        else:
+            strict = self.strict
+
+        if self.metadata.is_function_calling_model:
+            for tool_spec in tool_specs:
+                if tool_spec["type"] == "function":
+                    tool_spec["function"]["strict"] = strict
+                    tool_spec["function"]["parameters"][
+                        "additionalProperties"
+                    ] = False  # in current openai 1.40.0 it is always false.
+
         if isinstance(user_msg, str):
             user_msg = ChatMessage(role=MessageRole.USER, content=user_msg)
 

llama_index/llms/openai/utils.py

@@ -47,6 +47,7 @@ GPT4_MODELS: Dict[str, int] = {
     "gpt-4-turbo": 128000,
     "gpt-4o": 128000,
     "gpt-4o-2024-05-13": 128000,
+    "gpt-4o-2024-08-06": 128000,
     "gpt-4o-mini": 128000,
     "gpt-4o-mini-2024-07-18": 128000,
     # 0613 models (function calling):

{llama_index_llms_openai-0.1.27.dist-info → llama_index_llms_openai-0.1.29.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llama-index-llms-openai
-Version: 0.1.27
+Version: 0.1.29
 Summary: llama-index llms openai integration
 License: MIT
 Author: llama-index
@@ -12,6 +12,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: llama-index-core (>=0.10.57,<0.11.0)
+Requires-Dist: openai (>=1.40.0,<2.0.0)
 Description-Content-Type: text/markdown
 
 # LlamaIndex Llms Integration: Openai

llama_index_llms_openai-0.1.29.dist-info/RECORD

@@ -0,0 +1,6 @@
+llama_index/llms/openai/__init__.py,sha256=vm3cIBSGkBFlE77GyfyN0EhpJcnJZN95QMhPN53EkbE,148
+llama_index/llms/openai/base.py,sha256=XX74bUlw8I20lqlMjGvlga6y565c2RUXFM5b-UQDZtQ,33338
+llama_index/llms/openai/utils.py,sha256=VuDXkLR_BGVqoZc9IJqiJlVloZwG9Z7s1nGPAhlbvWE,13079
+llama_index_llms_openai-0.1.29.dist-info/METADATA,sha256=cmrnm6oQPK1sxJwDdDa3-elnKbosS4VzRdxPKNkBulM,650
+llama_index_llms_openai-0.1.29.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+llama_index_llms_openai-0.1.29.dist-info/RECORD,,

llama_index_llms_openai-0.1.27.dist-info/RECORD

@@ -1,6 +0,0 @@
-llama_index/llms/openai/__init__.py,sha256=vm3cIBSGkBFlE77GyfyN0EhpJcnJZN95QMhPN53EkbE,148
-llama_index/llms/openai/base.py,sha256=qh7RdyF2rjVnUEjSzejdpa-jbxQoqTeWaLHyaeK-Wo8,32503
-llama_index/llms/openai/utils.py,sha256=djxNH9V6ye93Mn9yI57pQ8ULHET2QqlVmpFsAcxJfdo,13046
-llama_index_llms_openai-0.1.27.dist-info/METADATA,sha256=BfC8R1QUIcNK94AzMrgnp271Pr5WzdAeAccKqrCPWwA,610
-llama_index_llms_openai-0.1.27.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-llama_index_llms_openai-0.1.27.dist-info/RECORD,,