@lobehub/chat 0.91.0 → 0.92.0

package/src/types/chatMessage.ts

@@ -1,6 +1,7 @@
 import { PluginRequestPayload } from '@lobehub/chat-plugin-sdk';
 
 import { ErrorType } from '@/types/fetch';
+import { Translate } from '@/types/translate';
 
 import { LLMRoleType } from './llm';
 import { BaseDataModel } from './meta';
@@ -18,6 +19,9 @@ export interface OpenAIFunctionCall {
   name: string;
 }
 
+export interface ChatTranslate extends Translate {
+  content?: string;
+}
 export interface ChatMessage extends BaseDataModel {
   /**
    * @title 内容
@@ -29,10 +33,7 @@ export interface ChatMessage extends BaseDataModel {
   extra?: {
     fromModel?: string;
     // 翻译
-    translate?: {
-      target: string;
-      to: string;
-    };
+    translate?: ChatTranslate;
   } & Record<string, any>;
 
   /**

package/src/types/translate.ts

@@ -0,0 +1,4 @@
+export interface Translate {
+  from: string;
+  to: string;
+}

package/src/prompts/chat.ts

@@ -1,32 +0,0 @@
-import { chatHelpers } from '@/store/session/helpers';
-import { LanguageModel } from '@/types/llm';
-import { OpenAIChatMessage, OpenAIChatStreamPayload } from '@/types/openai/chat';
-
-export const promptSummaryTitle = async (
-  messages: OpenAIChatMessage[],
-): Promise<Partial<OpenAIChatStreamPayload>> => {
-  const finalMessages: OpenAIChatMessage[] = [
-    {
-      content:
-        '你是一名擅长会话的助理，你需要将用户的会话总结为 10 个字以内的标题，不需要包含标点符号',
-      role: 'system',
-    },
-    {
-      content: `${messages.map((message) => `${message.role}: ${message.content}`).join('\n')}
-
-请总结上述对话为10个字以内的标题，不需要包含标点符号`,
-      role: 'user',
-    },
-  ];
-  // 如果超过 4k，则使用 GPT3.5 16K 模型
-  const tokens = await chatHelpers.getMessagesTokenCount(finalMessages);
-  let model: LanguageModel | undefined = undefined;
-  if (tokens > 4000) {
-    model = LanguageModel.GPT3_5_16K;
-  }
-
-  return {
-    messages: finalMessages,
-    model,
-  };
-};