出差表单插件

src/main/java/cn/com/poc/thirdparty/resource/demand/ai/function/oa_form/TraveReimbursementlFormFunction.java

@@ -9,6 +9,8 @@ import cn.com.poc.thirdparty.resource.demand.ai.entity.largemodel.LargeModelResp
 import cn.com.poc.thirdparty.resource.demand.ai.function.AbstractFunctionResult;
 import cn.com.poc.thirdparty.resource.demand.ai.function.AbstractLargeModelFunction;
 import cn.com.poc.thirdparty.resource.demand.ai.function.entity.FunctionLLMConfig;
+import cn.com.poc.thirdparty.resource.demand.ai.function.entity.Parameters;
+import cn.com.poc.thirdparty.resource.demand.ai.function.entity.Properties;
 import cn.com.poc.thirdparty.service.LLMService;
 import cn.hutool.core.collection.ListUtil;
 import org.springframework.stereotype.Component;
@@ -33,6 +35,8 @@ public class TraveReimbursementlFormFunction extends AbstractLargeModelFunction
 
     private final FunctionLLMConfig functionLLMConfig = new FunctionLLMConfig.FunctionLLMConfigBuilder()
             .name("completion_of_travel_reimbursement_form")
+            .parameters(new Parameters("object")
+                    .addProperties("query", new Properties("string", "获取用户输入")))
             .description(DESC)
             .build();
 

src/main/java/cn/com/poc/thirdparty/resource/demand/ai/function/oa_form/TravelFormFunction.java

@@ -11,6 +11,8 @@ import cn.com.poc.thirdparty.resource.demand.ai.entity.largemodel.LargeModelResp
 import cn.com.poc.thirdparty.resource.demand.ai.function.AbstractFunctionResult;
 import cn.com.poc.thirdparty.resource.demand.ai.function.AbstractLargeModelFunction;
 import cn.com.poc.thirdparty.resource.demand.ai.function.entity.FunctionLLMConfig;
+import cn.com.poc.thirdparty.resource.demand.ai.function.entity.Parameters;
+import cn.com.poc.thirdparty.resource.demand.ai.function.entity.Properties;
 import cn.com.poc.thirdparty.service.LLMService;
 import cn.hutool.core.collection.ListUtil;
 import com.alibaba.fastjson.JSON;
@@ -39,6 +41,8 @@ public class TravelFormFunction extends AbstractLargeModelFunction {
     private final String MODEL = "qwen-long";
     private final FunctionLLMConfig functionLLMConfig = new FunctionLLMConfig.FunctionLLMConfigBuilder()
             .name("completion_of_travel_form")
+            .parameters(new Parameters("object")
+                    .addProperties("query", new Properties("string", "获取用户输入")))
             .description(DESC)
             .build();
 
@@ -84,7 +88,7 @@ public class TravelFormFunction extends AbstractLargeModelFunction {
      */
     @Override
     public AbstractFunctionResult<String> doFunction(String content, String identifier) {
-        content = "我要去北京出差";
+//        content = "我今天要去北京出差，主要是去客户那边交流一下";
         AbstractFunctionResult<String> result = new AbstractFunctionResult<>();
         if (StringUtils.isBlank(content)) {
             result.setFunctionResult(StringUtils.EMPTY);
@@ -107,6 +111,7 @@ public class TravelFormFunction extends AbstractLargeModelFunction {
             return result;
         }
         String functionResult = largeModelDemandResult.getMessage().replace("`", "").replace("json", "").replace("\n", "");
+        functionResult = functionResult.replace("客户拜访", "CustomerVisits").replace("会议", " Meetings").replace("培训", "Training");
         result.setFunctionResult(functionResult);
         result.setPromptContent(functionResult);
         return result;