seeseemelk/ollama4j
1
0
Fork 0
forked from Mirror/ollama4j
Code Issues Pull Requests Actions Packages Releases 1 Activity

Adds documentation for annotation based Tool registration

Browse Source
This commit is contained in:
Markus Klenke
2024-12-27 23:07:35 +01:00
parent 1b38466f44
commit db008de0ca
3 changed files with 144 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/test/java/io/github/ollama4j/integrationtests/TestRealAPIs.java
View File

@@ -317,7 +317,7 @@ class TestRealAPIs {
List<OllamaChatToolCalls> toolCalls = chatResult.getChatHistory().get(1).getToolCalls();
assertEquals(1, toolCalls.size());
OllamaToolCallsFunction function = toolCalls.get(0).getFunction();
assertEquals("computeMkeConstant", function.getName());
assertEquals("computeImportantConstant", function.getName());
assertEquals(1, function.getArguments().size());
Object noOfDigits = function.getArguments().get("noOfDigits");
assertNotNull(noOfDigits);

2
src/test/java/io/github/ollama4j/samples/AnnotatedTool.java
View File

@@ -8,7 +8,7 @@ import java.math.BigDecimal;
public class AnnotatedTool {
@ToolSpec(desc = "Computes the most important constant all around the globe!")
public String computeMkeConstant(@ToolProperty(name = "noOfDigits",desc = "Number of digits that shall be returned") Integer noOfDigits ){
public String computeImportantConstant(@ToolProperty(name = "noOfDigits",desc = "Number of digits that shall be returned") Integer noOfDigits ){
return BigDecimal.valueOf((long)(Math.random()*1000000L),noOfDigits).toString();
}