commit

yxl

2025-02-10 ee07b9f5e0ca5344bf5b78fcf87355d065d8c2b7

 src/views/chartView/index.vue

@@ -1,7 +1,12 @@
<template>

  <div class="chartBox">

    <div class="ChartBox">

      <JwChat-index :taleList="list" @enter="bindEnter" v-model="inputMsg" :toolConfig="tool">

      <JwChat-index

        :taleList="list"

        @enter="bindEnter"

        v-model="inputMsg"

        :toolConfig="tool"

      >

      </JwChat-index>

    </div>

  </div>

@@ -12,12 +17,19 @@
import msgData from "../../assets/js/msg/msgData";

import store from "../../store";

import bus from "./../../assets/js/bus.js";

import { getAgentThistoryUpdate, getAgentIntention, getAgentLayerList, getAgentThistorySave } from "../../api/api.js";

import {

  getAgentThistoryUpdate,

  getAgentIntention,

  getAgentLayerList,

  getAgentThistorySave,

  getRagAgentIntention,

} from "../../api/api.js";

import mapSetFunc from "../../assets/js/mapSetFunc.js";

import { nanoid } from 'nanoid'

import { nanoid } from "nanoid";

import mapServer from "../../assets/js/map/mapServer.js";

import mapImg from "../../assets/js/map/mapImg.js";

import mapModel from "../../assets/js/map/mapModel.js";

import { marked } from "marked";

export default {

  name: "jwChat",



@@ -32,20 +44,21 @@
      obj: null,

      tool: {

        show: ["file", "history", "img"],

        callback: this.toolEvent

        callback: this.toolEvent,

      },

      config: {

        img: "/image/cover.png",

        name: "JwChat",

        dept: "最简单、最便捷",

        callback: this.bindCover

      }

        callback: this.bindCover,

      },

      closeMap: false,

    };

  },

  computed: {

    Obj() {

      return store.state.setMsgData;

    }

    },

  },

  watch: {

    Obj(newVal, oldVal) {

@@ -63,7 +76,7 @@
          this.list[this.list.length - 1].text = newVal;

        }

      }

    }

    },

  },

  mounted() {

    this.uuid = nanoid();

@@ -78,7 +91,6 @@
    // modelLayer.init();

    setTimeout(() => {

      this.getMsg("ai", "您好，请输入指令");



    }, 1000);



    bus.$on("mapInfo", (res) => {

@@ -107,46 +119,53 @@
    createSocket() {

      this.wsSocket = new WebSocket(config.wsHost);

      this.wsSocket.onopen = (event) => {

        console.log('WebSocket连接成功');

        console.log("WebSocket连接成功");

        //定时任务，由于使用nginx反向代理，60秒无数据传输会断开

        var timer = setInterval(() => {

          if ( this.wsSocket.readyState == 1) {

          if (this.wsSocket.readyState == 1) {

            this.wsSocket.send("心跳包检测");

          } else {

            //IM连接已断开

          }

        }, 50 * 1000);



      };

      this.wsSocket.onmessage = (event) => {

        // console.log('Received message:', event.data);

        console.log(event);



        if (event.data != "连接成功") {

        if (

          event.data != "连接成功" ||

          event.data.indexOf("收到消息，消息内容：心跳包检测") <= 0

        ) {

          console.log(event.data);

          const val = JSON.parse(event.data)

          const val = JSON.parse(event.data);

          if (val.session_id == this.uuid) {

            this.setMapfunc(val, this.obj)

            this.setMapfunc(val, this.obj);

          }

        }



      }

      };

    },

    async getQueryLayerList() {

      getAgentLayerList().then(response => {

      const item = {

        sourceType: "Tileset",

        name: "MilitaryFacilities",

        url: "/data-api/gisserver/c3dserver/sedb_DESKTOP-15DOTHJ_SEM4550/tileset.json",

      };

      setTimeout(() => {

        mapServer.addServer(item);

      }, 1000);

      return;

      getAgentLayerList().then((response) => {

        if (response.code != 200) return;

        const items = response.data;

        for (var i in items) {

          const name = items[i].name;

          if (

            name != "FW20" && name != "FW12"

          ) {

            mapServer.addServer(items[i])

          if (name != "FW20" && name != "FW12") {

            console.log("items[i]", items[i]);

            mapServer.addServer(items[i]);

          }



        }



      })

      });

    },

    bindEnter() {

      const msg = this.inputMsg;

@@ -173,7 +192,9 @@
        2

      );

      mapImg.addImageLayer();

      var roleMsg1 = msgData.getAffairsData([{ type: "text", val: "相关信息已在地图展示" }]);

      var roleMsg1 = msgData.getAffairsData([

        { type: "text", val: "相关信息已在地图展示" },

      ]);

      this.refashMsg(roleMsg1.list);

    },

    getMsg(type, msg) {

@@ -186,98 +207,294 @@
      }

    },

    async setMsgStart(res) {

      console.log("res----", res);

      //输入@map打开地图

      if (res == "@map") {

        var rolMsg = [

          {

            type: "text",

            val: "地图已打开",

          },

        ];

        this.refashMsg(rolMsg);

        //打开地图

        this.$store.state.setScreenFlag = true;

        this.closeMap = true;

        earthCtrl.viewer.camera.flyTo({

          destination: {

            x: -1803769.3854512463,

            y: 29158371.286777183,

            z: 13465105.223495383,

          },

          orientation: {

            heading: 6.283185307179586,

            pitch: -1.5707963267948966,

            roll: 0,

          },

        });

      }

      //有map关键字定位

      else if (res.indexOf("@map") > -1) {

        this.closeMap = true;

        window.earthCtrl.camera.flyTo(

          121.5898036178841,

          23.93799083591751,

          2732.163687177577,

          4.501790216143808,

          -32.230361363631324,

          0.001313831409284339,

          2

        );

      }

      mapImg.removeInit();

      mapSetFunc.removeAll();

      this.$store.state.setListColor = [];

      this.$store.state.showLenged = false;

      window.earthCtrl.camera.flyTo(

        121.5898036178841,

        23.93799083591751,

        2732.163687177577,

        4.501790216143808,

        -32.230361363631324,

        0.001313831409284339,

        2

      );

      this.$store.state.showRelationBox = false;



      const formData = new FormData();

      formData.append("message", res);

      formData.append("lib_id", "No1");

      formData.append("session_id", this.uuid.toString());

      // const formData = new FormData();

      // formData.append("message", res);

      // formData.append("lib_id", "No1");

      // formData.append("session_id", this.uuid.toString());

      const formData = {

        message: res.replace(/\s+/g, ""),

        lib_id: "No1",

        session_id: this.uuid.toString(),

      };



      var obj = {

        "ai": "",

        "create_time": msgData.getDate1(),

        "human": res,

        "sessionid": this.uuid,

        "update_time": "",

        "userid": 0

      if (formData.message.indexOf("@map") > -1) {

        var obj = {

          ai: "",

          create_time: msgData.getDate1(),

          human: res,

          sessionid: this.uuid,

          update_time: "",

          userid: 0,

        };

        getAgentThistorySave(obj).then((response) => {

          if (response.code == 200) {

            obj.id = response.id;

            this.obj = obj;

          }

          this.setAgentIntention(obj, formData);

        });

      } else {

        this.setRagAgentIntention(formData);

      }



      getAgentThistorySave(

        obj

      ).then(response => {



        if (response.code == 200) {

          obj.id = response.id;

          this.obj = obj;

        }



        this.setAgentIntention(obj, formData)

      })









      // var obj = {

      //   ai: "",

      //   create_time: msgData.getDate1(),

      //   human: res,

      //   sessionid: this.uuid,

      //   update_time: "",

      //   userid: 0,

      // };

      // getAgentThistorySave(obj).then((response) => {

      //   if (response.code == 200) {

      //     obj.id = response.id;

      //     this.obj = obj;

      //   }

      //   this.setAgentIntention(obj, formData);

      // });

    },

    //地图相关问答

    async setAgentIntention(obj, formData) {

      getAgentIntention(formData).then((response) => {



        if (response.code != 200) {

      console.log("obj", formData);

      if (formData.message.indexOf("@map") > -1) {

        getAgentIntention(formData)

          .then((response) => {

            if (response.code != 200) {

              if (formData.message != "@map") {

                var rolMsg = [

                  {

                    type: "text",

                    val: "未查询到相关信息。",

                  },

                ];

                this.refashMsg(rolMsg);

              }

              return;

            }

            const type = response.type;

            //普通对话信息查询

            if (type == "message" || type == "Rag") {

              //关闭地图

              if (!this.closeMap) {

                this.$store.state.setScreenFlag = false;

              }

              var rolMsg = [

                {

                  type: "text",

                  val: type == "Rag" ? response.content : response.msg,

                },

              ];

              this.refashMsg(rolMsg);

            }

            //地图查询

            else {

              //打开地图

              this.$store.state.setScreenFlag = true;

              this.setMapfunc(response.content, obj);

            }

          })

          .catch((error) => {

            var rolMsg = [

              {

                type: "text",

                val: "未查询到相关信息。",

              },

            ];

            this.refashMsg(rolMsg);

          });

      } else {

        getRagAgentIntention(formData).then((res) => {});

      }

      // getAgentIntention(formData)

      //   .then((response) => {

      //     if (response.code != 200) {

      //       if (formData.message != "@map") {

      //         var rolMsg = [

      //           {

      //             type: "text",

      //             val: "未查询到相关信息。",

      //           },

      //         ];

      //         this.refashMsg(rolMsg);

      //       }

      //       return;

      //     }

      //     const type = response.type;

      //     //普通对话信息查询

      //     if (type == "message" || type == "Rag") {

      //       //关闭地图

      //       if (!this.closeMap) {

      //         this.$store.state.setScreenFlag = false;

      //       }

      //       var rolMsg = [

      //         {

      //           type: "text",

      //           val: type == "Rag" ? response.content : response.msg,

      //         },

      //       ];

      //       this.refashMsg(rolMsg);

      //     }

      //     //地图查询

      //     else {

      //       //打开地图

      //       this.$store.state.setScreenFlag = true;

      //       this.setMapfunc(response.content, obj);

      //     }

      //   })

      //   .catch((error) => {

      //     var rolMsg = [

      //       {

      //         type: "text",

      //         val: "抱歉，服务器似乎出了点问题，我们正在修复中。",

      //       },

      //     ];

      //     this.refashMsg(rolMsg);

      //   });

    },

    //普通对话问答

    async setRagAgentIntention(formData) {

      try {

        // 发送请求

        const params = new URLSearchParams(formData);

        const url = `/rag-api/rag?${params.toString()}`;

        let response = await fetch(url, {

          method: "get",

          responseType: "stream",

          headers: {

            "Content-Type": "application/json",

          },

        });

        // ok字段判断是否成功获取到数据流

        if (!response.ok) {

          var rolMsg = [

            {

              type: "text",

              val: "未查询到相关信息。"

            }

              val: "未查询到相关信息。",

            },

          ];

          this.refashMsg(rolMsg);

          return;

        }





        const type = response.type;

        if (type == "Map") {

          this.setMapfunc(response.content, obj);

        } else if (type == "Rag") {

          this.setMapRag(response, obj);

        // 用来获取一个可读的流的读取器（Reader）以流的方式处理响应体数据

        const reader = response.body.getReader();

        // 将流中的字节数据解码为文本字符串

        const textDecoder = new TextDecoder();

        let result = true;

        let sendMsg = true;

        let sqlValue = "";

        while (result) {

          // done表示流是否已经完成读取  value包含读取到的数据块

          const { done, value } = await reader.read();

          if (done) {

            result = false;

            break;

          }

          let resText = textDecoder.decode(value);

          //包含<think>停止输出

          //包含</think>继续输出

          console.log(resText, "返回的数据");

          if (resText.indexOf("<think>") > -1) {

            sendMsg = false;

          }

          if (resText.indexOf("</think>") > -1) {

            sendMsg = true;

            let position = resText.indexOf('</think>');

            resText = resText.substring(position + "</think>".length);

          }

          if (sendMsg) {

            sqlValue += resText;

            var rolMsg = [

              {

                type: "stream",

                val: marked(sqlValue),

              },

            ];

            this.refashMsg(rolMsg);

          }

        }

      });

        // console.log(sqlValue, "输出所有返回数据");

      } catch (err) {

        var rolMsg = [

          {

            type: "text",

            val: "未查询到相关信息。",

          },

        ];

        this.refashMsg(rolMsg);

      }

    },

    setMapfunc(res, obj) {

      this.$store.state.showLenged = false;



      switch (res.func) {

        case "queryRelationship":



          var roleMsg = null;

          if (res.data) {

            mapSetFunc.init(res)

            mapSetFunc.init(res);

            var a = `${res.msg}`;

            this.setUpdateHistoryInFor(obj, a);

            roleMsg = msgData.getMdData("markdown", a);

          } else {

            roleMsg = msgData.getAffairsData([{ type: "text", val: res.msg }]);



          }



          this.refashMsg(roleMsg.list);

          break;

        case "queryMeta":

          var a = `${res.data}`;

          // var a = `|数据类型 |数据量 |\n|------|------|\n|军事目标 |724|\n|`//`${res.data}`;

          const subData = res.data

            .filter((item) => item.subtype !== null)

            .map((item) => `| ${item.subtype} | ${item.count} |`)

            .join("\n");

          var a = `| 数据类型 | 数据量 |\n|------|------|\n` + subData;

          console.log("aaa", subData, a);

          this.setUpdateHistoryInFor(obj, a);

          var roleMsg = msgData.getMdData("markdown", a);

          this.refashMsg(roleMsg.list);

          mapSetFunc.init(res)

          mapSetFunc.init(res);

          break;

        case "combatSimulate":

          this.setMsgStart2();

@@ -287,11 +504,9 @@
          this.refashMsg(roleMsg1.list);

          break;

        case "batEnv":



          if (res.msg) {

            var a = `${res.msg}`;

            var roleMsg = msgData.getMdData("markdown", a);





            roleMsg.list[0].link = true;



@@ -303,15 +518,155 @@


          break;

        case "poiMap":

        case "aroundPoi":

          mapSetFunc.init(res)

          mapSetFunc.init(res);

          this.setUpdateHistoryInFor(obj, res.msg);

          var roleMsg = msgData.getAffairsData([{ type: "text", val: res.msg }]);

          var roleMsg = msgData.getAffairsData([

            { type: "text", val: "地图已显示" },

          ]);

          this.refashMsg(roleMsg.list);

          break;

        case "aroundPoi":

          let dataList = [

            {

              type: "可打击目标",

              color: "#FF3366",

              list: [],

            },

            {

              type: "不可打击目标",

              color: "#33FF66",

              list: [],

            },

            {

              type: "其他",

              color: "#3366FF",

              list: [],

            },

          ];

          res.data.forEach((item) => {

            if (item.ifstrike === "1") {

              dataList[0].list.push(item.uuid);

            } else if (item.ifstrike === "2") {

              dataList[1].list.push(item.uuid);

            } else {

              dataList[2].list.push(item.uuid);

            }

          });

          const resultData = dataList

            .map((item) => `| ${item.type} | ${item.list.length} |`)

            .join("\n");

          var a = `| 数据类型 | 数据量 |\n|------|------|\n` + resultData;

          console.log("aaa", a);

          this.setUpdateHistoryInFor(obj, a);

          var roleMsg = msgData.getMdData("markdown", a);

          this.refashMsg(roleMsg.list);

          const finalData = {

            func: "aroundPoi",

            data: dataList,

          };

          mapSetFunc.init(finalData);

          break;

        case "getStrike":

          let dataList1 = [

            {

              type: "可打击目标",

              color: "#FF3366",

              list: [],

            },

            {

              type: "不可打击目标",

              color: "#33FF66",

              list: [],

            },

            {

              type: "其他",

              color: "#3366FF",

              list: [],

            },

          ];

          res.data.forEach((item) => {

            if (item.ifstrike === "1") {

              dataList1[0].list.push(item.uuid);

            } else if (item.ifstrike === "2") {

              dataList1[1].list.push(item.uuid);

            } else {

              dataList1[2].list.push(item.uuid);

            }

          });

          const resultData1 = dataList1

            .map((item) => `| ${item.type} | ${item.list.length} |`)

            .join("\n");

          var a = `| 数据类型 | 数据量 |\n|------|------|\n` + resultData1;

          console.log("aaa", a);

          this.setUpdateHistoryInFor(obj, a);

          var roleMsg = msgData.getMdData("markdown", a);

          this.refashMsg(roleMsg.list);

          const finalData1 = {

            func: "aroundPoi",

            data: dataList1,

          };

          mapSetFunc.init(finalData1);

          break;

        case "getParam":

          mapSetFunc.init(res);

          console.log("res.data.targetinfo", res);

          this.setUpdateHistoryInFor(obj, res.msg);

          if (res.data.length === 1) {

            var roleMsg = msgData.getAffairsData([

              { type: "text", val: res.data[0].targetinfo },

            ]);

            this.refashMsg(roleMsg.list);

          } else {

            const markData = res.data

              .map((item) => `| ${item.name} | ${item.targetinfo} |`)

              .join("\n");

            let markAttr = `| 名称 | 属性信息 |\n|------|------|\n` + markData;

            var roleMsg = msgData.getMdData("markdown", markAttr);

            this.refashMsg(roleMsg.list);

          }

          break;

        case "aroundPoi_2":

          mapSetFunc.init(res);

          console.log("res.data.targetinfo", res);

          const resData = res.data

            .map((item) => `| ${item.name} | ${item.targetinfo} |`)

            .join("\n");

          var attr = `| 名称 | 属性信息 |\n|------|------|\n` + resData;

          console.log("aaa", attr);

          var roleMsg = msgData.getMdData("markdown", attr);

          this.setUpdateHistoryInFor(obj, res.msg);

          this.refashMsg(roleMsg.list);

          break;

        case "attackentity":

          mapSetFunc.init(res);

          this.setUpdateHistoryInFor(obj, res.msg);

          if (res.data.length === 1) {

            var roleMsg = msgData.getAffairsData([

              { type: "text", val: res.data[0].targetinfo },

            ]);

            this.refashMsg(roleMsg.list);

          } else {

            const markData = res.data

              .map((item) => `| ${item.name} | ${item.targetinfo} |`)

              .join("\n");

            let markAttr = `| 名称 | 属性信息 |\n|------|------|\n` + markData;

            var roleMsg = msgData.getMdData("markdown", markAttr);

            this.refashMsg(roleMsg.list);

          }

          break;

        case "getRelatedEntity":

          this.setUpdateHistoryInFor(obj, res.msg);

          this.handleRelationData(res.data);

          var roleMsg = msgData.getAffairsData([

            { type: "text", val: "页面已显示" },

          ]);

          this.refashMsg(roleMsg.list);

          break;

        default:

          this.setUpdateHistoryInFor(obj, res.msg);

          var roleMsg1 = msgData.getAffairsData([{ type: "text", val: res.msg }]);

          var roleMsg1 = msgData.getAffairsData([

            { type: "text", val: res.msg },

          ]);

          this.refashMsg(roleMsg1.list);

          break;

      }

@@ -319,10 +674,9 @@
    setUpdateHistoryInFor(obj, message) {

      obj.ai = message;

      obj.update_time = msgData.getDate1();

      getAgentThistoryUpdate(obj).then(response => {

      getAgentThistoryUpdate(obj).then((response) => {

        console.log(response);



      })

      });

    },

    setMapRag(res, objRes) {

      const val = res.content.split("\n");

@@ -331,8 +685,8 @@
        if (val[i]) {

          obj.push({

            type: "text",

            val: val[i]

          })

            val: val[i],

          });

        }

      }

      var listData = [];

@@ -340,52 +694,116 @@
        var obj_file = res.fileSrource;

        for (var i = 0; i < obj_file.length; i++) {

          const file_name = obj_file[i].metadata.split("/");

          const name = file_name[file_name.length - 1]

          const name = file_name[file_name.length - 1];

          if (listData.length == 0) {

            listData.push({

              name: name,

              val: [

                obj_file[i].page_content

              ]

            })

              val: [obj_file[i].page_content],

            });

          } else {

            var boolen = false;

            for (var k in listData) {

              if (listData[k].name == name) {

                boolen = true;

                listData[k].val.push(obj_file[i].page_content)

                listData[k].val.push(obj_file[i].page_content);

              }

            }

            if (boolen == false) {

              listData.push({

                name: name,

                val: [

                  obj_file[i].page_content

                ]

              })

                val: [obj_file[i].page_content],

              });

            }



          }

        }

      }

      if (listData.length > 0) {

        obj.push({

          type: "fileSrource",

          val: listData

        })

          val: listData,

        });

      }



      var roleMsg = msgData.getAffairsData(obj);



      this.setUpdateHistoryInFor(objRes, JSON.stringify(roleMsg.list));

      this.refashMsg(roleMsg.list);



    },

    refashMsg(res) {

      this.loading = false;

      this.list[this.list.length - 1].list = res;

    }

  }

    },

    //关系数据处理

    handleRelationData(data) {

      // 准备节点和边

      let graphData = [];

      let links = [];

      // 处理主类型(mainType)

      Object.keys(data.mainType).forEach((main) => {

        graphData.push({

          name: main,

          category: "mainType",

          symbolSize: 70,

          itemStyle: {

            color: "#4169E1",

          },

        });



        data.mainType[main].forEach((sec) => {

          graphData.push({

            name: sec,

            category: "secType",

            symbolSize: 60,

            itemStyle: {

              color: "#87CEFA",

            },

          });



          // 处理子类型(secType)

          if (data.secType[sec]) {

            data.secType[sec].forEach((sub) => {

              graphData.push({

                name: sub,

                category: "subType",

                symbolSize: 50,

                itemStyle: {

                  color: "#30ECA6",

                },

              });



              // 处理具体项(subtype)

              if (data.subtype[sub]) {

                data.subtype[sub].forEach((item) => {

                  graphData.push({

                    name: item.name,

                    category: "subTypeItem",

                    symbolSize: 50,

                    itemStyle: {

                      color: "#00BFFF",

                    },

                    uuid: item.uuid,

                    lontitude: item.lontitude,

                    lattitude: item.lattitude,

                  });

                  links.push({ source: sub, target: item.name });

                });

              }

              // 父子节点连接

              links.push({ source: sec, target: sub });

            });

          }

          // 父子节点连接

          links.push({ source: main, target: sec });

        });

      });

      console.log("nodes---", graphData, "links---", links);

      this.$store.state.showRelationBox = true;

      this.$store.state.relationData = {

        nodes: graphData,

        links: links,

      };

    },

  },

};

</script>

<style scoped>