diff --git a/libs/partners/groq/langchain_groq/chat_models.py b/libs/partners/groq/langchain_groq/chat_models.py index 9fd02f230b216..b54491ce32e7a 100644 --- a/libs/partners/groq/langchain_groq/chat_models.py +++ b/libs/partners/groq/langchain_groq/chat_models.py @@ -1116,9 +1116,6 @@ class AnswerWithJustification(BaseModel): """ # noqa: E501 _ = kwargs.pop("strict", None) - if kwargs: - msg = f"Received unsupported arguments {kwargs}" - raise ValueError(msg) is_pydantic_schema = _is_pydantic_class(schema) if method == "function_calling": if schema is None: @@ -1136,6 +1133,7 @@ class AnswerWithJustification(BaseModel): "kwargs": {"method": "function_calling"}, "schema": formatted_tool, }, + **kwargs, ) if is_pydantic_schema: output_parser: OutputParserLike = PydanticToolsParser( @@ -1168,6 +1166,7 @@ class AnswerWithJustification(BaseModel): llm = self.bind( response_format=response_format, ls_structured_output_format=ls_format_info, + **kwargs, ) output_parser = ( PydanticOutputParser(pydantic_object=schema) # type: ignore[type-var, arg-type] @@ -1182,6 +1181,7 @@ class AnswerWithJustification(BaseModel): "kwargs": {"method": "json_mode"}, "schema": schema, }, + **kwargs, ) output_parser = ( PydanticOutputParser(pydantic_object=schema) # type: ignore[type-var, arg-type]