richardblythman
commited on
Commit
•
2818f3e
1
Parent(s):
47eabea
Upload folder using huggingface_hub
Browse files- multiplayer_chat/run.py +2 -2
multiplayer_chat/run.py
CHANGED
@@ -4,8 +4,8 @@ logger = get_logger(__name__)
|
|
4 |
|
5 |
async def run(inputs, worker_nodes = None, orchestrator_node = None, flow_run = None, cfg: dict = None):
|
6 |
from naptha_sdk.agent_service import AgentService
|
7 |
-
chat_initiator = AgentService(name="chat_initiator", fn="chat",
|
8 |
-
chat_receiver = AgentService(name="chat_receiver", fn="chat",
|
9 |
|
10 |
response = await chat_initiator(inputs=inputs.prompt)
|
11 |
for i in range(10):
|
|
|
4 |
|
5 |
async def run(inputs, worker_nodes = None, orchestrator_node = None, flow_run = None, cfg: dict = None):
|
6 |
from naptha_sdk.agent_service import AgentService
|
7 |
+
chat_initiator = AgentService(name="chat_initiator", fn="chat", worker_node=worker_nodes[0], orchestrator_node=orchestrator_node, flow_run=flow_run)
|
8 |
+
chat_receiver = AgentService(name="chat_receiver", fn="chat", worker_node=worker_nodes[1], orchestrator_node=orchestrator_node, flow_run=flow_run)
|
9 |
|
10 |
response = await chat_initiator(inputs=inputs.prompt)
|
11 |
for i in range(10):
|