You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
25 lines
548 B
25 lines
548 B
from swarms.workers import Worker
|
|
from swarms.agents.meta_prompter import MetaPrompterAgent
|
|
from langchain.llms import OpenAI
|
|
|
|
# init llm
|
|
llm = OpenAI()
|
|
|
|
# init the meta prompter agent that optimized prompts
|
|
meta_optimizer = MetaPrompterAgent(llm=llm)
|
|
|
|
# init the worker agent
|
|
worker = Worker(llm)
|
|
|
|
# broad task to complete
|
|
task = "Create a feedforward in pytorch"
|
|
|
|
# optimize the prompt
|
|
optimized_prompt = meta_optimizer.run(task)
|
|
|
|
# run the optimized prompt with detailed instructions
|
|
result = worker.run(optimized_prompt)
|
|
|
|
# print
|
|
print(result)
|