@blueking/ai-blueking 0.2.15-beta.3 → 0.3.0

package/README.md

@@ -63,6 +63,12 @@ npm i @blueking/ai-blueking
   import AIBlueking, { ChatHelper, RoleType, MessageStatus } from '@blueking/ai-blueking';
   import '@blueking/ai-blueking/dist/vue3/style.css';
 
+  interface ISendData {
+    content: string; // 用户输入的内容
+    cite?: string;   // 引用的内容
+    prompt?: string; // 使用的 prompt 模板
+  }
+
   const loading = ref(false);
   const messages = ref([]);
 
@@ -131,19 +137,28 @@ npm i @blueking/ai-blueking
   };
 
   // 发送消息
-  const handleSend = (message: string) => {
+  const handleSend = (args: ISendData) => {
     // 记录当前消息记录
     const chatHistory = [...messages.value];
     // 添加一条消息
     messages.value.push({
       role: 'user',
-      content: message,
+      content: args.content,
+      cite: args.cite,
     });
+
+    // 根据参数构造输入内容
+    const input = args.prompt
+      ? args.prompt // 如果有 prompt，直接使用
+      : args.cite
+        ? `${args.content}: ${args.cite}` // 如果有 cite，拼接 content 和 cite
+        : args.content; // 否则只使用 content
+
     // ai 消息，id是唯一标识当前流，调用 chatHelper.stop 的时候需要传入
     chatHelper.stream(
       {
         inputs: {
-          input: message,
+          input,
           chat_history: chatHistory,
         },
       },
@@ -273,6 +288,10 @@ npm i @blueking/ai-blueking
 
   const handleSend = (val: string) => {
     console.log('trigger send', val);
+      // args 包含:
+      // - content: 用户输入的内容
+      // - cite: 引用的内容(可选)
+      // - prompt: 使用的 prompt 模板(可选)
   };
 
   const handleStop = () => {