github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-17 16:39:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

cohere[patch]: support kwargs in with_structured_output (#19736)

Browse Source 
**Description:** We'd like to support passing additional kwargs in
`with_structured_output`. I believe this is the accepted approach to
enable additional arguments on API calls.
This commit is contained in:
harry-cohere 2024-03-29 21:30:14 +00:00 committed by GitHub
parent ce0a588ae6
commit 23fcc14650
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/partners/cohere/langchain_cohere/chat_models.py
View File

@ -191,10 +191,8 @@ class ChatCohere(BaseChatModel, BaseCohere):
A Runnable that takes any ChatModel input and returns either a dict or A Runnable that takes any ChatModel input and returns either a dict or
Pydantic class as output. Pydantic class as output.
""" """
if kwargs:
raise ValueError(f"Received unsupported arguments {kwargs}")
is_pydantic_schema = isinstance(schema, type) and issubclass(schema, BaseModel) is_pydantic_schema = isinstance(schema, type) and issubclass(schema, BaseModel)
llm = self.bind_tools([schema]) llm = self.bind_tools([schema], **kwargs)
if is_pydantic_schema: if is_pydantic_schema:
output_parser: OutputParserLike = PydanticToolsParser( output_parser: OutputParserLike = PydanticToolsParser(
tools=[schema], first_tool_only=True tools=[schema], first_tool_only=True