@blueking/ai-blueking 0.2.0-beta.2 → 0.2.0-beta.3

package/README.md

@@ -23,8 +23,7 @@ npm i @blueking/ai-blueking
 
 <script lang="ts" setup>
   import { ref } from 'vue';
-  import AIBlueking from '@blueking/ai-blueking';
-  import { ChatHelper } from '@blueking/ai-blueking/dist/chat-helper';
+  import AIBlueking, { ChatHelper, RoleType, MessageStatus } from '@blueking/ai-blueking';
   import '@blueking/ai-blueking/dist/vue3/style.css';
 
   const loading = ref(false);
@@ -34,20 +33,20 @@ npm i @blueking/ai-blueking
   const handleStart = (id: number | string) => {
     loading.value = true;
     messages.value.push({
-      role: 'assistant',
+      role: RoleType.Assistant,
       content: '内容正在生成中...',
-      status: 'loading',
+      status: MessageStatus.Loading,
     });
   };
 
   // 接收消息
   const handleReceiveMessage = (message: string, id: number | string) => {
     const currentMessage = messages.value.at(-1);
-    if (currentMessage.status === 'loading') {
+    if (currentMessage.status === MessageStatus.Loading) {
       // 如果是loading状态，直接覆盖
       currentMessage.content = message;
-      currentMessage.status = 'success';
-    } else if (currentMessage.status === 'success') {
+      currentMessage.status = MessageStatus.Success;
+    } else if (currentMessage.status === MessageStatus.Success) {
       // 如果是后续消息，就追加消息
       currentMessage.content += message;
     }
@@ -58,13 +57,13 @@ npm i @blueking/ai-blueking
     loading.value = false;
     const currentMessage = messages.value.at(-1);
     // loading 情况下终止
-    if (currentMessage.status === 'loading') {
+    if (currentMessage.status === MessageStatus.Loading) {
       currentMessage.content = '聊天内容已中断';
     }
   };
 
   // 错误处理
-  const handleError = (message: string, code: string, id: number | string) => {
+  const handleError = (message: string, code: string | number, id: number | string) => {
     if (message.includes('user authentication failed')) {
       // 未登录，跳转登录
       const loginUrl = new URL(process.env.BK_LOGIN_URL);
@@ -73,13 +72,12 @@ npm i @blueking/ai-blueking
     } else {
       // 处理错误消息
       const currentMessage = messages.value.at(-1);
-      currentMessage.status = 'error';
+      currentMessage.status = MessageStatus.Error;
       currentMessage.content = message;
       loading.value = false;
     }
   };
 
-  // 需要将 <网关名> 替换成插件部署后生成的网关名
   const prefix = process.env.BK_API_URL_TMPL.replace('{api_name}', '<网关名>').replace('http', 'https');
   const chatHelper = new ChatHelper(
     `${prefix}/prod/bk_plugin/plugin_api/assistant/`,
@@ -148,7 +146,7 @@ npm i @blueking/ai-blueking
 <script setup lang="ts">
   import { ref } from 'vue';
 
-  import AIBlueking, { type IPrompt, type IMessage, RoleType } from '@blueking/ai-blueking';
+  import AIBlueking, { type IPrompt, type IMessage, RoleType, MessageStatus } from '@blueking/ai-blueking';
 
   import '@blueking/ai-blueking/dist/vue3/style.css';
 
@@ -163,7 +161,7 @@ npm i @blueking/ai-blueking
       content: '1+1=3',
       time: '2024-08-07 14:29',
       role: RoleType.Assistant,
-      status: 'error',
+      status: MessageStatus.Error,
     },
     {
       content: '不对',
@@ -174,7 +172,7 @@ npm i @blueking/ai-blueking
       content: '1+1=2',
       time: '2024-08-12 09:31',
       role: RoleType.Assistant,
-      status: 'loading',
+      status: MessageStatus.Loading,
     },
     {
       content: '对了',
@@ -298,6 +296,7 @@ npm i @blueking/bkui-library
     @close="handleClose"
     @send="handleSend"
     @scroll="handleScroll"
+    @stop="handleStop"
   />
 </template>
 <script lang="ts">
@@ -391,6 +390,9 @@ npm i @blueking/bkui-library
       handleScroll(event: Event) {
         console.log('trigger scroll', event);
       },
+      handleStop() {
+        console.log('trigger stop');
+      },
     },
   };
 </script>

package/dist/types/enum.d.ts

@@ -7,3 +7,8 @@ export declare enum RoleType {
     System = "system",
     User = "user"
 }
+export declare enum MessageStatus {
+    Error = "error",
+    Loading = "loading",
+    Success = "success"
+}

package/dist/types/index.d.ts

@@ -1,6 +1,6 @@
-import { RoleType } from './enum';
+import { RoleType, MessageStatus } from './enum';
 export interface IMessage extends IChatHistory {
-    status?: 'error' | 'loading' | 'success';
+    status?: MessageStatus;
     time?: number | string;
 }
 export interface IPrompt {