update java unittest

userpj · Aug 8, 2024 · fa5af9f · fa5af9f
1 parent cfd6c89
commit fa5af9f
Showing 1 changed file with 59 additions and 3 deletions.
diff --git a/java/src/test/java/com/baidubce/appbuilder/AppBuilderClientTest.java b/java/src/test/java/com/baidubce/appbuilder/AppBuilderClientTest.java
@@ -5,17 +5,21 @@
 import com.baidubce.appbuilder.console.appbuilderclient.AppList;
 
 import java.io.IOException;
+import java.util.ArrayList;
+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.baidubce.appbuilder.model.appbuilderclient.AppBuilderClientRunRequest;
 import org.junit.Before;
 import org.junit.Test;
 
 import static org.junit.Assert.*;
 
-
-public class AppBuilderClientTest{
+public class AppBuilderClientTest {
     String appId;
 
     @Before
@@ -41,7 +45,7 @@ public void AppBuilderClientRunTest() throws IOException, AppBuilderServerExcept
         assertNotNull(conversationId);
         String fileId = builder.uploadLocalFile(conversationId, "src/test/java/com/baidubce/appbuilder/files/test.pdf");
         assertNotNull(fileId);
-        AppBuilderClientIterator itor = builder.run("北京有多少小学生", conversationId, new String[]{fileId}, true);
+        AppBuilderClientIterator itor = builder.run("北京有多少小学生", conversationId, new String[] { fileId }, true);
         assertTrue(itor.hasNext());
         while (itor.hasNext()) {
             AppBuilderClientResult result = itor.next();
@@ -54,4 +58,56 @@ public void testCreateConversation_AppBuilderServerException() throws IOExceptio
         AppBuilderClient builder = new AppBuilderClient("appId");
         builder.createConversation();
     }
+
+    @Test
+    public void AppBuilderClientRunFuncTest() throws IOException, AppBuilderServerException {
+        AppBuilderClient builder = new AppBuilderClient(appId);
+        String conversationId = builder.createConversation();
+        assertNotNull(conversationId);
+        String fileId = builder.uploadLocalFile(conversationId, "src/test/java/com/baidubce/appbuilder/files/test.pdf");
+        assertNotNull(fileId);
+        AppBuilderClientRunRequest request = new AppBuilderClientRunRequest();
+        request.setAppId(appId);
+        request.setConversationID(conversationId);
+        request.setQuery(
+                "今天北京的天气怎么样?");
+        request.setStream(false);
+
+        String name = "get_cur_whether";
+        String desc = "这是一个获得指定地点天气的工具";
+        Map<String, Object> parameters = new HashMap<>();
+
+        Map<String, Object> location = new HashMap<>();
+        location.put("type", "string");
+        location.put("description", "省，市名，例如：河北省");
+
+        Map<String, Object> unit = new HashMap<>();
+        unit.put("type", "string");
+        List<String> enumValues = new ArrayList<>();
+        enumValues.add("摄氏度");
+        enumValues.add("华氏度");
+        unit.put("enum", enumValues);
+
+        Map<String, Object> properties = new HashMap<>();
+        properties.put("location", location);
+        properties.put("unit", unit);
+
+        parameters.put("type", "object");
+        parameters.put("properties", properties);
+        List<String> required = new ArrayList<>();
+        required.add("location");
+        parameters.put("required", required);
+
+        AppBuilderClientRunRequest.Tool.Function func = new AppBuilderClientRunRequest.Tool.Function(name, desc,
+                parameters);
+        AppBuilderClientRunRequest.Tool tool = new AppBuilderClientRunRequest.Tool("function", func);
+        request.setTools(new AppBuilderClientRunRequest.Tool[] { tool });
+
+        AppBuilderClientIterator itor = builder.run(request);
+        assertTrue(itor.hasNext());
+        while (itor.hasNext()) {
+            AppBuilderClientResult result = itor.next();
+            System.out.println(result);
+        }
+    }
 }