diff --git a/playground/worker/ultranode_example.py b/playground/worker/ultranode_example.py index a3f62c8a..6a5285d7 100644 --- a/playground/worker/ultranode_example.py +++ b/playground/worker/ultranode_example.py @@ -9,9 +9,7 @@ The server is served on localhost:8000. Users should be able to write text input as 'query' and url array as 'files', and check the response. Users input form should be delivered in JSON format. I want it to have neumorphism-style. Serve it on port 4500. - """ node = WorkerUltraUltraNode(objective) - result = node.execute() \ No newline at end of file diff --git a/playground/worker/worker.py b/playground/worker/worker.py new file mode 100644 index 00000000..a6e78dcc --- /dev/null +++ b/playground/worker/worker.py @@ -0,0 +1,21 @@ +from langchain.models import OpenAIChat +from swarms import Worker + +llm = OpenAIChat( + model_name='gpt-4', + openai_api_key="api-key", + temperature=0.5 +) + +node = Worker( + llm=llm, + ai_name="Optimus Prime", + ai_role="Worker in a swarm", + external_tools = None, + human_in_the_loop = False, + temperature = 0.5, +) + +task = "What were the winning boston marathon times for the past 5 years (ending in 2022)? Generate a table of the year, name, country of origin, and times." +response = node.run(task) +print(response)