From 9c247307c69582087b57eb85c3b9ee91a52eeab4 Mon Sep 17 00:00:00 2001 From: Kye Date: Fri, 6 Oct 2023 02:16:55 -0400 Subject: [PATCH] worker example Former-commit-id: 3d8fab880fb5117251af8b303d30a4c8d386f567 --- playground/worker/ultranode_example.py | 2 -- playground/worker/worker.py | 21 +++++++++++++++++++++ 2 files changed, 21 insertions(+), 2 deletions(-) create mode 100644 playground/worker/worker.py 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)