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

fix: add null check for finish reason in completion handling in BaseAgent

Browse Source
This commit is contained in:
extio-de
2026-05-10 10:14:45 +02:00
parent b5a43ef7ac
commit 2028d9bea5
1 changed files with 17 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/main/java/de/extio/lmlib/agent/BaseAgent.java
View File

@@ -202,6 +202,7 @@ public interface BaseAgent {
split.context().getGraph().add("(" + durationMs + " ms)"); split.context().getGraph().add("(" + durationMs + " ms)");
split.context().setLastCompletion(completion); split.context().setLastCompletion(completion);
split.context().getRequestStatistic().add(completion); split.context().getRequestStatistic().add(completion);
if (completion.finishReason() != null) {
switch (completion.finishReason()) { switch (completion.finishReason()) {
case TOKEN_LIMIT_REACHED: case TOKEN_LIMIT_REACHED:
split.context().getGraph().add("⏹"); split.context().getGraph().add("⏹");
@@ -218,6 +219,7 @@ public interface BaseAgent {
default: default:
break; break;
} }
}
if (!completion.toolCalls().isEmpty() && responseHandler instanceof final ToolCallingAgentResponseHandler toolCallingAgentResponseHandler) { if (!completion.toolCalls().isEmpty() && responseHandler instanceof final ToolCallingAgentResponseHandler toolCallingAgentResponseHandler) {
if (toolCallRounds >= 6) { if (toolCallRounds >= 6) {