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

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;
} }
} }