github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-21 14:18:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

langchain_community.chat_models.oci_generative_ai: Fix a bug when using optional parameters in tools (#28829)

Browse Source 
When using tools with optional parameters, the parameter `type` is not
longer available since langchain update to 0.3 (because of the pydantic
upgrade?) and there is now an `anyOf` field instead. This results in the
`type` being `None` in the chat request for the tool parameter, and the
LLM call fails with the error:

```
oci.exceptions.ServiceError: {'target_service': 'generative_ai_inference', 
'status': 400, 'code': '400', 
'opc-request-id': '...', 
'message': 'Parameter definition must have a type.', 
'operation_name': 'chat'
...
}
```

Example code that fails:

```
from langchain_community.chat_models.oci_generative_ai import ChatOCIGenAI
from langchain_core.tools import tool
from typing import Optional

llm = ChatOCIGenAI(
        model_id="cohere.command-r-plus",
        service_endpoint="https://inference.generativeai.us-chicago-1.oci.oraclecloud.com",
        compartment_id="ocid1.compartment.oc1...",
        auth_profile="your_profile",
        auth_type="API_KEY",
        model_kwargs={"temperature": 0, "max_tokens": 3000},
)

@tool
def test(example: Optional[str] = None):
    """This is the tool to use to test things

    Args:
        example: example variable, defaults to None
    """
    return "this is a test"

llm_with_tools = llm.bind_tools([test])

result = llm_with_tools.invoke("can you make a test for g")
```

This PR sets the param type to `any` in that case, and fixes the
problem.

Co-authored-by: Erick Friis <erick@langchain.dev>
This commit is contained in:
Emmanuel Leroy 2024-12-19 14:17:34 -08:00 committed by GitHub
parent c3ccd93c12
commit c8db5a19ce
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
libs/community/langchain_community/chat_models/oci_generative_ai.py
View File

@ -61,6 +61,7 @@ JSON_TO_PYTHON_TYPES = {
"integer": "int", "integer": "int",
"array": "List", "array": "List",
"object": "Dict", "object": "Dict",
"any": "any",
} }
@ -323,7 +324,7 @@ class CohereProvider(Provider):
if "description" in p_def if "description" in p_def
else "", else "",
type=JSON_TO_PYTHON_TYPES.get( type=JSON_TO_PYTHON_TYPES.get(
p_def.get("type"), p_def.get("type") p_def.get("type"), p_def.get("type", "any")
), ),
is_required="default" not in p_def, is_required="default" not in p_def,
) )
@ -342,7 +343,7 @@ class CohereProvider(Provider):
p_name: self.oci_tool_param( p_name: self.oci_tool_param(
description=p_def.get("description"), description=p_def.get("description"),
type=JSON_TO_PYTHON_TYPES.get( type=JSON_TO_PYTHON_TYPES.get(
p_def.get("type"), p_def.get("type") p_def.get("type"), p_def.get("type", "any")
), ),
is_required="default" not in p_def, is_required="default" not in p_def,
) )
@ -363,7 +364,7 @@ class CohereProvider(Provider):
p_name: self.oci_tool_param( p_name: self.oci_tool_param(
description=p_def.get("description"), description=p_def.get("description"),
type=JSON_TO_PYTHON_TYPES.get( type=JSON_TO_PYTHON_TYPES.get(
p_def.get("type"), p_def.get("type") p_def.get("type"), p_def.get("type", "any")
), ),
is_required=p_name in parameters.get("required", []), is_required=p_name in parameters.get("required", []),
) )