Mirror/ollama4j
1
0
Fork 1
mirror of https://github.com/amithkoujalgi/ollama4j.git synced 2025-05-15 03:47:13 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Adds BigDecimal type for ToolProperty typeCast

Browse Source
This commit is contained in:
Markus Klenke 2024-12-27 23:05:08 +01:00
parent 26ec00dab8
commit 1b38466f44
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/main/java/io/github/ollama4j/tools/ReflectionalToolFunction.java
View File

@ -5,6 +5,7 @@ import lombok.Getter;
import lombok.Setter;
import java.lang.reflect.Method;
import java.math.BigDecimal;
import java.util.LinkedHashMap;
import java.util.Map;
@ -38,13 +39,14 @@ public class ReflectionalToolFunction implements ToolFunction{
return null;
}
String inputValueString = inputValue.toString();
if("java.lang.Integer".equals(className)){
switch (className) {
case "java.lang.Integer":
return Integer.parseInt(inputValueString);
}
if("java.lang.Boolean".equals(className)){
case "java.lang.Boolean":
return Boolean.valueOf(inputValueString);
}
else {
case "java.math.BigDecimal":
return new BigDecimal(inputValueString);
default:
return inputValueString;
}
}