Merge pull request #48 from NapthaAI/chore/reverse-ws-grpc

Chore/reverse ws grpc
NapthaAI · Dec 4, 2024 · 295b7af · 295b7af
2 parents 4b61522 + 1f299ec
commit 295b7af
Show file tree

Hide file tree

Showing 6 changed files with 841 additions and 5 deletions.
diff --git a/naptha_sdk/agent.py b/naptha_sdk/agent.py
@@ -25,5 +25,5 @@ async def call_agent_func(self, *args, **kwargs):
             agent_deployment=self.orchestrator_run.agent_deployments[self.agent_index].model_dump(),
         )
 
-        agent_run = await self.worker_node.run_agent_and_poll(agent_run_input)
+        agent_run = await self.worker_node.run_agent_in_node(agent_run_input)
         return agent_run
diff --git a/naptha_sdk/client/grpc_server_pb2.py b/naptha_sdk/client/grpc_server_pb2.py