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

feat: Enhance tool call handling in BaseAgent and AgentTest for improved multi-tool support

Browse Source
This commit is contained in:
extio-de
2026-04-15 22:25:32 +02:00
parent b8f47f2745
commit c3ffc52fab
4 changed files with 107 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -68,6 +68,10 @@ public interface BaseAgent {
}
default ToolCallData toolCallData(final AgentContext context) {
final var configuredToolCallData = context.getValue("_toolCallData", ToolCallData.class);
if (configuredToolCallData != null) {
return configuredToolCallData;
}
final var toolDefinitions = this.toolDefinitions(context);
if (toolDefinitions == null || toolDefinitions.isEmpty()) {
final var rememberedToolCallData = context.getToolCallData();
@@ -157,7 +161,7 @@ public interface BaseAgent {
final boolean skipCache = split.context().isSkipCache() || split.context().isAlwaysSkipCache();
split.context().setSkipCache(false);
final var conversation = this.setupConversation(split);
final var toolCallData = completionClient.supportsToolCalling() ? this.toolCallData(split.context()) : null;
var toolCallData = completionClient.supportsToolCalling() ? this.toolCallData(split.context()) : null;
this.rememberToolCallData(split.context(), toolCallData);
boolean parseable = false;
@@ -214,6 +218,9 @@ public interface BaseAgent {
if (toolCallingAgentResponseHandler.handleToolCalls(completion, split.context(), toolCallResults)) {
toolCallRounds++;
this.appendToolCallResults(conversation, completion, toolCallResults.results());
if (toolCallData != null && toolCallData.hasTools()) {
toolCallData = ToolCallData.auto(toolCallData.tools());
}
continue;
}
}
@@ -450,6 +457,7 @@ public interface BaseAgent {
private void clearRememberedToolCallData(final AgentContext context) {
context.setToolCallData(null);
context.setValue("_toolCallData", null);
}
private void nextAgent(final AgentContext context, final boolean skipCompletion) {