| 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 |