@lobehub/chat 0.157.1 → 0.158.0

package/src/libs/agent-runtime/azureOpenai/index.test.ts

@@ -1,5 +1,6 @@
 // @vitest-environment node
 import { AzureKeyCredential, OpenAIClient } from '@azure/openai';
+import OpenAI from 'openai';
 import { Mock, afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
 import * as debugStreamModule from '../utils/debugStream';
@@ -50,7 +51,7 @@ describe('LobeAzureOpenAI', () => {
   });
 
   describe('chat', () => {
-    it('should return a StreamingTextResponse on successful API call', async () => {
+    it('should return a Response on successful API call', async () => {
       // Arrange
       const mockStream = new ReadableStream();
       const mockResponse = Promise.resolve(mockStream);
@@ -68,6 +69,140 @@ describe('LobeAzureOpenAI', () => {
       expect(result).toBeInstanceOf(Response);
     });
 
+    describe('streaming response', () => {
+      it('should handle multiple data chunks correctly', async () => {
+        const data = [
+          {
+            choices: [],
+            created: 0,
+            id: '',
+            model: '',
+            object: '',
+            prompt_filter_results: [
+              {
+                prompt_index: 0,
+                content_filter_results: {
+                  hate: { filtered: false, severity: 'safe' },
+                  self_harm: { filtered: false, severity: 'safe' },
+                  sexual: { filtered: false, severity: 'safe' },
+                  violence: { filtered: false, severity: 'safe' },
+                },
+              },
+            ],
+          },
+          {
+            choices: [
+              {
+                content_filter_results: {
+                  hate: { filtered: false, severity: 'safe' },
+                  self_harm: { filtered: false, severity: 'safe' },
+                  sexual: { filtered: false, severity: 'safe' },
+                  violence: { filtered: false, severity: 'safe' },
+                },
+                delta: { content: '你' },
+                finish_reason: null,
+                index: 0,
+                logprobs: null,
+              },
+            ],
+            created: 1715516381,
+            id: 'chatcmpl-9O2SzeGv5xy6yz0TcQNA1DHHLJ8N1',
+            model: 'gpt-35-turbo-16k',
+            object: 'chat.completion.chunk',
+            system_fingerprint: null,
+          },
+          {
+            choices: [
+              {
+                content_filter_results: {
+                  hate: { filtered: false, severity: 'safe' },
+                  self_harm: { filtered: false, severity: 'safe' },
+                  sexual: { filtered: false, severity: 'safe' },
+                  violence: { filtered: false, severity: 'safe' },
+                },
+                delta: { content: '好' },
+                finish_reason: null,
+                index: 0,
+                logprobs: null,
+              },
+            ],
+            created: 1715516381,
+            id: 'chatcmpl-9O2SzeGv5xy6yz0TcQNA1DHHLJ8N1',
+            model: 'gpt-35-turbo-16k',
+            object: 'chat.completion.chunk',
+            system_fingerprint: null,
+          },
+          {
+            choices: [
+              {
+                content_filter_results: {
+                  hate: { filtered: false, severity: 'safe' },
+                  self_harm: { filtered: false, severity: 'safe' },
+                  sexual: { filtered: false, severity: 'safe' },
+                  violence: { filtered: false, severity: 'safe' },
+                },
+                delta: { content: '！' },
+                finish_reason: null,
+                index: 0,
+                logprobs: null,
+              },
+            ],
+            created: 1715516381,
+            id: 'chatcmpl-9O2SzeGv5xy6yz0TcQNA1DHHLJ8N1',
+            model: 'gpt-35-turbo-16k',
+            object: 'chat.completion.chunk',
+            system_fingerprint: null,
+          },
+        ];
+
+        const mockStream = new ReadableStream({
+          start(controller) {
+            data.forEach((chunk) => controller.enqueue(chunk));
+            controller.close();
+          },
+        });
+        vi.spyOn(instance['client'], 'streamChatCompletions').mockResolvedValue(mockStream as any);
+
+        const result = await instance.chat({
+          stream: true,
+          max_tokens: 2048,
+          temperature: 0.6,
+          top_p: 1,
+          model: 'gpt-35-turbo-16k',
+          presence_penalty: 0,
+          frequency_penalty: 0,
+          messages: [{ role: 'user', content: '你好' }],
+        });
+
+        const decoder = new TextDecoder();
+        const reader = result.body!.getReader();
+        const stream: string[] = [];
+
+        while (true) {
+          const { value, done } = await reader.read();
+          if (done) break;
+          stream.push(decoder.decode(value));
+        }
+
+        expect(stream).toEqual(
+          [
+            'id: ',
+            'event: data',
+            'data: {"choices":[],"created":0,"id":"","model":"","object":"","prompt_filter_results":[{"prompt_index":0,"content_filter_results":{"hate":{"filtered":false,"severity":"safe"},"self_harm":{"filtered":false,"severity":"safe"},"sexual":{"filtered":false,"severity":"safe"},"violence":{"filtered":false,"severity":"safe"}}}]}\n',
+            'id: chatcmpl-9O2SzeGv5xy6yz0TcQNA1DHHLJ8N1',
+            'event: text',
+            'data: "你"\n',
+            'id: chatcmpl-9O2SzeGv5xy6yz0TcQNA1DHHLJ8N1',
+            'event: text',
+            'data: "好"\n',
+            'id: chatcmpl-9O2SzeGv5xy6yz0TcQNA1DHHLJ8N1',
+            'event: text',
+            'data: "！"\n',
+          ].map((item) => `${item}\n`),
+        );
+      });
+    });
+
     describe('Error', () => {
       it('should return AzureBizError with DeploymentNotFound error', async () => {
         // Arrange
@@ -165,7 +300,6 @@ describe('LobeAzureOpenAI', () => {
   });
 
   describe('private method', () => {
-
     describe('tocamelCase', () => {
       it('should convert string to camel case', () => {
         const key = 'image_url';
@@ -179,41 +313,41 @@ describe('LobeAzureOpenAI', () => {
     describe('camelCaseKeys', () => {
       it('should convert object keys to camel case', () => {
         const obj = {
-          "frequency_penalty": 0,
-          "messages": [
+          frequency_penalty: 0,
+          messages: [
             {
-              "role": "user",
-              "content": [
+              role: 'user',
+              content: [
                 {
-                  "type": "image_url",
-                  "image_url": {
-                    "url": "<image URL>"
-                  }
-                }
-              ]
-            }
-          ]
+                  type: 'image_url',
+                  image_url: {
+                    url: '<image URL>',
+                  },
+                },
+              ],
+            },
+          ],
         };
 
         const newObj = instance['camelCaseKeys'](obj);
 
         expect(newObj).toEqual({
-          "frequencyPenalty": 0,
-          "messages": [
+          frequencyPenalty: 0,
+          messages: [
             {
-              "role": "user",
-              "content": [
+              role: 'user',
+              content: [
                 {
-                  "type": "image_url",
-                  "imageUrl": {
-                    "url": "<image URL>"
-                  }
-                }
-              ]
-            }
-          ]
+                  type: 'image_url',
+                  imageUrl: {
+                    url: '<image URL>',
+                  },
+                },
+              ],
+            },
+          ],
         });
       });
     });
-  })
+  });
 });

package/src/libs/agent-runtime/utils/streams/openai.ts

@@ -14,7 +14,11 @@ import {
 
 export const transformOpenAIStream = (chunk: OpenAI.ChatCompletionChunk): StreamProtocolChunk => {
   // maybe need another structure to add support for multiple choices
+
   const item = chunk.choices[0];
+  if (!item) {
+    return { data: chunk, id: chunk.id, type: 'data' };
+  }
 
   if (typeof item.delta?.content === 'string') {
     return { data: item.delta.content, id: chunk.id, type: 'text' };

package/src/locales/default/auth.ts

@@ -1,6 +1,8 @@
 export default {
   login: '登录',
   loginOrSignup: '登录 / 注册',
+  profile: '个人资料',
+  security: '安全',
   signout: '退出登录',
   signup: '注册',
 };

package/src/locales/default/chat.ts

@@ -5,6 +5,7 @@ export default {
   agentDefaultMessage:
     '你好，我是 **{{name}}**，你可以立即与我开始对话，也可以前往 [助手设置](/chat/settings#session={{id}}) 完善我的信息。',
   agentDefaultMessageWithSystemRole: '你好，我是 **{{name}}**，{{systemRole}}，让我们开始对话吧！',
+  agentsAndConversations: '助手与会话',
   backToBottom: '跳转至当前',
   clearCurrentMessages: '清空当前会话消息',
   confirmClearCurrentMessages: '即将清空当前会话消息，清空后将无法找回，请确认你的操作',

package/src/locales/default/common.ts

@@ -163,6 +163,7 @@ export default {
   userPanel: {
     anonymousNickName: '匿名用户',
     billing: '账单管理',
+    data: '数据存储',
     defaultNickname: '社区版用户',
     discord: '社区支持',
     docs: '使用文档',

package/src/store/user/slices/auth/selectors.ts

@@ -1,6 +1,6 @@
 import { t } from 'i18next';
 
-import { enableAuth } from '@/const/auth';
+import { enableAuth, enableClerk } from '@/const/auth';
 import { UserStore } from '@/store/user';
 import { LobeUser } from '@/types/user';
 
@@ -43,4 +43,5 @@ const isLogin = (s: UserStore) => {
 export const authSelectors = {
   isLogin,
   isLoginWithAuth: (s: UserStore) => s.isSignedIn,
+  isLoginWithClerk: (s: UserStore) => s.isSignedIn && enableClerk,
 };

package/src/app/(auth)/profile/[[...slugs]]/PageTitle.tsx

@@ -1,13 +0,0 @@
-'use client';
-
-import { memo } from 'react';
-import { useTranslation } from 'react-i18next';
-
-import PageTitle from '@/components/PageTitle';
-
-const Title = memo(() => {
-  const { t } = useTranslation('auth');
-
-  return <PageTitle title={t('signup')} />;
-});
-export default Title;

package/src/app/(auth)/profile/[[...slugs]]/page.tsx

@@ -1,14 +0,0 @@
-import { UserProfile } from '@clerk/nextjs';
-
-import PageTitle from './PageTitle';
-
-const Page = () => {
-  return (
-    <>
-      <PageTitle />
-      <UserProfile />
-    </>
-  );
-};
-
-export default Page;

package/src/app/(main)/(mobile)/me/features/Cate.tsx

@@ -1,33 +0,0 @@
-'use client';
-
-import { useRouter } from 'next/navigation';
-import { memo } from 'react';
-import { Flexbox } from 'react-layout-kit';
-import urlJoin from 'url-join';
-
-import { useCategory } from '@/app/(main)/settings/hooks/useCategory';
-import Cell from '@/components/Cell';
-import Divider from '@/components/Cell/Divider';
-
-const SettingCate = memo(() => {
-  const settingItems = useCategory({ mobile: true });
-  const router = useRouter();
-
-  return (
-    <Flexbox width={'100%'}>
-      {settingItems?.map(({ key, icon, label, type }: any, index) => {
-        if (type === 'divider') return <Divider key={index} />;
-        return (
-          <Cell
-            icon={icon}
-            key={key}
-            label={label}
-            onClick={() => router.push(urlJoin('/settings', key))}
-          />
-        );
-      })}
-    </Flexbox>
-  );
-});
-
-export default SettingCate;

package/src/app/(main)/(mobile)/me/features/ExtraCate.tsx

@@ -1,24 +0,0 @@
-'use client';
-
-import { memo } from 'react';
-import { Flexbox } from 'react-layout-kit';
-
-import Cell from '@/components/Cell';
-import Divider from '@/components/Cell/Divider';
-
-import { useExtraCate } from './useExtraCate';
-
-const ExtraCate = memo(() => {
-  const mainItems = useExtraCate();
-
-  return (
-    <Flexbox width={'100%'}>
-      {mainItems?.map(({ key, icon, label, type, onClick }: any, index) => {
-        if (type === 'divider') return <Divider key={index} />;
-        return <Cell icon={icon} key={key} label={label} onClick={onClick} />;
-      })}
-    </Flexbox>
-  );
-});
-
-export default ExtraCate;

package/src/app/(main)/(mobile)/me/features/useExtraCate.tsx

@@ -1,62 +0,0 @@
-import { DiscordIcon, Icon } from '@lobehub/ui';
-import { Book, Feather, HardDriveDownload, HardDriveUpload } from 'lucide-react';
-import { useTranslation } from 'react-i18next';
-
-import { type MenuProps } from '@/components/Menu';
-import { DISCORD, DOCUMENTS, FEEDBACK } from '@/const/url';
-import DataImporter from '@/features/DataImporter';
-import { configService } from '@/services/config';
-
-export const useExtraCate = () => {
-  const { t } = useTranslation(['common', 'setting']);
-
-  const iconSize = { fontSize: 20 };
-
-  const exports: MenuProps['items'] = [
-    {
-      icon: <Icon icon={HardDriveUpload} size={iconSize} />,
-      key: 'import',
-      label: <DataImporter>{t('import')}</DataImporter>,
-    },
-    {
-      icon: <Icon icon={HardDriveDownload} size={iconSize} />,
-      key: 'export',
-      label: t('export'),
-      onClick: configService.exportAll,
-    },
-    {
-      type: 'divider',
-    },
-  ];
-
-  const helps: MenuProps['items'] = [
-    {
-      icon: <Icon icon={Book} size={iconSize} />,
-      key: 'docs',
-      label: t('document'),
-      onClick: () => window.open(DOCUMENTS, '__blank'),
-    },
-    {
-      icon: <Icon icon={Feather} size={iconSize} />,
-      key: 'feedback',
-      label: t('feedback'),
-      onClick: () => window.open(FEEDBACK, '__blank'),
-    },
-    {
-      icon: <Icon icon={DiscordIcon} size={iconSize} />,
-      key: 'discord',
-      label: 'Discord',
-      onClick: () => window.open(DISCORD, '__blank'),
-    },
-  ];
-
-  const mainItems = [
-    {
-      type: 'divider',
-    },
-    ...exports,
-    ...helps,
-  ].filter(Boolean) as MenuProps['items'];
-
-  return mainItems;
-};