@lobehub/chat 1.116.3 → 1.117.0

package/src/config/aiModels/openrouter.ts

@@ -11,6 +11,39 @@ const openrouterChatModels: AIChatModelCard[] = [
     id: 'openrouter/auto',
     type: 'chat',
   },
+  {
+    abilities: {
+      imageOutput: true,
+      vision: true,
+    },
+    contextWindowTokens: 32_768 + 8192,
+    description: 'Gemini 2.5 Flash 实验模型，支持图像生成',
+    displayName: 'Gemini 2.5 Flash Image Preview',
+    id: 'google/gemini-2.5-flash-image-preview',
+    maxOutput: 8192,
+    pricing: {
+      units: [
+        { name: 'imageOutput', rate: 30, strategy: 'fixed', unit: 'millionTokens' },
+        { name: 'textInput', rate: 0.3, strategy: 'fixed', unit: 'millionTokens' },
+        { name: 'textOutput', rate: 2.5, strategy: 'fixed', unit: 'millionTokens' },
+      ],
+    },
+    releasedAt: '2025-08-26',
+    type: 'chat',
+  },
+  {
+    abilities: {
+      imageOutput: true,
+      vision: true,
+    },
+    contextWindowTokens: 32_768 + 8192,
+    description: 'Gemini 2.5 Flash 实验模型，支持图像生成',
+    displayName: 'Gemini 2.5 Flash Image Preview (free)',
+    id: 'google/gemini-2.5-flash-image-preview:free',
+    maxOutput: 8192,
+    releasedAt: '2025-08-26',
+    type: 'chat',
+  },
   {
     abilities: {
       reasoning: true,

package/src/config/aiModels/vertexai.ts

@@ -126,21 +126,21 @@ const vertexaiChatModels: AIChatModelCard[] = [
       imageOutput: true,
       vision: true,
     },
-    contextWindowTokens: 32_768 + 32_768,
+    contextWindowTokens: 32_768 + 8192,
     description:
       'Gemini 2.5 Flash Image Preview 是 Google 最新、最快、最高效的原生多模态模型，它允许您通过对话生成和编辑图像。',
     displayName: 'Gemini 2.5 Flash Image Preview',
     enabled: true,
     id: 'gemini-2.5-flash-image-preview',
-    maxOutput: 32_768,
+    maxOutput: 8192,
     pricing: {
       units: [
         { name: 'textInput', rate: 0.3, strategy: 'fixed', unit: 'millionTokens' },
         { name: 'textOutput', rate: 2.5, strategy: 'fixed', unit: 'millionTokens' },
-        { name: 'imageOutput', rate: 3, strategy: 'fixed', unit: 'millionTokens' },
+        { name: 'imageOutput', rate: 30, strategy: 'fixed', unit: 'millionTokens' },
       ],
     },
-    releasedAt: '2025-08-27',
+    releasedAt: '2025-08-26',
     type: 'chat',
   },
   {

package/src/features/Conversation/Extras/Usage/UsageDetail/index.tsx

@@ -61,6 +61,12 @@ const TokenDetail = memo<TokenDetailProps>(({ meta, model, provider }) => {
         ? detailTokens.outputReasoning.credit
         : detailTokens.outputReasoning.token,
     },
+    !!detailTokens.outputImage && {
+      color: theme.purple,
+      id: 'outputImage',
+      title: t('messages.tokenDetails.outputImage'),
+      value: isShowCredit ? detailTokens.outputImage.credit : detailTokens.outputImage.token,
+    },
     !!detailTokens.outputAudio && {
       color: theme.cyan9,
       id: 'outputAudio',

package/src/features/Conversation/Extras/Usage/UsageDetail/tokens.test.ts

@@ -143,6 +143,44 @@ describe('getDetailsToken', () => {
     });
   });
 
+  it('should handle outputImageTokens correctly', () => {
+    const usage = {
+      inputTextTokens: 100,
+      outputImageTokens: 60,
+      outputReasoningTokens: 30,
+      totalOutputTokens: 200,
+      totalTokens: 300,
+    } as ModelTokensUsage;
+
+    const result = getDetailsToken(usage, mockModelCard);
+
+    expect(result.outputImage).toEqual({
+      credit: 1, // 60 * 0.02 = 1.2 -> 1
+      id: 'outputImage',
+      token: 60,
+    });
+
+    expect(result.outputReasoning).toEqual({
+      credit: 1, // 30 * 0.02 = 0.6 -> 1
+      token: 30,
+    });
+
+    expect(result.outputText).toEqual({
+      credit: 2, // (200 - 30 - 60) * 0.02 = 2.2 -> 2
+      token: 110,
+    });
+
+    expect(result.totalOutput).toEqual({
+      credit: 4, // 200 * 0.02 = 4
+      token: 200,
+    });
+
+    expect(result.totalTokens).toEqual({
+      credit: 4, // total credit equals totalOutputCredit here
+      token: 300,
+    });
+  });
+
   it('should handle inputCitationTokens correctly', () => {
     const usage: ModelTokensUsage = {
       inputCitationTokens: 75,

package/src/features/Conversation/Extras/Usage/UsageDetail/tokens.ts

@@ -21,9 +21,14 @@ export const getDetailsToken = (
 
   const outputReasoningTokens = usage.outputReasoningTokens || (usage as any).reasoningTokens || 0;
 
+  const outputImageTokens = usage.outputImageTokens || (usage as any).imageTokens || 0;
+
   const outputTextTokens = usage.outputTextTokens
     ? usage.outputTextTokens
-    : totalOutputTokens - outputReasoningTokens - (usage.outputAudioTokens || 0);
+    : totalOutputTokens -
+      outputReasoningTokens -
+      (usage.outputAudioTokens || 0) -
+      outputImageTokens;
 
   const inputWriteCacheTokens = usage.inputWriteCacheTokens || 0;
   const inputCacheTokens = usage.inputCachedTokens || (usage as any).cachedTokens || 0;
@@ -93,6 +98,13 @@ export const getDetailsToken = (
           token: usage.outputAudioTokens,
         }
       : undefined,
+    outputImage: !!outputImageTokens
+      ? {
+          credit: calcCredit(outputImageTokens, formatPrice.output),
+          id: 'outputImage',
+          token: outputImageTokens,
+        }
+      : undefined,
     outputReasoning: !!outputReasoningTokens
       ? {
           credit: calcCredit(outputReasoningTokens, formatPrice.output),

package/src/features/Conversation/components/ChatItem/ShareMessageModal/ShareText/index.tsx

@@ -1,3 +1,4 @@
+import { exportFile } from '@lobechat/utils/client';
 import { Button, copyToClipboard } from '@lobehub/ui';
 import { App } from 'antd';
 import isEqual from 'fast-deep-equal';
@@ -10,7 +11,6 @@ import { useIsMobile } from '@/hooks/useIsMobile';
 import { useChatStore } from '@/store/chat';
 import { topicSelectors } from '@/store/chat/selectors';
 import { ChatMessage } from '@/types/message';
-import { exportFile } from '@/utils/client/exportFile';
 
 import { useStyles } from '../style';
 import Preview from './Preview';

package/src/features/Conversation/components/ChatItem/index.tsx

@@ -6,6 +6,7 @@ import { MouseEventHandler, ReactNode, memo, use, useCallback, useMemo } from 'r
 import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
 
+import { HtmlPreviewAction } from '@/components/HtmlPreview';
 import { isDesktop } from '@/const/version';
 import ChatItem from '@/features/ChatItem';
 import { VirtuosoContext } from '@/features/Conversation/components/VirtualizedList/VirtuosoContext';
@@ -33,6 +34,14 @@ import { normalizeThinkTags, processWithArtifact } from './utils';
 const rehypePlugins = markdownElements.map((element) => element.rehypePlugin).filter(Boolean);
 const remarkPlugins = markdownElements.map((element) => element.remarkPlugin).filter(Boolean);
 
+const isHtmlCode = (content: string, language: string) => {
+  return (
+    language === 'html' ||
+    (language === '' && content.includes('<html>')) ||
+    (language === '' && content.includes('<!DOCTYPE html>'))
+  );
+};
+
 const useStyles = createStyles(({ css, prefixCls }) => ({
   loading: css`
     opacity: 0.6;
@@ -175,6 +184,20 @@ const Item = memo<ChatListItemProps>(
       () => ({
         animated,
         citations: item?.role === 'user' ? undefined : item?.search?.citations,
+        componentProps: {
+          highlight: {
+            actionsRender: ({ content, actionIconSize, language, originalNode }: any) => {
+              const showHtmlPreview = isHtmlCode(content, language);
+
+              return (
+                <>
+                  {showHtmlPreview && <HtmlPreviewAction content={content} size={actionIconSize} />}
+                  {originalNode}
+                </>
+              );
+            },
+          },
+        },
         components,
         customRender: markdownCustomRender,
         enableCustomFootnotes: item?.role === 'assistant',

package/src/features/ShareModal/ShareJSON/index.tsx

@@ -1,3 +1,5 @@
+import { FORM_STYLE } from '@lobechat/const';
+import { exportFile } from '@lobechat/utils/client';
 import { Button, Form, type FormItemProps, copyToClipboard } from '@lobehub/ui';
 import { App, Switch } from 'antd';
 import isEqual from 'fast-deep-equal';
@@ -6,13 +8,11 @@ import { memo, useState } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
 
-import { FORM_STYLE } from '@/const/layoutTokens';
 import { useIsMobile } from '@/hooks/useIsMobile';
 import { useAgentStore } from '@/store/agent';
 import { agentSelectors } from '@/store/agent/selectors';
 import { useChatStore } from '@/store/chat';
 import { chatSelectors, topicSelectors } from '@/store/chat/selectors';
-import { exportFile } from '@/utils/client/exportFile';
 
 import { useStyles } from '../style';
 import Preview from './Preview';

package/src/features/ShareModal/ShareText/index.tsx

@@ -1,3 +1,4 @@
+import { exportFile } from '@lobechat/utils/client';
 import { Button, Form, type FormItemProps, copyToClipboard } from '@lobehub/ui';
 import { App, Switch } from 'antd';
 import isEqual from 'fast-deep-equal';
@@ -12,7 +13,6 @@ import { useAgentStore } from '@/store/agent';
 import { agentSelectors } from '@/store/agent/selectors';
 import { useChatStore } from '@/store/chat';
 import { chatSelectors, topicSelectors } from '@/store/chat/selectors';
-import { exportFile } from '@/utils/client/exportFile';
 
 import { useStyles } from '../style';
 import Preview from './Preview';

package/src/libs/oidc-provider/adapter.ts

@@ -164,7 +164,7 @@ class OIDCAdapter {
       log('[%s] Setting userId: %s', this.name, payload.accountId);
     } else {
       try {
-        const { getUserAuth } = await import('@/utils/server/auth');
+        const { getUserAuth } = await import('@lobechat/utils/server');
         try {
           const { userId } = await getUserAuth();
           if (userId) {

package/src/libs/trpc/edge/middleware/jwtPayload.test.ts

@@ -1,11 +1,11 @@
 // @vitest-environment node
+import * as utils from '@lobechat/utils/server';
 import { TRPCError } from '@trpc/server';
 import { beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { createCallerFactory } from '@/libs/trpc/edge';
 import { AuthContext, createContextInner } from '@/libs/trpc/edge/context';
 import { edgeTrpc as trpc } from '@/libs/trpc/edge/init';
-import * as utils from '@/utils/server/xor';
 
 import { jwtPayloadChecker } from './jwtPayload';
 

package/src/libs/trpc/edge/middleware/jwtPayload.ts

@@ -1,7 +1,6 @@
+import { getXorPayload } from '@lobechat/utils/server';
 import { TRPCError } from '@trpc/server';
 
-import { getXorPayload } from '@/utils/server/xor';
-
 import { edgeTrpc } from '../init';
 
 export const jwtPayloadChecker = edgeTrpc.middleware(async (opts) => {

package/src/libs/trpc/lambda/middleware/keyVaults.ts

@@ -1,7 +1,6 @@
+import { getXorPayload } from '@lobechat/utils/server';
 import { TRPCError } from '@trpc/server';
 
-import { getXorPayload } from '@/utils/server/xor';
-
 import { trpc } from '../init';
 
 export const keyVaults = trpc.middleware(async (opts) => {

package/src/locales/default/chat.ts

@@ -128,6 +128,7 @@ export default {
       inputWriteCached: '输入缓存写入',
       output: '输出',
       outputAudio: '音频输出',
+      outputImage: '图像输出',
       outputText: '文本输出',
       outputTitle: '输出明细',
       reasoning: '深度思考',

package/src/locales/default/components.ts

@@ -75,6 +75,18 @@ export default {
   GoBack: {
     back: '返回',
   },
+  HtmlPreview: {
+    actions: {
+      download: '下载',
+      preview: '预览',
+    },
+    iframeTitle: 'HTML 预览',
+    mode: {
+      code: '代码',
+      preview: '预览',
+    },
+    title: 'HTML 预览',
+  },
   ImageUpload: {
     actions: {
       changeImage: '点击更换图片',

package/src/middleware.ts

@@ -1,4 +1,5 @@
 import { clerkMiddleware, createRouteMatcher } from '@clerk/nextjs/server';
+import { parseDefaultThemeFromCountry } from '@lobechat/utils/server';
 import debug from 'debug';
 import { NextRequest, NextResponse } from 'next/server';
 import { UAParser } from 'ua-parser-js';
@@ -11,11 +12,10 @@ import { LOBE_THEME_APPEARANCE } from '@/const/theme';
 import { appEnv } from '@/envs/app';
 import NextAuthEdge from '@/libs/next-auth/edge';
 import { Locales } from '@/locales/resources';
-import { parseBrowserLanguage } from '@/utils/locale';
-import { parseDefaultThemeFromCountry } from '@/utils/server/geo';
-import { RouteVariants } from '@/utils/server/routeVariants';
 
 import { oidcEnv } from './envs/oidc';
+import { parseBrowserLanguage } from './utils/locale';
+import { RouteVariants } from './utils/server/routeVariants';
 
 // Create debug logger instances
 const logDefault = debug('middleware:default');

package/src/server/routers/tools/search.test.ts

@@ -9,7 +9,7 @@ import { SEARCH_SEARXNG_NOT_CONFIG } from '@/types/tool/search';
 import { searchRouter } from './search';
 
 // Mock JWT verification
-vi.mock('@/utils/server/xor', () => ({
+vi.mock('@lobechat/utils/server', () => ({
   getXorPayload: vi.fn().mockReturnValue({ userId: '1' }),
 }));
 

package/src/services/config.ts

@@ -1,11 +1,9 @@
+import { BRANDING_NAME, isDeprecatedEdition, isServerMode } from '@lobechat/const';
+import { downloadFile, exportJSONFile } from '@lobechat/utils/client';
 import dayjs from 'dayjs';
 
-import { BRANDING_NAME } from '@/const/branding';
-import { isDeprecatedEdition, isServerMode } from '@/const/version';
 import { CURRENT_CONFIG_VERSION } from '@/migrations';
 import { ImportPgDataStructure } from '@/types/export';
-import { downloadFile } from '@/utils/client/downloadFile';
-import { exportJSONFile } from '@/utils/client/exportFile';
 
 import { exportService } from './export';
 import { configService as deprecatedExportService } from './export/_deprecated';

package/src/utils/client/switchLang.ts

@@ -1,7 +1,7 @@
-import { LOBE_LOCALE_COOKIE } from '@lobechat/const';
 import { setCookie } from '@lobechat/utils';
 import { changeLanguage } from 'i18next';
 
+import { LOBE_LOCALE_COOKIE } from '@/const/locale';
 import { LocaleMode } from '@/types/locale';
 
 export const switchLang = (locale: LocaleMode) => {

package/{packages/utils/src → src/utils}/server/pageProps.ts

@@ -1,6 +1,7 @@
 import { translation } from '@/server/translation';
 import { DynamicLayoutProps } from '@/types/next';
-import { RouteVariants } from '@/utils/server/routeVariants';
+
+import { RouteVariants } from './routeVariants';
 
 export const parsePageMetaProps = async (props: DynamicLayoutProps) => {
   const { locale: hl, isMobile } = await RouteVariants.getVariantsFromProps(props);

package/tsconfig.json

@@ -21,7 +21,7 @@
       "@/libs/model-runtime": ["./packages/model-runtime/src/index.ts"],
       "@/libs/model-runtime/*": ["./packages/model-runtime/src/*"],
       "@/database/*": ["./packages/database/src/*", "./src/database/*"],
-      "@/const/*": ["./packages/const/src/*"],
+      "@/const/*": ["./packages/const/src/*", "./src/const/*"],
       "@/utils/*": ["./packages/utils/src/*", "./src/utils/*"],
       "@/types/*": ["./packages/types/src/*", "./src/types/*"],
       "@/*": ["./src/*"],

package/vitest.config.mts

@@ -13,6 +13,7 @@ export default defineConfig({
       '@/database/_deprecated': resolve(__dirname, './src/database/_deprecated'),
       '@/database': resolve(__dirname, './packages/database/src'),
       '@/utils/client/switchLang': resolve(__dirname, './src/utils/client/switchLang'),
+      '@/const/locale': resolve(__dirname, './src/const/locale'),
       // TODO: after refactor the errorResponse, we can remove it
       '@/utils/errorResponse': resolve(__dirname, './src/utils/errorResponse'),
       '@/utils': resolve(__dirname, './packages/utils/src'),

package/packages/model-runtime/src/UniformRuntime/index.ts

@@ -1,117 +0,0 @@
-/**
- * @see https://github.com/lobehub/lobe-chat/discussions/6563
- */
-import { LobeRuntimeAI } from '../BaseAI';
-import { LobeOpenAI } from '../openai';
-import { providerRuntimeMap } from '../runtimeMap';
-import {
-  type ChatCompletionErrorPayload,
-  ChatMethodOptions,
-  ChatStreamPayload,
-  EmbeddingsOptions,
-  EmbeddingsPayload,
-  TextToImagePayload,
-  TextToSpeechPayload,
-} from '../types';
-
-export interface RuntimeItem {
-  id: string;
-  models?: string[];
-  runtime: LobeRuntimeAI;
-}
-
-interface ProviderInitParams extends Record<string, any> {
-  accessKeyId?: string;
-  accessKeySecret?: string;
-  apiKey?: string;
-  apiVersion?: string;
-  baseURL?: string;
-  baseURLOrAccountID?: string;
-  dangerouslyAllowBrowser?: boolean;
-  region?: string;
-  sessionToken?: string;
-}
-
-interface ProviderInstance {
-  apiType: keyof typeof providerRuntimeMap;
-  models?: string[];
-  params: ProviderInitParams;
-  runtime?: typeof LobeOpenAI;
-}
-
-interface UniformRuntimeOptions {
-  chat?: {
-    handleError?: (error: any) => Omit<ChatCompletionErrorPayload, 'provider'> | undefined;
-  };
-}
-
-class UniformRuntime {
-  private _runtimes: RuntimeItem[];
-  private _options: UniformRuntimeOptions;
-
-  constructor(id: string, providers: ProviderInstance[], options: UniformRuntimeOptions) {
-    if (providers.length === 0) {
-      throw new Error('empty providers');
-    }
-
-    this._runtimes = providers.map((options) => {
-      const providerAI = options.runtime ?? providerRuntimeMap[options.apiType] ?? LobeOpenAI;
-      const runtime: LobeRuntimeAI = new providerAI({ ...options.params, id });
-
-      return { id: options.apiType, models: options.models, runtime };
-    });
-
-    this._options = options;
-  }
-
-  // 检查下是否能匹配到特定模型，否则默认使用第一个 runtime
-  getRuntimeByModel(model: string) {
-    const runtimeItem =
-      this._runtimes.find((runtime) => runtime.models && runtime.models.includes(model)) ||
-      this._runtimes[0];
-
-    return runtimeItem.runtime;
-  }
-
-  async chat(payload: ChatStreamPayload, options?: ChatMethodOptions) {
-    try {
-      const runtime = this.getRuntimeByModel(payload.model);
-
-      return await runtime.chat!(payload, options);
-    } catch (e) {
-      if (this._options.chat?.handleError) {
-        const error = this._options.chat.handleError(e);
-
-        if (error) {
-          throw error;
-        }
-      }
-
-      throw e;
-    }
-  }
-
-  async textToImage(payload: TextToImagePayload) {
-    const runtime = this.getRuntimeByModel(payload.model);
-
-    return runtime.textToImage?.(payload);
-  }
-
-  async models() {
-    return this._runtimes[0].runtime.models?.();
-  }
-
-  async embeddings(payload: EmbeddingsPayload, options?: EmbeddingsOptions) {
-    const runtime = this.getRuntimeByModel(payload.model);
-
-    return runtime.embeddings?.(payload, options);
-  }
-
-  async textToSpeech(payload: TextToSpeechPayload, options?: EmbeddingsOptions) {
-    const runtime = this.getRuntimeByModel(payload.model);
-
-    return runtime.textToSpeech?.(payload, options);
-  }
-}
-
-export default UniformRuntime;