file imports + removing boilerplate

pull/39/head
Kye 1 year ago
parent b42d23cb59
commit dd43b60fc4

@ -2,10 +2,9 @@ from __future__ import annotations
from enum import Enum from enum import Enum
from abc import ABC, abstractmethod 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 pydantic import BaseModel
from swarms.utils.logger import logger
from langchain.llms.base import BaseLLM from langchain.llms.base import BaseLLM
from langchain.agents.agent import AgentExecutor from langchain.agents.agent import AgentExecutor

@ -1,5 +1,6 @@
from swarms.agents.tools.base import BaseToolSet, SessionGetter, ToolScope, tool
from swarms.utils.logger import logger from swarms.utils.logger import logger
from swarms.agents.tools.base import BaseToolSet, tool, SessionGetter, ToolScope
class ExitConversation(BaseToolSet): class ExitConversation(BaseToolSet):
@tool( @tool(

@ -1,5 +1,3 @@
######################## ######################################################## file system
from langchain.agents.agent_toolkits import FileManagementToolkit from langchain.agents.agent_toolkits import FileManagementToolkit
from tempfile import TemporaryDirectory from tempfile import TemporaryDirectory
@ -17,8 +15,3 @@ file_management_tools = FileManagementToolkit(
).get_tools() ).get_tools()
read_tool, write_tool, list_tool = file_management_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({})

@ -17,6 +17,7 @@ from transformers import (
CLIPSegForImageSegmentation, CLIPSegForImageSegmentation,
CLIPSegProcessor, CLIPSegProcessor,
) )
from swarms.agents.prompts.prompts import IMAGE_PROMPT from swarms.agents.prompts.prompts import IMAGE_PROMPT
from swarms.agents.tools.base import tool from swarms.agents.tools.base import tool
from swarms.agents.tools.main import BaseToolSet from swarms.agents.tools.main import BaseToolSet

@ -1,10 +1,9 @@
import requests import requests
from bs4 import BeautifulSoup 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.utils.logger import logger
from swarms.agents.tools.base import tool
class RequestsGet(BaseToolSet): class RequestsGet(BaseToolSet):

Loading…
Cancel
Save