main
Kye 2 years ago
parent 9408bba916
commit 962f1092db

@ -21,8 +21,8 @@ from langchain.agents import ZeroShotAgent, Tool, AgentExecutor
from langchain import OpenAI, SerpAPIWrapper, LLMChain from langchain import OpenAI, SerpAPIWrapper, LLMChain
from swarms.agents.workers.auto_agent import agent from swarms.agents.workers.auto_agent import agent_worker
worker_agent = agent worker_agent = agent_worker
# Define your embedding model # Define your embedding model
embeddings_model = OpenAIEmbeddings() embeddings_model = OpenAIEmbeddings()

@ -90,7 +90,7 @@ tools = [
# HumanInputRun(), # Activate if you want the permit asking for help from the human # 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_name="WorkerX",
ai_role="Assistant", ai_role="Assistant",
tools=tools, 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. 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}"]) # agent.run([f"{tree_of_thoughts_prompt} {input_problem}"])
Loading…
Cancel
Save