github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-10 23:41:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

core[patch]: support bedrock converse -> openai tool (#27754)

Browse Source
This commit is contained in:
Bagatur
2024-10-30 12:20:39 -07:00
committed by GitHub
parent 3dfdb3e6fb
commit 94ea950c6c
2 changed files with 39 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
libs/core/langchain_core/utils/function_calling.py
View File

@@ -341,8 +341,8 @@ def convert_to_openai_function(
A dictionary, Pydantic BaseModel class, TypedDict class, a LangChain
Tool object, or a Python function. If a dictionary is passed in, it is
assumed to already be a valid OpenAI function, a JSON schema with
top-level 'title' and 'description' keys specified, or an Anthropic format
tool.
top-level 'title' and 'description' keys specified, an Anthropic format
tool, or an Amazon Bedrock Converse format tool.
strict:
If True, model output is guaranteed to exactly match the JSON Schema
provided in the function definition. If None, ``strict`` argument will not
@@ -362,6 +362,10 @@ def convert_to_openai_function(
.. versionchanged:: 0.3.13
Support for Anthropic format tools added.
.. versionchanged:: 0.3.14
Support for Amazon Bedrock Converse format tools added.
"""
from langchain_core.tools import BaseTool
@@ -389,6 +393,13 @@ def convert_to_openai_function(
"description": function["description"],
"parameters": function["input_schema"],
}
# an Amazon Bedrock Converse format tool
elif isinstance(function, dict) and "toolSpec" in function:
oai_function = {
"name": function["toolSpec"]["name"],
"description": function["toolSpec"]["description"],
"parameters": function["toolSpec"]["inputSchema"]["json"],
}
elif isinstance(function, type) and is_basemodel_subclass(function):
oai_function = cast(dict, convert_pydantic_to_openai_function(function))
elif is_typeddict(function):