forked from Mirror/ollama4j
Extends chat API to automatically load registered Tools
This commit is contained in:
committed by
Markus Klenke
parent
903a8176cd
commit
e9c33ab0b2
@@ -775,8 +775,8 @@ public class OllamaAPI {
|
||||
result = requestCaller.callSync(request);
|
||||
}
|
||||
|
||||
//add registered Tools to Request
|
||||
|
||||
// add all registered tools to Request
|
||||
request.setTools(toolRegistry.getRegisteredSpecs().stream().map(Tools.ToolSpecification::getToolPrompt).collect(Collectors.toList()));
|
||||
|
||||
return new OllamaChatResult(result.getResponse(), result.getResponseTime(), result.getHttpStatusCode(), request.getMessages());
|
||||
}
|
||||
|
||||
@@ -1,5 +1,6 @@
|
||||
package io.github.ollama4j.tools;
|
||||
|
||||
import java.util.Collection;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
@@ -14,4 +15,8 @@ public class ToolRegistry {
|
||||
public void addTool (String name, Tools.ToolSpecification specification) {
|
||||
tools.put(name, specification);
|
||||
}
|
||||
|
||||
public Collection<Tools.ToolSpecification> getRegisteredSpecs(){
|
||||
return tools.values();
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user