Jayashree Sridhar commited on
Commit
b73fa18
·
1 Parent(s): 81484f0

modeified basetool importing library

Browse files
agents/tools/knowledge_tools.py CHANGED
@@ -1,7 +1,8 @@
1
  # .base_tool import BaseTool
2
  from utils.knowledge_base import KnowledgeBase
3
  #from pydantic import BaseModel, PrivateAttr
4
- from crewai_tools.tools import BaseTool
 
5
 
6
  class SearchKnowledgeTool(BaseTool):
7
  name: str = "search_knowledge"
 
1
  # .base_tool import BaseTool
2
  from utils.knowledge_base import KnowledgeBase
3
  #from pydantic import BaseModel, PrivateAttr
4
+ from crewai_tools.tools.base import BaseTool
5
+
6
 
7
  class SearchKnowledgeTool(BaseTool):
8
  name: str = "search_knowledge"
agents/tools/llm_tools.py CHANGED
@@ -1,7 +1,8 @@
1
  #from .base_tool import BaseTool
2
  from models.tinygpt2_model import TinyGPT2Model
3
  #from pydantic import BaseModel, PrivateAttr
4
- from crewai_tools.tools import BaseTool
 
5
 
6
  class MistralChatTool(BaseTool):
7
  name: str = "mistral_chat"
 
1
  #from .base_tool import BaseTool
2
  from models.tinygpt2_model import TinyGPT2Model
3
  #from pydantic import BaseModel, PrivateAttr
4
+ from crewai_tools.tools.base import BaseTool
5
+
6
 
7
  class MistralChatTool(BaseTool):
8
  name: str = "mistral_chat"
agents/tools/validation_tools.py CHANGED
@@ -9,7 +9,8 @@ import json
9
  from transformers import pipeline
10
  import torch
11
  #from pydantic import BaseModel, PrivateAttr
12
- from crewai_tools.tools import BaseTool
 
13
 
14
  # @dataclass
15
  class ValidationResult:
 
9
  from transformers import pipeline
10
  import torch
11
  #from pydantic import BaseModel, PrivateAttr
12
+ from crewai_tools.tools.base import BaseTool
13
+
14
 
15
  # @dataclass
16
  class ValidationResult:
agents/tools/voice_tools.py CHANGED
@@ -8,7 +8,9 @@ import tempfile
8
  import soundfile as sf
9
  import torch
10
  #from pydantic import BaseModel, PrivateAttr
11
- from crewai_tools.tools import BaseTool
 
 
12
  class MultilingualVoiceProcessor:
13
 
14
  def __init__(self, model_name="openai/whisper-base", device=None):
 
8
  import soundfile as sf
9
  import torch
10
  #from pydantic import BaseModel, PrivateAttr
11
+ from crewai_tools.tools.base import BaseTool
12
+
13
+
14
  class MultilingualVoiceProcessor:
15
 
16
  def __init__(self, model_name="openai/whisper-base", device=None):