File size: 679 Bytes
a805bcd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from naptha_sdk.utils import get_logger
logger = get_logger(__name__)
def run(inputs, worker_nodes = None, orchestrator_node = None, flow_run = None, cfg: dict = None):
from naptha_sdk.agent_service import AgentService
chat_initiator = AgentService(naptha, name="chat_initiator", fn=chat, worker_node_url="http://node.naptha.ai:7001")
chat_receiver = AgentService(naptha, name="chat_receiver", fn=chat, worker_node_url="http://node1.naptha.ai:7001")
response = await chat_initiator(inputs=inputs.prompt)
for i in range(10):
response = await chat_receiver(inputs=response)
response = await chat_initiator(inputs=response)
return response |