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

modeified basetool importing library

Browse files
agents/tools/knowledge_tools.py CHANGED
@@ -1,7 +1,7 @@
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):
 
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):
agents/tools/llm_tools.py CHANGED
@@ -1,7 +1,7 @@
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):
 
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):
agents/tools/validation_tools.py CHANGED
@@ -9,7 +9,7 @@ import json
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
 
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
agents/tools/voice_tools.py CHANGED
@@ -8,7 +8,7 @@ import tempfile
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:
 
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: