diff --git a/integrations/langchain/src/databricks_langchain/chat_models.py b/integrations/langchain/src/databricks_langchain/chat_models.py
index a4fa1621..f063b069 100644
--- a/integrations/langchain/src/databricks_langchain/chat_models.py
+++ b/integrations/langchain/src/databricks_langchain/chat_models.py
@@ -975,10 +975,9 @@ 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]
-                    ),
+                    "schema": schema.model_json_schema() if is_pydantic_schema else schema,  # type: ignore[union-attr]
                 },
             }
             llm = self.bind(response_format=response_format)