接口调整处理

leutu

2024-06-27 b8b0b633bbe96658ba72ff78744591e5e1bdafd4

 src/main/java/com/yb/controller/AiController.java

@@ -41,8 +41,9 @@
    @GetMapping("/chat")
    @Operation(summary = "大模型对话")
    public String chat(@RequestParam(value = "message",defaultValue = "Hi") String message){
        return chatModel.call("请用中文回答如下问题,如果有地名出现，回到内容包括经纬度并使用json返回,"+message);
        return chatModel.call(message);
    }


    @GetMapping("/intent")
    @Operation(summary = "意图请求")
@@ -50,6 +51,7 @@
        String json = okHttpUtil.getMessage(intent+"="+message);
        Map<String,String> hashMap = new HashMap<>() ;
        iteraJson(json,hashMap);
        intentSwitch(hashMap,message);
        return  JSON.toJSON(hashMap).toString() ;
    }
    public  boolean iteraJson(String str, Map res){
@@ -72,21 +74,26 @@
    @GetMapping("/mix")
    @Operation(summary = "意图与大模型请求")
    public String mix(@RequestParam(value = "message",defaultValue = "Hi") String message){
        String json = okHttpUtil.postFastgpt(message);
        String json = okHttpUtil.getMessage(intent+"="+message);
        System.out.println(message);
//        JSONObject jsonObject = JSONObject.parseObject(json);
        Map<String,String> hashMap = new HashMap<>() ;
        iteraJson(json,hashMap);
        String place = null ;
        if( json != null) {

        String place = hashMap.get("place");
            iteraJson(json, hashMap);
            intentSwitch(hashMap,message);
            place = hashMap.get("place");
        }
        if( place != null ){
            String llm = chatModel.call("请用中文回答如下问题,如果有地名出现，回到内容包括经纬度并使用json返回,"+place);
            String llm = chatModel.call(place);
           hashMap.put("content",llm);

          json =   JSON.toJSON(hashMap).toString() ;
        }else{
            json = hashMap.put("content", chatModel.call("请用中文回答如下问题,如果有地名出现，回到内容包括经纬度并使用json返回,"+message));
            json = hashMap.put("content", chatModel.call(message));
        }
        hashMap.put("code","200");
        System.out.println(JSON.toJSON(hashMap).toString());
        System.out.println("====================================");
        return JSON.toJSON(hashMap).toString() ;
@@ -98,7 +105,7 @@
    @Operation(summary = "知识库查询")
    public String fastgpt(@RequestParam(value = "message",defaultValue = "三维实景") String message){

        okHttpUtil.requesFast(message);
        //okHttpUtil.requesFast(message);
        String json = okHttpUtil.requesFast(message);

        return  json ;
@@ -113,4 +120,58 @@

        return  json ;
    }
    /*
  def anomalyCheck(time,place):    【映射到意图1】
  """异常情况查询。查询在某个时间某个地点的异常情况，与往日有何不同。查看某个地点的卫星图片。"""

  scanWeapon();
  =======================================
  def weaponParameters(weapons):   【映射到意图2】
  """武器详细参数查询。查询武器的详细信息。查询武器的具体参数。"""
  raise ToolException(f"没有名称为{weapons}的武器。")

  weaponParameters();
  =========================================

  def searchTarget(place, entity):    【映射到意图3,4,5,6
【1. 输入参数为：政治目标，社会目标，军事目标，民宅，医院对应意图3.4
searchTarget();
2.输入参数为：可打击目标/不可打击目标对应意图5
searchStrikeableTarget();
3.输入参数为：美国空军指挥中心，对应意图6】
  """查询某地的政治目标/社会目标/经济目标/军事目标/民宅/医院/可打击目标/不可打击目标等建筑实体的情况"""
  USAFCommandCenterQuery();

def attackanddefenceCenter(weapons: str, place: str):  #【映射到意图7.8。
    这里让你判断是杀伤性武器还是防御性武器，杀伤性武器的就对对应意图7杀伤半径，
    防御性武器就对应意图8防御半径。】
  """该工具的输入必须为武器名称和地点。武器模拟仿真，给出某种场景。武器分析，地图仿真指令。或者查询武器的杀伤半径/防空半径/杀伤范围/防空范围。"""
attackUSAFCommandCenter();
defenceRangeTyphoonSystem();

   */
    private void intentSwitch(Map<String,String>hash,String message){
        if( hash.get("name") == null ) return ;
        if( hash.get("name").equals("anomalyCheck")) hash.put("name","scanWeapon");
        if( hash.get("name").equals("weaponParameters")) hash.put("name","weaponParameters");
        if( hash.get("name").equals("searchTarget")) {
            String entity = hash.get("entity");
            if( entity.indexOf("政治")>=0 || entity.indexOf("社会")>=0||entity.indexOf("军事")>=0 ||
                    entity.indexOf("民宅")>=0 || entity.indexOf("医院")>=0 || entity.indexOf("其它")>=0)
                hash.put("name","searchTarget");
            if( entity.indexOf("打击")>=0 )
                hash.put("name","searchStrikeableTarget");
            if( entity.indexOf("美国")>=0 || entity.indexOf("空军")>=0||entity.indexOf("指挥")>=0
                ||entity.indexOf("中心")>=0)
                hash.put("name","USAFCommandCenterQuery");

        }
        if( hash.get("name").equals("attackanddefenceCenter")) {
            if( message.indexOf("打击")>=0 || message.indexOf("杀伤")>=0 || message.indexOf("摧毁")>=0 ||
                    message.indexOf("破坏")>=0)
                hash.put("name","attackUSAFCommandCenter");
            else
                hash.put("name","defenceRangeTyphoonSystem");
        }
    }
}