From 66517b53fafe4f447c292612c7a08f671751c7aa Mon Sep 17 00:00:00 2001 From: Kye Date: Thu, 3 Aug 2023 11:17:38 -0400 Subject: [PATCH] file imports + removing boilerplate Former-commit-id: b3c5795fbec4f62fadc3c2ae99ba7330a4325bad --- swarms/agents/tools/base.py | 3 +-- swarms/agents/tools/exit_conversation.py | 3 ++- swarms/agents/tools/file_mangagement.py | 7 ------- swarms/agents/tools/models.py | 1 + swarms/agents/tools/requests.py | 5 ++--- 5 files changed, 6 insertions(+), 13 deletions(-) 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):