diff --git a/lib/crewai-tools/src/crewai_tools/tools/ai_mind_tool/ai_mind_tool.py b/lib/crewai-tools/src/crewai_tools/tools/ai_mind_tool/ai_mind_tool.py index 4d48c3e06f..a102c464f1 100644 --- a/lib/crewai-tools/src/crewai_tools/tools/ai_mind_tool/ai_mind_tool.py +++ b/lib/crewai-tools/src/crewai_tools/tools/ai_mind_tool/ai_mind_tool.py @@ -52,7 +52,6 @@ def __init__(self, api_key: str | None = None, **kwargs): try: from minds.client import Client # type: ignore - from minds.datasources import DatabaseConfig # type: ignore except ImportError as e: raise ImportError( "`minds_sdk` package not found, please run `pip install minds-sdk`" @@ -63,14 +62,18 @@ def __init__(self, api_key: str | None = None, **kwargs): # Convert the datasources to DatabaseConfig objects. datasources = [] for datasource in self.datasources: - config = DatabaseConfig( - name=f"{AIMindToolConstants.DATASOURCE_NAME_PREFIX}_{secrets.token_hex(5)}", + db_name = f"{AIMindToolConstants.DATASOURCE_NAME_PREFIX}_{secrets.token_hex(5)}" + + minds_client.datasources.create( + name=db_name, engine=datasource["engine"], description=datasource["description"], - connection_data=datasource["connection_data"], - tables=datasource["tables"], + connection_data=datasource["connection_data"] ) - datasources.append(config) + datasources.append({ + 'name': db_name, + 'tables': datasource["tables"] + }) # Generate a random name for the Mind. name = f"{AIMindToolConstants.MIND_NAME_PREFIX}_{secrets.token_hex(5)}"