Mirror/ollama4j
1
0
Fork 1
mirror of https://github.com/amithkoujalgi/ollama4j.git synced 2025-05-15 11:57:12 +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

18
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,14 +39,15 @@ 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) {
return Integer.parseInt(inputValueString); case "java.lang.Integer":
} return Integer.parseInt(inputValueString);
if("java.lang.Boolean".equals(className)){ case "java.lang.Boolean":
return Boolean.valueOf(inputValueString); return Boolean.valueOf(inputValueString);
} case "java.math.BigDecimal":
else { return new BigDecimal(inputValueString);
return inputValueString; default:
return inputValueString;
} }
} }