seeseemelk/ollama4j
1
0
Fork 0
forked from Mirror/ollama4j
Code Issues Pull Requests Actions Packages Releases 1 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
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 lombok.Setter;
import java.lang.reflect.Method; import java.lang.reflect.Method;
import java.math.BigDecimal;
import java.util.LinkedHashMap; import java.util.LinkedHashMap;
import java.util.Map; import java.util.Map;
@@ -38,13 +39,14 @@ public class ReflectionalToolFunction implements ToolFunction{
return null; return null;
} }
String inputValueString = inputValue.toString(); String inputValueString = inputValue.toString();
if("java.lang.Integer".equals(className)){ switch (className) {
case "java.lang.Integer":
return Integer.parseInt(inputValueString); return Integer.parseInt(inputValueString);
} case "java.lang.Boolean":
if("java.lang.Boolean".equals(className)){
return Boolean.valueOf(inputValueString); return Boolean.valueOf(inputValueString);
} case "java.math.BigDecimal":
else { return new BigDecimal(inputValueString);
default:
return inputValueString; return inputValueString;
} }
} }