diff --git a/java/src/test/java/com/baidubce/appbuilder/AppBuilderClientTest.java b/java/src/test/java/com/baidubce/appbuilder/AppBuilderClientTest.java index 3e8cad3d..7154e4d0 100644 --- a/java/src/test/java/com/baidubce/appbuilder/AppBuilderClientTest.java +++ b/java/src/test/java/com/baidubce/appbuilder/AppBuilderClientTest.java @@ -5,14 +5,13 @@ import com.baidubce.appbuilder.console.appbuilderclient.AppList; import java.io.IOException; -import java.util.ArrayList; +import java.nio.file.Paths; +import java.nio.file.Files; import java.util.HashMap; -import java.util.List; import java.util.Map; import com.baidubce.appbuilder.model.appbuilderclient.AppBuilderClientIterator; import com.baidubce.appbuilder.model.appbuilderclient.AppBuilderClientResult; import com.baidubce.appbuilder.model.appbuilderclient.AppListRequest; -import com.google.gson.Gson; import com.baidubce.appbuilder.model.appbuilderclient.AppBuilderClientRunRequest; import org.junit.Before; import org.junit.Test; @@ -69,33 +68,7 @@ public void AppBuilderClientRunFuncTest() throws IOException, AppBuilderServerEx AppBuilderClientRunRequest request = new AppBuilderClientRunRequest(appId, conversationId, "今天北京的天气怎么样?", false); - // 如果你本地的java版本更高,可以使用文本块特性,简化字符串构造 - String toolJson = "{\n" + - " \"type\": \"function\",\n" + - " \"function\": {\n" + - " \"name\": \"get_cur_whether\",\n" + - " \"description\": \"这是一个获得指定地点天气的工具\",\n" + - " \"parameters\": {\n" + - " \"type\": \"object\",\n" + - " \"properties\": {\n" + - " \"location\": {\n" + - " \"type\": \"string\",\n" + - " \"description\": \"省,市名,例如:河北省\"\n" + - " },\n" + - " \"unit\": {\n" + - " \"type\": \"string\",\n" + - " \"enum\": [\n" + - " \"摄氏度\",\n" + - " \"华氏度\"\n" + - " ]\n" + - " }\n" + - " },\n" + - " \"required\": [\n" + - " \"location\"\n" + - " ]\n" + - " }\n" + - " }\n" + - "}"; + String toolJson = new String(Files.readAllBytes(Paths.get("src/test/java/com/baidubce/appbuilder/files/toolcall.json"))); request.setTools(toolJson); AppBuilderClientIterator itor = builder.run(request); @@ -123,11 +96,7 @@ public void AppBuilderClientRunToolChoiceTest() throws IOException, AppBuilderSe String conversationId = builder.createConversation(); assertNotNull(conversationId); - AppBuilderClientRunRequest request = new AppBuilderClientRunRequest(); - request.setAppId(appId); - request.setConversationID(conversationId); - request.setQuery("你能干什么"); - request.setStream(false); + AppBuilderClientRunRequest request = new AppBuilderClientRunRequest(appId, conversationId, "你能干什么", false); request.setEndUserId("java_test_user_0"); Map input = new HashMap<>(); input.put("city", "北京"); diff --git a/java/src/test/java/com/baidubce/appbuilder/files/toolcall.json b/java/src/test/java/com/baidubce/appbuilder/files/toolcall.json new file mode 100644 index 00000000..f926a1ae --- /dev/null +++ b/java/src/test/java/com/baidubce/appbuilder/files/toolcall.json @@ -0,0 +1,26 @@ +{ + "type": "function", + "function": { + "name": "get_cur_whether", + "description": "这是一个获得指定地点天气的工具", + "parameters": { + "type": "object", + "properties": { + "location": { + "type": "string", + "description": "省,市名,例如:河北省" + }, + "unit": { + "type": "string", + "enum": [ + "摄氏度", + "华氏度" + ] + } + }, + "required": [ + "location" + ] + } + } +} \ No newline at end of file