[chore]: tool_choice overiding for with_structured_output

langchain-ai · Jan 21, 2025 · 1e347d3 · 1e347d3
1 parent 06456c1
commit 1e347d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libs/core/langchain_core/language_models/chat_models.py b/libs/core/langchain_core/language_models/chat_models.py
@@ -1239,7 +1239,7 @@ class AnswerWithJustification(BaseModel):
         if self.bind_tools is BaseChatModel.bind_tools:
             msg = "with_structured_output is not implemented for this model."
             raise NotImplementedError(msg)
-        llm = self.bind_tools([schema], tool_choice="any")
+        llm = self.bind_tools([schema], tool_choice=kwargs.get("tool_choice", "any"))
         if isinstance(schema, type) and is_basemodel_subclass(schema):
             output_parser: OutputParserLike = PydanticToolsParser(
                 tools=[cast(TypeBaseModel, schema)], first_tool_only=True