From 962f1092db2c4b0647c79a7cda1101b32745d39f Mon Sep 17 00:00:00 2001 From: Kye Date: Mon, 3 Jul 2023 18:27:10 -0400 Subject: [PATCH] clean up --- swarms/agents/swarms.py | 4 ++-- swarms/agents/workers/auto_agent.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/swarms/agents/swarms.py b/swarms/agents/swarms.py index 4fd928fd..ac6a05d9 100644 --- a/swarms/agents/swarms.py +++ b/swarms/agents/swarms.py @@ -21,8 +21,8 @@ from langchain.agents import ZeroShotAgent, Tool, AgentExecutor from langchain import OpenAI, SerpAPIWrapper, LLMChain -from swarms.agents.workers.auto_agent import agent -worker_agent = agent +from swarms.agents.workers.auto_agent import agent_worker +worker_agent = agent_worker # Define your embedding model embeddings_model = OpenAIEmbeddings() diff --git a/swarms/agents/workers/auto_agent.py b/swarms/agents/workers/auto_agent.py index 33860b5a..aa204d52 100644 --- a/swarms/agents/workers/auto_agent.py +++ b/swarms/agents/workers/auto_agent.py @@ -90,7 +90,7 @@ tools = [ # HumanInputRun(), # Activate if you want the permit asking for help from the human ] -agent = AutoGPT.from_llm_and_tools( +agent_worker = AutoGPT.from_llm_and_tools( ai_name="WorkerX", ai_role="Assistant", tools=tools, @@ -99,7 +99,7 @@ agent = AutoGPT.from_llm_and_tools( human_in_the_loop=True, # Set to True if you want to add feedback at each step. ) -agent.chain.verbose = True +agent_worker.chain.verbose = True @@ -131,4 +131,4 @@ Possible next steps: """ -agent.run([f"{tree_of_thoughts_prompt} {input_problem}"]) \ No newline at end of file +# agent.run([f"{tree_of_thoughts_prompt} {input_problem}"]) \ No newline at end of file