Fix for handling big messages from Python agent (LangStream#749)

vectorize-io · Dec 13, 2023 · 4a9f731 · 4a9f731
1 parent 5d6dbac
commit 4a9f731
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/...nts/langstream-agent-grpc/src/main/java/ai/langstream/agents/grpc/GrpcAgentProcessor.java b/...nts/langstream-agent-grpc/src/main/java/ai/langstream/agents/grpc/GrpcAgentProcessor.java
@@ -59,7 +59,7 @@ public synchronized void onNewSchemaToSend(Schema schema) {
     @Override
     public void start() throws Exception {
         super.start();
-        request = AgentServiceGrpc.newStub(channel).withWaitForReady().process(responseObserver);
+        request = asyncStub.process(responseObserver);
         restarting.set(false);
         startFailedButDevelopmentMode = false;
     }

diff --git a/...m-agents/langstream-agent-grpc/src/main/java/ai/langstream/agents/grpc/GrpcAgentSink.java b/...m-agents/langstream-agent-grpc/src/main/java/ai/langstream/agents/grpc/GrpcAgentSink.java
@@ -53,7 +53,7 @@ public void onNewSchemaToSend(Schema schema) {
     @Override
     public void start() throws Exception {
         super.start();
-        request = AgentServiceGrpc.newStub(channel).withWaitForReady().write(responseObserver);
+        request = asyncStub.write(responseObserver);
         restarting.set(false);
         startFailedButDevelopmentMode = false;
     }

diff --git a/...agents/langstream-agent-grpc/src/main/java/ai/langstream/agents/grpc/GrpcAgentSource.java b/...agents/langstream-agent-grpc/src/main/java/ai/langstream/agents/grpc/GrpcAgentSource.java
@@ -53,7 +53,7 @@ public void onNewSchemaToSend(Schema schema) {
     @Override
     public void start() throws Exception {
         super.start();
-        request = AgentServiceGrpc.newStub(channel).withWaitForReady().read(responseObserver);
+        request = asyncStub.read(responseObserver);
         restarting.set(false);
         startFailedButDevelopmentMode = false;
     }