refactor: update to be compliant with new serialization implementation

bsorrentino · Oct 24, 2024 · baf6fae · baf6fae
1 parent a14cdfc
commit baf6fae
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/...j/src/main/java/org/bsc/langgraph4j/langchain4j/serializer/std/UserMessageSerializer.java b/...j/src/main/java/org/bsc/langgraph4j/langchain4j/serializer/std/UserMessageSerializer.java
@@ -1,19 +1,19 @@
 package org.bsc.langgraph4j.langchain4j.serializer.std;
 
 import dev.langchain4j.data.message.UserMessage;
-import org.bsc.langgraph4j.serializer.std.BaseSerializer;
+import org.bsc.langgraph4j.serializer.std.NullableObjectSerializer;
 
 import java.io.IOException;
 import java.io.ObjectInput;
 import java.io.ObjectOutput;
 
-public class UserMessageSerializer extends BaseSerializer<UserMessage> {
+public class UserMessageSerializer implements NullableObjectSerializer<UserMessage> {
     @Override
     public void write(UserMessage object, ObjectOutput out) throws IOException {
 
         if( object.hasSingleText() ) {
             out.writeUTF( object.singleText() );
-            writeUTFNullable( object.name(), out);
+            writeNullableUTF( object.name(), out);
             return;
         }
         throw new IllegalArgumentException( "Unsupported content type: " + object.type() );
@@ -22,8 +22,9 @@ public void write(UserMessage object, ObjectOutput out) throws IOException {
     @Override
     public UserMessage read(ObjectInput in) throws IOException, ClassNotFoundException {
         String text = in.readUTF();
-        return readUTFNullable(in)
-                .map( name -> UserMessage.from(name, text) ).orElseGet( () -> UserMessage.from(text) );
+        return readNullableUTF(in)
+                .map( name -> UserMessage.from(name, text) )
+                .orElseGet( () -> UserMessage.from(text) );
 
     }
 }
diff --git a/studio-jetty/src/test/java/org/bsc/langgraph4j/AgentExecutorStreamingServer.java b/studio-jetty/src/test/java/org/bsc/langgraph4j/AgentExecutorStreamingServer.java
@@ -35,7 +35,7 @@ public static void main(String[] args) throws Exception {
         var app = new AgentExecutor().graphBuilder()
                 .chatLanguageModel(llm)
                 .objectsWithTools(listOf(new TestTool()))
-                .stateSerializer( JSONStateSerializer.of(objectMapper) )
+                .stateSerializer( AgentExecutor.Serializers.JSON.object() )
                 .build();