@difizen/libro-ai-native 0.2.41-next.3 → 0.2.41-next.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@difizen/libro-ai-native",
-  "version": "0.2.41-next.3",
+  "version": "0.2.41-next.5",
   "description": "",
   "keywords": [
     "libro",
@@ -32,7 +32,7 @@
     "src"
   ],
   "dependencies": {
-    "@difizen/libro-jupyter": "0.2.41-next.3",
+    "@difizen/libro-jupyter": "0.2.41-next.5",
     "@ant-design/icons": "^5.4.0",
     "react-zoom-pan-pinch": "^3.6.1",
     "@difizen/mana-app": "latest",

package/src/ai-native-command-contribution.tsx

@@ -198,10 +198,12 @@ export class LibroAINativeCommandContribution
           );
         libroAINativeForCellView.showAI = true;
         addCellAIClassname(cell);
-        libroAINativeForCellView.chatStream({
-          content: l10n.getLang()
+        const code =
+          l10n.getLang() === 'en-US'
             ? `Could you please explain this piece of code?：${cell.model.value}，Provide the reasons and the results of the optimization.`
-            : `帮忙优化一下这段代码：${cell.model.value}，给出原因以及优化结果`,
+            : `帮忙优化一下这段代码：${cell.model.value}，给出原因以及优化结果`;
+        libroAINativeForCellView.chatStream({
+          content: code,
         });
         // this.libroService.active?.enterEditMode();
       },
@@ -229,11 +231,13 @@ export class LibroAINativeCommandContribution
             cell,
           );
         libroAINativeForCellView.showAI = true;
+        const code =
+          l10n.getLang() === 'en-US'
+            ? `Please help optimize this piece of code: ${cell.model.value},provide reasons and the optimized result.`
+            : `帮忙优化一下这段代码：${cell.model.value}，给出原因以及优化结果`;
         addCellAIClassname(cell);
         libroAINativeForCellView.chatStream({
-          content: l10n.getLang()
-            ? `Please help optimize this piece of code: ${cell.model.value},provide reasons and the optimized result.`
-            : `帮忙优化一下这段代码：${cell.model.value}，给出原因以及优化结果`,
+          content: code,
         });
         // this.libroService.active?.enterEditMode();
       },

package/src/ai-native-for-cell-view.tsx

@@ -16,6 +16,7 @@ import { inject } from '@difizen/mana-app';
 import { BaseView, transient, view } from '@difizen/mana-app';
 import { l10n } from '@difizen/mana-l10n';
 import { Button } from 'antd';
+import type { ParsedEvent } from 'eventsource-parser/stream';
 import { EventSourceParserStream } from 'eventsource-parser/stream';
 import breaks from 'remark-breaks';
 import remarkGfm from 'remark-gfm';
@@ -100,7 +101,7 @@ export class LibroAINativeForCellView extends BaseView {
     const url = `/libro/api/chatstream`;
     const msg = {
       chat_key: chat_key,
-      prompt: content,
+      prompt: encodeURIComponent(content),
       language: language,
     };
     const res = await this.fetcher.post<ReadableStream<Uint8Array>>(url, msg, {
@@ -130,9 +131,13 @@ export class LibroAINativeForCellView extends BaseView {
       });
       this.libroAIChatMessageItemModel = msgItem;
       let alreayDone = false;
+      let error: ParsedEvent = { data: '', type: 'event' };
       try {
         while (!alreayDone) {
           const { value, done } = await reader.read();
+          if (value) {
+            error = value;
+          }
           if (done) {
             alreayDone = true;
             msgItem.handleEventData({
@@ -146,6 +151,7 @@ export class LibroAINativeForCellView extends BaseView {
           msgItem.handleEventData(event);
         }
       } catch {
+        console.error('libro-ai-error:' + error.data);
         msgItem.handleEventData({
           type: 'error',
         });

package/src/index.less

@@ -265,7 +265,7 @@
 
 .libro-ai-native-for-cell-container {
   display: flex;
-  padding-top: 18px;
+  padding: 18px 0;
   justify-content: space-between;
 }
 

package/src/libro-ai-native-chat-service.ts

@@ -2,6 +2,7 @@ import type { IChatEvent, IChatMessageItem } from '@difizen/magent-chat';
 import { ChatEvent } from '@difizen/magent-chat';
 import { LibroChatService } from '@difizen/magent-libro';
 import { singleton } from '@difizen/mana-app';
+import type { ParsedEvent } from 'eventsource-parser/stream';
 import { EventSourceParserStream } from 'eventsource-parser/stream';
 
 import type { LibroAINativeChatMessageItemOption } from './protocol.js';
@@ -14,7 +15,7 @@ export class LibroAINativeChatService extends LibroChatService {
   ): Promise<IChatMessageItem[]> => {
     const { content, system_prompt } = option;
     const res = await this.fetcher.post<any>(`/libro/api/chat`, {
-      prompt: content,
+      prompt: encodeURIComponent(content),
       system_prompt,
     });
 
@@ -39,7 +40,7 @@ export class LibroAINativeChatService extends LibroChatService {
 
     const url = `/libro/api/chatstream`;
     const msg = {
-      prompt: content,
+      prompt: encodeURIComponent(content),
       system_prompt,
     };
     const res = await this.fetcher.post<ReadableStream<Uint8Array>>(url, msg, {
@@ -69,6 +70,7 @@ export class LibroAINativeChatService extends LibroChatService {
         content: '',
       });
       let alreayDone = false;
+      let error: ParsedEvent = { data: '', type: 'event' };
       try {
         while (!alreayDone) {
           const { value, done } = await reader.read();
@@ -80,13 +82,18 @@ export class LibroAINativeChatService extends LibroChatService {
 
             break;
           }
+          if (value) {
+            error = value;
+          }
           const data = JSON.parse(value.data);
           const event = ChatEvent.format(value.event || 'chunk', data);
           eventCallback(event);
         }
       } catch {
+        console.error('libro-ai-error:' + error.data);
         eventCallback({
           type: 'error',
+          message: error.data,
         });
       }
       return;