@lobehub/lobehub 2.0.0-next.46 → 2.0.0-next.48

package/packages/database/migrations/meta/_journal.json

@@ -308,6 +308,20 @@
       "when": 1762251112601,
       "tag": "0043_add_ai_model_settings",
       "breakpoints": true
+    },
+    {
+      "idx": 44,
+      "version": "7",
+      "when": 1762870034882,
+      "tag": "0044_high_toxin",
+      "breakpoints": true
+    },
+    {
+      "idx": 45,
+      "version": "7",
+      "when": 1762911968658,
+      "tag": "0045_add_tool_intervention",
+      "breakpoints": true
     }
   ],
   "version": "6"

package/packages/database/src/core/migrations.json

@@ -223,7 +223,10 @@
     "hash": "9646161fa041354714f823d726af27247bcd6e60fa3be5698c0d69f337a5700b"
   },
   {
-    "sql": ["DROP TABLE \"user_budgets\";", "\nDROP TABLE \"user_subscriptions\";"],
+    "sql": [
+      "DROP TABLE \"user_budgets\";",
+      "\nDROP TABLE \"user_subscriptions\";"
+    ],
     "bps": true,
     "folderMillis": 1729699958471,
     "hash": "7dad43a2a25d1aec82124a4e53f8d82f8505c3073f23606c1dc5d2a4598eacf9"
@@ -295,7 +298,9 @@
     "hash": "845a692ceabbfc3caf252a97d3e19a213bc0c433df2689900135f9cfded2cf49"
   },
   {
-    "sql": ["ALTER TABLE \"messages\" ADD COLUMN \"reasoning\" jsonb;"],
+    "sql": [
+      "ALTER TABLE \"messages\" ADD COLUMN \"reasoning\" jsonb;"
+    ],
     "bps": true,
     "folderMillis": 1737609172353,
     "hash": "2cb36ae4fcdd7b7064767e04bfbb36ae34518ff4bb1b39006f2dd394d1893868"
@@ -510,7 +515,9 @@
     "hash": "a7ccf007fd185ff922823148d1eae6fafe652fc98d2fd2793f84a84f29e93cd1"
   },
   {
-    "sql": ["ALTER TABLE \"ai_providers\" ADD COLUMN \"config\" jsonb;"],
+    "sql": [
+      "ALTER TABLE \"ai_providers\" ADD COLUMN \"config\" jsonb;"
+    ],
     "bps": true,
     "folderMillis": 1749309388370,
     "hash": "39cea379f08ee4cb944875c0b67f7791387b508c2d47958bb4cd501ed1ef33eb"
@@ -628,7 +635,9 @@
     "hash": "1ba9b1f74ea13348da98d6fcdad7867ab4316ed565bf75d84d160c526cdac14b"
   },
   {
-    "sql": ["ALTER TABLE \"agents\" ADD COLUMN IF NOT EXISTS \"virtual\" boolean DEFAULT false;"],
+    "sql": [
+      "ALTER TABLE \"agents\" ADD COLUMN IF NOT EXISTS \"virtual\" boolean DEFAULT false;"
+    ],
     "bps": true,
     "folderMillis": 1759116400580,
     "hash": "433ddae88e785f2db734e49a4c115eee93e60afe389f7919d66e5ba9aa159a37"
@@ -678,13 +687,17 @@
     "hash": "4bdc6505797d7a33b622498c138cfd47f637239f6905e1c484cd01d9d5f21d6b"
   },
   {
-    "sql": ["ALTER TABLE \"user_settings\" ADD COLUMN IF NOT EXISTS \"image\" jsonb;"],
+    "sql": [
+      "ALTER TABLE \"user_settings\" ADD COLUMN IF NOT EXISTS \"image\" jsonb;"
+    ],
     "bps": true,
     "folderMillis": 1760108430562,
     "hash": "ce09b301abb80f6563abc2f526bdd20b4f69bae430f09ba2179b9e3bfec43067"
   },
   {
-    "sql": ["ALTER TABLE \"documents\" ADD COLUMN IF NOT EXISTS \"editor_data\" jsonb;"],
+    "sql": [
+      "ALTER TABLE \"documents\" ADD COLUMN IF NOT EXISTS \"editor_data\" jsonb;"
+    ],
     "bps": true,
     "folderMillis": 1761554153406,
     "hash": "bf2f21293e90e11cf60a784cf3ec219eafa95f7545d7d2f9d1449c0b0949599a"
@@ -762,5 +775,21 @@
     "bps": true,
     "folderMillis": 1762251112601,
     "hash": "923ccbdf46c32be9a981dabd348e6923b4a365444241e9b8cc174bf5b914cbc5"
+  },
+  {
+    "sql": [
+      "ALTER TABLE \"agents\" ADD COLUMN IF NOT EXISTS \"market_identifier\" text;\n"
+    ],
+    "bps": true,
+    "folderMillis": 1762870034882,
+    "hash": "4178aacb4b8892b7fd15d29209bbf9b1d1f9d7c406ba796f27542c0bcd919680"
+  },
+  {
+    "sql": [
+      "ALTER TABLE \"message_plugins\" ADD COLUMN IF NOT EXISTS \"intervention\" jsonb;\n"
+    ],
+    "bps": true,
+    "folderMillis": 1762911968658,
+    "hash": "552a032cc0e595277232e70b5f9338658585bafe9481ae8346a5f322b673a68b"
   }
-]
+]

package/packages/database/src/models/file.ts

@@ -25,8 +25,22 @@ export class FileModel {
     this.db = db;
   }
 
+  /**
+   * Get file by ID without userId filter (public access)
+   * Use this for scenarios like file proxy where file should be accessible by ID alone
+   *
+   * @param db - Database instance
+   * @param id - File ID
+   * @returns File record or undefined
+   */
+  static async getFileById(db: LobeChatDatabase, id: string): Promise<FileItem | undefined> {
+    return db.query.files.findFirst({
+      where: eq(files.id, id),
+    });
+  }
+
   create = async (
-    params: Omit<NewFile, 'id' | 'userId'> & { knowledgeBaseId?: string },
+    params: Omit<NewFile, 'id' | 'userId'> & { id?: string; knowledgeBaseId?: string },
     insertToGlobalFiles?: boolean,
     trx?: Transaction,
   ) => {

package/packages/database/src/models/message.ts

@@ -8,6 +8,7 @@ import {
   ChatVideoItem,
   CreateMessageParams,
   DBMessageItem,
+  MessagePluginItem,
   ModelRankItem,
   NewMessageQueryParams,
   QueryMessageParams,
@@ -23,7 +24,6 @@ import { merge } from '@/utils/merge';
 import { today } from '@/utils/time';
 
 import {
-  MessagePluginItem,
   chunks,
   documents,
   embeddings,

package/packages/database/src/models/session.ts

@@ -256,6 +256,31 @@ export class SessionModel {
         if (existResult) return existResult;
       }
 
+      // Extract and properly map fields for agent creation from DiscoverAssistantDetail
+      const {
+        // MetaData fields (from discover assistant)
+        title,
+        description,
+        tags = [],
+        avatar,
+        backgroundColor,
+        // LobeAgentConfig fields
+        model,
+        params,
+        systemRole,
+        provider,
+        plugins = [],
+        openingMessage,
+        openingQuestions = [],
+        // TTS config
+        tts,
+        // Chat config
+        chatConfig,
+        // Field name mapping
+        examples, // maps to fewShots
+        identifier, // maps to marketIdentifier
+        marketIdentifier,
+      } = config as any;
       if (type === 'group') {
         const result = await trx
           .insert(sessions)
@@ -276,9 +301,24 @@ export class SessionModel {
       const newAgents = await trx
         .insert(agents)
         .values({
-          ...config,
+          avatar,
+          backgroundColor,
+          chatConfig: chatConfig || {},
           createdAt: new Date(),
+          description,
+          fewShots: examples || null, // Map examples to fewShots field
           id: idGenerator('agents'),
+          marketIdentifier: identifier || marketIdentifier,
+          model: typeof model === 'string' ? model : null,
+          openingMessage,
+          openingQuestions,
+          params: params || {},
+          plugins,
+          provider,
+          systemRole,
+          tags,
+          title,
+          tts: tts || {},
           updatedAt: new Date(),
           userId: this.userId,
         })
@@ -595,6 +635,7 @@ export class SessionModel {
         avatar: agent?.avatar ?? avatar ?? undefined,
         backgroundColor: agent?.backgroundColor ?? backgroundColor ?? undefined,
         description: agent?.description ?? description ?? undefined,
+        marketIdentifier: agent?.marketIdentifier ?? undefined,
         tags: agent?.tags ?? undefined,
         title: agent?.title ?? title ?? undefined,
       },

package/packages/database/src/repositories/aiInfra/index.test.ts

@@ -25,7 +25,7 @@ beforeEach(async () => {
   vi.clearAllMocks();
 
   repo = new AiInfraRepos(clientDB as any, userId, mockProviderConfigs);
-});
+}, 30000);
 
 describe('AiInfraRepos', () => {
   describe('getAiProviderList', () => {

package/packages/database/src/repositories/dataExporter/index.test.ts

@@ -157,7 +157,7 @@ describe('DataExporterRepos', () => {
 
     // 插入测试数据
     await setupTestData();
-  });
+  }, 30000);
 
   afterEach(async () => {
     await db.delete(users);

package/packages/database/src/repositories/tableViewer/index.test.ts

@@ -16,7 +16,7 @@ const mockDB = {
 beforeEach(async () => {
   await initializeDB();
   vi.clearAllMocks();
-});
+}, 30000);
 
 describe('TableViewerRepo', () => {
   describe('getAllTables', () => {

package/packages/database/src/schemas/agent.ts

@@ -37,6 +37,7 @@ export const agents = pgTable(
     tags: jsonb('tags').$type<string[]>().default([]),
     avatar: text('avatar'),
     backgroundColor: text('background_color'),
+    marketIdentifier: text('market_identifier'),
 
     plugins: jsonb('plugins').$type<string[]>().default([]),
 

package/packages/database/src/schemas/message.ts

@@ -1,5 +1,5 @@
 /* eslint-disable sort-keys-fix/sort-keys-fix  */
-import { GroundingSearch, ModelReasoning } from '@lobechat/types';
+import { GroundingSearch, ModelReasoning, ToolIntervention } from '@lobechat/types';
 import {
   boolean,
   index,
@@ -11,7 +11,7 @@ import {
   uniqueIndex,
   uuid,
 } from 'drizzle-orm/pg-core';
-import { createInsertSchema, createSelectSchema } from 'drizzle-zod';
+import { createInsertSchema } from 'drizzle-zod';
 
 import { idGenerator } from '../utils/idGenerator';
 import { timestamps, varchar255 } from './_helpers';
@@ -146,10 +146,10 @@ export const messagePlugins = pgTable(
       .primaryKey(),
 
     toolCallId: text('tool_call_id'),
-    type: text('type', {
-      enum: ['default', 'markdown', 'standalone', 'builtin'],
-    }).default('default'),
+    type: text('type').default('default'),
 
+    // Human intervention fields
+    intervention: jsonb('intervention').$type<ToolIntervention>(),
     apiName: text('api_name'),
     arguments: text('arguments'),
     identifier: text('identifier'),
@@ -168,9 +168,6 @@ export const messagePlugins = pgTable(
   }),
 );
 
-export type MessagePluginItem = typeof messagePlugins.$inferSelect;
-export const updateMessagePluginSchema = createSelectSchema(messagePlugins);
-
 export const messageTTS = pgTable(
   'message_tts',
   {

package/packages/electron-client-ipc/src/events/index.ts

@@ -2,7 +2,11 @@ import { LocalSystemDispatchEvents } from './localSystem';
 import { MenuDispatchEvents } from './menu';
 import { NotificationDispatchEvents } from './notification';
 import { ProtocolBroadcastEvents, ProtocolDispatchEvents } from './protocol';
-import { RemoteServerBroadcastEvents, RemoteServerDispatchEvents } from './remoteServer';
+import {
+  
+  RemoteServerBroadcastEvents,
+  RemoteServerDispatchEvents,
+} from './remoteServer';
 import { DesktopSettingsDispatchEvents } from './settings';
 import { ShortcutDispatchEvents } from './shortcut';
 import { SystemBroadcastEvents, SystemDispatchEvents } from './system';
@@ -51,4 +55,5 @@ export type MainBroadcastParams<T extends MainBroadcastEventKey> = Parameters<
   MainBroadcastEvents[T]
 >[0];
 
+export type { MarketAuthorizationParams } from './remoteServer';
 export type { OpenSettingsWindowOptions } from './windows';

package/packages/electron-client-ipc/src/events/remoteServer.ts

@@ -1,6 +1,10 @@
 import { DataSyncConfig } from '../types/dataSync';
 import { ProxyTRPCRequestParams, ProxyTRPCRequestResult } from '../types/proxyTRPCRequest';
 
+export interface MarketAuthorizationParams {
+  authUrl: string;
+}
+
 /**
  * 远程服务器配置相关的事件
  */
@@ -21,6 +25,10 @@ export interface RemoteServerDispatchEvents {
     error?: string;
     success: boolean;
   };
+  requestMarketAuthorization: (params: MarketAuthorizationParams) => {
+    error?: string;
+    success: boolean;
+  };
   setRemoteServerConfig: (config: DataSyncConfig) => boolean;
 }
 

package/packages/fetch-sse/package.json

@@ -0,0 +1,29 @@
+{
+  "name": "@lobechat/fetch-sse",
+  "version": "1.0.0",
+  "private": true,
+  "description": "SSE fetch utilities with streaming support",
+  "exports": {
+    ".": {
+      "types": "./src/index.ts",
+      "default": "./src/index.ts"
+    },
+    "./parseError": {
+      "types": "./src/parseError.ts",
+      "default": "./src/parseError.ts"
+    }
+  },
+  "main": "./src/index.ts",
+  "types": "./src/index.ts",
+  "scripts": {
+    "test": "vitest",
+    "test:coverage": "vitest --coverage --silent='passed-only'"
+  },
+  "dependencies": {
+    "@lobechat/const": "workspace:*",
+    "@lobechat/model-runtime": "workspace:*",
+    "@lobechat/types": "workspace:*",
+    "@lobechat/utils": "workspace:*",
+    "i18next": "^24.2.1"
+  }
+}

package/packages/{utils/src/fetch → fetch-sse/src}/__tests__/fetchSSE.test.ts

@@ -1,10 +1,10 @@
 import { MESSAGE_CANCEL_FLAT } from '@lobechat/const';
 import { ChatMessageError } from '@lobechat/types';
+import { FetchEventSourceInit } from '@lobechat/utils/client/fetchEventSource/index';
+import { fetchEventSource } from '@lobechat/utils/client/fetchEventSource/index';
+import { sleep } from '@lobechat/utils/sleep';
 import { afterEach, describe, expect, it, vi } from 'vitest';
 
-import { FetchEventSourceInit } from '../../client/fetchEventSource';
-import { fetchEventSource } from '../../client/fetchEventSource';
-import { sleep } from '../../sleep';
 import { fetchSSE } from '../fetchSSE';
 
 // 模拟 i18next
@@ -12,7 +12,7 @@ vi.mock('i18next', () => ({
   t: vi.fn((key) => `translated_${key}`),
 }));
 
-vi.mock('../../client/fetchEventSource', () => ({
+vi.mock('@lobechat/utils/client/fetchEventSource/index', () => ({
   fetchEventSource: vi.fn(),
 }));
 

package/packages/{utils/src/fetch → fetch-sse/src}/__tests__/parseError.test.ts

@@ -1,14 +1,14 @@
 import { ErrorResponse } from '@lobechat/types';
-import { afterEach, describe, expect, it, vi } from 'vitest';
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { getMessageError } from '../parseError';
 
-// 模拟 i18next
+// Mock i18next
 vi.mock('i18next', () => ({
   t: vi.fn((key) => `translated_${key}`),
 }));
 
-// 模拟 Response
+// Mock Response
 const createMockResponse = (body: any, ok: boolean, status: number = 200) => ({
   ok,
   status,
@@ -38,11 +38,14 @@ const createMockResponse = (body: any, ok: boolean, status: number = 200) => ({
   },
 });
 
-// 在每次测试后清理所有模拟
 afterEach(() => {
   vi.restoreAllMocks();
 });
 
+beforeEach(() => {
+  vi.clearAllMocks();
+});
+
 describe('getMessageError', () => {
   it('should handle business error correctly', async () => {
     const mockErrorResponse: ErrorResponse = {

package/packages/{utils/src/fetch → fetch-sse/src}/fetchSSE.ts

@@ -12,9 +12,9 @@ import {
   ResponseAnimation,
   ResponseAnimationStyle,
 } from '@lobechat/types';
+import { fetchEventSource } from '@lobechat/utils/client/fetchEventSource/index';
+import { nanoid } from '@lobechat/utils/uuid';
 
-import { fetchEventSource } from '../client/fetchEventSource';
-import { nanoid } from '../uuid';
 import { getMessageError } from './parseError';
 
 type SSEFinishType = 'done' | 'error' | 'abort';

package/packages/{utils/src/fetch → fetch-sse/src}/parseError.ts

@@ -1,7 +1,7 @@
 import { ChatMessageError, ErrorResponse, ErrorType } from '@lobechat/types';
 import { t } from 'i18next';
 
-export const getMessageError = async (response: Response) => {
+export const getMessageError = async (response: Response): Promise<ChatMessageError> => {
   let chatMessageError: ChatMessageError;
 
   // try to get the biz error
@@ -9,13 +9,13 @@ export const getMessageError = async (response: Response) => {
     const data = (await response.json()) as ErrorResponse;
     chatMessageError = {
       body: data.body,
-      message: t(`response.${data.errorType}` as any, { ns: 'error' }),
+      message: t(`response.${data.errorType}`, { ns: 'error' }),
       type: data.errorType,
     };
   } catch {
     // if not return, then it's a common error
     chatMessageError = {
-      message: t(`response.${response.status}` as any, { ns: 'error' }),
+      message: t(`response.${response.status}`, { ns: 'error' }),
       type: response.status as ErrorType,
     };
   }

package/packages/model-bank/src/aiModels/mistral.ts

@@ -45,7 +45,8 @@ const mistralChatModels: AIChatModelCard[] = [
       vision: true,
     },
     contextWindowTokens: 128_000,
-    description: 'Magistral Small 1.2 是Mistral AI于2025年9月发布的开源小型推理模型，具有视觉支持。',
+    description:
+      'Magistral Small 1.2 是Mistral AI于2025年9月发布的开源小型推理模型，具有视觉支持。',
     displayName: 'Magistral Small 1.2',
     id: 'magistral-small-2509',
     pricing: {

package/packages/model-runtime/src/core/contextBuilders/anthropic.test.ts

@@ -1,8 +1,8 @@
+import { imageUrlToBase64 } from '@lobechat/utils';
 import { OpenAI } from 'openai';
-import { describe, expect, it, vi } from 'vitest';
+import { beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { OpenAIChatMessage, UserMessageContentPart } from '../../types/chat';
-import { imageUrlToBase64 } from '../../utils/imageToBase64';
 import { parseDataUri } from '../../utils/uriParser';
 import {
   buildAnthropicBlock,
@@ -12,16 +12,22 @@ import {
 } from './anthropic';
 
 // Mock the parseDataUri function since it's an implementation detail
-vi.mock('../../utils/uriParser', () => ({
-  parseDataUri: vi.fn().mockReturnValue({
-    mimeType: 'image/jpeg',
-    base64: 'base64EncodedString',
-    type: 'base64',
-  }),
+vi.mock('../../utils/uriParser');
+vi.mock('@lobechat/utils', () => ({
+  imageUrlToBase64: vi.fn(),
 }));
-vi.mock('../../utils/imageToBase64');
 
 describe('anthropicHelpers', () => {
+  beforeEach(() => {
+    vi.resetAllMocks();
+    // Set default mock implementation for parseDataUri
+    vi.mocked(parseDataUri).mockReturnValue({
+      mimeType: 'image/jpeg',
+      base64: 'base64EncodedString',
+      type: 'base64',
+    });
+  });
+
   describe('buildAnthropicBlock', () => {
     it('should return the content as is for text type', async () => {
       const content: UserMessageContentPart = { type: 'text', text: 'Hello!' };
@@ -52,7 +58,7 @@ describe('anthropicHelpers', () => {
         base64: null,
         type: 'url',
       });
-      vi.mocked(imageUrlToBase64).mockResolvedValue({
+      vi.mocked(imageUrlToBase64).mockResolvedValueOnce({
         base64: 'convertedBase64String',
         mimeType: 'image/jpg',
       });
@@ -82,7 +88,7 @@ describe('anthropicHelpers', () => {
         base64: null,
         type: 'url',
       });
-      vi.mocked(imageUrlToBase64).mockResolvedValue({
+      vi.mocked(imageUrlToBase64).mockResolvedValueOnce({
         base64: 'convertedBase64String',
         mimeType: 'image/png',
       });

package/packages/model-runtime/src/core/contextBuilders/anthropic.ts

@@ -1,8 +1,8 @@
 import Anthropic from '@anthropic-ai/sdk';
+import { imageUrlToBase64 } from '@lobechat/utils';
 import OpenAI from 'openai';
 
 import { OpenAIChatMessage, UserMessageContentPart } from '../../types';
-import { imageUrlToBase64 } from '../../utils/imageToBase64';
 import { parseDataUri } from '../../utils/uriParser';
 
 export const buildAnthropicBlock = async (

package/packages/model-runtime/src/core/contextBuilders/google.test.ts

@@ -1,9 +1,9 @@
 // @vitest-environment node
 import { Type as SchemaType } from '@google/genai';
+import * as imageToBase64Module from '@lobechat/utils';
 import { describe, expect, it, vi } from 'vitest';
 
 import { ChatCompletionTool, OpenAIChatMessage, UserMessageContentPart } from '../../types';
-import * as imageToBase64Module from '../../utils/imageToBase64';
 import { parseDataUri } from '../../utils/uriParser';
 import {
   buildGoogleMessage,

package/packages/model-runtime/src/core/contextBuilders/google.ts

@@ -5,9 +5,9 @@ import {
   Part,
   Type as SchemaType,
 } from '@google/genai';
+import { imageUrlToBase64 } from '@lobechat/utils';
 
 import { ChatCompletionTool, OpenAIChatMessage, UserMessageContentPart } from '../../types';
-import { imageUrlToBase64 } from '../../utils/imageToBase64';
 import { safeParseJSON } from '../../utils/safeParseJSON';
 import { parseDataUri } from '../../utils/uriParser';
 
@@ -64,12 +64,9 @@ export const buildGooglePart = async (
       }
 
       if (type === 'url') {
-        // For video URLs, we need to fetch and convert to base64
+        // Use imageUrlToBase64 for SSRF protection (works for any binary data including videos)
         // Note: This might need size/duration limits for practical use
-        const response = await fetch(content.video_url.url);
-        const arrayBuffer = await response.arrayBuffer();
-        const base64 = Buffer.from(arrayBuffer).toString('base64');
-        const mimeType = response.headers.get('content-type') || 'video/mp4';
+        const { base64, mimeType } = await imageUrlToBase64(content.video_url.url);
 
         return {
           inlineData: { data: base64, mimeType },

package/packages/model-runtime/src/core/contextBuilders/openai.test.ts

@@ -1,8 +1,8 @@
+import { imageUrlToBase64 } from '@lobechat/utils';
 import OpenAI from 'openai';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { OpenAIChatMessage } from '../../types';
-import { imageUrlToBase64 } from '../../utils/imageToBase64';
 import { parseDataUri } from '../../utils/uriParser';
 import {
   convertImageUrlToFile,
@@ -12,7 +12,9 @@ import {
 } from './openai';
 
 // 模拟依赖
-vi.mock('../../utils/imageToBase64');
+vi.mock('@lobechat/utils', () => ({
+  imageUrlToBase64: vi.fn(),
+}));
 vi.mock('../../utils/uriParser');
 
 describe('convertMessageContent', () => {

package/packages/model-runtime/src/core/contextBuilders/openai.ts

@@ -1,8 +1,8 @@
+import { imageUrlToBase64 } from '@lobechat/utils';
 import OpenAI, { toFile } from 'openai';
 
 import { disableStreamModels, systemToUserModels } from '../../const/models';
 import { ChatStreamPayload, OpenAIChatMessage } from '../../types';
-import { imageUrlToBase64 } from '../../utils/imageToBase64';
 import { parseDataUri } from '../../utils/uriParser';
 
 export const convertMessageContent = async (

package/packages/model-runtime/src/core/openaiCompatibleFactory/createImage.test.ts

@@ -1,9 +1,9 @@
 // @vitest-environment node
+import * as imageToBase64Module from '@lobechat/utils';
 import OpenAI from 'openai';
 import { beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { CreateImagePayload } from '../../types/image';
-import * as imageToBase64Module from '../../utils/imageToBase64';
 import * as uriParserModule from '../../utils/uriParser';
 import { createOpenAICompatibleImage } from './createImage';
 

package/packages/model-runtime/src/core/openaiCompatibleFactory/createImage.ts

@@ -1,3 +1,4 @@
+import { imageUrlToBase64 } from '@lobechat/utils';
 import { cleanObject } from '@lobechat/utils/object';
 import createDebug from 'debug';
 import { RuntimeImageGenParamsValue } from 'model-bank';
@@ -5,7 +6,6 @@ import OpenAI from 'openai';
 
 import { CreateImagePayload, CreateImageResponse } from '../../types/image';
 import { getModelPricing } from '../../utils/getModelPricing';
-import { imageUrlToBase64 } from '../../utils/imageToBase64';
 import { parseDataUri } from '../../utils/uriParser';
 import { convertImageUrlToFile } from '../contextBuilders/openai';
 import { convertOpenAIImageUsage } from '../usageConverters/openai';

package/packages/model-runtime/src/core/openaiCompatibleFactory/index.test.ts

@@ -1,15 +1,12 @@
 // @vitest-environment node
-import {
-  AgentRuntimeErrorType,
-  ChatStreamCallbacks,
-  ChatStreamPayload,
-  LobeOpenAICompatibleRuntime,
-} from '@lobechat/model-runtime';
 import { ModelProvider } from 'model-bank';
 import OpenAI from 'openai';
 import type { Stream } from 'openai/streaming';
 import { Mock, afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
+import { LobeOpenAICompatibleRuntime } from '../../core/BaseAI';
+import { ChatStreamCallbacks, ChatStreamPayload } from '../../types/chat';
+import { AgentRuntimeErrorType } from '../../types/error';
 import * as debugStreamModule from '../../utils/debugStream';
 import * as openaiHelpers from '../contextBuilders/openai';
 import { createOpenAICompatibleRuntime } from './index';