@lobehub/chat 1.28.6 → 1.29.0

package/src/app/(main)/settings/system-agent/index.tsx

@@ -12,6 +12,7 @@ const Page = () => {
     <>
       <SystemAgentForm systemAgentKey="topic" />
       <SystemAgentForm systemAgentKey="translation" />
+      <SystemAgentForm systemAgentKey="historyCompress" />
       <SystemAgentForm systemAgentKey="agentMeta" />
       {isServerMode && enableKnowledgeBase && (
         <SystemAgentForm

package/src/chains/__tests__/__snapshots__/summaryHistory.test.ts.snap

@@ -0,0 +1,21 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`chainSummaryHistory > should use the default model if the token count is below the GPT-3.5 limit 1`] = `
+{
+  "messages": [
+    {
+      "content": "You're an assistant who's good at extracting key takeaways from conversations and summarizing them. Please summarize according to the user's needs. The content you need to summarize is located in the <chat_history> </chat_history> group of xml tags. The summary needs to maintain the original language.",
+      "role": "system",
+    },
+    {
+      "content": "<chat_history>
+<assistant>Hello, how can I assist you?</assistant>
+<user>I need help with my account.</user>
+</chat_history>
+
+Please summarize the above conversation and retain key information. The summarized content will be used as context for subsequent prompts, and should be limited to 400 tokens.",
+      "role": "user",
+    },
+  ],
+}
+`;

package/src/chains/__tests__/summaryHistory.test.ts

@@ -0,0 +1,24 @@
+import { Mock, describe, expect, it, vi } from 'vitest';
+
+import { chatHelpers } from '@/store/chat/helpers';
+import { globalHelpers } from '@/store/user/helpers';
+import { ChatMessage } from '@/types/message';
+import { OpenAIChatMessage } from '@/types/openai/chat';
+
+import { chainSummaryHistory } from '../summaryHistory';
+
+describe('chainSummaryHistory', () => {
+  it('should use the default model if the token count is below the GPT-3.5 limit', async () => {
+    // Arrange
+    const messages = [
+      { content: 'Hello, how can I assist you?', role: 'assistant' },
+      { content: 'I need help with my account.', role: 'user' },
+    ] as ChatMessage[];
+
+    // Act
+    const result = chainSummaryHistory(messages);
+
+    // Assert
+    expect(result).toMatchSnapshot();
+  });
+});

package/src/chains/summaryHistory.ts

@@ -0,0 +1,19 @@
+import { chatHistoryPrompts } from '@/prompts/chatMessages';
+import { ChatMessage } from '@/types/message';
+import { ChatStreamPayload } from '@/types/openai/chat';
+
+export const chainSummaryHistory = (messages: ChatMessage[]): Partial<ChatStreamPayload> => ({
+  messages: [
+    {
+      content: `You're an assistant who's good at extracting key takeaways from conversations and summarizing them. Please summarize according to the user's needs. The content you need to summarize is located in the <chat_history> </chat_history> group of xml tags. The summary needs to maintain the original language.`,
+      role: 'system',
+    },
+    {
+      content: `${chatHistoryPrompts(messages)}
+
+Please summarize the above conversation and retain key information. The summarized content will be used as context for subsequent prompts, and should be limited to 400 tokens.`,
+
+      role: 'user',
+    },
+  ],
+});

package/src/const/settings/agent.ts

@@ -17,7 +17,9 @@ export const DEFAULT_AGENT_CHAT_CONFIG: LobeAgentChatConfig = {
   autoCreateTopicThreshold: 2,
   displayMode: 'chat',
   enableAutoCreateTopic: true,
-  historyCount: 1,
+  enableCompressHistory: true,
+  enableHistoryCount: true,
+  historyCount: 8,
 };
 
 export const DEFAULT_AGENT_CONFIG: LobeAgentConfig = {

package/src/const/settings/systemAgent.ts

@@ -22,6 +22,7 @@ export const DEFAULT_QUERY_REWRITE_SYSTEM_AGENT_ITEM: QueryRewriteSystemAgent =
 
 export const DEFAULT_SYSTEM_AGENT_CONFIG: UserSystemAgentConfig = {
   agentMeta: DEFAULT_SYSTEM_AGENT_ITEM,
+  historyCompress: DEFAULT_SYSTEM_AGENT_ITEM,
   queryRewrite: DEFAULT_QUERY_REWRITE_SYSTEM_AGENT_ITEM,
   topic: DEFAULT_SYSTEM_AGENT_ITEM,
   translation: DEFAULT_SYSTEM_AGENT_ITEM,

package/src/database/client/models/__tests__/session.test.ts

@@ -24,7 +24,6 @@ describe('SessionModel', () => {
       group: 'testGroup',
       meta: {},
       config: DEFAULT_AGENT_CONFIG,
-      // ... other properties based on LobeAgentSession
     };
   });
 

package/src/database/server/migrations/0011_add_topic_history_summary.sql

@@ -0,0 +1,2 @@
+ALTER TABLE "topics" ADD COLUMN "history_summary" text;--> statement-breakpoint
+ALTER TABLE "topics" ADD COLUMN "metadata" jsonb;