mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-13 01:58:30 +00:00
Fix #2390: Add databricks-sdk dependency to tools extra
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
@@ -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"
|
||||
]
|
||||
|
||||
13
tests/tools/test_databricks_import.py
Normal file
13
tests/tools/test_databricks_import.py
Normal 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}")
|
||||
Reference in New Issue
Block a user