refactor: Update modelCategory method to accept AgentContext and adjust related calls

2025-07-13 10:42:02 +02:00
parent 70f7a06da8
commit a176e6b6c4
3 changed files with 18 additions and 10 deletions
--- a/src/main/java/de/extio/lmlib/agent/Agent.java
+++ b/src/main/java/de/extio/lmlib/agent/Agent.java
@@ -36,6 +36,14 @@ public record Agent(String name,
        return BaseAgent.super.agentType(context);
    }
    
+    @Override
+    public ModelCategory modelCategory(final AgentContext context) {
+        if (modelCategory != null) {
+            return modelCategory;
+        }
+        return BaseAgent.super.modelCategory(context);
+    }
+
    @Override
    public void preProcess(final AgentContext context) {
        if (preProcessor != null) {