diff --git a/swarms/agents/tools/base.py b/swarms/agents/tools/base.py index 8a2e05b4..b345431c 100644 --- a/swarms/agents/tools/base.py +++ b/swarms/agents/tools/base.py @@ -2,10 +2,9 @@ from __future__ import annotations from enum import Enum from abc import ABC, abstractmethod -from typing import Any, Callable, Optional, Type, Union, Tuple +from typing import Any, Callable, Optional, Type, Tuple from pydantic import BaseModel -from swarms.utils.logger import logger from langchain.llms.base import BaseLLM from langchain.agents.agent import AgentExecutor diff --git a/swarms/agents/tools/exit_conversation.py b/swarms/agents/tools/exit_conversation.py index d5027a30..dcaeeba4 100644 --- a/swarms/agents/tools/exit_conversation.py +++ b/swarms/agents/tools/exit_conversation.py @@ -1,5 +1,6 @@ +from swarms.agents.tools.base import BaseToolSet, SessionGetter, ToolScope, tool from swarms.utils.logger import logger -from swarms.agents.tools.base import BaseToolSet, tool, SessionGetter, ToolScope + class ExitConversation(BaseToolSet): @tool( diff --git a/swarms/agents/tools/file_mangagement.py b/swarms/agents/tools/file_mangagement.py index 1637a6a8..b9c2041a 100644 --- a/swarms/agents/tools/file_mangagement.py +++ b/swarms/agents/tools/file_mangagement.py @@ -1,5 +1,3 @@ -######################## ######################################################## file system - from langchain.agents.agent_toolkits import FileManagementToolkit from tempfile import TemporaryDirectory @@ -17,8 +15,3 @@ file_management_tools = FileManagementToolkit( ).get_tools() read_tool, write_tool, list_tool = file_management_tools -# write_tool.run({"file_path": "example.txt", "text": "Hello World!"}) - -# # List files in the working directory -# list_tool.run({}) - diff --git a/swarms/agents/tools/models.py b/swarms/agents/tools/models.py index d4eae22f..9ae5abe5 100644 --- a/swarms/agents/tools/models.py +++ b/swarms/agents/tools/models.py @@ -17,6 +17,7 @@ from transformers import ( CLIPSegForImageSegmentation, CLIPSegProcessor, ) + from swarms.agents.prompts.prompts import IMAGE_PROMPT from swarms.agents.tools.base import tool from swarms.agents.tools.main import BaseToolSet diff --git a/swarms/agents/tools/requests.py b/swarms/agents/tools/requests.py index a2f5d8c9..4c76b454 100644 --- a/swarms/agents/tools/requests.py +++ b/swarms/agents/tools/requests.py @@ -1,10 +1,9 @@ import requests - from bs4 import BeautifulSoup -from swarms.agents.tools.base import BaseToolSet + +from swarms.agents.tools.base import BaseToolSet, tool from swarms.utils.logger import logger -from swarms.agents.tools.base import tool class RequestsGet(BaseToolSet):