@lobehub/lobehub 2.0.0-next.141 → 2.0.0-next.143

package/apps/desktop/src/preload/streamer.test.ts

@@ -0,0 +1,365 @@
+import type { ProxyTRPCRequestParams } from '@lobechat/electron-client-ipc';
+import { beforeEach, describe, expect, it, vi } from 'vitest';
+
+// Mock electron module
+const mockIpcRendererOn = vi.fn();
+const mockIpcRendererOnce = vi.fn();
+const mockIpcRendererSend = vi.fn();
+const mockIpcRendererRemoveAllListeners = vi.fn();
+
+vi.mock('electron', () => ({
+  ipcRenderer: {
+    on: mockIpcRendererOn,
+    once: mockIpcRendererOnce,
+    removeAllListeners: mockIpcRendererRemoveAllListeners,
+    send: mockIpcRendererSend,
+  },
+}));
+
+// Mock uuid
+vi.mock('uuid', () => ({
+  v4: vi.fn(() => 'test-request-id-123'),
+}));
+
+const { onStreamInvoke } = await import('./streamer');
+
+describe('onStreamInvoke', () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it('should set up stream listeners and send start event', () => {
+    const params: ProxyTRPCRequestParams = {
+      input: { query: 'test' },
+      path: 'test.endpoint',
+      type: 'query',
+    };
+
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    onStreamInvoke(params, callbacks);
+
+    // Verify listeners are registered
+    expect(mockIpcRendererOn).toHaveBeenCalledWith(
+      'stream:data:test-request-id-123',
+      expect.any(Function),
+    );
+    expect(mockIpcRendererOnce).toHaveBeenCalledWith(
+      'stream:end:test-request-id-123',
+      expect.any(Function),
+    );
+    expect(mockIpcRendererOnce).toHaveBeenCalledWith(
+      'stream:error:test-request-id-123',
+      expect.any(Function),
+    );
+    expect(mockIpcRendererOnce).toHaveBeenCalledWith(
+      'stream:response:test-request-id-123',
+      expect.any(Function),
+    );
+
+    // Verify start event is sent
+    expect(mockIpcRendererSend).toHaveBeenCalledWith('stream:start', {
+      ...params,
+      requestId: 'test-request-id-123',
+    });
+  });
+
+  it('should invoke onData callback when data is received', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {},
+      path: 'test',
+      type: 'query',
+    };
+
+    onStreamInvoke(params, callbacks);
+
+    // Get the data listener callback
+    const dataListener = mockIpcRendererOn.mock.calls.find((call) =>
+      call[0].includes('stream:data'),
+    )?.[1];
+
+    // Simulate data event
+    const testData = Buffer.from('test data');
+    dataListener?.(null, testData);
+
+    expect(callbacks.onData).toHaveBeenCalledWith(new Uint8Array(testData));
+  });
+
+  it('should invoke onResponse callback when response is received', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {},
+      path: 'test',
+      type: 'query',
+    };
+
+    onStreamInvoke(params, callbacks);
+
+    // Get the response listener callback
+    const responseListener = mockIpcRendererOnce.mock.calls.find((call) =>
+      call[0].includes('stream:response'),
+    )?.[1];
+
+    // Simulate response event
+    const testResponse = {
+      headers: { 'content-type': 'application/json' },
+      status: 200,
+      statusText: 'OK',
+    };
+    responseListener?.(null, testResponse);
+
+    expect(callbacks.onResponse).toHaveBeenCalledWith(testResponse);
+  });
+
+  it('should invoke onEnd callback and cleanup when stream ends', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {},
+      path: 'test',
+      type: 'query',
+    };
+
+    onStreamInvoke(params, callbacks);
+
+    // Get the end listener callback
+    const endListener = mockIpcRendererOnce.mock.calls.find((call) =>
+      call[0].includes('stream:end'),
+    )?.[1];
+
+    // Simulate end event
+    endListener?.(null);
+
+    expect(callbacks.onEnd).toHaveBeenCalled();
+
+    // Verify cleanup
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:data:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:end:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:error:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:response:test-request-id-123',
+    );
+  });
+
+  it('should invoke onError callback and cleanup when error occurs', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {},
+      path: 'test',
+      type: 'query',
+    };
+
+    onStreamInvoke(params, callbacks);
+
+    // Get the error listener callback
+    const errorListener = mockIpcRendererOnce.mock.calls.find((call) =>
+      call[0].includes('stream:error'),
+    )?.[1];
+
+    // Simulate error event
+    const testError = new Error('Stream processing failed');
+    errorListener?.(null, testError);
+
+    expect(callbacks.onError).toHaveBeenCalledWith(testError);
+
+    // Verify cleanup
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:data:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:end:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:error:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:response:test-request-id-123',
+    );
+  });
+
+  it('should return cleanup function that removes all listeners', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {},
+      path: 'test',
+      type: 'query',
+    };
+
+    const cleanup = onStreamInvoke(params, callbacks);
+
+    // Call cleanup function
+    cleanup();
+
+    // Verify all listeners are removed
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:data:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:end:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:error:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:response:test-request-id-123',
+    );
+  });
+
+  it('should handle multiple data chunks', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {},
+      path: 'test',
+      type: 'query',
+    };
+
+    onStreamInvoke(params, callbacks);
+
+    const dataListener = mockIpcRendererOn.mock.calls.find((call) =>
+      call[0].includes('stream:data'),
+    )?.[1];
+
+    // Simulate multiple data chunks
+    const chunk1 = Buffer.from('chunk1');
+    const chunk2 = Buffer.from('chunk2');
+    const chunk3 = Buffer.from('chunk3');
+
+    dataListener?.(null, chunk1);
+    dataListener?.(null, chunk2);
+    dataListener?.(null, chunk3);
+
+    expect(callbacks.onData).toHaveBeenCalledTimes(3);
+    expect(callbacks.onData).toHaveBeenNthCalledWith(1, new Uint8Array(chunk1));
+    expect(callbacks.onData).toHaveBeenNthCalledWith(2, new Uint8Array(chunk2));
+    expect(callbacks.onData).toHaveBeenNthCalledWith(3, new Uint8Array(chunk3));
+  });
+
+  it('should handle complex request parameters', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {
+        filters: { active: true },
+        query: 'complex query',
+        sort: { field: 'date', order: 'desc' },
+      },
+      path: 'complex.nested.endpoint',
+      type: 'mutation',
+    };
+
+    onStreamInvoke(params, callbacks);
+
+    expect(mockIpcRendererSend).toHaveBeenCalledWith('stream:start', {
+      ...params,
+      requestId: 'test-request-id-123',
+    });
+  });
+
+  it('should not invoke callbacks after cleanup', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {},
+      path: 'test',
+      type: 'query',
+    };
+
+    const cleanup = onStreamInvoke(params, callbacks);
+
+    // Cleanup immediately
+    cleanup();
+
+    // Try to trigger callbacks after cleanup (this simulates late events)
+    const dataListener = mockIpcRendererOn.mock.calls.find((call) =>
+      call[0].includes('stream:data'),
+    )?.[1];
+
+    // Since listeners are removed, this shouldn't do anything
+    // The actual behavior depends on electron's implementation
+    // But we can verify cleanup was called
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledTimes(4);
+  });
+
+  it('should handle empty buffer data', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {},
+      path: 'test',
+      type: 'query',
+    };
+
+    onStreamInvoke(params, callbacks);
+
+    const dataListener = mockIpcRendererOn.mock.calls.find((call) =>
+      call[0].includes('stream:data'),
+    )?.[1];
+
+    const emptyBuffer = Buffer.from('');
+    dataListener?.(null, emptyBuffer);
+
+    expect(callbacks.onData).toHaveBeenCalledWith(new Uint8Array(emptyBuffer));
+  });
+});

package/apps/desktop/vitest.config.mts

@@ -5,6 +5,7 @@ export default defineConfig({
   test: {
     alias: {
       '@': resolve(__dirname, './src/main'),
+      '~common': resolve(__dirname, './src/common'),
     },
     coverage: {
       all: false,

package/changelog/v1.json

@@ -1,4 +1,22 @@
 [
+  {
+    "children": {
+      "features": [
+        "Support market cloud endpoint mcp."
+      ]
+    },
+    "date": "2025-12-02",
+    "version": "2.0.0-next.143"
+  },
+  {
+    "children": {
+      "fixes": [
+        "Remove internal apiMode param from chat completion API requests."
+      ]
+    },
+    "date": "2025-12-01",
+    "version": "2.0.0-next.142"
+  },
   {
     "children": {
       "fixes": [

package/docs/self-hosting/environment-variables/model-provider.mdx

@@ -121,6 +121,37 @@ If you need to use Azure OpenAI to provide model services, you can refer to the
 - Default: `-`
 - Example: `-all,+gemini-1.5-flash-latest,+gemini-1.5-pro-latest`
 
+## Vertex AI
+
+### `VERTEXAI_CREDENTIALS`
+
+- Type: Required
+- Description: A JSON string of your Google Cloud service account key, you can get the key from [here](/docs/usage/providers/vertexai).
+- Default: -
+- Example: `{"type": "service_account", "project_id": "your-gcp-project-id", ...}`
+
+### `VERTEXAI_PROJECT`
+
+- Type: Optional
+- Description: Your Google Cloud project ID. If not set, it will be obtained from the `project_id` field in `VERTEXAI_CREDENTIALS`.
+- Default: -
+- Example: `your-gcp-project-id`
+
+### `VERTEXAI_LOCATION`
+
+- Type: Optional
+- Description: The region where your Vertex AI model is located.
+- Default: `global`
+- Example: `us-central1`
+
+
+### `VERTEXAI_MODEL_LIST`
+
+- Type: Optional
+- Description: Used to control the model list, use `+` to add a model, use `-` to hide a model, use `model_name=display_name` to customize the display name of a model, separated by commas. Definition syntax rules see [model-list][model-list]
+- Default: `-`
+- Example: `-all,+gemini-1.5-flash-latest,+gemini-1.5-pro-latest`
+
 ## Anthropic AI
 
 ### `ANTHROPIC_API_KEY`

package/docs/self-hosting/environment-variables/model-provider.zh-CN.mdx

@@ -119,6 +119,36 @@ LobeChat 在部署时提供了丰富的模型服务商相关的环境变量，
 - 默认值：`-`
 - 示例：`-all,+gemini-1.5-flash-latest,+gemini-1.5-pro-latest`
 
+## Vertex AI
+
+### `VERTEXAI_CREDENTIALS`
+
+- 类型：必选
+- 描述：Google Cloud 服务账号密钥的 JSON 字符串。用于认证和授权访问 Vertex AI 服务，获取方法请参考 [这里](/zh/docs/usage/providers/vertexai)
+- 默认值：-
+- 示例：`{"type": "service_account", "project_id": "your-gcp-project-id", ...}`
+
+### `VERTEXAI_PROJECT`
+
+- 类型：可选
+- 描述：你的 Google Cloud 项目 ID。如果未设置，将从 `VERTEXAI_CREDENTIALS` 中的 `project_id` 字段获取。
+- 默认值：-
+- 示例：`your-gcp-project-id`
+
+### `VERTEXAI_LOCATION`
+
+- 类型：可选
+- 描述：你的 Vertex AI 模型所在的区域。
+- 默认值：`global`
+- 示例：`us-central1`
+
+### `VERTEXAI_MODEL_LIST`
+
+- 类型：可选
+- 描述：用来控制模型列表，使用 `+` 增加一个模型，使用 `-` 来隐藏一个模型，使用 `模型名=展示名<扩展配置>` 来自定义模型的展示名，用英文逗号隔开。模型定义语法规则见 [模型列表][model-list]
+- 默认值：`-`
+- 示例：`-all,+gemini-1.5-flash-latest,+gemini-1.5-pro-latest`
+
 ## Anthropic AI
 
 ### `ANTHROPIC_API_KEY`

package/locales/ar/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "إلغاء",
+    "confirm": "تأكيد التفويض",
+    "description": {
+      "and": "و",
+      "prefix": "بالنقر على تأكيد التفويض، فإنك توافق على",
+      "privacy": "سياسة الخصوصية",
+      "terms": "شروط الخدمة"
+    },
+    "title": "تأكيد التفويض"
+  },
   "callback": {
     "buttons": {
       "close": "إغلاق النافذة"
@@ -33,8 +44,10 @@
     "stateMissing": "لم يتم العثور على حالة التفويض، يرجى المحاولة مرة أخرى."
   },
   "messages": {
+    "authorized": "تم تفويض خدمة LobeHub بنجاح",
     "loading": "جارٍ بدء عملية التفويض...",
     "success": {
+      "cloudMcpInstall": "تم التفويض بنجاح! يمكنك الآن تثبيت إضافة Cloud MCP.",
       "submit": "تم التفويض بنجاح! يمكنك الآن نشر المساعد.",
       "upload": "تم التفويض بنجاح! يمكنك الآن نشر إصدار جديد."
     }

package/locales/bg-BG/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "Отказ",
+    "confirm": "Разреши използването",
+    "description": {
+      "and": "и",
+      "prefix": "С натискане на „Разреши използването“ се съгласявате с",
+      "privacy": "Политиката за поверителност",
+      "terms": "Условията за ползване"
+    },
+    "title": "Потвърждаване на оторизацията"
+  },
   "callback": {
     "buttons": {
       "close": "Затвори прозореца"
@@ -33,8 +44,10 @@
     "stateMissing": "Състоянието на упълномощаване не бе намерено, моля опитайте отново."
   },
   "messages": {
+    "authorized": "Успешно упълномощаване на услугата LobeHub",
     "loading": "Стартиране на процеса по упълномощаване...",
     "success": {
+      "cloudMcpInstall": "Успешно разрешение! Вече можете да инсталирате приставката Cloud MCP.",
       "submit": "Упълномощаването е успешно! Вече можете да публикувате помощник.",
       "upload": "Упълномощаването е успешно! Вече можете да публикувате нова версия."
     }

package/locales/de-DE/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "Abbrechen",
+    "confirm": "Autorisieren",
+    "description": {
+      "and": "und",
+      "prefix": "Mit dem Klick auf „Autorisieren“ stimmen Sie den",
+      "privacy": "Datenschutzbestimmungen",
+      "terms": "Nutzungsbedingungen"
+    },
+    "title": "Autorisierung bestätigen"
+  },
   "callback": {
     "buttons": {
       "close": "Fenster schließen"
@@ -33,8 +44,10 @@
     "stateMissing": "Autorisierungsstatus nicht gefunden, bitte erneut versuchen."
   },
   "messages": {
+    "authorized": "LobeHub-Dienst erfolgreich autorisiert",
     "loading": "Autorisierungsvorgang wird gestartet...",
     "success": {
+      "cloudMcpInstall": "Autorisierung erfolgreich! Sie können jetzt das Cloud MCP-Plugin installieren.",
       "submit": "Autorisierung erfolgreich! Du kannst jetzt einen Assistenten veröffentlichen.",
       "upload": "Autorisierung erfolgreich! Du kannst jetzt eine neue Version veröffentlichen."
     }

package/locales/en-US/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "Cancel",
+    "confirm": "Authorize",
+    "description": {
+      "and": "and",
+      "prefix": "By clicking Authorize, you agree to",
+      "privacy": "Privacy Policy",
+      "terms": "Terms of Service"
+    },
+    "title": "Confirm Authorization"
+  },
   "callback": {
     "buttons": {
       "close": "Close Window"
@@ -33,8 +44,10 @@
     "stateMissing": "Authorization state not found. Please try again."
   },
   "messages": {
+    "authorized": "LobeHub service authorized successfully",
     "loading": "Starting authorization process...",
     "success": {
+      "cloudMcpInstall": "Authorization successful! You can now install the Cloud MCP plugin.",
       "submit": "Authorization successful! You can now publish your assistant.",
       "upload": "Authorization successful! You can now publish a new version."
     }

package/locales/es-ES/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "Cancelar",
+    "confirm": "Autorizar uso",
+    "description": {
+      "and": "y",
+      "prefix": "Al hacer clic en autorizar uso, se considera que aceptas",
+      "privacy": "la Política de Privacidad",
+      "terms": "los Términos de Servicio"
+    },
+    "title": "Confirmar autorización"
+  },
   "callback": {
     "buttons": {
       "close": "Cerrar ventana"
@@ -33,8 +44,10 @@
     "stateMissing": "No se encontró el estado de autenticación, por favor inténtalo de nuevo."
   },
   "messages": {
+    "authorized": "Autorización del servicio LobeHub realizada con éxito",
     "loading": "Iniciando el proceso de autenticación...",
     "success": {
+      "cloudMcpInstall": "¡Autorización exitosa! Ahora puedes instalar el complemento Cloud MCP.",
       "submit": "¡Autenticación exitosa! Ahora puedes publicar asistentes.",
       "upload": "¡Autenticación exitosa! Ahora puedes publicar una nueva versión."
     }

package/locales/fa-IR/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "لغو",
+    "confirm": "تأیید مجوز استفاده",
+    "description": {
+      "and": "و",
+      "prefix": "با کلیک بر روی تأیید مجوز استفاده، شما با",
+      "privacy": "سیاست حفظ حریم خصوصی",
+      "terms": "شرایط خدمات"
+    },
+    "title": "تأیید مجوز"
+  },
   "callback": {
     "buttons": {
       "close": "بستن پنجره"
@@ -33,8 +44,10 @@
     "stateMissing": "وضعیت احراز هویت یافت نشد، لطفاً دوباره تلاش کنید."
   },
   "messages": {
+    "authorized": "دسترسی به سرویس LobeHub با موفقیت تأیید شد",
     "loading": "در حال آغاز فرآیند احراز هویت...",
     "success": {
+      "cloudMcpInstall": "مجوز با موفقیت صادر شد! اکنون می‌توانید افزونه Cloud MCP را نصب کنید.",
       "submit": "احراز هویت با موفقیت انجام شد! اکنون می‌توانید دستیار را منتشر کنید.",
       "upload": "احراز هویت با موفقیت انجام شد! اکنون می‌توانید نسخه جدیدی منتشر کنید."
     }

package/locales/fr-FR/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "Annuler",
+    "confirm": "Autoriser l'utilisation",
+    "description": {
+      "and": "et",
+      "prefix": "En cliquant sur « Autoriser l'utilisation », vous acceptez",
+      "privacy": "la Politique de confidentialité",
+      "terms": "les Conditions d'utilisation"
+    },
+    "title": "Confirmer l'autorisation"
+  },
   "callback": {
     "buttons": {
       "close": "Fermer la fenêtre"
@@ -33,8 +44,10 @@
     "stateMissing": "État d'autorisation introuvable, veuillez réessayer."
   },
   "messages": {
+    "authorized": "Autorisation du service LobeHub réussie",
     "loading": "Lancement du processus d'autorisation...",
     "success": {
+      "cloudMcpInstall": "Autorisation réussie ! Vous pouvez maintenant installer le plugin Cloud MCP.",
       "submit": "Autorisation réussie ! Vous pouvez maintenant publier un assistant.",
       "upload": "Autorisation réussie ! Vous pouvez maintenant publier une nouvelle version."
     }

package/locales/it-IT/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "Annulla",
+    "confirm": "Autorizza l'utilizzo",
+    "description": {
+      "and": "e",
+      "prefix": "Facendo clic su 'Autorizza l'utilizzo' si accetta",
+      "privacy": "l'Informativa sulla privacy",
+      "terms": "i Termini di servizio"
+    },
+    "title": "Conferma autorizzazione"
+  },
   "callback": {
     "buttons": {
       "close": "Chiudi finestra"
@@ -33,8 +44,10 @@
     "stateMissing": "Stato di autorizzazione non trovato, riprova."
   },
   "messages": {
+    "authorized": "Autorizzazione al servizio LobeHub avvenuta con successo",
     "loading": "Avvio del processo di autorizzazione in corso...",
     "success": {
+      "cloudMcpInstall": "Autorizzazione avvenuta con successo! Ora puoi installare il plugin Cloud MCP.",
       "submit": "Autorizzazione riuscita! Ora puoi pubblicare l'assistente.",
       "upload": "Autorizzazione riuscita! Ora puoi pubblicare una nuova versione."
     }