@lobehub/chat 1.134.3 → 1.134.4

package/packages/prompts/src/chains/__tests__/__snapshots__/answerWithContext.test.ts.snap

@@ -0,0 +1,164 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`chainAnswerWithContext > should filter out empty context strings 1`] = `
+{
+  "messages": [
+    {
+      "content": "You are a helpful assistant specialized in Test. Your task is to answer questions based on the provided context passages.
+
+IMPORTANT RULES:
+- First, check if the context is relevant to the question topic
+- If the context is about a COMPLETELY DIFFERENT topic than the question:
+  * State what topic the context is about
+  * Clearly state "The provided context does not contain information about [question topic]"
+  * Do NOT answer using your general knowledge
+- If the context is related to the question topic (even if information is limited):
+  * ALWAYS use the context information as a foundation
+  * You SHOULD supplement with your general knowledge to provide a complete, helpful answer
+  * For "how to" questions, MUST provide practical, actionable steps combining context + your expertise
+  * The context provides the topic relevance - you provide the comprehensive answer
+  * Example: If context mentions "Docker is for containerization", and question is "How to deploy with Docker?", you should explain deployment steps using your knowledge
+- Answer in the same language as the question
+- Use markdown formatting for better readability
+
+The provided context passages:
+
+<context>
+Valid context
+Another valid context
+</context>
+
+Question to answer:
+
+Test question",
+      "role": "user",
+    },
+  ],
+}
+`;
+
+exports[`chainAnswerWithContext > should generate correct payload with context and knowledge 1`] = `
+{
+  "messages": [
+    {
+      "content": "You are a helpful assistant specialized in AI/Machine Learning. Your task is to answer questions based on the provided context passages.
+
+IMPORTANT RULES:
+- First, check if the context is relevant to the question topic
+- If the context is about a COMPLETELY DIFFERENT topic than the question:
+  * State what topic the context is about
+  * Clearly state "The provided context does not contain information about [question topic]"
+  * Do NOT answer using your general knowledge
+- If the context is related to the question topic (even if information is limited):
+  * ALWAYS use the context information as a foundation
+  * You SHOULD supplement with your general knowledge to provide a complete, helpful answer
+  * For "how to" questions, MUST provide practical, actionable steps combining context + your expertise
+  * The context provides the topic relevance - you provide the comprehensive answer
+  * Example: If context mentions "Docker is for containerization", and question is "How to deploy with Docker?", you should explain deployment steps using your knowledge
+- Answer in the same language as the question
+- Use markdown formatting for better readability
+
+The provided context passages:
+
+<context>
+Context passage 1
+Context passage 2
+</context>
+
+Question to answer:
+
+What is artificial intelligence?",
+      "role": "user",
+    },
+  ],
+}
+`;
+
+exports[`chainAnswerWithContext > should handle empty context array 1`] = `
+{
+  "messages": [
+    {
+      "content": "You are a helpful assistant specialized in AI. Please answer the following question using your knowledge.
+
+Answer in the same language as the question and use markdown formatting for better readability.
+
+Question to answer:
+
+What is AI?",
+      "role": "user",
+    },
+  ],
+}
+`;
+
+exports[`chainAnswerWithContext > should handle multiple knowledge areas 1`] = `
+{
+  "messages": [
+    {
+      "content": "You are a helpful assistant specialized in AI/ML/NLP/Computer Vision. Your task is to answer questions based on the provided context passages.
+
+IMPORTANT RULES:
+- First, check if the context is relevant to the question topic
+- If the context is about a COMPLETELY DIFFERENT topic than the question:
+  * State what topic the context is about
+  * Clearly state "The provided context does not contain information about [question topic]"
+  * Do NOT answer using your general knowledge
+- If the context is related to the question topic (even if information is limited):
+  * ALWAYS use the context information as a foundation
+  * You SHOULD supplement with your general knowledge to provide a complete, helpful answer
+  * For "how to" questions, MUST provide practical, actionable steps combining context + your expertise
+  * The context provides the topic relevance - you provide the comprehensive answer
+  * Example: If context mentions "Docker is for containerization", and question is "How to deploy with Docker?", you should explain deployment steps using your knowledge
+- Answer in the same language as the question
+- Use markdown formatting for better readability
+
+The provided context passages:
+
+<context>
+Context
+</context>
+
+Question to answer:
+
+Tell me about these fields",
+      "role": "user",
+    },
+  ],
+}
+`;
+
+exports[`chainAnswerWithContext > should handle single knowledge area 1`] = `
+{
+  "messages": [
+    {
+      "content": "You are a helpful assistant specialized in Technology. Your task is to answer questions based on the provided context passages.
+
+IMPORTANT RULES:
+- First, check if the context is relevant to the question topic
+- If the context is about a COMPLETELY DIFFERENT topic than the question:
+  * State what topic the context is about
+  * Clearly state "The provided context does not contain information about [question topic]"
+  * Do NOT answer using your general knowledge
+- If the context is related to the question topic (even if information is limited):
+  * ALWAYS use the context information as a foundation
+  * You SHOULD supplement with your general knowledge to provide a complete, helpful answer
+  * For "how to" questions, MUST provide practical, actionable steps combining context + your expertise
+  * The context provides the topic relevance - you provide the comprehensive answer
+  * Example: If context mentions "Docker is for containerization", and question is "How to deploy with Docker?", you should explain deployment steps using your knowledge
+- Answer in the same language as the question
+- Use markdown formatting for better readability
+
+The provided context passages:
+
+<context>
+Single context
+</context>
+
+Question to answer:
+
+How does it work?",
+      "role": "user",
+    },
+  ],
+}
+`;

package/packages/prompts/src/chains/__tests__/__snapshots__/pickEmoji.test.ts.snap

@@ -0,0 +1,58 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`chainPickEmoji > should return a payload with the correct structure and embedded user content 1`] = `
+{
+  "messages": [
+    {
+      "content": "You are an emoji expert who selects the most appropriate emoji to represent concepts, emotions, or topics.
+
+Rules:
+- Output ONLY a single emoji (1-2 characters maximum)
+- Focus on the CONTENT meaning, not the language it's written in
+- Choose an emoji that best represents the core topic, activity, or subject matter
+- Prioritize topic-specific emojis over generic emotion emojis (e.g., for sports, use 🏃 instead of 😅)
+- For work/projects, use work-related emojis (💼, 🚀, 💪) not cultural symbols
+- For pure emotions without specific topics, use face emojis (happy: 🎉, sad: 😢, thinking: 🤔)
+- For activities or subjects, use object or symbol emojis that represent the main topic
+- No explanations or additional text",
+      "role": "system",
+    },
+    {
+      "content": "I am a copywriting master who helps name design and art works with literary depth",
+      "role": "user",
+    },
+    {
+      "content": "✒️",
+      "role": "assistant",
+    },
+    {
+      "content": "I am a code wizard who converts JavaScript code to TypeScript",
+      "role": "user",
+    },
+    {
+      "content": "🧙‍♂️",
+      "role": "assistant",
+    },
+    {
+      "content": "I just got a promotion at work",
+      "role": "user",
+    },
+    {
+      "content": "🎉",
+      "role": "assistant",
+    },
+    {
+      "content": "I am a business plan expert who helps with startup strategies and marketing",
+      "role": "user",
+    },
+    {
+      "content": "🚀",
+      "role": "assistant",
+    },
+    {
+      "content": "你是一名星际探索者",
+      "role": "user",
+    },
+  ],
+}
+`;

package/packages/prompts/src/chains/__tests__/__snapshots__/summaryTitle.test.ts.snap

@@ -0,0 +1,26 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`chainSummaryTitle > should use the default model if the token count is below the GPT-3.5 limit 1`] = `
+{
+  "messages": [
+    {
+      "content": "You are a professional conversation summarizer. Generate a concise title that captures the essence of the conversation.
+
+Rules:
+- Output ONLY the title text, no explanations or additional context
+- Maximum 10 words
+- Maximum 50 characters
+- No punctuation marks
+- Use the language specified by the locale code: en-US
+- The title should accurately reflect the main topic of the conversation
+- Keep it short and to the point",
+      "role": "system",
+    },
+    {
+      "content": "assistant: Hello, how can I assist you?
+user: I need help with my account.",
+      "role": "user",
+    },
+  ],
+}
+`;

package/packages/prompts/src/chains/__tests__/__snapshots__/translate.test.ts.snap

@@ -0,0 +1,22 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`chainTranslate > should create a payload with system and user messages for translation 1`] = `
+{
+  "messages": [
+    {
+      "content": "You are a professional translator. Translate the input text to zh-CN.
+
+Rules:
+- Output ONLY the translated text, no explanations or additional context
+- Preserve technical terms, code identifiers, API keys, and proper nouns exactly as they appear
+- Maintain the original formatting and structure
+- Use natural, idiomatic expressions in the target language",
+      "role": "system",
+    },
+    {
+      "content": "This is a test sentence.",
+      "role": "user",
+    },
+  ],
+}
+`;

package/packages/prompts/src/chains/__tests__/answerWithContext.test.ts

@@ -3,98 +3,53 @@ import { describe, expect, it } from 'vitest';
 import { chainAnswerWithContext } from '../answerWithContext';
 
 describe('chainAnswerWithContext', () => {
-  it('should generate correct chat payload with context and knowledge', () => {
-    const testParams = {
+  it('should generate correct payload with context and knowledge', () => {
+    const result = chainAnswerWithContext({
       context: ['Context passage 1', 'Context passage 2'],
       knowledge: ['AI', 'Machine Learning'],
       question: 'What is artificial intelligence?',
-    };
+    });
 
-    const result = chainAnswerWithContext(testParams);
-
-    expect(result.messages).toHaveLength(1);
-    expect(result.messages![0].role).toBe('user');
-    expect(result.messages![0].content).toContain('AI/Machine Learning');
-    expect(result.messages![0].content).toContain('Context passage 1');
-    expect(result.messages![0].content).toContain('Context passage 2');
-    expect(result.messages![0].content).toContain('What is artificial intelligence?');
+    expect(result).toMatchSnapshot();
   });
 
   it('should handle single knowledge area', () => {
-    const testParams = {
+    const result = chainAnswerWithContext({
       context: ['Single context'],
       knowledge: ['Technology'],
       question: 'How does it work?',
-    };
-
-    const result = chainAnswerWithContext(testParams);
+    });
 
-    expect(result.messages![0].content).toContain('Technology');
+    expect(result).toMatchSnapshot();
   });
 
   it('should handle multiple knowledge areas', () => {
-    const testParams = {
+    const result = chainAnswerWithContext({
       context: ['Context'],
       knowledge: ['AI', 'ML', 'NLP', 'Computer Vision'],
       question: 'Tell me about these fields',
-    };
+    });
 
-    const result = chainAnswerWithContext(testParams);
-
-    expect(result.messages![0].content).toContain('AI/ML/NLP/Computer Vision');
+    expect(result).toMatchSnapshot();
   });
 
   it('should handle empty context array', () => {
-    const testParams = {
+    const result = chainAnswerWithContext({
       context: [],
       knowledge: ['AI'],
       question: 'What is AI?',
-    };
-
-    const result = chainAnswerWithContext(testParams);
+    });
 
-    expect(result.messages![0].content).toContain('<Context>');
-    expect(result.messages![0].content).toContain('</Context>');
+    expect(result).toMatchSnapshot();
   });
 
-  it('should include proper context formatting', () => {
-    const testParams = {
-      context: ['First passage', 'Second passage'],
+  it('should filter out empty context strings', () => {
+    const result = chainAnswerWithContext({
+      context: ['Valid context', '', '  ', 'Another valid context'],
       knowledge: ['Test'],
       question: 'Test question',
-    };
-
-    const result = chainAnswerWithContext(testParams);
-
-    expect(result.messages![0].content).toContain(
-      '<Context>\nFirst passage\nSecond passage\n</Context>',
-    );
-  });
-
-  it('should include proper instructions about using passages', () => {
-    const testParams = {
-      context: ['Context'],
-      knowledge: ['Knowledge'],
-      question: 'Question',
-    };
-
-    const result = chainAnswerWithContext(testParams);
-    const content = result.messages![0].content;
-
-    expect(content).toContain('passages might not be relevant');
-    expect(content).toContain('please only use the passages that are relevant');
-    expect(content).toContain('answer using your knowledge');
-  });
-
-  it('should include markdown formatting instruction', () => {
-    const testParams = {
-      context: ['Context'],
-      knowledge: ['Knowledge'],
-      question: 'Question',
-    };
-
-    const result = chainAnswerWithContext(testParams);
+    });
 
-    expect(result.messages![0].content).toContain('follow markdown syntax');
+    expect(result).toMatchSnapshot();
   });
 });

package/packages/prompts/src/chains/__tests__/pickEmoji.test.ts

@@ -1,46 +1,11 @@
-import { ChatStreamPayload } from '@lobechat/types';
 import { describe, expect, it } from 'vitest';
 
 import { chainPickEmoji } from '../pickEmoji';
 
-// 描述测试块
 describe('chainPickEmoji', () => {
-  // 测试用例：验证函数返回的结构
   it('should return a payload with the correct structure and embedded user content', () => {
-    // 用户输入的内容
-    const userContent = '你是一名星际探索者';
+    const result = chainPickEmoji('你是一名星际探索者');
 
-    // 预期的返回值结构
-    const expectedPayload: Partial<ChatStreamPayload> = {
-      messages: [
-        {
-          content:
-            '你是一名擅长进行概念抽象的设计师与 Emoji 专家，你需要根据角色能力的描述抽象出一个表达物理实体的概念 Emoji 作为角色头像, 格式要求如下：\n输入: {文本作为JSON引用字符串}\n输出: {一个Emoji}',
-          role: 'system',
-        },
-        {
-          content: `输入: {你是一名文案大师，帮我为一些设计 / 艺术作品起名，名字需要有文学内涵，注重精炼和赋子意境，表达作品的情景氛国，使名称既简洁又富有诗意。}`,
-          role: 'user',
-        },
-        { content: '✒️', role: 'assistant' },
-        {
-          content: `输入: {你是一名代码巫师，请将下面的代码转成 ts，不要修改实现。如果原本 js 中没有定义的全局变量，需要补充 declare 的类型声明。}`,
-          role: 'user',
-        },
-        { content: '🧙‍♂️', role: 'assistant' },
-        {
-          content: `输入: {你是一名创业计划撰写专家，可以提供包括创意名称、简短的标语、目标用户画像、用户痛点、主要价值主张、销售/营销渠道、收入流、成本结构等计划生成。}`,
-          role: 'user',
-        },
-        { content: '🚀', role: 'assistant' },
-        { content: `输入: {${userContent}}`, role: 'user' },
-      ],
-    };
-
-    // 执行函数并获取结果
-    const result = chainPickEmoji(userContent);
-
-    // 断言结果是否符合预期
-    expect(result).toEqual(expectedPayload);
+    expect(result).toMatchSnapshot();
   });
 });

package/packages/prompts/src/chains/__tests__/summaryTitle.test.ts

@@ -11,25 +11,11 @@ describe('chainSummaryTitle', () => {
       { content: 'I need help with my account.', role: 'user' },
     ];
     const currentLanguage = 'en-US';
-    const tokenCount = 10000; // Arbitrary token count below the GPT-3.5 limit
 
     // Act
-    const result = await chainSummaryTitle(messages, currentLanguage);
+    const result = chainSummaryTitle(messages, currentLanguage);
 
     // Assert
-    expect(result).toEqual({
-      messages: [
-        {
-          content: '你是一名擅长会话的助理，你需要将用户的会话总结为 10 个字以内的标题',
-          role: 'system',
-        },
-        {
-          content: `assistant: Hello, how can I assist you?\nuser: I need help with my account.
-
-请总结上述对话为10个字以内的标题，不需要包含标点符号，输出语言语种为：${currentLanguage}`,
-          role: 'user',
-        },
-      ],
-    });
+    expect(result).toMatchSnapshot();
   });
 });

package/packages/prompts/src/chains/__tests__/translate.test.ts

@@ -12,17 +12,6 @@ describe('chainTranslate', () => {
     const result = chainTranslate(content, targetLang);
 
     // Assert
-    expect(result).toEqual({
-      messages: [
-        {
-          content: '你是一名擅长翻译的助理，你需要将输入的语言翻译为目标语言',
-          role: 'system',
-        },
-        {
-          content: `请将以下内容 ${content}，翻译为 ${targetLang} `,
-          role: 'user',
-        },
-      ],
-    });
+    expect(result).toMatchSnapshot();
   });
 });

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

@@ -1,33 +1,57 @@
 import { ChatStreamPayload } from '@lobechat/types';
 
+interface AnswerWithContext {
+  context: string[];
+  knowledge: string[];
+  question: string;
+}
 export const chainAnswerWithContext = ({
   context,
   knowledge,
   question,
-}: {
-  context: string[];
-  knowledge: string[];
-  question: string;
-}): Partial<ChatStreamPayload> => ({
-  messages: [
-    {
-      content: `You are also a helpful assistant good answering questions related to ${knowledge.join('/')}. And you'll be provided with a question and several passages that might be relevant. And currently your task is to provide answer based on the question and passages.
+}: AnswerWithContext): Partial<ChatStreamPayload> => {
+  const filteredContext = context.filter((c) => c.trim());
+  const hasContext = filteredContext.length > 0;
+
+  return {
+    messages: [
+      {
+        content: hasContext
+          ? `You are a helpful assistant specialized in ${knowledge.join('/')}. Your task is to answer questions based on the provided context passages.
+
+IMPORTANT RULES:
+- First, check if the context is relevant to the question topic
+- If the context is about a COMPLETELY DIFFERENT topic than the question:
+  * State what topic the context is about
+  * Clearly state "The provided context does not contain information about [question topic]"
+  * Do NOT answer using your general knowledge
+- If the context is related to the question topic (even if information is limited):
+  * ALWAYS use the context information as a foundation
+  * You SHOULD supplement with your general knowledge to provide a complete, helpful answer
+  * For "how to" questions, MUST provide practical, actionable steps combining context + your expertise
+  * The context provides the topic relevance - you provide the comprehensive answer
+  * Example: If context mentions "Docker is for containerization", and question is "How to deploy with Docker?", you should explain deployment steps using your knowledge
+- Answer in the same language as the question
+- Use markdown formatting for better readability
+
+The provided context passages:
 
-Note that passages might not be relevant to the question, please only use the passages that are relevant. Or if there is no relevant passage, please answer using your knowledge.
+<context>
+${filteredContext.join('\n')}
+</context>
 
-Answer should use the same original language as the question and follow markdown syntax.
+Question to answer:
 
-The provided passages as context:
+${question}`
+          : `You are a helpful assistant specialized in ${knowledge.join('/')}. Please answer the following question using your knowledge.
 
-<Context>
-${context.join('\n')}
-</Context>
+Answer in the same language as the question and use markdown formatting for better readability.
 
-The question to answer is:
+Question to answer:
 
-${question}
-`,
-      role: 'user',
-    },
-  ],
-});
+${question}`,
+        role: 'user',
+      },
+    ],
+  };
+};

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

@@ -7,25 +7,39 @@ import { ChatStreamPayload } from '@lobechat/types';
 export const chainPickEmoji = (content: string): Partial<ChatStreamPayload> => ({
   messages: [
     {
-      content:
-        '你是一名擅长进行概念抽象的设计师与 Emoji 专家，你需要根据角色能力的描述抽象出一个表达物理实体的概念 Emoji 作为角色头像, 格式要求如下：\n输入: {文本作为JSON引用字符串}\n输出: {一个Emoji}',
+      content: `You are an emoji expert who selects the most appropriate emoji to represent concepts, emotions, or topics.
+
+Rules:
+- Output ONLY a single emoji (1-2 characters maximum)
+- Focus on the CONTENT meaning, not the language it's written in
+- Choose an emoji that best represents the core topic, activity, or subject matter
+- Prioritize topic-specific emojis over generic emotion emojis (e.g., for sports, use 🏃 instead of 😅)
+- For work/projects, use work-related emojis (💼, 🚀, 💪) not cultural symbols
+- For pure emotions without specific topics, use face emojis (happy: 🎉, sad: 😢, thinking: 🤔)
+- For activities or subjects, use object or symbol emojis that represent the main topic
+- No explanations or additional text`,
       role: 'system',
     },
     {
-      content: `输入: {你是一名文案大师，帮我为一些设计 / 艺术作品起名，名字需要有文学内涵，注重精炼和赋子意境，表达作品的情景氛国，使名称既简洁又富有诗意。}`,
+      content: 'I am a copywriting master who helps name design and art works with literary depth',
       role: 'user',
     },
     { content: '✒️', role: 'assistant' },
     {
-      content: `输入: {你是一名代码巫师，请将下面的代码转成 ts，不要修改实现。如果原本 js 中没有定义的全局变量，需要补充 declare 的类型声明。}`,
+      content: 'I am a code wizard who converts JavaScript code to TypeScript',
       role: 'user',
     },
     { content: '🧙‍♂️', role: 'assistant' },
     {
-      content: `输入: {你是一名创业计划撰写专家，可以提供包括创意名称、简短的标语、目标用户画像、用户痛点、主要价值主张、销售/营销渠道、收入流、成本结构等计划生成。}`,
+      content: 'I just got a promotion at work',
+      role: 'user',
+    },
+    { content: '🎉', role: 'assistant' },
+    {
+      content: 'I am a business plan expert who helps with startup strategies and marketing',
       role: 'user',
     },
     { content: '🚀', role: 'assistant' },
-    { content: `输入: {${content}}`, role: 'user' },
+    { content, role: 'user' },
   ],
 });

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

@@ -3,19 +3,24 @@ import { ChatStreamPayload, OpenAIChatMessage } from '@/types/openai/chat';
 export const chainSummaryTitle = (
   messages: OpenAIChatMessage[],
   locale: string,
-): Partial<ChatStreamPayload> => {
-  return {
-    messages: [
-      {
-        content: '你是一名擅长会话的助理，你需要将用户的会话总结为 10 个字以内的标题',
-        role: 'system',
-      },
-      {
-        content: `${messages.map((message) => `${message.role}: ${message.content}`).join('\n')}
+): Partial<ChatStreamPayload> => ({
+  messages: [
+    {
+      content: `You are a professional conversation summarizer. Generate a concise title that captures the essence of the conversation.
 
-请总结上述对话为10个字以内的标题，不需要包含标点符号，输出语言语种为：${locale}`,
-        role: 'user',
-      },
-    ],
-  };
-};
+Rules:
+- Output ONLY the title text, no explanations or additional context
+- Maximum 10 words
+- Maximum 50 characters
+- No punctuation marks
+- Use the language specified by the locale code: ${locale}
+- The title should accurately reflect the main topic of the conversation
+- Keep it short and to the point`,
+      role: 'system',
+    },
+    {
+      content: messages.map((message) => `${message.role}: ${message.content}`).join('\n'),
+      role: 'user',
+    },
+  ],
+});