andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 00:28:31 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fixed a few bugs, type hints and imports

Browse Source
This commit is contained in:
Minura Punchihewa
2025-01-03 01:17:11 +05:30
parent 555638a654
commit faff58ba1c
1 changed files with 13 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/crewai_tools/tools/ai_minds_tool/ai_minds_tool.py
View File

@@ -1,5 +1,5 @@
import secrets import secrets
from typing import Dict, Optional, Text, Type from typing import Any, Dict, List, Optional, Text, Type
from crewai.tools import BaseTool from crewai.tools import BaseTool
from openai import OpenAI from openai import OpenAI
@@ -23,13 +23,13 @@ class AIMindTool(BaseTool):
) )
args_schema: Type[BaseModel] = AIMindInputSchema args_schema: Type[BaseModel] = AIMindInputSchema
api_key: Optional[str] = None api_key: Optional[str] = None
datasources: Optional[Dict] = None datasources: Optional[List[Dict[str, Any]]] = None
mind_name: Optional[Text] = None mind_name: Optional[Text] = None
def __init__(self, api_key: Optional[Text] = None, **kwargs): def __init__(self, api_key: Optional[Text] = None, **kwargs):
super().__init__(**kwargs) super().__init__(api_key=api_key, **kwargs)
try: try:
from minds_sdk import Client # type: ignore from minds.client import Client # type: ignore
from minds.datasources import DatabaseConfig # type: ignore from minds.datasources import DatabaseConfig # type: ignore
except ImportError: except ImportError:
raise ImportError( raise ImportError(
@@ -41,15 +41,14 @@ class AIMindTool(BaseTool):
# Convert the datasources to DatabaseConfig objects. # Convert the datasources to DatabaseConfig objects.
datasources = [] datasources = []
for datasource in self.datasources: for datasource in self.datasources:
if datasource["type"] == "database": config = DatabaseConfig(
config = DatabaseConfig( name=f"cai_ds_{secrets.token_hex(5)}",
name=datasource["name"], engine=datasource["engine"],
engine=datasource["engine"], description=datasource["description"],
description=datasource["description"], connection_data=datasource["connection_data"],
connection_data=datasource["connection_data"], tables=datasource["tables"],
tables=datasource["tables"], )
) datasources.append(config)
datasources.append(config)
# Generate a random name for the Mind. # Generate a random name for the Mind.
name = f"cai_mind_{secrets.token_hex(5)}" name = f"cai_mind_{secrets.token_hex(5)}"
@@ -68,7 +67,7 @@ class AIMindTool(BaseTool):
# The Minds API is OpenAI compatible and therefore, the OpenAI client can be used. # The Minds API is OpenAI compatible and therefore, the OpenAI client can be used.
openai_client = OpenAI(base_url="https://mdb.ai/", api_key=self.api_key) openai_client = OpenAI(base_url="https://mdb.ai/", api_key=self.api_key)
completion = openai_client.create( completion = openai_client.chat.completions.create(
model=self.mind_name, model=self.mind_name,
messages=[{"role": "user", "content": query}], messages=[{"role": "user", "content": query}],
stream=False, stream=False,