Skip to content

Commit c4b90d6

Browse files
committed
refactor(langchain4jToolNodeTest.java): remove deprecation in update tool parameters definition
1 parent 9e70311 commit c4b90d6

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

langchain4j/src/test/java/org/bsc/langgraph4j/ToolNodeTest.java

+5-3
Original file line numberDiff line numberDiff line change
@@ -4,14 +4,14 @@
44
import com.google.gson.internal.LinkedTreeMap;
55
import dev.langchain4j.agent.tool.*;
66
import dev.langchain4j.data.message.ToolExecutionResultMessage;
7+
import dev.langchain4j.model.chat.request.json.JsonObjectSchema;
78
import dev.langchain4j.service.tool.ToolExecutor;
89
import org.bsc.langgraph4j.langchain4j.tool.ToolNode;
910
import org.junit.jupiter.api.Test;
1011

1112
import java.util.Map;
1213
import java.util.Optional;
1314

14-
import static dev.langchain4j.agent.tool.JsonSchemaProperty.description;
1515
import static java.lang.String.format;
1616
import static org.junit.jupiter.api.Assertions.assertEquals;
1717
import static org.junit.jupiter.api.Assertions.assertTrue;
@@ -47,8 +47,10 @@ public void invokeToolNode() {
4747

4848
toolSpecification = ToolSpecification.builder()
4949
.name("specialSumTwoNumbers")
50-
.addParameter("operand1", JsonSchemaProperty.type("float"), description("Operand 1 for specialK operation"))
51-
.addParameter("operand2", JsonSchemaProperty.type("float"), description("Operand 2 for specialK operation"))
50+
.parameters(JsonObjectSchema.builder()
51+
.addNumberProperty("operand1","Operand 1 for specialK operation" )
52+
.addNumberProperty( "operand2", "Operand 2 for specialK operation" )
53+
.build())
5254
.description("Returns a Float - sum of two numbers")
5355
.build();
5456

0 commit comments

Comments
 (0)