Fix #2390: Add databricks-sdk dependency to tools extra

Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-03-17 18:17:38 +00:00
parent e723e5ca3f
commit ad561be4c6
2 changed files with 14 additions and 1 deletions

View File

@@ -45,7 +45,7 @@ Documentation = "https://docs.crewai.com"
Repository = "https://github.com/crewAIInc/crewAI"
[project.optional-dependencies]
tools = ["crewai-tools>=0.37.0"]
tools = ["crewai-tools>=0.37.0", "databricks-sdk>=0.46.0"]
embeddings = [
"tiktoken~=0.7.0"
]

View File

@@ -0,0 +1,13 @@
import pytest
def test_databricks_sdk_import():
"""Test that databricks-sdk can be imported without errors.
This test verifies that the databricks-sdk dependency is properly installed
when using the tools extra, which is required by the databricks_query_tool.
"""
try:
import databricks.sdk
assert True
except ImportError as e:
pytest.fail(f"Failed to import databricks.sdk: {e}")