diff --git a/litellm/utils.py b/litellm/utils.py index 0fd21f09919..9b82f5d3214 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -2408,6 +2408,7 @@ def supports_response_schema( litellm.LlmProviders.FIREWORKS_AI, litellm.LlmProviders.LM_STUDIO, litellm.LlmProviders.NEBIUS, + litellm.LlmProviders.DATABRICKS, ] if custom_llm_provider in PROVIDERS_GLOBALLY_SUPPORT_RESPONSE_SCHEMA: diff --git a/tests/litellm_utils_tests/test_utils.py b/tests/litellm_utils_tests/test_utils.py index f585aadbfcb..b4e99ee3fa4 100644 --- a/tests/litellm_utils_tests/test_utils.py +++ b/tests/litellm_utils_tests/test_utils.py @@ -1065,6 +1065,7 @@ def test_parse_content_for_reasoning(content, expected_reasoning, expected_conte ("vertex_ai/gemini-1.5-pro", True), ("gemini/gemini-1.5-pro", True), ("predibase/llama3-8b-instruct", True), + ("databricks/databricks-meta-llama-3-1-70b-instruct", True), ("gpt-3.5-turbo", False), ("groq/llama-3.3-70b-versatile", False), ],