@blueking/ai-ui-sdk 0.0.1-beta.2 → 0.0.1-beta.4

package/dist/common/chart-helper.d.ts

@@ -21,11 +21,11 @@ export declare class ChatHelper {
         handleEnd: HandleEnd;
         handleError: HandleError;
     });
-    stream({ sessionCode, url, headers, param, }: {
+    stream({ sessionCode, url, headers, data, }: {
         sessionCode: string;
         url: string;
         headers?: Record<string, string>;
-        param?: Record<string, any>;
+        data?: Record<string, any>;
     }): Promise<void>;
     stop(sessionCode: string): void;
 }

package/dist/common/type-transform.d.ts

@@ -0,0 +1,7 @@
+import type { ISessionContent, ISessionPrompt } from '../types/type';
+/**
+ * 将 sessionContent 转换为 sessionPrompt
+ * @param sessionContent sessionContent
+ * @returns sessionPrompt
+ */
+export declare const transferSessionContent2SessionPrompt: (sessionContent: ISessionContent) => ISessionPrompt;

package/dist/main.d.ts

@@ -1,4 +1,4 @@
-import type { ISessionContent } from './types/type.ts';
+import type { ISessionContent, ISession, ISessionPrompt } from './types/type.ts';
 import { SessionContentRole, SessionContentStatus } from './types/enum';
 type SessionContentsMap = {
     [key: string]: ISessionContent[];
@@ -6,17 +6,19 @@ type SessionContentsMap = {
 type SessionLoadingMap = {
     [key: string]: boolean;
 };
-export type { ISessionContent, };
-export { SessionContentRole, SessionContentStatus, };
+export type * from './types/type';
+export * from './types/enum';
+export * from './common/type-transform';
 export interface AICallbacks {
     handleStart?: (sessionCode: string, sessionContent: ISessionContent) => void;
     handleText?: (sessionCode: string, sessionContent: ISessionContent) => void;
     handleReferenceDoc?: (sessionCode: string, sessionContent: ISessionContent) => void;
     handleThink?: (sessionCode: string, sessionContent: ISessionContent) => void;
     handleEnd?: (sessionCode: string, sessionContent: ISessionContent) => void;
-    handleError?: (sessionCode: string, code: string | undefined, sessionContent: ISessionContent) => void;
+    handleError?: (sessionCode: string, sessionContent: ISessionContent, code: string | undefined) => void;
 }
-export declare const useAI: ({ handleStart, handleText, handleReferenceDoc, handleThink, handleEnd, handleError, }: AICallbacks) => {
+export declare const useAI: ({ handleStart, handleText, handleReferenceDoc, handleThink, handleEnd, handleError, }?: AICallbacks) => {
+    currentSession: import("vue").Ref<ISession | undefined, ISession | undefined>;
     sessionContents: import("vue").Ref<{
         id?: number | undefined;
         createdAt?: string | undefined;
@@ -46,14 +48,17 @@ export declare const useAI: ({ handleStart, handleText, handleReferenceDoc, hand
     }[]>;
     sessionContentsMap: SessionContentsMap;
     sessionLoadingMap: import("vue").Ref<SessionLoadingMap, SessionLoadingMap>;
-    chat: ({ sessionCode, param, url, headers, }: {
+    prompts: import("vue").ComputedRef<ISessionPrompt[]>;
+    chat: ({ sessionCode, data, url, headers, }: {
         sessionCode: string;
-        param?: Record<string, any>;
+        data?: Record<string, any>;
         url: string;
         headers?: Record<string, string>;
     }) => void;
+    stopChat: (sessionCode: string) => void;
     getSessionContentBySessionCode: (sessionCode: string) => ISessionContent;
     getSessionContentsBySessionCode: (sessionCode: string) => ISessionContent[];
-    changeSessionCode: (sessionCode: string) => void;
+    setCurrentSession: (session: ISession) => void;
     setSessionContents: (data: ISessionContent[]) => void;
 };
+export declare const useStyle: () => void;

package/dist/main.js

@@ -1,4 +1,4 @@
-!function(e,t){"object"==typeof exports&&"object"==typeof module?module.exports=t(require("vue")):"function"==typeof define&&define.amd?define(["vue"],t):"object"==typeof exports?exports["ai-ui-sdk"]=t(require("vue")):e["ai-ui-sdk"]=t(e.vue)}("undefined"!=typeof self?self:this,e=>(()=>{"use strict";var t,n,o,i,r={380:t=>{t.exports=e}},a={};function l(e){var t=a[e];if(void 0!==t)return t.exports;var n=a[e]={exports:{}};return r[e](n,n.exports,l),n.exports}l.d=(e,t)=>{for(var n in t)l.o(t,n)&&!l.o(e,n)&&Object.defineProperty(e,n,{enumerable:!0,get:t[n]})},l.o=(e,t)=>Object.prototype.hasOwnProperty.call(e,t),l.r=e=>{"undefined"!=typeof Symbol&&Symbol.toStringTag&&Object.defineProperty(e,Symbol.toStringTag,{value:"Module"}),Object.defineProperty(e,"__esModule",{value:!0})},l.p="/";var d={};l.r(d),l.d(d,{SessionContentRole:()=>o,SessionContentStatus:()=>n,default:()=>w,useAI:()=>y});var s={};l.r(s),l.d(s,{In:()=>o,c_:()=>n,OD:()=>y});var c=l(380);!function(e){e.System="system",e.Assistant="assistant",e.User="user",e.Guide="guide",e.Pause="pause",e.UserImage="user-image",e.Hidden="hidden",e.HiddenUser="hidden-user",e.HiddenAssistant="hidden-assistant",e.HiddenSystem="hidden-system",e.HiddenGuide="hidden-guide",e.TemplateUser="template-user",e.TemplateAssistant="template-assistant",e.TemplateSystem="template-system",e.TemplateGuide="template-guide",e.TemplateHidden="template-hidden"}(t||(t={})),function(e){e.Fail="fail",e.Loading="loading",e.Success="success"}(n||(n={})),function(e){e.Ai="ai",e.User="user",e.Time="time",e.System="system",e.Role="role",e.Hidden="hidden",e.Guide="guide",e.Pause="pause",e.TokenExpired="token-expired",e.UserImage="user-image",e.HiddenUser="hidden-user",e.HiddenAi="hidden-ai",e.HiddenRole="hidden-role",e.HiddenGuide="hidden-guide",e.TemplateUser="template-user",e.TemplateAi="template-ai",e.TemplateRole="template-role",e.TemplateGuide="template-guide",e.TemplateHidden="template-hidden",e.ImageNotSupported="image-not-supported"}(o||(o={})),function(e){e.TokenExpired="80003",e.ImageNotSupported="82003",e.UnAuthorizedPreviewFile="80401",e.TagRepeat="1513405",e[e.Aborted=20]="Aborted",e[e.UnLogin=401]="UnLogin",e.IamNoPermission="IAM_NO_PERMISSION"}(i||(i={}));let u=e=>{let t=`<section class="knowledge-head click-close">
+!function(e,t){"object"==typeof exports&&"object"==typeof module?module.exports=t(require("vue")):"function"==typeof define&&define.amd?define(["vue"],t):"object"==typeof exports?exports["ai-ui-sdk"]=t(require("vue")):e["ai-ui-sdk"]=t(e.vue)}("undefined"!=typeof self?self:this,e=>(()=>{"use strict";var t,n,o,i,r,l,a,d,s,u,c={380:t=>{t.exports=e}},p={};function h(e){var t=p[e];if(void 0!==t)return t.exports;var n=p[e]={exports:{}};return c[e](n,n.exports,h),n.exports}h.d=(e,t)=>{for(var n in t)h.o(t,n)&&!h.o(e,n)&&Object.defineProperty(e,n,{enumerable:!0,get:t[n]})},h.o=(e,t)=>Object.prototype.hasOwnProperty.call(e,t),h.r=e=>{"undefined"!=typeof Symbol&&Symbol.toStringTag&&Object.defineProperty(e,Symbol.toStringTag,{value:"Module"}),Object.defineProperty(e,"__esModule",{value:!0})},h.p="/";var m={};h.r(m),h.d(m,{ApiValueType:()=>a,EnumCharacter:()=>d,FileLinkType:()=>r,FileStatus:()=>l,HttpErrorCode:()=>i,KnowledgePathType:()=>s,KnowledgeType:()=>u,SessionContentRole:()=>o,SessionContentStatus:()=>n,SessionPromptRole:()=>t,default:()=>P,transferSessionContent2SessionPrompt:()=>F,useAI:()=>U,useStyle:()=>H});var f={};h.r(f),h.d(f,{Cy:()=>a,hb:()=>d,rs:()=>r,BM:()=>l,eo:()=>i,sh:()=>s,zW:()=>u,In:()=>o,c_:()=>n,bJ:()=>t,DA:()=>F,OD:()=>U,X3:()=>H});var g=h(380);!function(e){e.System="system",e.Assistant="assistant",e.User="user",e.Guide="guide",e.Pause="pause",e.UserImage="user-image",e.Hidden="hidden",e.HiddenUser="hidden-user",e.HiddenAssistant="hidden-assistant",e.HiddenSystem="hidden-system",e.HiddenGuide="hidden-guide",e.TemplateUser="template-user",e.TemplateAssistant="template-assistant",e.TemplateSystem="template-system",e.TemplateGuide="template-guide",e.TemplateHidden="template-hidden"}(t||(t={})),function(e){e.Fail="fail",e.Loading="loading",e.Success="success"}(n||(n={})),function(e){e.Ai="ai",e.User="user",e.Time="time",e.System="system",e.Role="role",e.Hidden="hidden",e.Guide="guide",e.Pause="pause",e.TokenExpired="token-expired",e.UserImage="user-image",e.HiddenUser="hidden-user",e.HiddenAi="hidden-ai",e.HiddenRole="hidden-role",e.HiddenGuide="hidden-guide",e.TemplateUser="template-user",e.TemplateAi="template-ai",e.TemplateRole="template-role",e.TemplateGuide="template-guide",e.TemplateHidden="template-hidden",e.ImageNotSupported="image-not-supported"}(o||(o={})),function(e){e.TokenExpired="80003",e.ImageNotSupported="82003",e.UnAuthorizedPreviewFile="80401",e.TagRepeat="1513405",e[e.Aborted=20]="Aborted",e[e.UnLogin=401]="UnLogin",e.IamNoPermission="IAM_NO_PERMISSION"}(i||(i={})),function(e){e.External="external",e.BkRepo="bk_repo"}(r||(r={})),function(e){e.Init="init",e.Uploading="uploading",e.Uploaded="uploaded",e.Deleting="deleting",e.Deleted="deleted",e.Failed="failed"}(l||(l={})),function(e){e.String="string",e.Number="number",e.Boolean="boolean",e.Array="array",e.Object="object"}(a||(a={})),function(e){e.All="all",e.System="system",e.User="user",e.Public="public",e.Space="space"}(d||(d={})),function(e){e.File="file",e.Folder="folder"}(s||(s={})),function(e){e.File="file",e.Manual="manual",e.Link="link"}(u||(u={}));let v=e=>{let t=`<section class="knowledge-head click-close">
     <svg
       class="bkaidev-wenzhang"
     >
@@ -9,14 +9,13 @@
   </section>
   <ul class="knowledge-body">`;return e.forEach(e=>{let{file_path:n,preview_path:o}=e.metadata,i=n.split("/").pop();t+=`<li
       class="knowledge-item"
-      title="${i} (${o})"
+      title="${i} (${o||n})"
     >
       <i class="bkaidev-icon bkaidev-zhishiku"></i>
-      <a href="${o}" target="_blank" class="knowledge-link g-flex-truncate">
+      <a href="${o||n}" target="_blank" class="knowledge-link g-flex-truncate">
         ${i}
       </a>
-      <span class="knowledge-path g-flex-truncate">${n}</span>
-    </li>`}),t+="</ul>"},p=()=>{let e=null,t=()=>{let t=`.knowledge-tips {
+    </li>`}),t+="</ul>"},b=()=>{let e=null,t=()=>{let t=`.knowledge-tips {
       position: relative;
       padding: 6px 8px;
       margin-bottom: 14px;
@@ -129,22 +128,17 @@
           font-size: 14px;
           margin-right: 6px;
         }
-        .knowledge-path {
-          margin-left: auto;
-          color: #979BA5;
-          text-align: right;
-        }
         &:hover {
           background: #EAEBF0;
         }
       }
-    }`;(e=document.createElement("style")).textContent=t+n,document.head.appendChild(e)},n=()=>{e&&(e.remove(),e=null)};(0,c.onBeforeMount)(()=>{t()}),(0,c.onBeforeUnmount)(()=>{n()})},h=e=>{try{return JSON.parse(e),!0}catch(e){return!1}},g=(e,t,n,o)=>{let i="内容正在生成中..."===e||n?"":e;if(i.includes('<section class="think-head click-close">')){let e=i.match(/<section class="think-body">([\s\S]*?)<\/section>/);if(e){let n=e[1],o=n.replace(/\n$/g,`${t}
+    }`;(e=document.createElement("style")).textContent=t+n,document.head.appendChild(e)},n=()=>{e&&(e.remove(),e=null)};(0,g.onBeforeMount)(()=>{t()}),(0,g.onBeforeUnmount)(()=>{n()})},k=e=>{try{return JSON.parse(e),!0}catch(e){return!1}},y=(e,t,n,o)=>{let i="内容正在生成中..."===e||n?"":e;if(i.includes('<section class="think-head click-close">')){let e=i.match(/<section class="think-body">([\s\S]*?)<\/section>/);if(e){let n=e[1],o=n.replace(/\n$/g,`${t}
 `);i=i.replace(n,o)}}else i+=`<section class="think-head click-close">
       <i class="bkaidev-icon bkaidev-sikao"></i>思考中...<i class="bkaidev-icon bkaidev-angle-up"></i>
     </section>
     <section class="think-body">
       ${t}
-</section>`;return o&&(i=i.replace("思考中...",`已完成思考 (耗时：${function(e){let t=Math.floor(e/36e5),n=Math.floor(e%36e5/6e4),o=Math.floor(e%6e4/1e3),i=e%1e3,r=[];return t>0&&r.push(`${t}h`),n>0&&r.push(`${n}min`),o>0&&r.push(`${o}s`),i>0&&r.push(`${i.toFixed(2)}ms`),r.join(" ")}(o)})`)),i},f=e=>{let t="正在思考..."===e,n=e.match(/<section class="think-body">([\s\S]*?)<\/section>$/),o=n&&""===n[1].trim();return t||o},m=()=>{let e=null,t=()=>{let t=`.think-head {
+</section>`;return o&&(i=i.replace("思考中...",`已完成思考 (耗时：${function(e){let t=Math.floor(e/36e5),n=Math.floor(e%36e5/6e4),o=Math.floor(e%6e4/1e3),i=e%1e3,r=[];return t>0&&r.push(`${t}h`),n>0&&r.push(`${n}min`),o>0&&r.push(`${o}s`),i>0&&r.push(`${i.toFixed(2)}ms`),r.join(" ")}(o)})`)),i},x=e=>{let t="正在思考..."===e,n=e.match(/<section class="think-body">([\s\S]*?)<\/section>$/),o=n&&""===n[1].trim();return t||o},S=()=>{let e=null,t=()=>{let t=`.think-head {
       display: flex;
       align-items: center;
       padding: 0 8px 0 6px;
@@ -192,4 +186,4 @@
       &~ .knowledge-head {
         margin-bottom: 8px;
       }
-    }`;(e=document.createElement("style")).textContent=t,document.head.appendChild(e)},n=()=>{e&&(e.remove(),e=null)};(0,c.onBeforeMount)(()=>{t()}),(0,c.onBeforeUnmount)(()=>{n()})};function b(e,t,n,o,i,r,a){try{var l=e[r](a),d=l.value}catch(e){n(e);return}l.done?t(d):Promise.resolve(d).then(o,i)}function k(e){return function(){var t=this,n=arguments;return new Promise(function(o,i){var r=e.apply(t,n);function a(e){b(r,o,i,a,l,"next",e)}function l(e){b(r,o,i,a,l,"throw",e)}a(void 0)})}}function v(e,t,n){return t in e?Object.defineProperty(e,t,{value:n,enumerable:!0,configurable:!0,writable:!0}):e[t]=n,e}class x{stream({sessionCode:e,url:t,headers:n,param:o}){var i=this;return k(function*(){var r,a;yield null===(r=i.handleStart)||void 0===r?void 0:r.call(i,e);let l=new AbortController;i.controllerMap[e]=l,fetch(t,{method:"post",signal:l.signal,headers:function(e){for(var t=1;t<arguments.length;t++){var n=null!=arguments[t]?arguments[t]:{},o=Object.keys(n);"function"==typeof Object.getOwnPropertySymbols&&(o=o.concat(Object.getOwnPropertySymbols(n).filter(function(e){return Object.getOwnPropertyDescriptor(n,e).enumerable}))),o.forEach(function(t){v(e,t,n[t])})}return e}({"Content-Type":"application/json"},n),mode:"cors",credentials:"include",body:JSON.stringify(o)}).then((a=k(function*(t){let n=t.body.pipeThrough(new window.TextDecoderStream).getReader(),o="";for(;;)try{let{value:r,done:a}=yield n.read();if(!t.ok){i.handleError(e,r||t.statusText,t.status);break}if(a){i.handleEnd(e);break}(o+r.toString()).split("\n").forEach(n=>{let r=n.replace("data:","").trim();if(h(r)){var a,l;let{event:n,content:d,cover:s,documents:c,result:u,code:p,elapsed_time:h,message:g}=JSON.parse(r);if(!1===u||200!==t.status){i.handleError(e,g||"模型调用失败",p);return}switch(n){case"text":i.handleText(e,d,s);break;case"reference_doc":null===(a=i.handleReferenceDoc)||void 0===a||a.call(i,e,c,s);break;case"think":null===(l=i.handleThink)||void 0===l||l.call(i,e,d,s,h);break;case"done":i.handleEnd(e,s?d:"");break;case"error":i.handleError(e,g||"模型调用失败",p)}o=""}else r&&(o=r)})}catch(t){(null==t?void 0:t.code)!==20&&i.handleError(e,`模型调用失败：${t.message}`,t.code);break}}),function(e){return a.apply(this,arguments)}))})()}stop(e){var t,n;return null===(n=this.controllerMap[e])||void 0===n||null===(t=n.abort)||void 0===t||t.call(n),this.handleEnd(e)}constructor({handleStart:e,handleText:t,handleReferenceDoc:n,handleThink:o,handleEnd:i,handleError:r}){v(this,"handleStart",void 0),v(this,"handleText",void 0),v(this,"handleReferenceDoc",void 0),v(this,"handleThink",void 0),v(this,"handleEnd",void 0),v(this,"handleError",void 0),v(this,"controllerMap",void 0),this.handleStart=e,this.handleText=t,this.handleReferenceDoc=n,this.handleThink=o,this.handleEnd=i,this.handleError=r,this.controllerMap={}}}let y=({handleStart:e,handleText:t,handleReferenceDoc:r,handleThink:a,handleEnd:l,handleError:d})=>{p(),m();let s="",h=(0,c.ref)({}),b=(0,c.ref)([]),k={},v=new x({handleStart:function(t){h.value[t]=!0;let i={sessionCode:t,role:o.Ai,status:n.Loading,content:"内容正在生成中..."};return!function(e,t){w(e).push(t)}(t,i),null==e?void 0:e(t,i)},handleText:function(e,o,i){let r=y(e);if("内容正在生成中..."===r.content)r.content=o;else{if(r.status!==n.Loading)return;r.content=i?o:r.content+o}return null==t?void 0:t(e,r)},handleReferenceDoc:function(e,t,n){let o=y(e),i=u(t);return o.content=n?i:o.content+i,null==r?void 0:r(e,o)},handleThink:function(e,t,n,o){let i=y(e);return i.content=g(i.content,t,n,o),null==a?void 0:a(e,i)},handleEnd:function(e,t){let o=y(e);if(o.status===n.Loading)return h.value[e]=!1,t&&(o.content=t),o.status=n.Success,null==l?void 0:l(e,o);("内容正在生成中..."===o.content||f(o.content))&&S(e,"聊天内容已中断")},handleError:S});function y(e){var t;return s===e?b.value.at(-1):null===(t=k[e])||void 0===t?void 0:t.at(-1)}function w(e){return s===e?b.value:k[e]}function S(e,t,r){let a=y(e);return a.status=n.Fail,a.content=t,h.value[e]=!1,r===i.TokenExpired&&(a.content="抱歉，您的剩余 Token 不足，无法返回回答内容，请先清空当前会话（上下文仍会作为历史记录保留)）",a.role=o.TokenExpired),r===i.ImageNotSupported&&(a.content="抱歉，当前模型不支持图片内容解析",a.role=o.ImageNotSupported),null==d?void 0:d(e,r,a)}return{sessionContents:b,sessionContentsMap:k,sessionLoadingMap:h,chat:function({sessionCode:e,param:t,url:n,headers:o}){v.stream({sessionCode:e,url:n,param:t,headers:o})},getSessionContentBySessionCode:y,getSessionContentsBySessionCode:w,changeSessionCode:function(e){s=e,k[e]||(k[e]=[]),b.value=k[e]},setSessionContents:function(e){s&&(k[s]=e,b.value=e)}}},w=s.default;if("undefined"!=typeof window){let{currentScript:e}=window.document,t=e&&e.src.match(/(.+\/)[^/]+\.js(\?.*)?$/);t&&(l.p=t[1])}return d})());
+    }`;(e=document.createElement("style")).textContent=t,document.head.appendChild(e)},n=()=>{e&&(e.remove(),e=null)};(0,g.onBeforeMount)(()=>{t()}),(0,g.onBeforeUnmount)(()=>{n()})};function T(e,t,n,o,i,r,l){try{var a=e[r](l),d=a.value}catch(e){n(e);return}a.done?t(d):Promise.resolve(d).then(o,i)}function w(e){return function(){var t=this,n=arguments;return new Promise(function(o,i){var r=e.apply(t,n);function l(e){T(r,o,i,l,a,"next",e)}function a(e){T(r,o,i,l,a,"throw",e)}l(void 0)})}}function E(e,t,n){return t in e?Object.defineProperty(e,t,{value:n,enumerable:!0,configurable:!0,writable:!0}):e[t]=n,e}class A{stream({sessionCode:e,url:t,headers:n,data:o}){var i=this;return w(function*(){var r,l;yield null===(r=i.handleStart)||void 0===r?void 0:r.call(i,e);let a=new AbortController;i.controllerMap[e]=a,fetch(t,{method:"post",signal:a.signal,headers:function(e){for(var t=1;t<arguments.length;t++){var n=null!=arguments[t]?arguments[t]:{},o=Object.keys(n);"function"==typeof Object.getOwnPropertySymbols&&(o=o.concat(Object.getOwnPropertySymbols(n).filter(function(e){return Object.getOwnPropertyDescriptor(n,e).enumerable}))),o.forEach(function(t){E(e,t,n[t])})}return e}({"Content-Type":"application/json"},n),mode:"cors",credentials:"include",body:JSON.stringify(o)}).then((l=w(function*(t){let n=t.body.pipeThrough(new window.TextDecoderStream).getReader(),o="";for(;;)try{let{value:r,done:l}=yield n.read();if(!t.ok){i.handleError(e,r||t.statusText,t.status);break}if(l){i.handleEnd(e);break}(o+r.toString()).split("\n").forEach(n=>{let r=n.replace("data:","").trim();if(k(r)){var l,a;let{event:n,content:d,cover:s,documents:u,result:c,code:p,elapsed_time:h,message:m}=JSON.parse(r);if(!1===c||200!==t.status){i.handleError(e,m||"模型调用失败",p);return}switch(n){case"text":i.handleText(e,d,s);break;case"reference_doc":null===(l=i.handleReferenceDoc)||void 0===l||l.call(i,e,u,s);break;case"think":null===(a=i.handleThink)||void 0===a||a.call(i,e,d,s,h);break;case"done":i.handleEnd(e,s?d:"");break;case"error":i.handleError(e,m||"模型调用失败",p)}o=""}else r&&(o=r)})}catch(t){(null==t?void 0:t.code)!==20&&i.handleError(e,`模型调用失败：${t.message}`,t.code);break}}),function(e){return l.apply(this,arguments)}))})()}stop(e){var t,n;return null===(n=this.controllerMap[e])||void 0===n||null===(t=n.abort)||void 0===t||t.call(n),this.handleEnd(e)}constructor({handleStart:e,handleText:t,handleReferenceDoc:n,handleThink:o,handleEnd:i,handleError:r}){E(this,"handleStart",void 0),E(this,"handleText",void 0),E(this,"handleReferenceDoc",void 0),E(this,"handleThink",void 0),E(this,"handleEnd",void 0),E(this,"handleError",void 0),E(this,"controllerMap",void 0),this.handleStart=e,this.handleText=t,this.handleReferenceDoc=n,this.handleThink=o,this.handleEnd=i,this.handleError=r,this.controllerMap={}}}let F=e=>{let n={[o.Ai]:t.Assistant,[o.User]:t.User,[o.System]:t.System,[o.Hidden]:t.Hidden,[o.Guide]:t.Guide,[o.Time]:t.System,[o.Role]:t.System,[o.HiddenAi]:t.HiddenAssistant,[o.HiddenGuide]:t.HiddenGuide,[o.HiddenRole]:t.HiddenSystem,[o.HiddenUser]:t.HiddenUser,[o.TemplateAi]:t.TemplateAssistant,[o.TemplateGuide]:t.TemplateGuide,[o.TemplateRole]:t.TemplateSystem,[o.TemplateUser]:t.TemplateUser,[o.TemplateHidden]:t.TemplateHidden,[o.Pause]:t.Pause,[o.TokenExpired]:t.Pause,[o.ImageNotSupported]:t.Pause,[o.UserImage]:t.UserImage};return{content:e.content,role:n[e.role]}},U=({handleStart:e,handleText:t,handleReferenceDoc:r,handleThink:l,handleEnd:a,handleError:d}={})=>{let s="内容正在生成中...",u=(0,g.ref)(),c=(0,g.ref)({}),p=(0,g.ref)([]),h={},m=(0,g.computed)(()=>{let e=[];for(let t=p.value.length-1;t>=0;t--){let i=p.value[t],r=t+1,l=p.value[r];for(;l&&![o.Ai,o.TokenExpired,o.ImageNotSupported,o.Pause,o.Guide].includes(l.role);)r+=1,l=p.value[r];if(i.role===o.System)break;i.status===n.Fail||(null==l?void 0:l.status)===n.Fail&&[o.User,o.UserImage].includes(i.role)||[o.Time,o.System].includes(i.role)||e.unshift(i)}return e}),f=(0,g.computed)(()=>{var e,t,n;let i=[],r=[],l=[],a=p.value.findLastIndex(e=>e.role===o.System&&["已启用角色","已启用模型"].some(t=>e.content.includes(t))),d=0;null===(n=u.value)||void 0===n||null===(t=n.roleInfo)||void 0===t||null===(e=t.content)||void 0===e||e.forEach(e=>{let t=p.value[a+1+d],n=p.value[a+2+d];if((null==t?void 0:t.content)===e.content&&t.role!==o.System&&(i.push(F(t)),d+=1,(null==t?void 0:t.role)===o.Pause)){for(;n&&n.role===o.System;)d+=1,n=p.value[a+1+d];for(;n&&[o.User,o.UserImage].includes(n.role);)i.push(F(n)),d+=1,n=p.value[a+1+d]}}),r.push(...m.value.map(F));let s=e=>{let t=!0;for(let o=e;o<i.length;o++){var n;i[o].content!==(null===(n=r[o-e])||void 0===n?void 0:n.content)&&(t=!1)}return t};for(let e=0;e<i.length;e++){let t=i[e];if(s(e))break;l.push(t)}return l.push(...r),l}),b=new A({handleStart:function(t){c.value[t]=!0;let i={sessionCode:t,role:o.Ai,status:n.Loading,content:s};return!function(e,t){S(e).push(t)}(t,i),null==e?void 0:e(t,i)},handleText:function(e,o,i){let r=k(e);if(r.content===s)r.content=o;else{if(r.status!==n.Loading)return;r.content=i?o:r.content+o}return null==t?void 0:t(e,r)},handleReferenceDoc:function(e,t,n){let o=k(e),i=v(t);return o.content=n?i:o.content+i,null==r?void 0:r(e,o)},handleThink:function(e,t,n,o){let i=k(e);return i.content=y(i.content,t,n,o),null==l?void 0:l(e,i)},handleEnd:function(e,t){let o=k(e);if(o.status===n.Loading)return c.value[e]=!1,t&&(o.content=t),o.status=n.Success,null==a?void 0:a(e,o);(o.content===s||x(o.content))&&T(e,"聊天内容已中断")},handleError:T});function k(e){var t,n;return(null===(t=u.value)||void 0===t?void 0:t.sessionCode)===e?p.value.at(-1):null===(n=h[e])||void 0===n?void 0:n.at(-1)}function S(e){var t;return(null===(t=u.value)||void 0===t?void 0:t.sessionCode)===e?p.value:h[e]}function T(e,t,r){let l=k(e);return l.status=n.Fail,l.content=t,c.value[e]=!1,r===i.TokenExpired&&(l.content="抱歉，您的剩余 Token 不足，无法返回回答内容，请先清空当前会话（上下文仍会作为历史记录保留)）",l.role=o.TokenExpired),r===i.ImageNotSupported&&(l.content="抱歉，当前模型不支持图片内容解析",l.role=o.ImageNotSupported),null==d?void 0:d(e,l,r)}return{currentSession:u,sessionContents:p,sessionContentsMap:h,sessionLoadingMap:c,prompts:f,chat:function({sessionCode:e,data:t,url:n,headers:o}){b.stream({sessionCode:e,url:n,data:t,headers:o})},stopChat:function(e){b.stop(e)},getSessionContentBySessionCode:k,getSessionContentsBySessionCode:S,setCurrentSession:function(e){u.value=e,h[e.sessionCode]||(h[e.sessionCode]=[]),p.value=h[e.sessionCode]},setSessionContents:function(e){u.value&&(h[u.value.sessionCode]=e,p.value=e)}}},H=()=>{b(),S()},P=f.default;if("undefined"!=typeof window){let{currentScript:e}=window.document,t=e&&e.src.match(/(.+\/)[^/]+\.js(\?.*)?$/);t&&(h.p=t[1])}return m})());

package/dist/types/enum.d.ts

@@ -52,3 +52,38 @@ export declare enum HttpErrorCode {
     UnLogin = 401,
     IamNoPermission = "IAM_NO_PERMISSION"
 }
+export declare enum FileLinkType {
+    External = "external",
+    BkRepo = "bk_repo"
+}
+export declare enum FileStatus {
+    Init = "init",
+    Uploading = "uploading",
+    Uploaded = "uploaded",
+    Deleting = "deleting",
+    Deleted = "deleted",
+    Failed = "failed"
+}
+export declare enum ApiValueType {
+    String = "string",
+    Number = "number",
+    Boolean = "boolean",
+    Array = "array",
+    Object = "object"
+}
+export declare enum EnumCharacter {
+    All = "all",
+    System = "system",
+    User = "user",
+    Public = "public",
+    Space = "space"
+}
+export declare enum KnowledgePathType {
+    File = "file",
+    Folder = "folder"
+}
+export declare enum KnowledgeType {
+    File = "file",
+    Manual = "manual",
+    Link = "link"
+}

package/dist/types/type.d.ts

@@ -1,4 +1,4 @@
-import { SessionContentRole, SessionPromptRole, SessionContentStatus } from './enum.ts';
+import { SessionContentRole, SessionPromptRole, SessionContentStatus, FileLinkType, FileStatus, ApiValueType, EnumCharacter, KnowledgePathType, KnowledgeType } from './enum';
 export type Document = {
     metadata: {
         file_path: string;
@@ -6,6 +6,180 @@ export type Document = {
         preview_path: string;
     };
 };
+export interface IContent {
+    role: SessionPromptRole;
+    content: string;
+}
+export interface IVariable {
+    name: string;
+    value: string;
+}
+export interface IToolRule {
+    func: string;
+    message: string;
+    value: string;
+}
+export interface IToolParam {
+    name: string;
+    required: boolean;
+    type: ApiValueType;
+    default: string | number | boolean;
+    description: string;
+    validate: {
+        enable: boolean;
+        rules: IToolRule[];
+    };
+}
+export interface IResourcePermission {
+    accessModel?: boolean;
+    accessModelStrategy?: boolean;
+    viewSpace?: boolean;
+    manageSpace?: boolean;
+    viewAgent?: boolean;
+    createAgent?: boolean;
+    manageAgent?: boolean;
+    viewPrompt?: boolean;
+    createPrompt?: boolean;
+    managePrompt?: boolean;
+    viewCollection?: boolean;
+    createCollection?: boolean;
+    manageCollection?: boolean;
+    viewTool?: boolean;
+    createTool?: boolean;
+    manageTool?: boolean;
+    viewKnowledgebase?: boolean;
+    createKnowledgebase?: boolean;
+    manageKnowledgebase?: boolean;
+    viewQueryTemplate?: boolean;
+    createQueryTemplate?: boolean;
+    manageQueryTemplate?: boolean;
+}
+export interface ITool {
+    spaceId?: string;
+    header: IToolParam[];
+    query?: IToolParam[];
+    body?: IToolParam[];
+    creator: string;
+    createTime: string;
+    updateTime: string;
+    toolCode: string;
+    toolName: string;
+    icon: string;
+    description: string;
+    method: string;
+    url: string;
+    generateType: EnumCharacter;
+    id?: number;
+    status?: string;
+    isSensitive?: boolean;
+    permission?: IResourcePermission;
+}
+export interface ILinkFile {
+    url: string;
+    fileName: string;
+    fileSize: number;
+    fileType: string;
+    linkType: FileLinkType;
+    fileNumber?: number;
+    id?: string;
+    files?: File[];
+    loadingPercentage?: number;
+    headers?: Record<string, string>;
+}
+export interface IKnowledgeLinkFile extends ILinkFile {
+    name: string;
+    handleType: string[];
+    splitter: string;
+    pathType: KnowledgePathType;
+    indexConfig?: IKnowledgeIndexConfig;
+}
+export interface IKnowledgeIndexApi {
+    index_name: string;
+    index_type: string;
+    status?: 'normal' | 'init' | 'deleting';
+    index_config: {
+        type?: 'user' | 'system';
+        data_type?: string;
+        columns: string[];
+    };
+}
+export interface IKnowledgeIndexConfig {
+    vector_indexes: IKnowledgeIndexApi[];
+    scalar_indexes: IKnowledgeIndexApi[];
+    full_text_indexes: IKnowledgeIndexApi[];
+}
+export interface IKnowledge {
+    id?: number;
+    spaceId?: string;
+    url: string;
+    content: string;
+    tagNames: string[];
+    splitter: string;
+    handleType: string[];
+    updateFrequency?: number;
+    headers?: Record<string, string>;
+    name: string;
+    fileName: string;
+    fileSize: number;
+    fileType: string;
+    filePath: string;
+    pathType: KnowledgePathType;
+    createdType: KnowledgeType;
+    status?: FileStatus;
+    knowledgebaseId: number;
+    knowledgebaseName: string;
+    parentAnchorPath?: string;
+    permission?: IResourcePermission;
+    anchorPath?: string;
+    linkFiles?: IKnowledgeLinkFile[];
+    updatedAt?: string;
+    updatedBy?: string;
+    errorMessage?: string;
+    indexConfig?: IKnowledgeIndexConfig;
+}
+export interface IKnowledgebase {
+    id?: number;
+    spaceId?: string;
+    anchorPath?: string;
+    parentAnchorPath?: string;
+    filePath: string;
+    fileName?: string;
+    fileType?: string;
+    handleType: string[];
+    updateFrequency?: number;
+    name: string;
+    generateType?: EnumCharacter;
+    pathType?: KnowledgePathType;
+    createdType?: KnowledgeType;
+    number: number;
+    folderNumber?: number;
+    url?: string;
+    updatedBy?: string;
+    indexConfig?: IKnowledgeIndexConfig;
+    permission?: IResourcePermission;
+    children?: Array<IKnowledgebase>;
+}
+export interface ISession {
+    sessionCode: string;
+    sessionName: string;
+    model: string;
+    roleInfo?: {
+        collectionId: number;
+        collectionName: string;
+        content: IContent[];
+        variables: IVariable[];
+    };
+    sessionProperty?: {
+        isAutoClear?: boolean;
+        isAutoCalcPrompt?: boolean;
+    };
+    tools?: ITool[];
+    anchorPathResources?: {
+        knowledges: IKnowledge[];
+        knowledgebases: IKnowledgebase[];
+        knowledgeFolders: IKnowledgebase[];
+    };
+}
 export interface ISessionContent {
     id?: number;
     createdAt?: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blueking/ai-ui-sdk",
-  "version": "0.0.1-beta.2",
+  "version": "0.0.1-beta.4",
   "description": "蓝鲸AI UI SDK",
   "main": "dist/main.js",
   "types": "dist/main.d.ts",