new requirements.txt

pull/45/head
Kye 1 year ago
parent 184fe4354f
commit 45c7fc2cc1

@ -1,114 +1,133 @@
transformers # transformers
openai # openai
langchain==0.0.240 # langchain==0.0.240
torch # torch
torchvision # torchvision
asyncio # asyncio
nest_asyncio # nest_asyncio
bs4 # bs4
playwright # playwright
duckduckgo-search # duckduckgo-search
faiss-cpu # faiss-cpu
wget==3.2 # wget==3.2
# codeinterpreterapi # # codeinterpreterapi
accelerate # accelerate
addict # addict
albumentations # albumentations
basicsr # basicsr
controlnet-aux # controlnet-aux
diffusers==0.14.0 # diffusers==0.14.0
langchain-experimental # langchain-experimental
einops # einops
imageio # imageio
imageio-ffmpeg # imageio-ffmpeg
kornia # kornia
numpy # numpy
omegaconf # omegaconf
open_clip_torch # open_clip_torch
opencv-python # opencv-python
prettytable # prettytable
safetensors # safetensors
google-generativeai # google-generativeai
test-tube # test-tube
timm # timm
webdataset # webdataset
yapf # yapf
httpx # httpx
ggl # ggl
arxiv # arxiv
google-api-python-client # google-api-python-client
google-auth-httplib2 # google-auth-httplib2
beautifulsoup4==4.11.2 # beautifulsoup4==4.11.2
O365 # O365
pytube # pytube
pydub # pydub
llama-index # llama-index
fastapi==0.94.1 # fastapi==0.94.1
pydantic==1.10.6 # pydantic==1.10.6
tenacity==8.2.2 # tenacity==8.2.2
python-dotenv # python-dotenv
boto3 # boto3
uvicorn==0.21.1 # uvicorn==0.21.1
python-ptrace # python-ptrace
jinja2==3.1.2 # jinja2==3.1.2
python-multipart==0.0.6 # python-multipart==0.0.6
celery==5.3.1 # celery==5.3.1
redis==4.6.0 # redis==4.6.0
sentencepiece # sentencepiece
bitsandbytes==0.41.0 # bitsandbytes==0.41.0
psycopg2-binary==2.9.5 # psycopg2-binary==2.9.5
google-search-results==2.4.2 # google-search-results==2.4.2
black==23.7.0 # black==23.7.0
Pillow # Pillow
selenium # selenium
# ONNI AGENT REQUIREMENTS # # ONNI AGENT REQUIREMENTS
diffusers # diffusers
controlnet_aux # controlnet_aux
tiktoken # tiktoken
espnet==202304 # espnet==202304
espnet_model_zoo==0.1.7 # espnet_model_zoo==0.1.7
# flask==2.2.3 # # flask==2.2.3
# flask_cors==3.0.10 # # flask_cors==3.0.10
waitress==2.1.2 # waitress==2.1.2
asteroid # asteroid
speechbrain # speechbrain
timm # timm
typeguard # typeguard
pytesseract # pytesseract
huggingface_hub # huggingface_hub
fastapi_cache # fastapi_cache
fastapi-limiter # fastapi-limiter
sphinx_rtd_theme # sphinx_rtd_theme
pegasusx # pegasusx
oceandb # oceandb
# simpleaichat
simpleaichat transformers
langchain==0.2.1
nest_asyncio
pegasusx
google-generativeai
oceandb
langchain-experimental
playwright
duckduckgo_search
wget==3.2
simpleaichat
httpx
ggl
beautifulsoup4
celery
redis
google-search-results==2.4.2
Pillow

@ -1,10 +1,11 @@
from swarms.swarms.swarms import HierarchicalSwarm, swarm
from swarms.workers.worker_node import WorkerNodeInitializer, WorkerNode, worker_node #swarms
from swarms.boss.boss_node import BossNodeInitializer, BossNode
# worker
#boss
#models #models
from swarms.agents.models.anthropic import Anthropic
from swarms.agents.models.huggingface import HuggingFaceLLM
from swarms.agents.models.palm import GooglePalm
from swarms.agents.models.petals import Petals
from swarms.agents.models.openai import OpenAI

@ -18,9 +18,9 @@ from ptrace.debugger import (
) )
from ptrace.func_call import FunctionCallOptions from ptrace.func_call import FunctionCallOptions
from ptrace.syscall import PtraceSyscall from ptrace.syscall import PtraceSyscall
from ptrace.tools import signal_to_exitcode from ptrace.Tools import signal_to_exitcode
from swarms.agents.tools.base import BaseToolSet, SessionGetter, ToolScope, tool from swarms.agents.Tools.base import BaseToolSet, SessionGetter, ToolScope, Tool
from swarms.utils.logger import logger from swarms.utils.logger import logger
from swarms.utils.main import ANSI, Color, Style # test from swarms.utils.main import ANSI, Color, Style # test
@ -34,7 +34,7 @@ def verify(func):
try: try:
filepath = args[0].filepath filepath = args[0].filepath
except AttributeError: except AttributeError:
raise Exception("This tool doesn't have filepath. Please check your code.") raise Exception("This Tool doesn't have filepath. Please check your code.")
if not str(Path(filepath).resolve()).startswith(str(Path().resolve())): if not str(Path(filepath).resolve()).startswith(str(Path().resolve())):
return "You can't access file outside of playground." return "You can't access file outside of playground."
return func(*args, **kwargs) return func(*args, **kwargs)
@ -201,7 +201,7 @@ class Terminal(BaseToolSet):
def __init__(self): def __init__(self):
self.sessions: Dict[str, List[SyscallTracer]] = {} self.sessions: Dict[str, List[SyscallTracer]] = {}
@tool( @Tool(
name="Terminal", name="Terminal",
description="Executes commands in a terminal." description="Executes commands in a terminal."
"If linux errno occurs, we have to solve the problem with the terminal. " "If linux errno occurs, we have to solve the problem with the terminal. "
@ -629,7 +629,7 @@ class CodePatcher:
class CodeEditor(BaseToolSet): class CodeEditor(BaseToolSet):
@tool( @Tool(
name="CodeEditor.READ", name="CodeEditor.READ",
description="Read and understand code. " description="Read and understand code. "
"Input should be filename and line number group. ex. test.py|1-10 " "Input should be filename and line number group. ex. test.py|1-10 "
@ -647,7 +647,7 @@ class CodeEditor(BaseToolSet):
) )
return output return output
@tool( @Tool(
name="CodeEditor.SUMMARY", name="CodeEditor.SUMMARY",
description="Summary code. " description="Summary code. "
"Read the code structured into a tree. " "Read the code structured into a tree. "
@ -667,10 +667,10 @@ class CodeEditor(BaseToolSet):
) )
return output return output
@tool( @Tool(
name="CodeEditor.APPEND", name="CodeEditor.APPEND",
description="Append code to the existing file. " description="Append code to the existing file. "
"If the code is completed, use the Terminal tool to execute it, if not, append the code through the this tool. " "If the code is completed, use the Terminal Tool to execute it, if not, append the code through the this Tool. "
"Input should be filename and code to append. " "Input should be filename and code to append. "
"Input code must be the code that should be appended, NOT whole code. " "Input code must be the code that should be appended, NOT whole code. "
"ex. test.py\nprint('hello world')\n " "ex. test.py\nprint('hello world')\n "
@ -689,10 +689,10 @@ class CodeEditor(BaseToolSet):
) )
return output return output
@tool( @Tool(
name="CodeEditor.WRITE", name="CodeEditor.WRITE",
description="Write code to create a new tool. " description="Write code to create a new Tool. "
"If the code is completed, use the Terminal tool to execute it, if not, append the code through the CodeEditor.APPEND tool. " "If the code is completed, use the Terminal Tool to execute it, if not, append the code through the CodeEditor.APPEND Tool. "
"Input should be formatted like: " "Input should be formatted like: "
"<filename>\n<code>\n\n" "<filename>\n<code>\n\n"
"Here is an example: " "Here is an example: "
@ -712,7 +712,7 @@ class CodeEditor(BaseToolSet):
) )
return output return output
@tool( @Tool(
name="CodeEditor.PATCH", name="CodeEditor.PATCH",
description="Patch the code to correct the error if an error occurs or to improve it. " description="Patch the code to correct the error if an error occurs or to improve it. "
"Input is a list of patches. The patch is separated by {seperator}. ".format( "Input is a list of patches. The patch is separated by {seperator}. ".format(
@ -742,7 +742,7 @@ class CodeEditor(BaseToolSet):
) )
return output return output
@tool( @Tool(
name="CodeEditor.DELETE", name="CodeEditor.DELETE",
description="Delete code in file for a new start. " description="Delete code in file for a new start. "
"Input should be filename." "Input should be filename."

@ -1,9 +1,9 @@
from swarms.agents.tools.base import BaseToolSet, SessionGetter, ToolScope, tool from swarms.agents.tools.base import BaseToolSet, SessionGetter, ToolScope, Tool
from swarms.utils.logger import logger from swarms.utils.logger import logger
class ExitConversation(BaseToolSet): class ExitConversation(BaseToolSet):
@tool( @Tool(
name="Exit Conversation", name="Exit Conversation",
description="A tool to exit the conversation. " description="A tool to exit the conversation. "
"Use this when you want to exit the conversation. " "Use this when you want to exit the conversation. "

Loading…
Cancel
Save