From 04c0d589d861fb4a7785d78c6fcbd5d681310b2d Mon Sep 17 00:00:00 2001 From: Will Xu Date: Fri, 12 Dec 2025 11:11:37 +0800 Subject: [PATCH 1/2] adjust schema type --- .../langchain/src/databricks_langchain/chat_models.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/integrations/langchain/src/databricks_langchain/chat_models.py b/integrations/langchain/src/databricks_langchain/chat_models.py index a4fa1621..28e078f4 100644 --- a/integrations/langchain/src/databricks_langchain/chat_models.py +++ b/integrations/langchain/src/databricks_langchain/chat_models.py @@ -976,9 +976,7 @@ class AnswerWithJustification(BaseModel): "type": "json_schema", "json_schema": { "strict": True, - "schema": ( - schema.model_json_schema() if is_pydantic_schema else schema # type: ignore[union-attr] - ), + "schema": schema.model_json_schema() if is_pydantic_schema else schema, # type: ignore[union-attr] }, } llm = self.bind(response_format=response_format) From 783bbbea1f495d2813a20f14697ed2cf0b6dfcf1 Mon Sep 17 00:00:00 2001 From: Will Xu Date: Fri, 12 Dec 2025 11:11:58 +0800 Subject: [PATCH 2/2] add name for response_format --- integrations/langchain/src/databricks_langchain/chat_models.py | 1 + 1 file changed, 1 insertion(+) diff --git a/integrations/langchain/src/databricks_langchain/chat_models.py b/integrations/langchain/src/databricks_langchain/chat_models.py index 28e078f4..f063b069 100644 --- a/integrations/langchain/src/databricks_langchain/chat_models.py +++ b/integrations/langchain/src/databricks_langchain/chat_models.py @@ -975,6 +975,7 @@ class AnswerWithJustification(BaseModel): response_format = { "type": "json_schema", "json_schema": { + "name": kwargs.get("schema_name", "json_schema"), "strict": True, "schema": schema.model_json_schema() if is_pydantic_schema else schema, # type: ignore[union-attr] },