Jayashree Sridhar commited on
Commit
6d99a60
·
1 Parent(s): 05b62e8

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.tool import BaseTool
 
5
 
6
 
7
  class SearchKnowledgeTool(BaseTool):
 
1
  # .base_tool import BaseTool
2
  from utils.knowledge_base import KnowledgeBase
3
  #from pydantic import BaseModel, PrivateAttr
4
+ from crewai_tools import BaseTool
5
+
6
 
7
 
8
  class SearchKnowledgeTool(BaseTool):
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.tool import BaseTool
 
5
 
6
 
7
  class MistralChatTool(BaseTool):
 
1
  #from .base_tool import BaseTool
2
  from models.tinygpt2_model import TinyGPT2Model
3
  #from pydantic import BaseModel, PrivateAttr
4
+ from crewai_tools import BaseTool
5
+
6
 
7
 
8
  class MistralChatTool(BaseTool):
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.tool import BaseTool
 
13
 
14
 
15
  # @dataclass
 
9
  from transformers import pipeline
10
  import torch
11
  #from pydantic import BaseModel, PrivateAttr
12
+ from crewai_tools import BaseTool
13
+
14
 
15
 
16
  # @dataclass
agents/tools/voice_tools.py CHANGED
@@ -8,7 +8,8 @@ import tempfile
8
  import soundfile as sf
9
  import torch
10
  #from pydantic import BaseModel, PrivateAttr
11
- from crewai.tools.tool import BaseTool
 
12
 
13
 
14
  class MultilingualVoiceProcessor:
 
8
  import soundfile as sf
9
  import torch
10
  #from pydantic import BaseModel, PrivateAttr
11
+ from crewai_tools import BaseTool
12
+
13
 
14
 
15
  class MultilingualVoiceProcessor: