@lobehub/chat 1.114.0 → 1.114.2

package/{src → packages/prompts/src}/chains/__tests__/summaryDescription.test.ts

@@ -1,23 +1,14 @@
-import { Mock, describe, expect, it, vi } from 'vitest';
-
-import { globalHelpers } from '@/store/global/helpers';
+import { describe, expect, it } from 'vitest';
 
 import { chainSummaryDescription } from '../summaryDescription';
 
-// Mock the globalHelpers.getCurrentLanguage function
-vi.mock('@/store/global/helpers', () => ({
-  globalHelpers: {
-    getCurrentLanguage: vi.fn(() => 'en-US'),
-  },
-}));
-
 describe('chainSummaryDescription', () => {
   it('should return a payload with the correct structure and content based on the input', () => {
     const inputContent = '你是一个专业的JavaScript开发者';
     const expectedOutput = {
       messages: [
         {
-          content: `你是一名擅长技能总结的助理，你需要将用户的输入的内容总结为一个角色技能简介，不超过 20 个字。内容需要确保信息清晰、逻辑清晰，并有效地传达角色的技能和经验，需要并翻译为目标语言:${globalHelpers.getCurrentLanguage()}。格式要求如下：\n输入: {文本作为JSON引用字符串} [locale]\n输出: {简介}`,
+          content: `你是一名擅长技能总结的助理，你需要将用户的输入的内容总结为一个角色技能简介，不超过 20 个字。内容需要确保信息清晰、逻辑清晰，并有效地传达角色的技能和经验，需要并翻译为目标语言:en-US。格式要求如下：\n输入: {文本作为JSON引用字符串} [locale]\n输出: {简介}`,
           role: 'system',
         },
         {
@@ -64,17 +55,14 @@ nav: API
       temperature: 0,
     };
 
-    const result = chainSummaryDescription(inputContent);
+    const result = chainSummaryDescription(inputContent, 'en-US');
 
     expect(result).toEqual(expectedOutput);
   });
 
   it('should handle different languages', () => {
-    // Override the getCurrentLanguage mock to return a different language
-    (globalHelpers.getCurrentLanguage as Mock).mockReturnValueOnce('zh-CN');
-
     const inputContent = '你是一个专业的JavaScript开发者';
-    const result = chainSummaryDescription(inputContent);
+    const result = chainSummaryDescription(inputContent, 'zh-CN');
 
     // Verify that the language in the output matches the mocked language
     expect(result.messages![0].content).toContain('翻译为目标语言:zh-CN');

package/{src → packages/prompts/src}/chains/__tests__/summaryHistory.test.ts

@@ -1,10 +1,6 @@
+import { ChatMessage } from '@lobechat/types';
 import { Mock, describe, expect, it, vi } from 'vitest';
 
-import { chatHelpers } from '@/store/chat/helpers';
-import { globalHelpers } from '@/store/global/helpers';
-import { ChatMessage } from '@/types/message';
-import { OpenAIChatMessage } from '@/types/openai/chat';
-
 import { chainSummaryHistory } from '../summaryHistory';
 
 describe('chainSummaryHistory', () => {

package/{src → packages/prompts/src}/chains/__tests__/summaryTags.test.ts

@@ -1,25 +1,15 @@
 import { Mock, describe, expect, it } from 'vitest';
 
-import { globalHelpers } from '@/store/global/helpers';
-
 import { chainSummaryTags } from '../summaryTags';
 
-// Mock the getCurrentLanguage function
-vi.mock('@/store/global/helpers', () => ({
-  globalHelpers: {
-    getCurrentLanguage: vi.fn(),
-  },
-}));
-
 describe('chainSummaryTags', () => {
   it('should create a payload with system and user messages including the provided content and current language', () => {
     // Arrange
     const content = '这是一段测试文本';
     const currentLanguage = 'en-US';
-    (globalHelpers.getCurrentLanguage as Mock).mockReturnValue(currentLanguage);
 
     // Act
-    const result = chainSummaryTags(content);
+    const result = chainSummaryTags(content, currentLanguage);
 
     // Assert
     expect(result).toEqual({
@@ -47,8 +37,5 @@ describe('chainSummaryTags', () => {
         { content: `输入: {${content}} [${currentLanguage}]`, role: 'user' },
       ],
     });
-
-    // Verify that the getCurrentLanguage function was called
-    expect(globalHelpers.getCurrentLanguage).toHaveBeenCalled();
   });
 });

package/{src → packages/prompts/src}/chains/__tests__/summaryTitle.test.ts

@@ -1,25 +1,8 @@
-import { Mock, describe, expect, it, vi } from 'vitest';
-
-import { chatHelpers } from '@/store/chat/helpers';
-import { globalHelpers } from '@/store/global/helpers';
-import { OpenAIChatMessage } from '@/types/openai/chat';
+import { OpenAIChatMessage } from '@lobechat/types';
+import { describe, expect, it, vi } from 'vitest';
 
 import { chainSummaryTitle } from '../summaryTitle';
 
-// Mock the getCurrentLanguage function
-vi.mock('@/store/global/helpers', () => ({
-  globalHelpers: {
-    getCurrentLanguage: vi.fn(),
-  },
-}));
-
-// Mock the chatHelpers.getMessagesTokenCount function
-vi.mock('@/store/chat/helpers', () => ({
-  chatHelpers: {
-    getMessagesTokenCount: vi.fn(),
-  },
-}));
-
 describe('chainSummaryTitle', () => {
   it('should use the default model if the token count is below the GPT-3.5 limit', async () => {
     // Arrange
@@ -29,11 +12,9 @@ describe('chainSummaryTitle', () => {
     ];
     const currentLanguage = 'en-US';
     const tokenCount = 10000; // Arbitrary token count below the GPT-3.5 limit
-    (globalHelpers.getCurrentLanguage as Mock).mockReturnValue(currentLanguage);
-    (chatHelpers.getMessagesTokenCount as Mock).mockResolvedValue(tokenCount);
 
     // Act
-    const result = await chainSummaryTitle(messages);
+    const result = await chainSummaryTitle(messages, currentLanguage);
 
     // Assert
     expect(result).toEqual({

package/{src → packages/prompts/src}/chains/abstractChunk.ts

@@ -1,5 +1,5 @@
-import { DEFAULT_MODEL } from '@/const/settings';
-import { ChatStreamPayload } from '@/types/openai/chat';
+import { DEFAULT_MODEL } from '@lobechat/const';
+import { ChatStreamPayload } from '@lobechat/types';
 
 export const chainAbstractChunkText = (text: string): Partial<ChatStreamPayload> => {
   return {

package/{src → packages/prompts/src}/chains/answerWithContext.ts

@@ -1,4 +1,4 @@
-import { ChatStreamPayload } from '@/types/openai/chat';
+import { ChatStreamPayload } from '@lobechat/types';
 
 export const chainAnswerWithContext = ({
   context,

package/packages/prompts/src/chains/index.ts

@@ -0,0 +1,12 @@
+export * from './abstractChunk';
+export * from './answerWithContext';
+export * from './langDetect';
+export * from './pickEmoji';
+export * from './rewriteQuery';
+export * from './summaryAgentName';
+export * from './summaryDescription';
+export * from './summaryGenerationTitle';
+export * from './summaryHistory';
+export * from './summaryTags';
+export * from './summaryTitle';
+export * from './translate';

package/{src → packages/prompts/src}/chains/langDetect.ts

@@ -1,4 +1,4 @@
-import { ChatStreamPayload } from '@/types/openai/chat';
+import { ChatStreamPayload } from '@lobechat/types';
 
 export const chainLangDetect = (content: string): Partial<ChatStreamPayload> => ({
   messages: [

package/{src → packages/prompts/src}/chains/pickEmoji.ts

@@ -1,4 +1,4 @@
-import { ChatStreamPayload } from '@/types/openai/chat';
+import { ChatStreamPayload } from '@lobechat/types';
 
 /**
  * pick emoji for user prompt

package/{src → packages/prompts/src}/chains/rewriteQuery.ts

@@ -1,5 +1,5 @@
-import { DEFAULT_REWRITE_QUERY } from '@/const/settings';
-import { ChatStreamPayload } from '@/types/openai/chat';
+import { DEFAULT_REWRITE_QUERY } from '@lobechat/const';
+import { ChatStreamPayload } from '@lobechat/types';
 
 export const chainRewriteQuery = (
   query: string,

package/{src → packages/prompts/src}/chains/summaryAgentName.ts

@@ -1,11 +1,12 @@
-import { globalHelpers } from '@/store/global/helpers';
-import { ChatStreamPayload } from '@/types/openai/chat';
+import { ChatStreamPayload } from '@lobechat/types';
 
 /**
  * summary agent name for user prompt
- * @param content
  */
-export const chainSummaryAgentName = (content: string): Partial<ChatStreamPayload> => ({
+export const chainSummaryAgentName = (
+  content: string,
+  locale: string,
+): Partial<ChatStreamPayload> => ({
   messages: [
     {
       content: `你是一名擅长起名的起名大师，名字需要有文学内涵，注重精炼和赋子意境，你需要将用户的描述总结为 10 个字以内的角色，并翻译为目标语言。格式要求如下：\n输入: {文本作为JSON引用字符串} [locale]\n输出: {角色名}`,
@@ -30,6 +31,6 @@ export const chainSummaryAgentName = (content: string): Partial<ChatStreamPayloa
       role: 'user',
     },
     { content: '邮件优化助理', role: 'assistant' },
-    { content: `输入: {${content}} [${globalHelpers.getCurrentLanguage()}]`, role: 'user' },
+    { content: `输入: {${content}} [${locale}]`, role: 'user' },
   ],
 });

package/{src → packages/prompts/src}/chains/summaryDescription.ts

@@ -1,10 +1,12 @@
-import { globalHelpers } from '@/store/global/helpers';
-import { ChatStreamPayload } from '@/types/openai/chat';
+import { ChatStreamPayload } from '@lobechat/types';
 
-export const chainSummaryDescription = (content: string): Partial<ChatStreamPayload> => ({
+export const chainSummaryDescription = (
+  content: string,
+  locale: string,
+): Partial<ChatStreamPayload> => ({
   messages: [
     {
-      content: `你是一名擅长技能总结的助理，你需要将用户的输入的内容总结为一个角色技能简介，不超过 20 个字。内容需要确保信息清晰、逻辑清晰，并有效地传达角色的技能和经验，需要并翻译为目标语言:${globalHelpers.getCurrentLanguage()}。格式要求如下：\n输入: {文本作为JSON引用字符串} [locale]\n输出: {简介}`,
+      content: `你是一名擅长技能总结的助理，你需要将用户的输入的内容总结为一个角色技能简介，不超过 20 个字。内容需要确保信息清晰、逻辑清晰，并有效地传达角色的技能和经验，需要并翻译为目标语言:${locale}。格式要求如下：\n输入: {文本作为JSON引用字符串} [locale]\n输出: {简介}`,
       role: 'system',
     },
     {
@@ -43,7 +45,7 @@ nav: API
       role: 'user',
     },
     { content: '擅长创业计划撰写与咨询', role: 'assistant' },
-    { content: `输入: {${content}} [${globalHelpers.getCurrentLanguage()}]`, role: 'user' },
+    { content: `输入: {${content}} [${locale}]`, role: 'user' },
   ],
   temperature: 0,
 });

package/{src → packages/prompts/src}/chains/summaryGenerationTitle.ts

@@ -1,19 +1,17 @@
-import { globalHelpers } from '@/store/global/helpers';
-import { ChatStreamPayload } from '@/types/openai/chat';
+import { ChatStreamPayload } from '@lobechat/types';
 
 export const chainSummaryGenerationTitle = (
   prompts: string[],
   modal: 'image' | 'video',
+  locale: string,
 ): Partial<ChatStreamPayload> => {
-  const lang = globalHelpers.getCurrentLanguage();
-
   // Format multiple prompts for better readability
   const formattedPrompts = prompts.map((prompt, index) => `${index + 1}. ${prompt}`).join('\n');
 
   return {
     messages: [
       {
-        content: `你是一位资深的 AI 艺术创作者和语言大师。你需要根据用户提供的 AI ${modal} prompt 总结出一个标题。这个标题应简洁地描述创作的核心内容，将用于标识和管理该系列作品。字数需控制在10个字以内，不需要包含标点符号，输出语言为：${lang}。`,
+        content: `你是一位资深的 AI 艺术创作者和语言大师。你需要根据用户提供的 AI ${modal} prompt 总结出一个标题。这个标题应简洁地描述创作的核心内容，将用于标识和管理该系列作品。字数需控制在10个字以内，不需要包含标点符号，输出语言为：${locale}。`,
         role: 'system',
       },
       {

package/{src → packages/prompts/src}/chains/summaryHistory.ts

@@ -1,7 +1,6 @@
-import { chatHistoryPrompts } from '@lobechat/prompts';
-import { ChatMessage } from '@lobechat/types';
+import { ChatMessage, ChatStreamPayload } from '@lobechat/types';
 
-import { ChatStreamPayload } from '@/types/openai/chat';
+import { chatHistoryPrompts } from '../prompts';
 
 export const chainSummaryHistory = (messages: ChatMessage[]): Partial<ChatStreamPayload> => ({
   messages: [

package/{src → packages/prompts/src}/chains/summaryTags.ts

@@ -1,7 +1,6 @@
-import { globalHelpers } from '@/store/global/helpers';
-import { ChatStreamPayload } from '@/types/openai/chat';
+import { ChatStreamPayload } from '@lobechat/types';
 
-export const chainSummaryTags = (content: string): Partial<ChatStreamPayload> => ({
+export const chainSummaryTags = (content: string, locale: string): Partial<ChatStreamPayload> => ({
   messages: [
     {
       content:
@@ -23,6 +22,6 @@ export const chainSummaryTags = (content: string): Partial<ChatStreamPayload> =>
       role: 'user',
     },
     { content: 'entrepreneurship,planning,consulting', role: 'assistant' },
-    { content: `输入: {${content}} [${globalHelpers.getCurrentLanguage()}]`, role: 'user' },
+    { content: `输入: {${content}} [${locale}]`, role: 'user' },
   ],
 });

package/{src → packages/prompts/src}/chains/summaryTitle.ts

@@ -1,9 +1,9 @@
-import { globalHelpers } from '@/store/global/helpers';
 import { ChatStreamPayload, OpenAIChatMessage } from '@/types/openai/chat';
 
-export const chainSummaryTitle = (messages: OpenAIChatMessage[]): Partial<ChatStreamPayload> => {
-  const lang = globalHelpers.getCurrentLanguage();
-
+export const chainSummaryTitle = (
+  messages: OpenAIChatMessage[],
+  locale: string,
+): Partial<ChatStreamPayload> => {
   return {
     messages: [
       {
@@ -13,7 +13,7 @@ export const chainSummaryTitle = (messages: OpenAIChatMessage[]): Partial<ChatSt
       {
         content: `${messages.map((message) => `${message.role}: ${message.content}`).join('\n')}
 
-请总结上述对话为10个字以内的标题，不需要包含标点符号，输出语言语种为：${lang}`,
+请总结上述对话为10个字以内的标题，不需要包含标点符号，输出语言语种为：${locale}`,
         role: 'user',
       },
     ],

package/{src → packages/prompts/src}/chains/translate.ts

@@ -1,4 +1,4 @@
-import { ChatStreamPayload } from '@/types/openai/chat';
+import { ChatStreamPayload } from '@lobechat/types';
 
 export const chainTranslate = (
   content: string,

package/packages/prompts/src/index.ts

@@ -1,5 +1,2 @@
-export * from './chatMessages';
-export * from './files';
-export * from './knowledgeBaseQA';
-export * from './plugin';
-export * from './systemRole';
+export * from './chains';
+export * from './prompts';

package/packages/prompts/src/{files → prompts/files}/index.test.ts

@@ -1,7 +1,6 @@
+import { ChatFileItem, ChatImageItem } from '@lobechat/types';
 import { describe, expect, it } from 'vitest';
 
-import { ChatFileItem, ChatImageItem } from '@/types/message';
-
 import { filesPrompts } from './index';
 
 describe('filesPrompts', () => {

package/packages/prompts/src/prompts/index.ts

@@ -0,0 +1,5 @@
+export * from './chatMessages';
+export * from './files';
+export * from './knowledgeBaseQA';
+export * from './plugin';
+export * from './systemRole';

package/packages/prompts/src/{knowledgeBaseQA → prompts/knowledgeBaseQA}/index.test.ts

@@ -1,8 +1,6 @@
+import { ChatSemanticSearchChunk, KnowledgeItem, KnowledgeType } from '@lobechat/types';
 import { describe, expect, it } from 'vitest';
 
-import { ChatSemanticSearchChunk } from '@/types/chunk';
-import { KnowledgeItem, KnowledgeType } from '@/types/knowledgeBase';
-
 import { knowledgeBaseQAPrompts } from './index';
 
 describe('knowledgeBaseQAPrompts', () => {

package/packages/prompts/src/{plugin → prompts/plugin}/index.test.ts

@@ -1,3 +1,5 @@
+import { describe, expect, it } from 'vitest';
+
 import { pluginPrompts } from './index';
 import { Tool } from './tools';
 

package/packages/prompts/vitest.config.ts

@@ -0,0 +1,7 @@
+import { defineConfig } from 'vitest/config';
+
+export default defineConfig({
+  test: {
+    environment: 'happy-dom',
+  },
+});

package/packages/types/src/index.ts

@@ -1,6 +1,10 @@
 export * from './artifact';
 export * from './chunk';
+export * from './clientDB';
 export * from './fetch';
 export * from './knowledgeBase';
 export * from './message';
+// FIXME: I think we need a refactor for the "openai" types
+// it more likes the UI message payload
+export * from './openai/chat';
 export * from './trace';

package/packages/web-crawler/package.json

@@ -4,6 +4,10 @@
   "private": true,
   "main": "src/index.ts",
   "types": "src/index.ts",
+  "scripts": {
+    "test": "vitest",
+    "test:coverage": "vitest --coverage"
+  },
   "dependencies": {
     "@mozilla/readability": "^0.6.0",
     "happy-dom": "^17.0.0",

package/packages/web-crawler/src/crawImpl/__tests__/jina.test.ts

@@ -1,4 +1,4 @@
-import { describe, expect, it, vi } from 'vitest';
+import { beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { jina } from '../jina';
 

package/packages/web-crawler/src/crawImpl/__tests__/search1api.test.ts

@@ -1,4 +1,4 @@
-import { describe, expect, it, vi } from 'vitest';
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { NetworkConnectionError, PageNotFoundError, TimeoutError } from '../../utils/errorType';
 import * as withTimeoutModule from '../../utils/withTimeout';

package/packages/web-crawler/src/utils/__tests__/withTimeout.test.ts

@@ -1,4 +1,4 @@
-import { describe, expect, it, vi } from 'vitest';
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { TimeoutError } from '../errorType';
 import { DEFAULT_TIMEOUT, withTimeout } from '../withTimeout';

package/packages/web-crawler/src/utils/appUrlRules.test.ts

@@ -1,3 +1,5 @@
+import { describe, expect, it } from 'vitest';
+
 import { applyUrlRules } from './appUrlRules';
 
 describe('applyUrlRules', () => {

package/packages/web-crawler/src/utils/htmlToMarkdown.test.ts

@@ -1,6 +1,6 @@
 import { readFileSync } from 'node:fs';
 import * as path from 'node:path';
-import { expect } from 'vitest';
+import { describe, expect, it } from 'vitest';
 
 import { FilterOptions } from '../type';
 import { htmlToMarkdown } from './htmlToMarkdown';

package/packages/web-crawler/vitest.config.ts

@@ -0,0 +1,7 @@
+import { defineConfig } from 'vitest/config';
+
+export default defineConfig({
+  test: {
+    environment: 'node',
+  },
+});

package/src/app/[variants]/(main)/settings/provider/(detail)/[id]/ClientMode.tsx

@@ -7,11 +7,15 @@ import { Flexbox } from 'react-layout-kit';
 import Loading from '@/components/Loading/BrandTextLoading';
 import { useClientDataSWR } from '@/libs/swr';
 import { aiProviderService } from '@/services/aiProvider';
+import { useAiInfraStore } from '@/store/aiInfra';
 
 import ModelList from '../../features/ModelList';
 import ProviderConfig from '../../features/ProviderConfig';
 
 const ClientMode = memo<{ id: string }>(({ id }) => {
+  const useFetchAiProviderItem = useAiInfraStore((s) => s.useFetchAiProviderItem);
+  useFetchAiProviderItem(id);
+
   const { data, isLoading } = useClientDataSWR('get-client-provider', () =>
     aiProviderService.getAiProviderById(id),
   );

package/src/features/AgentSetting/store/action.ts

@@ -1,13 +1,16 @@
+import {
+  chainPickEmoji,
+  chainSummaryAgentName,
+  chainSummaryDescription,
+  chainSummaryTags,
+} from '@lobechat/prompts';
 import { TraceNameMap, TracePayload, TraceTopicType } from '@lobechat/types';
 import { getSingletonAnalyticsOptional } from '@lobehub/analytics';
 import type { PartialDeep } from 'type-fest';
 import { StateCreator } from 'zustand/vanilla';
 
-import { chainPickEmoji } from '@/chains/pickEmoji';
-import { chainSummaryAgentName } from '@/chains/summaryAgentName';
-import { chainSummaryDescription } from '@/chains/summaryDescription';
-import { chainSummaryTags } from '@/chains/summaryTags';
 import { chatService } from '@/services/chat';
+import { globalHelpers } from '@/store/global/helpers';
 import { useUserStore } from '@/store/user';
 import { systemAgentSelectors } from '@/store/user/slices/settings/selectors';
 import { LobeAgentChatConfig, LobeAgentConfig } from '@/types/agent';
@@ -118,7 +121,10 @@ export const store: StateCreator<Store, [['zustand/devtools', never]]> = (set, g
         updateLoadingState('description', loading);
       },
       onMessageHandle: streamUpdateMetaString('description'),
-      params: merge(get().internal_getSystemAgentForMeta(), chainSummaryDescription(systemRole)),
+      params: merge(
+        get().internal_getSystemAgentForMeta(),
+        chainSummaryDescription(systemRole, globalHelpers.getCurrentLanguage()),
+      ),
       trace: get().getCurrentTracePayload({ traceName: TraceNameMap.SummaryAgentDescription }),
     });
   },
@@ -145,7 +151,10 @@ export const store: StateCreator<Store, [['zustand/devtools', never]]> = (set, g
       onMessageHandle: streamUpdateMetaArray('tags'),
       params: merge(
         get().internal_getSystemAgentForMeta(),
-        chainSummaryTags([meta.title, meta.description, systemRole].filter(Boolean).join(',')),
+        chainSummaryTags(
+          [meta.title, meta.description, systemRole].filter(Boolean).join(','),
+          globalHelpers.getCurrentLanguage(),
+        ),
       ),
       trace: get().getCurrentTracePayload({ traceName: TraceNameMap.SummaryAgentTags }),
     });
@@ -172,7 +181,10 @@ export const store: StateCreator<Store, [['zustand/devtools', never]]> = (set, g
       onMessageHandle: streamUpdateMetaString('title'),
       params: merge(
         get().internal_getSystemAgentForMeta(),
-        chainSummaryAgentName([meta.description, systemRole].filter(Boolean).join(',')),
+        chainSummaryAgentName(
+          [meta.description, systemRole].filter(Boolean).join(','),
+          globalHelpers.getCurrentLanguage(),
+        ),
       ),
       trace: get().getCurrentTracePayload({ traceName: TraceNameMap.SummaryAgentTitle }),
     });

package/src/server/routers/async/ragEval.ts

@@ -1,9 +1,9 @@
 import { ModelProvider } from '@lobechat/model-runtime';
+import { chainAnswerWithContext } from '@lobechat/prompts';
 import { TRPCError } from '@trpc/server';
 import OpenAI from 'openai';
 import { z } from 'zod';
 
-import { chainAnswerWithContext } from '@/chains/answerWithContext';
 import { DEFAULT_EMBEDDING_MODEL, DEFAULT_MODEL } from '@/const/settings';
 import { ChunkModel } from '@/database/models/chunk';
 import { EmbeddingModel } from '@/database/models/embedding';

package/src/store/chat/slices/aiChat/actions/memory.ts

@@ -1,7 +1,7 @@
+import { chainSummaryHistory } from '@lobechat/prompts';
 import { TraceNameMap } from '@lobechat/types';
 import { StateCreator } from 'zustand/vanilla';
 
-import { chainSummaryHistory } from '@/chains/summaryHistory';
 import { chatService } from '@/services/chat';
 import { topicService } from '@/services/topic';
 import { ChatStore } from '@/store/chat';

package/src/store/chat/slices/aiChat/actions/rag.ts

@@ -1,6 +1,6 @@
+import { chainRewriteQuery } from '@lobechat/prompts';
 import { StateCreator } from 'zustand/vanilla';
 
-import { chainRewriteQuery } from '@/chains/rewriteQuery';
 import { chatService } from '@/services/chat';
 import { ragService } from '@/services/rag';
 import { useAgentStore } from '@/store/agent';

package/src/store/chat/slices/thread/action.ts

@@ -1,10 +1,10 @@
 /* eslint-disable sort-keys-fix/sort-keys-fix, typescript-sort-keys/interface */
 // Disable the auto sort key eslint rule to make the code more logic and readable
+import { chainSummaryTitle } from '@lobechat/prompts';
 import isEqual from 'fast-deep-equal';
 import { SWRResponse, mutate } from 'swr';
 import { StateCreator } from 'zustand/vanilla';
 
-import { chainSummaryTitle } from '@/chains/summaryTitle';
 import { LOADING_FLAT, THREAD_DRAFT_ID } from '@/const/message';
 import { isDeprecatedEdition } from '@/const/version';
 import { useClientDataSWR } from '@/libs/swr';
@@ -12,6 +12,7 @@ import { chatService } from '@/services/chat';
 import { threadService } from '@/services/thread';
 import { threadSelectors } from '@/store/chat/selectors';
 import { ChatStore } from '@/store/chat/store';
+import { globalHelpers } from '@/store/global/helpers';
 import { useSessionStore } from '@/store/session';
 import { useUserStore } from '@/store/user';
 import { systemAgentSelectors } from '@/store/user/selectors';
@@ -280,7 +281,7 @@ export const chatThreadMessage: StateCreator<
 
         internal_updateThreadTitleInSummary(threadId, output);
       },
-      params: merge(threadConfig, chainSummaryTitle(messages)),
+      params: merge(threadConfig, chainSummaryTitle(messages, globalHelpers.getCurrentLanguage())),
     });
   },
 

package/src/store/chat/slices/topic/action.ts

@@ -1,13 +1,13 @@
 /* eslint-disable sort-keys-fix/sort-keys-fix, typescript-sort-keys/interface */
 // Note: To make the code more logic and readable, we just disable the auto sort key eslint rule
 // DON'T REMOVE THE FIRST LINE
+import { chainSummaryTitle } from '@lobechat/prompts';
 import { TraceNameMap } from '@lobechat/types';
 import isEqual from 'fast-deep-equal';
 import { t } from 'i18next';
 import useSWR, { SWRResponse, mutate } from 'swr';
 import { StateCreator } from 'zustand/vanilla';
 
-import { chainSummaryTitle } from '@/chains/summaryTitle';
 import { message } from '@/components/AntdStaticMethods';
 import { LOADING_FLAT } from '@/const/message';
 import { useClientDataSWR } from '@/libs/swr';
@@ -16,6 +16,7 @@ import { messageService } from '@/services/message';
 import { topicService } from '@/services/topic';
 import { CreateTopicParams } from '@/services/topic/type';
 import type { ChatStore } from '@/store/chat';
+import { globalHelpers } from '@/store/global/helpers';
 import { useUserStore } from '@/store/user';
 import { systemAgentSelectors } from '@/store/user/selectors';
 import { ChatMessage } from '@/types/message';
@@ -167,7 +168,7 @@ export const chatTopic: StateCreator<
 
         internal_updateTopicTitleInSummary(topicId, output);
       },
-      params: merge(topicConfig, chainSummaryTitle(messages)),
+      params: merge(topicConfig, chainSummaryTitle(messages, globalHelpers.getCurrentLanguage())),
       trace: get().getCurrentTracePayload({ traceName: TraceNameMap.SummaryTopicTitle, topicId }),
     });
   },

package/src/store/chat/slices/translate/action.test.ts

@@ -1,8 +1,8 @@
+import { chainLangDetect } from '@lobechat/prompts';
+import { chainTranslate } from '@lobechat/prompts';
 import { act, renderHook } from '@testing-library/react';
 import { Mock, afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { chainLangDetect } from '@/chains/langDetect';
-import { chainTranslate } from '@/chains/translate';
 import { chatService } from '@/services/chat';
 import { messageService } from '@/services/message';
 import { messageMapKey } from '@/store/chat/utils/messageMapKey';