@lobehub/chat 1.91.2 → 1.92.0

package/src/services/chat.ts

@@ -41,6 +41,7 @@ import { createErrorResponse } from '@/utils/errorResponse';
 import { FetchSSEOptions, fetchSSE, getMessageError } from '@/utils/fetch';
 import { genToolCallingName } from '@/utils/toolCall';
 import { createTraceHeader, getTraceId } from '@/utils/trace';
+import { parsePlaceholderVariablesMessages } from '@/utils/client/parserPlaceholder';
 
 import { createHeaderWithAuth, createPayloadWithKeyVaults } from './_auth';
 import { API_ENDPOINTS } from './_url';
@@ -172,14 +173,18 @@ class ChatService {
 
     // =================== 0. process search =================== //
     const chatConfig = agentChatConfigSelectors.currentChatConfig(getAgentStoreState());
-
+    const aiInfraStoreState = getAiInfraStoreState();
     const enabledSearch = chatConfig.searchMode !== 'off';
+    const isProviderHasBuiltinSearch = aiProviderSelectors.isProviderHasBuiltinSearch(
+      payload.provider!,
+    )(aiInfraStoreState);
     const isModelHasBuiltinSearch = aiModelSelectors.isModelHasBuiltinSearch(
       payload.model,
       payload.provider!,
-    )(getAiInfraStoreState());
+    )(aiInfraStoreState);
 
-    const useModelSearch = isModelHasBuiltinSearch && chatConfig.useModelBuiltinSearch;
+    const useModelSearch =
+      (isProviderHasBuiltinSearch || isModelHasBuiltinSearch) && chatConfig.useModelBuiltinSearch;
 
     const useApplicationBuiltinSearchTool = enabledSearch && !useModelSearch;
 
@@ -189,11 +194,14 @@ class ChatService {
       pluginIds.push(WebBrowsingManifest.identifier);
     }
 
-    // ============  1. preprocess messages   ============ //
+    // ============  1. preprocess placeholder variables   ============ //
+    const parsedMessages = parsePlaceholderVariablesMessages(messages);
+
+    // ============  2. preprocess messages   ============ //
 
     const oaiMessages = this.processMessages(
       {
-        messages,
+        messages: parsedMessages,
         model: payload.model,
         provider: payload.provider!,
         tools: pluginIds,
@@ -201,28 +209,28 @@ class ChatService {
       options,
     );
 
-    // ============  2. preprocess tools   ============ //
+    // ============  3. preprocess tools   ============ //
 
     const tools = this.prepareTools(pluginIds, {
       model: payload.model,
       provider: payload.provider!,
     });
 
-    // ============  3. process extend params   ============ //
+    // ============  4. process extend params   ============ //
 
     let extendParams: Record<string, any> = {};
 
     const isModelHasExtendParams = aiModelSelectors.isModelHasExtendParams(
       payload.model,
       payload.provider!,
-    )(getAiInfraStoreState());
+    )(aiInfraStoreState);
 
     // model
     if (isModelHasExtendParams) {
       const modelExtendParams = aiModelSelectors.modelExtendParams(
         payload.model,
         payload.provider!,
-      )(getAiInfraStoreState());
+      )(aiInfraStoreState);
       // if model has extended params, then we need to check if the model can use reasoning
 
       if (modelExtendParams!.includes('enableReasoning')) {

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

@@ -1,7 +1,6 @@
 /* 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 { produce } from 'immer';
-import { template } from 'lodash-es';
 import { StateCreator } from 'zustand/vanilla';
 
 import { LOADING_FLAT, MESSAGE_CANCEL_FLAT } from '@/const/message';
@@ -13,7 +12,7 @@ import { messageService } from '@/services/message';
 import { useAgentStore } from '@/store/agent';
 import { agentChatConfigSelectors, agentSelectors } from '@/store/agent/selectors';
 import { getAgentStoreState } from '@/store/agent/store';
-import { aiModelSelectors } from '@/store/aiInfra';
+import { aiModelSelectors, aiProviderSelectors } from '@/store/aiInfra';
 import { getAiInfraStoreState } from '@/store/aiInfra/store';
 import { chatHelpers } from '@/store/chat/helpers';
 import { ChatStore } from '@/store/chat/store';
@@ -299,7 +298,8 @@ export const generateAIChat: StateCreator<
     // create a new array to avoid the original messages array change
     const messages = [...originalMessages];
 
-    const { model, provider, chatConfig } = agentSelectors.currentAgentConfig(getAgentStoreState());
+    const agentStoreState = getAgentStoreState();
+    const { model, provider, chatConfig } = agentSelectors.currentAgentConfig(agentStoreState);
 
     let fileChunks: MessageSemanticSearchChunk[] | undefined;
     let ragQueryId;
@@ -323,7 +323,7 @@ export const generateAIChat: StateCreator<
         chunks,
         userQuery: lastMsg.content,
         rewriteQuery,
-        knowledge: agentSelectors.currentEnabledKnowledge(getAgentStoreState()),
+        knowledge: agentSelectors.currentEnabledKnowledge(agentStoreState),
       });
 
       // 3. add the retrieve context messages to the messages history
@@ -355,14 +355,25 @@ export const generateAIChat: StateCreator<
     if (!assistantId) return;
 
     // 3. place a search with the search working model if this model is not support tool use
+    const aiInfraStoreState = getAiInfraStoreState();
     const isModelSupportToolUse = aiModelSelectors.isModelSupportToolUse(
       model,
       provider!,
-    )(getAiInfraStoreState());
-    const isAgentEnableSearch = agentChatConfigSelectors.isAgentEnableSearch(getAgentStoreState());
+    )(aiInfraStoreState);
+    const isProviderHasBuiltinSearch = aiProviderSelectors.isProviderHasBuiltinSearch(provider!)(
+      aiInfraStoreState,
+    );
+    const isModelHasBuiltinSearch = aiModelSelectors.isModelHasBuiltinSearch(
+      model,
+      provider!,
+    )(aiInfraStoreState);
+    const useModelBuiltinSearch = agentChatConfigSelectors.useModelBuiltinSearch(agentStoreState);
+    const useModelSearch =
+      (isProviderHasBuiltinSearch || isModelHasBuiltinSearch) && useModelBuiltinSearch;
+    const isAgentEnableSearch = agentChatConfigSelectors.isAgentEnableSearch(agentStoreState);
 
-    if (isAgentEnableSearch && !isModelSupportToolUse) {
-      const { model, provider } = agentChatConfigSelectors.searchFCModel(getAgentStoreState());
+    if (isAgentEnableSearch && !useModelSearch && !isModelSupportToolUse) {
+      const { model, provider } = agentChatConfigSelectors.searchFCModel(agentStoreState);
 
       let isToolsCalling = false;
       let isError = false;
@@ -460,10 +471,10 @@ export const generateAIChat: StateCreator<
     }
 
     // 6. summary history if context messages is larger than historyCount
-    const historyCount = agentChatConfigSelectors.historyCount(getAgentStoreState());
+    const historyCount = agentChatConfigSelectors.historyCount(agentStoreState);
 
     if (
-      agentChatConfigSelectors.enableHistoryCount(getAgentStoreState()) &&
+      agentChatConfigSelectors.enableHistoryCount(agentStoreState) &&
       chatConfig.enableCompressHistory &&
       originalMessages.length > historyCount
     ) {
@@ -495,8 +506,6 @@ export const generateAIChat: StateCreator<
     const agentConfig = agentSelectors.currentAgentConfig(getAgentStoreState());
     const chatConfig = agentChatConfigSelectors.currentChatConfig(getAgentStoreState());
 
-    const compiler = template(chatConfig.inputTemplate, { interpolate: /{{([\S\s]+?)}}/g });
-
     // ================================== //
     //   messages uniformly preprocess    //
     // ================================== //
@@ -511,29 +520,12 @@ export const generateAIChat: StateCreator<
       historyCount,
     });
 
-    // 2. replace inputMessage template
-    preprocessMsgs = !chatConfig.inputTemplate
-      ? preprocessMsgs
-      : preprocessMsgs.map((m) => {
-          if (m.role === 'user') {
-            try {
-              return { ...m, content: compiler({ text: m.content }) };
-            } catch (error) {
-              console.error(error);
-
-              return m;
-            }
-          }
-
-          return m;
-        });
-
-    // 3. add systemRole
+    // 2. add systemRole
     if (agentConfig.systemRole) {
       preprocessMsgs.unshift({ content: agentConfig.systemRole, role: 'system' } as ChatMessage);
     }
 
-    // 4. handle max_tokens
+    // 3. handle max_tokens
     agentConfig.params.max_tokens = chatConfig.enableMaxTokens
       ? agentConfig.params.max_tokens
       : undefined;

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

@@ -33,6 +33,24 @@ afterEach(() => {
 });
 
 describe('userProfileSelectors', () => {
+  describe('fullName', () => {
+    it('should return user fullName if exist', () => {
+      const store: UserStore = {
+        user: { fullName: 'John Doe' },
+      } as UserStore;
+
+      expect(userProfileSelectors.fullName(store)).toBe('John Doe');
+    });
+
+    it('should return empty string if not exist', () => {
+      const store: UserStore = {
+        user: { fullName: undefined },
+      } as UserStore;
+
+      expect(userProfileSelectors.fullName(store)).toBe('');
+    });
+  });
+
   describe('nickName', () => {
     it('should return default nickname when auth is disabled and not desktop', () => {
       enableAuth = false;

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

@@ -36,6 +36,7 @@ const username = (s: UserStore) => {
 };
 
 export const userProfileSelectors = {
+  fullName: (s: UserStore): string => s.user?.fullName || '',
   nickName,
   userAvatar: (s: UserStore): string => s.user?.avatar || '',
   userId: (s: UserStore) => s.user?.id,

package/src/utils/client/parserPlaceholder.test.ts

@@ -0,0 +1,326 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { parsePlaceholderVariablesMessages, VARIABLE_GENERATORS } from './parserPlaceholder';
+
+// Mock dependencies
+vi.mock('@/utils/uuid', () => ({
+  uuid: () => 'mocked-uuid-12345'
+}));
+
+vi.mock('@/store/user', () => ({
+  useUserStore: {
+    getState: () => ({})
+  }
+}));
+
+vi.mock('@/store/user/selectors', () => ({
+  userProfileSelectors: {
+    nickName: () => 'Test User',
+    username: () => 'testuser',
+    fullName: () => 'Test Full Name'
+  }
+}));
+
+vi.mock('@/store/agent/store', () => ({
+  getAgentStoreState: () => ({})
+}));
+
+vi.mock('@/store/agent/selectors', () => ({
+  agentChatConfigSelectors: {
+    currentChatConfig: () => ({
+      inputTemplate: 'Hello {{username}}!'
+    })
+  }
+}));
+
+describe('parsePlaceholderVariablesMessages', () => {
+  beforeEach(() => {
+    // Mock Date for consistent testing
+    vi.useFakeTimers();
+    vi.setSystemTime(new Date('2025-06-06T06:06:06.666Z'));
+
+    // Mock Math.random for consistent random values
+    vi.spyOn(Math, 'random').mockReturnValue(0.5);
+  });
+
+  afterEach(() => {
+    vi.useRealTimers();
+    vi.restoreAllMocks();
+  });
+
+  describe('string content messages', () => {
+    it('should replace template variables in string content', () => {
+      const messages = [
+        {
+          id: '1',
+          content: 'Hello {{username}}, today is {{date}}'
+        }
+      ];
+
+      const result = parsePlaceholderVariablesMessages(messages);
+
+      expect(result[0].content).toContain('testuser');
+      expect(result[0].content).toContain(new Date().toLocaleDateString());
+    });
+
+    it('should handle multiple variables in one message', () => {
+      const messages = [
+        {
+          id: '1',
+          content: 'Time: {{time}}, Date: {{date}}, User: {{nickname}}'
+        }
+      ];
+
+      const result = parsePlaceholderVariablesMessages(messages);
+
+      expect(result[0].content).toContain('Test User');
+      expect(result[0].content).toMatch(/Time: .+, Date: .+, User: Test User/);
+    });
+
+    it('should preserve message structure when replacing variables', () => {
+      const messages = [
+        {
+          id: '1',
+          role: 'user',
+          content: 'Hello {{username}}'
+        }
+      ];
+
+      const result = parsePlaceholderVariablesMessages(messages);
+
+      expect(result[0]).toEqual({
+        id: '1',
+        role: 'user',
+        content: 'Hello testuser'
+      });
+    });
+  });
+
+  describe('array content messages', () => {
+    it('should replace variables in text type array elements', () => {
+      const messages = [
+        {
+          id: '1',
+          content: [
+            {
+              type: 'text',
+              text: 'Hello {{username}}'
+            },
+            {
+              type: 'image_url',
+              image_url: 'image.jpg'
+            }
+          ]
+        }
+      ];
+
+      const result = parsePlaceholderVariablesMessages(messages);
+
+      expect(result[0].content[0].text).toBe('Hello testuser');
+      expect(result[0].content[1]).toEqual({
+        type: 'image_url',
+        image_url: 'image.jpg'
+      });
+    });
+
+    it('should handle multiple text elements with variables', () => {
+      const messages = [
+        {
+          id: '1',
+          content: [
+            {
+              type: 'text',
+              text: 'Date: {{date}}'
+            },
+            {
+              type: 'text',
+              text: 'Time: {{time}}'
+            },
+            {
+              type: 'image_url',
+              image_url: 'test.jpg'
+            }
+          ]
+        }
+      ];
+
+      const result = parsePlaceholderVariablesMessages(messages);
+
+      expect(result[0].content[0].text).toContain(new Date().toLocaleDateString());
+      expect(result[0].content[1].text).toContain(new Date().toLocaleTimeString());
+      expect(result[0].content[2]).toEqual({
+        type: 'image_url',
+        image_url: 'test.jpg'
+      });
+    });
+
+    it('should preserve non-text array elements unchanged', () => {
+      const messages = [
+        {
+          id: '1',
+          content: [
+            {
+              type: 'image_url',
+              image_url: 'image.jpg',
+            },
+            {
+              type: 'image_url',
+              name: 'image2.jpg'
+            }
+          ]
+        }
+      ];
+
+      const result = parsePlaceholderVariablesMessages(messages);
+
+      expect(result[0].content).toEqual([
+        {
+          type: 'image_url',
+          image_url: 'image.jpg'
+        },
+        {
+          type: 'image_url',
+          name: 'image2.jpg'
+        }
+      ]);
+    });
+  });
+
+  describe('edge cases', () => {
+    it('should handle empty messages array', () => {
+      const result = parsePlaceholderVariablesMessages([]);
+      expect(result).toEqual([]);
+    });
+
+    it('should handle messages without content', () => {
+      const messages = [
+        { id: '1' },
+        { id: '2', content: null },
+        { id: '3', content: undefined }
+      ];
+
+      const result = parsePlaceholderVariablesMessages(messages);
+
+      expect(result).toEqual([
+        { id: '1' },
+        { id: '2', content: null },
+        { id: '3', content: undefined }
+      ]);
+    });
+
+    it('should handle empty string content', () => {
+      const messages = [
+        { id: '1', content: '' }
+      ];
+
+      const result = parsePlaceholderVariablesMessages(messages);
+
+      expect(result[0].content).toBe('');
+    });
+
+    it('should handle content without variables', () => {
+      const messages = [
+        { id: '1', content: 'Hello world!' },
+        { 
+          id: '2', 
+          content: [
+            { type: 'text', text: 'No variables here' },
+            { type: 'image_url', image_url: 'test.jpg' }
+          ]
+        }
+      ];
+
+      const result = parsePlaceholderVariablesMessages(messages);
+
+      expect(result[0].content).toBe('Hello world!');
+      expect(result[1].content[0].text).toBe('No variables here');
+    });
+
+    it('should handle unknown variable types', () => {
+      const messages = [
+        { id: '1', content: 'Hello {{unknown_variable}}!' }
+      ];
+
+      const result = parsePlaceholderVariablesMessages(messages);
+
+      // Unknown variables should remain unchanged
+      expect(result[0].content).toBe('Hello {{unknown_variable}}!');
+    });
+
+    it('should handle nested variables (input_template)', () => {
+      const messages = [
+        { id: '1', content: 'Template: {{input_template}}' }
+      ];
+
+      const result = parsePlaceholderVariablesMessages(messages);
+
+      // Should resolve nested variables in input_template
+      expect(result[0].content).toBe('Template: Hello testuser!');
+    });
+  });
+
+  describe('specific variable types', () => {
+    it('should handle time variables', () => {
+      const messages = [
+        { 
+          id: '1', 
+          content: 'Year: {{year}}, Month: {{month}}, Day: {{day}}' 
+        }
+      ];
+
+      const result = parsePlaceholderVariablesMessages(messages);
+
+      expect(result[0].content).toContain('Year: 2025');
+      expect(result[0].content).toContain('Month: 06');
+      expect(result[0].content).toContain('Day: 06');
+    });
+
+    it('should handle random variables', () => {
+      const messages = [
+        { 
+          id: '1', 
+          content: 'Random: {{random}}, Bool: {{random_bool}}, UUID: {{uuid}}' 
+        }
+      ];
+
+      const result = parsePlaceholderVariablesMessages(messages);
+
+      expect(result[0].content).toContain('Random: 500001'); // Math.random() * 1000000 + 1 with 0.5
+      expect(result[0].content).toContain('Bool: false'); // Math.random() > 0.5 with 0.5
+      expect(result[0].content).toContain('UUID: mocked-uuid-12345');
+    });
+
+    it('should handle user variables', () => {
+      const messages = [
+        { 
+          id: '1', 
+          content: 'User: {{username}}, Nickname: {{nickname}}' 
+        }
+      ];
+
+      const result = parsePlaceholderVariablesMessages(messages);
+
+      expect(result[0].content).toBe('User: testuser, Nickname: Test User');
+    });
+  });
+
+  describe('multiple messages', () => {
+    it('should process multiple messages correctly', () => {
+      const messages = [
+        { id: '1', content: 'Hello {{username}}' },
+        { 
+          id: '2', 
+          content: [
+            { type: 'text', text: 'Today is {{date}}' }
+          ]
+        },
+        { id: '3', content: 'Time: {{time}}' }
+      ];
+
+      const result = parsePlaceholderVariablesMessages(messages);
+
+      expect(result[0].content).toBe('Hello testuser');
+      expect(result[1].content[0].text).toContain(new Date().toLocaleDateString());
+      expect(result[2].content).toContain(new Date().toLocaleTimeString());
+    });
+  });
+});