fix compule

LangStream · Oct 25, 2023 · cf84e78 · cf84e78
1 parent c3b7467
commit cf84e78
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 6 deletions.
diff --git a/...ngstream-agent-grpc/src/main/java/ai/langstream/agents/grpc/PythonGrpcAgentProcessor.java b/...ngstream-agent-grpc/src/main/java/ai/langstream/agents/grpc/PythonGrpcAgentProcessor.java
@@ -30,7 +30,7 @@ public void init(Map<String, Object> configuration) throws Exception {
 
     @Override
     public void start() throws Exception {
-        server = new PythonGrpcServer(agentContext.getCodeDirectory(), configuration, agentContext);
+        server = new PythonGrpcServer(agentContext.getCodeDirectory(), configuration, agentId(), agentContext);
         channel = server.start();
         super.start();
     }

diff --git a/...ts/langstream-agent-grpc/src/main/java/ai/langstream/agents/grpc/PythonGrpcAgentSink.java b/...ts/langstream-agent-grpc/src/main/java/ai/langstream/agents/grpc/PythonGrpcAgentSink.java
@@ -30,7 +30,7 @@ public void init(Map<String, Object> configuration) throws Exception {
 
     @Override
     public void start() throws Exception {
-        server = new PythonGrpcServer(agentContext.getCodeDirectory(), configuration, agentContext);
+        server = new PythonGrpcServer(agentContext.getCodeDirectory(), configuration, agentId(), agentContext);
         channel = server.start();
         super.start();
     }

diff --git a/.../langstream-agent-grpc/src/main/java/ai/langstream/agents/grpc/PythonGrpcAgentSource.java b/.../langstream-agent-grpc/src/main/java/ai/langstream/agents/grpc/PythonGrpcAgentSource.java
@@ -30,7 +30,7 @@ public void init(Map<String, Object> configuration) throws Exception {
 
     @Override
     public void start() throws Exception {
-        server = new PythonGrpcServer(agentContext.getCodeDirectory(), configuration, agentContext);
+        server = new PythonGrpcServer(agentContext.getCodeDirectory(), configuration, agentId(), agentContext);
         channel = server.start();
         super.start();
     }

diff --git a/...gents/langstream-agent-grpc/src/main/java/ai/langstream/agents/grpc/PythonGrpcServer.java b/...gents/langstream-agent-grpc/src/main/java/ai/langstream/agents/grpc/PythonGrpcServer.java
@@ -33,12 +33,14 @@ public class PythonGrpcServer {
 
     private final Path codeDirectory;
     private final Map<String, Object> configuration;
+    private final String agentId;
     private final AgentContext agentContext;
     private Process pythonProcess;
 
-    public PythonGrpcServer(Path codeDirectory, Map<String, Object> configuration, AgentContext agentContext) {
+    public PythonGrpcServer(Path codeDirectory, Map<String, Object> configuration, String agentId, AgentContext agentContext) {
         this.codeDirectory = codeDirectory;
         this.configuration = configuration;
+        this.agentId = agentId;
         this.agentContext = agentContext;
     }
 
@@ -100,7 +102,7 @@ public ManagedChannel start() throws Exception {
 
     private AgentContextConfiguration computeAgentContextConfiguration() {
         final Optional<Path> persistentStateDirectoryForAgent =
-                agentContext.getPersistentStateDirectoryForAgent(agentContext.getAgentId());
+                agentContext.getPersistentStateDirectoryForAgent(agentId);
 
         final String persistentStateDirectory = persistentStateDirectoryForAgent.map(p -> p.toFile().getAbsolutePath()).orElse(null);
         AgentContextConfiguration agentContextConfiguration = new AgentContextConfiguration(persistentStateDirectory);

diff --git a/langstream-runtime/langstream-runtime-impl/src/main/python/Pipfile b/langstream-runtime/langstream-runtime-impl/src/main/python/Pipfile
@@ -1,4 +1,4 @@
-    [packages]
+[packages]
 grpcio = "*"
 fastavro = "*"