exo/lm-lib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix TextAgentResponseHandler, Added afterChunkUpdate()

Browse Source
This commit is contained in:
extio-de
2025-06-29 17:29:37 +02:00
parent b761018f9f
commit c5e2ace604
2 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pom.xml
View File

@@ -4,7 +4,7 @@
<artifactId>lm-lib</artifactId>
<groupId>de.extio</groupId>
<name>lm-lib</name>
<version>1.6.0</version>
<version>1.6.2</version>
<properties>
<java.version>21</java.version>

13
src/main/java/de/extio/lmlib/agent/responsehandler/TextAgentResponseHandler.java
View File

@@ -1,6 +1,7 @@
package de.extio.lmlib.agent.responsehandler;
import java.util.Objects;
import java.util.function.Consumer;
import java.util.function.Function;
import org.slf4j.Logger;
@@ -23,15 +24,18 @@ public class TextAgentResponseHandler implements StreamedAgentResponseHandler {
private final Function<String, String> chunkTransformer;
private final Consumer<AgentContext> afterChunkUpdate;
public TextAgentResponseHandler(final String key) {
this(key, null, null);
this(key, null, null, null);
}
public TextAgentResponseHandler(final String key, final Function<String, String> transformer, final Function<String, String> chunkTransformer) {
public TextAgentResponseHandler(final String key, final Function<String, String> transformer, final Function<String, String> chunkTransformer, final Consumer<AgentContext> afterChunkUpdate) {
this.key = key;
this.chunkKey = key.concat("_chunk");
this.transformer = transformer;
this.chunkTransformer = chunkTransformer;
this.afterChunkUpdate = afterChunkUpdate;
}
@Override
@@ -57,8 +61,11 @@ public class TextAgentResponseHandler implements StreamedAgentResponseHandler {
chunk = this.chunkTransformer.apply(chunk);
}
context.setStringValue(this.chunkKey, chunk);
context.setStringValue(this.key, Objects.requireNonNullElse(context.getStringValue(this.key), "".concat(chunk)));
context.setStringValue(this.key, Objects.requireNonNullElse(context.getStringValue(this.key), "").concat(chunk));
context.setStringValue(UPDATE_KEY, this.key);
if (afterChunkUpdate != null) {
afterChunkUpdate.accept(context);
}
return true;
}