@nebulars/sseengine 1.0.0 → 1.3.44

package/src/store/menus.js

@@ -0,0 +1,45 @@
+export default ({ http }) => {
+  return {
+    async MENUS_UPDATE({ state }, params = {}) {
+      const { menus } = state;
+      Object.assign(menus, params);
+      return { menus };
+    },
+
+    async MENUS_INIT({ state }) {
+      return {
+        list: [],
+      };
+    },
+
+    async MENUS_GET({ state }) {
+      const { list, menus } = state;
+      const { data } = await http.api.sessionPage(menus);
+      const { items } = data;
+
+      if (!items.length) {
+        menus.more = false;
+      }
+
+      return { list: menus.page === 1 ? items : list.concat(items), menus };
+    },
+
+    async MENUS_DEL({ state }, { session_id }) {
+      const { data: success } = await http.api.sessionDelete({ session_id });
+      const { list } = state;
+
+      const index = list.findIndex(({ session_id: id }) => id === session_id);
+
+      if (~index) {
+        list.splice(index, 1);
+      }
+
+      return { success, list };
+    },
+
+    async MENUS_RENAME({}, { session_id, title }) {
+      const { data: success } = await http.api.sessionRename({ session_id, title });
+      return success;
+    },
+  };
+};

package/src/store/modify.less

@@ -0,0 +1,127 @@
+@space: FinQA;
+
+/**
+ * Layout
+ * ======== ======== ========
+ */
+.@{space}--Container {
+  position: relative;
+  background-color: transparent;
+}
+
+.@{space}--Header,
+.@{space}--Footer,
+.@{space}--Main {
+  display: flex;
+}
+
+.@{space}--Header,
+.@{space}--Footer {
+  flex-direction: row;
+}
+
+.@{space}--Main {
+  flex-direction: column;
+}
+
+.@{space}--Main,
+.@{space}--Chat {
+  flex: 1;
+}
+
+.@{space}--Header,
+.@{space}--Footer,
+.@{space}--Main {
+  align-items: center;
+}
+
+.@{space}--Header,
+.@{space}--Footer {
+  justify-content: space-between;
+}
+
+.@{space}--Header,
+.@{space}--Footer,
+.@{space}--Main {
+  transition: all @fast;
+}
+
+/**
+ * Interface
+ * ======== ======== ========
+ */
+.@{space}--Header,
+.@{space}--Footer,
+.@{space}--Chat,
+.@{space}--Mention {
+  padding-inline: @gap;
+}
+
+.@{space}--Input {
+  padding: @gap;
+}
+
+.@{space}--Header,
+.@{space}--Footer,
+.@{space}--Main {
+  width: 100%;
+}
+
+.@{space}--Input {
+  height: 100%;
+}
+
+.@{space}--Header {
+  height: @size-header;
+}
+
+.@{space}--Footer {
+  height: @size-footer;
+}
+
+.@{space}--Main {
+}
+
+.@{space}--Chat,
+.@{space}--Mention {
+  width: @size-main;
+}
+
+.@{space}--Mention {
+  // height: @size-mention;
+}
+
+.@{space}--Welcome {
+  height: 100%;
+}
+
+/**
+ * Colour
+ * ======== ======== ========
+ */
+.@{space}--Input {
+  background-color: @color-input;
+}
+
+/**
+ * Responsive
+ * ======== ======== ========
+ */
+.@{space}--Main {
+}
+
+/**
+ * Modify
+ * ======== ======== ========
+ */
+.@{space}--Header {
+  background-color: @color-input;
+}
+
+.@{space}--Input {
+  border-radius: @gap;
+}
+
+.@{space}--Mention {
+  position: relative;
+}

package/src/store/presets.js

@@ -0,0 +1,95 @@
+export default [
+  {
+    "option_name": "模型",
+    "field_name": "model_name",
+    "type": "single selection dropdown",
+    "defaults": "gpt-4.1",
+    "selections": [
+      {
+        "show_name": "gpt-4.1",
+        "field_value": "gpt-4.1"
+      }
+    ],
+    "description": "选择agent执行的llm"
+  },
+  {
+    "option_name": "工具选择",
+    "field_name": "tools",
+    "type": "multiple selection dropdown",
+    "defaults": [
+      "MixRetriever",
+      "Calculator",
+      "FinDataFetcher",
+      "BatchQuery",
+      "EntityListFinder",
+      "TimeParser",
+      "ContextFiller"
+    ],
+    "selections": [
+      {
+        "show_name": "混合检索工具",
+        "field_value": "MixRetriever"
+      },
+      {
+        "show_name": "计算工具",
+        "field_value": "Calculator"
+      },
+      {
+        "show_name": "金融数据获取工具",
+        "field_value": "FinDataFetcher"
+      },
+      {
+        "show_name": "批量数据获取工具",
+        "field_value": "BatchQuery"
+      },
+      {
+        "show_name": "金融实体获取工具",
+        "field_value": "EntityListFinder"
+      },
+      {
+        "show_name": "时间解析工具",
+        "field_value": "TimeParser"
+      },
+      {
+        "show_name": "上下文填充工具",
+        "field_value": "ContextFiller"
+      }
+    ],
+    "description": "选择智能体可使用的工具，以增强其能力"
+  },
+  {
+    "option_name": "业务数据",
+    "field_name": "data_sources",
+    "type": "multiple selection dropdown",
+    "defaults": [
+      "FinQA数据库",
+      "机构研究",
+      "公告",
+      "专家信息",
+      "互联网信息"
+    ],
+    "selections": [
+      {
+        "show_name": "FinQA数据库",
+        "field_value": "FinQA数据库"
+      },
+      {
+        "show_name": "机构研究",
+        "field_value": "机构研究"
+      },
+      {
+        "show_name": "公告",
+        "field_value": "公告"
+      },
+      {
+        "show_name": "专家信息",
+        "field_value": "专家信息"
+      },
+      {
+        "show_name": "互联网信息",
+        "field_value": "互联网信息"
+      }
+    ],
+    "description": "选择业务数据来源"
+  }
+]

package/src/store/remote.js

@@ -0,0 +1,34 @@
+export default ({ scrolling, kitchen }) => {
+  return {
+    async SCROLLING() {
+      await kitchen.sleep(3);
+      scrolling(`#FinQAChat`, 'end');
+    },
+
+    async PROMPT_UPDATE({}, prompt) {
+      return { prompt };
+    },
+
+    async FOCUS_ON() {
+      return { focus: true };
+    },
+
+    async FOCUS_OFF() {
+      return { focus: false };
+    },
+
+    async PRODUCE_ON({ dispatch }) {
+      await dispatch('CODE_UPDATE');
+      return { produce: true };
+    },
+
+    async PRODUCE_OFF({ dispatch }) {
+      await dispatch('CODE_UPDATE');
+      return { produce: false };
+    },
+
+    async CODE_UPDATE({}, code = 0) {
+      return { code };
+    },
+  };
+};

package/src/store/session.js

@@ -0,0 +1,57 @@
+// Use Presets
+import { default as presets } from './presets';
+
+export default ({ http, dom }) => {
+  return {
+    /**
+     * @param chat_type
+     *  @prop { String } intent_chat
+     *  @prop { String } db_chat
+     *  @prop { String } mix_search
+     * ========== ========== ==========
+     */
+    async PRESETS_GET({}, session_id) {
+      const chat_type = 'intent_chat';
+      // const { data: presets } = await http.api.utilsPresetOptions({ session_id, chat_type });
+      return { presets };
+    },
+
+    async OPTION_SET({}, presets) {
+      const options = Object.fromEntries(presets.map(({ field_name, defaults }) => [field_name, defaults]));
+      return { options };
+    },
+
+    async OPTION_UPDATE({ state }, { session_id }) {
+      const { data } = await http.api.sessionUpdateParams({ session_id, params: state.options });
+      return data;
+    },
+
+    async ID_CREATE({ state }, prompt) {
+      const { prompt: title = prompt, options } = state;
+      const { data: session_id } = await http.api.sessionCreate({ title, params: options });
+      return { session_id };
+    },
+
+    async NOTES_UPDATE({ state }, { id }) {
+      const { notes } = state;
+      notes[id] = true;
+      return { notes };
+    },
+
+    async CURRENT_SET({ state }, current = null) {
+      const { info } = state;
+      info.current = current;
+
+      return { info };
+    },
+
+    async RAG_UPDATE({ state }, rag) {
+      return { rag };
+    },
+
+    async COMPLIANCE({}, answer_id) {
+      const { data } = await http.api.sseUpdateCompliance({ answer_id, compliance: 1 });
+      return data;
+    },
+  };
+};

package/src/store/sevani.js

@@ -0,0 +1,15 @@
+export default ({ http }) => {
+  return {
+    async SEVANI_UPDATE({}, { target: sevani }) {
+      return { sevani };
+    },
+
+    async SEVANI_CLEAN({}, classname) {
+      Array.from(document.querySelectorAll(`.${classname}`)).forEach(e => e.classList.remove('hover'));
+    },
+
+    async SEVANI_FIND({ state }, classname) {
+      return Array.from(state.sevani.children).find(e => e.classList.contains(classname));
+    },
+  };
+};

package/src/store/sse.js

@@ -0,0 +1,223 @@
+export default ({ $api, sse, http, deepfind, last, toast }) => {
+  // // Hard Code
+  // const getLastNode = () => dom.last(['query-id', 'answer-id']);
+
+  // // Hard Code
+  // const getNextNode = source => (source.children.length ? source.children[0] : source);
+
+  // Update Store
+  return {
+    async STREAM_CACHE({}, stream) {
+      return { stream };
+    },
+
+    async SSE_STOP({ state, dispatch }, session_id) {
+      // SSE Stop
+      const { data } = await http.api.sseStopSse(session_id);
+
+      // Get Status
+      const { status } = data;
+
+      // Get Tower
+      const { tower } = state;
+
+      // Check Stopping
+      if (status === 'stopping') {
+        // Stop Produce
+        await dispatch('PRODUCE_OFF');
+
+        // Clean Current
+        await dispatch('CURRENT_SET');
+
+        // Tower is Live
+        if (tower) {
+          // Close
+          tower.close();
+
+          // Update as Null
+          return { tower: null };
+        }
+      }
+    },
+
+    /**
+     * Scenario 1: New Question      - Parent is Null
+     * Scenario 2: Continue Question - Parent is Last
+     * Scenario 3: Node Question     - Parent is Item.Parent
+     * ======== ======== ======== ======== ======== ========
+     */
+    async SSE_CONNECT({ state, dispatch }, { session_id, content, parent = null }) {
+      // Update Option First
+      await dispatch('OPTION_UPDATE', { session_id });
+
+      // Set Link
+      const link = `${$api.sse}?rag=${state.rag}&session_id=${session_id}&parent=${parent}&content=${encodeURIComponent(content || state.stream)}`;
+
+      // SSE
+      const tower = sse(link, {
+        // Connect Open
+        async open(data, event) {
+          // Start Produce
+          await dispatch('PRODUCE_ON');
+        },
+
+        // Connect Error
+        async error(data, event) {
+          // Show Toast - 因为每次 SSE 都会执行 Error，故暂时隐藏报错
+          // toast.error(`${data || '发生错误，请刷新重试'}`);
+
+          // Stop Produce
+          await dispatch('PRODUCE_OFF');
+
+          // Clean Current
+          await dispatch('CURRENT_SET');
+        },
+
+        // Connect Message
+        async message({ type, data, msg }) {
+          // Abnormal Status
+          if (['success', 'close', 'timeout'].includes(type)) {
+            // Stop Produce
+            await dispatch('PRODUCE_OFF');
+
+            // Clean Current
+            await dispatch('CURRENT_SET');
+          }
+
+          // Success as Endness
+          if (type === 'success') {
+            return;
+          }
+
+          // Close as Endness
+          if (type === 'close') {
+            return;
+          }
+
+          // Timeout as Endness
+          if (type === 'timeout') {
+            // Error Tip
+            return toast.error(msg);
+          }
+
+          // Update Code
+          if (type === 'status_code') {
+            await dispatch('CODE_UPDATE', data.status_code);
+          }
+
+          // Update Query
+          await dispatch('QUERY_UPDATE', { ...data, type, parent });
+
+          // Scrolling
+          // await dispatch('SCROLLING');
+        },
+      });
+
+      // Update Tower
+      return { tower };
+    },
+
+    /**
+     * @name QUERY_UPDATE
+     * ========== ========== ==========
+     * @param conversation_id - from Service
+     * @param msg - special notice
+     * @param content
+     * @param role - 1 or 2
+     * @param type
+     *  @prop { String } ready
+     *  @prop { String } message
+     *  @prop { String } success
+     *  @prop { String } thinks
+     *  @prop { String } references
+     *  @prop { String } workflows
+     * @param think
+     * @param references
+     * @param workflows
+     * ========== ========== ==========
+     */
+    async QUERY_UPDATE({ state, dispatch }, { conversation_id: id, x_trace_id = null, msg, content = '', role = 2, type, parent, parent_origin, think, references, workflows, item = null }) {
+      // Get Params
+      const { info, options: params } = state;
+
+      // SSE Keep Message
+      if (id) {
+        item = deepfind(info.history, id);
+        // item = info.history.find(item => {
+        //   return item.id === id;
+        // });
+      }
+
+      // Time First
+      if (!item) {
+        // Reset Parent
+        parent = info.current || parent || parent_origin;
+
+        // Create Item
+        item = {
+          id,
+          x_trace_id,
+          parent,
+          content,
+          role,
+          charts: [],
+          thinks: [],
+          references: [],
+          workflows: [],
+          params,
+          active: 1,
+          type,
+          children: [],
+        };
+
+        // Has Root
+        if (parent) {
+          const { children } = deepfind(info.history, parent);
+          children.push(item);
+        }
+
+        // Is Root
+        else {
+          // Hard code for Bug
+          if (info.history.length) {
+            const { children } = info.history[0];
+            children.push(item);
+          }
+
+          // Empty History
+          else {
+            info.history.push(item);
+          }
+        }
+      }
+
+      if (id) {
+        // Only For First Answer after Query
+        await dispatch('CURRENT_SET', id);
+      }
+
+      if (think) {
+        item.thinks.push(think);
+      }
+
+      if (references) {
+        item.references.push(...references);
+      }
+
+      if (workflows) {
+        item.workflows.push(...workflows);
+      }
+
+      if (item.content !== content) {
+        item.content += content;
+      }
+
+      if (type === 'success') {
+        await dispatch('PRODUCE_OFF');
+        await dispatch('CURRENT_SET');
+      }
+
+      return { info };
+    },
+  };
+};

package/src/store/sup.js

@@ -0,0 +1,7 @@
+export default ({ http }) => {
+  return {
+    async SUP_UPDATE({ dispatch }, sup) {
+      return { sup };
+    },
+  };
+};

package/src/components/fqa-icon/index.vue

@@ -1,36 +0,0 @@
-<style lang="less" scoped>
-.fqa-icon {
-  transition: all @effect;
-}
-</style>
-
-<template>
-  <component class="fqa-icon" :is="icon" :style="{ fontSize: `${size}px`, color, cursor, transitionDuration: `${duration}s` }" />
-</template>
-
-<script>
-export default {
-  props: {
-    icon: {
-      type: String,
-      default: '',
-    },
-    size: {
-      type: [Number, String],
-      default: 14,
-    },
-    color: {
-      type: String,
-      default: 'inherit',
-    },
-    cursor: {
-      type: String,
-      default: 'default',
-    },
-    duration: {
-      type: [Number, String],
-      default: 0.36,
-    },
-  },
-};
-</script>