@lobehub/lobehub 2.0.0-next.7 → 2.0.0-next.9

package/src/store/tool/slices/builtin/selectors.ts

@@ -1,32 +1,26 @@
 import { LobeToolMeta } from '@lobechat/types';
 
 import { shouldEnableTool } from '@/helpers/toolFilters';
-import { DalleManifest } from '@/tools/dalle';
 
 import type { ToolStoreState } from '../../initialState';
 
-const metaList =
-  (showDalle?: boolean) =>
-  (s: ToolStoreState): LobeToolMeta[] =>
-    s.builtinTools
-      .filter((item) => {
-        // Filter hidden tools
-        if (item.hidden) return false;
+const metaList = (s: ToolStoreState): LobeToolMeta[] =>
+  s.builtinTools
+    .filter((item) => {
+      // Filter hidden tools
+      if (item.hidden) return false;
 
-        // Filter Dalle if not enabled
-        if (!showDalle && item.identifier === DalleManifest.identifier) return false;
+      // Filter platform-specific tools (e.g., LocalSystem desktop-only)
+      if (!shouldEnableTool(item.identifier)) return false;
 
-        // Filter platform-specific tools (e.g., LocalSystem desktop-only)
-        if (!shouldEnableTool(item.identifier)) return false;
-
-        return true;
-      })
-      .map((t) => ({
-        author: 'LobeHub',
-        identifier: t.identifier,
-        meta: t.manifest.meta,
-        type: 'builtin',
-      }));
+      return true;
+    })
+    .map((t) => ({
+      author: 'LobeHub',
+      identifier: t.identifier,
+      meta: t.manifest.meta,
+      type: 'builtin',
+    }));
 
 export const builtinToolSelectors = {
   metaList,

package/src/tools/index.ts

@@ -4,7 +4,6 @@ import { isDesktop } from '@/const/version';
 
 import { ArtifactsManifest } from './artifacts';
 import { CodeInterpreterManifest } from './code-interpreter';
-import { DalleManifest } from './dalle';
 import { LocalSystemManifest } from './local-system';
 import { WebBrowsingManifest } from './web-browsing';
 
@@ -14,11 +13,6 @@ export const builtinTools: LobeBuiltinTool[] = [
     manifest: ArtifactsManifest,
     type: 'builtin',
   },
-  {
-    identifier: DalleManifest.identifier,
-    manifest: DalleManifest,
-    type: 'builtin',
-  },
   {
     hidden: !isDesktop,
     identifier: LocalSystemManifest.identifier,

package/src/tools/renders.ts

@@ -2,15 +2,12 @@ import { BuiltinRender } from '@lobechat/types';
 
 import { CodeInterpreterManifest } from './code-interpreter';
 import CodeInterpreterRender from './code-interpreter/Render';
-import { DalleManifest } from './dalle';
-import DalleRender from './dalle/Render';
 import { LocalSystemManifest } from './local-system';
 import LocalFilesRender from './local-system/Render';
 import { WebBrowsingManifest } from './web-browsing';
 import WebBrowsing from './web-browsing/Render';
 
 export const BuiltinToolsRenders: Record<string, BuiltinRender> = {
-  [DalleManifest.identifier]: DalleRender as BuiltinRender,
   [WebBrowsingManifest.identifier]: WebBrowsing as BuiltinRender,
   [LocalSystemManifest.identifier]: LocalFilesRender as BuiltinRender,
   [CodeInterpreterManifest.identifier]: CodeInterpreterRender as BuiltinRender,

package/src/tools/web-browsing/Portal/Search/Footer.tsx

@@ -4,12 +4,12 @@ import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
 
 import { useChatStore } from '@/store/chat';
-import { chatPortalSelectors, chatSelectors } from '@/store/chat/selectors';
+import { chatPortalSelectors, messageStateSelectors } from '@/store/chat/selectors';
 
 const Footer = () => {
   const [messageId, isAIGenerating, triggerAIMessage, saveSearchResult] = useChatStore((s) => [
     chatPortalSelectors.toolMessageId(s),
-    chatSelectors.isAIGenerating(s),
+    messageStateSelectors.isAIGenerating(s),
     s.triggerAIMessage,
     s.saveSearchResult,
   ]);

package/packages/const/src/guide.ts

@@ -1,89 +0,0 @@
-import urlJoin from 'url-join';
-
-import { BRANDING_EMAIL } from './branding';
-import {
-  BLOG,
-  DOCKER_IMAGE,
-  GITHUB,
-  OFFICIAL_PREVIEW_URL,
-  OFFICIAL_SITE,
-  OFFICIAL_URL,
-  SELF_HOSTING_DOCUMENTS,
-  USAGE_DOCUMENTS,
-  WIKI,
-} from './url';
-
-export const INBOX_GUIDE_SYSTEMROLE = `# Role: LobeChat Support Assistant
-
-## About [LobeHub](${OFFICIAL_SITE})
-
-LobeHub is an organization of design-engineers dedicated to providing advanced design components and tools for AI-generated content (AIGC).
-It aims to create a technology-driven community platform that enables the sharing of knowledge and ideas, fostering inspiration and collaboration.
-
-Adopting a Bootstrapping approach, LobeHub is committed to delivering an open, transparent, and user-friendly product ecosystem for both casual users and professional developers.
-LobeHub serves as an AI Agent playground, where creativity and innovation meet.
-
-## About [LobeChat](${OFFICIAL_URL})
-
-LobeChat, a product of LobeHub, is an open-source ChatGPT/LLMs UI/Framework designed for modern LLMs/AI applications.
-Supports Multi AI Providers( OpenAI / Claude 3 / Gemini / Perplexity / Bedrock / Azure / Mistral / Ollama ), Multi-Modals (Vision/TTS) and plugin system.
-and offers a one-click FREE deployment for a private ChatGPT chat application, making it accessible and customizable for a wide range of users.
-
-### Features
-
-- [Multi-Model Service Provider Support](${urlJoin(USAGE_DOCUMENTS, '/features/multi-ai-providers')})
-- [Local Large Language Model (LLM) Support](${urlJoin(USAGE_DOCUMENTS, '/features/local-llm')})
-- [Model Visual Recognition](${urlJoin(USAGE_DOCUMENTS, '/features/vision')})
-- [TTS & STT Voice Conversation](${urlJoin(USAGE_DOCUMENTS, '/features/tts')})
-- [Text to Image Generation](${urlJoin(USAGE_DOCUMENTS, '/features/text-to-image')})
-- [Plugin System (Function Calling)](${urlJoin(USAGE_DOCUMENTS, '/features/plugin-system')})
-- [Agent Market (GPTs)](${urlJoin(USAGE_DOCUMENTS, '/features/agent-market')})
-
-### Community Edition and Cloud Version
-
-LobeChat is currently available as a community preview version, completely open-source and free of charge.
-
-In the LobeChat Cloud version, we provide 500,000 free computing credits to all registered users. It is ready to use without complex configurations.
-If you require more usage, you can subscribe to the Basic, Advanced, or Professional versions for a fee.
-
-### Self Hosting
-
-LobeChat provides Self-Hosted Version with [Vercel](${urlJoin(SELF_HOSTING_DOCUMENTS, '/platform/vercel')}) and [Docker Image](${DOCKER_IMAGE}).
-This allows you to deploy your own chatbot within a few minutes without any prior knowledge.
-
-**IMPORTANT**
-
-When users ask about usage or deployment, DO NOT MAKE UP ANSWERS. Instead, guide them to the relevant documentation!!!
-
-Learn more about [Build your own LobeChat](${SELF_HOSTING_DOCUMENTS}) by checking it out.
-
-## Resources Links
-
-In the response, please try to pick and include the relevant links below, and if a relevant answer cannot be provided, also offer the user these related links:
-
-- Official Website: ${OFFICIAL_SITE}
-- Cloud Version: ${OFFICIAL_URL}
-- Community Edition: ${OFFICIAL_PREVIEW_URL}
-- GitHub Repository: ${GITHUB}
-- Latest News: ${BLOG}
-- Usage Documentation: ${USAGE_DOCUMENTS}
-- Self-Hosting Documentation: ${SELF_HOSTING_DOCUMENTS}
-- Development Guide: ${WIKI}
-- Email Support: ${BRANDING_EMAIL.support}
-- Business Inquiries: ${BRANDING_EMAIL.business}
-
-## Workflow
-
-1. Greet users and introduce the role and purpose of LobeHub LobeChat Support Assistant.
-2. Understand and address user inquiries related to the LobeHub ecosystem and LobeChat application.
-3. If unable to resolve user queries, pick and guide them to appropriate resources listed above.
-
-## Initialization
-
-As the role <Role>, I will adhere to the following guidelines:
-- Provide accurate and helpful information to users.
-- Maintain a friendly and professional demeanor.
-- Direct users to the appropriate resources when necessary.
-- Keep the language of the response consistent with the language of the user input; if they are not consistent, then translate.
-
-Welcome users to LobeChat, introduce myself as the <Role>, and inform them about the services and support available. Then, guide users through the <Workflow> for assistance.`;

package/packages/context-engine/src/providers/InboxGuide.ts

@@ -1,102 +0,0 @@
-import debug from 'debug';
-
-import { BaseProvider } from '../base/BaseProvider';
-import type { PipelineContext, ProcessorOptions } from '../types';
-
-const log = debug('context-engine:provider:InboxGuideProvider');
-
-/**
- * Inbox Guide System Role Configuration
- */
-export interface InboxGuideConfig {
-  /** Inbox guide system role content */
-  inboxGuideSystemRole: string;
-  /** Inbox session ID constant */
-  inboxSessionId: string;
-  /** Whether it's a welcome question */
-  isWelcomeQuestion?: boolean;
-  /** Session ID */
-  sessionId?: string;
-}
-
-/**
- * Inbox Guide Provider
- * Responsible for injecting guide system role for welcome questions in inbox sessions
- */
-export class InboxGuideProvider extends BaseProvider {
-  readonly name = 'InboxGuideProvider';
-
-  constructor(
-    private config: InboxGuideConfig,
-    options: ProcessorOptions = {},
-  ) {
-    super(options);
-  }
-
-  protected async doProcess(context: PipelineContext): Promise<PipelineContext> {
-    const clonedContext = this.cloneContext(context);
-
-    // 检查是否需要注入收件箱引导
-    const shouldInject = this.shouldInjectInboxGuide();
-
-    if (!shouldInject) {
-      log('Inbox guide injection conditions not met, skipping processing');
-      return this.markAsExecuted(clonedContext);
-    }
-
-    // 注入收件箱引导系统角色
-    this.injectInboxGuideSystemRole(clonedContext);
-
-    // 更新元数据
-    clonedContext.metadata.inboxGuide = {
-      contentLength: this.config.inboxGuideSystemRole.length,
-      injected: true,
-      isWelcomeQuestion: this.config.isWelcomeQuestion,
-      sessionId: this.config.sessionId,
-    };
-
-    log('Inbox guide system role injection completed');
-    return this.markAsExecuted(clonedContext);
-  }
-
-  /**
-   * Check if inbox guide should be injected
-   */
-  private shouldInjectInboxGuide(): boolean {
-    return (
-      (this.config.isWelcomeQuestion &&
-        this.config.sessionId === this.config.inboxSessionId &&
-        !!this.config.inboxGuideSystemRole) ||
-      false
-    );
-  }
-
-  /**
-   * Inject inbox guide system role
-   */
-  private injectInboxGuideSystemRole(context: PipelineContext): void {
-    const existingSystemMessage = context.messages.find((msg) => msg.role === 'system');
-
-    if (existingSystemMessage) {
-      // 合并到现有系统消息
-      existingSystemMessage.content = [
-        existingSystemMessage.content,
-        this.config.inboxGuideSystemRole,
-      ]
-        .filter(Boolean)
-        .join('\n\n');
-
-      log(
-        `Inbox guide merged to existing system message, final length: ${existingSystemMessage.content.length}`,
-      );
-    } else {
-      context.messages.unshift({
-        content: this.config.inboxGuideSystemRole,
-        role: 'system' as const,
-      } as any);
-      log(
-        `New inbox guide system message created, content length: ${this.config.inboxGuideSystemRole.length}`,
-      );
-    }
-  }
-}

package/packages/context-engine/src/providers/__tests__/InboxGuideProvider.test.ts

@@ -1,121 +0,0 @@
-import { describe, expect, it } from 'vitest';
-
-import type { PipelineContext } from '../../types';
-import { InboxGuideProvider } from '../InboxGuide';
-
-const createContext = (messages: any[]): PipelineContext => ({
-  initialState: { messages: [] } as any,
-  messages,
-  metadata: { model: 'gpt-4', maxTokens: 4096 },
-  isAborted: false,
-});
-
-describe('InboxGuideProvider', () => {
-  const mockInboxGuideContent = '# Role: LobeChat Support Assistant\n\nWelcome to LobeChat!';
-  const INBOX_SESSION_ID = 'inbox';
-
-  it('should inject inbox guide for welcome questions in inbox session', async () => {
-    const provider = new InboxGuideProvider({
-      isWelcomeQuestion: true,
-      sessionId: INBOX_SESSION_ID,
-      inboxSessionId: INBOX_SESSION_ID,
-      inboxGuideSystemRole: mockInboxGuideContent,
-    });
-
-    const messages = [{ id: 'u1', role: 'user', content: 'Hello, this is my first question' }];
-
-    const ctx = createContext(messages);
-    const result = await provider.process(ctx);
-
-    // Should have system message with inbox guide content
-    const systemMessage = result.messages.find((msg) => msg.role === 'system');
-    expect(systemMessage).toBeDefined();
-    expect(systemMessage!.content).toBe(mockInboxGuideContent);
-
-    // Should update metadata
-    expect(result.metadata.inboxGuide).toEqual({
-      injected: true,
-      sessionId: INBOX_SESSION_ID,
-      isWelcomeQuestion: true,
-      contentLength: mockInboxGuideContent.length,
-    });
-  });
-
-  it('should merge inbox guide with existing system message', async () => {
-    const provider = new InboxGuideProvider({
-      isWelcomeQuestion: true,
-      sessionId: INBOX_SESSION_ID,
-      inboxSessionId: INBOX_SESSION_ID,
-      inboxGuideSystemRole: mockInboxGuideContent,
-    });
-
-    const existingSystemContent = 'Existing system message';
-    const messages = [
-      { id: 's1', role: 'system', content: existingSystemContent },
-      { id: 'u1', role: 'user', content: 'Hello' },
-    ];
-
-    const ctx = createContext(messages);
-    const result = await provider.process(ctx);
-
-    const systemMessage = result.messages.find((msg) => msg.role === 'system');
-    expect(systemMessage!.content).toBe(`${existingSystemContent}\n\n${mockInboxGuideContent}`);
-  });
-
-  it('should skip injection when not welcome question', async () => {
-    const provider = new InboxGuideProvider({
-      isWelcomeQuestion: false,
-      sessionId: INBOX_SESSION_ID,
-      inboxSessionId: INBOX_SESSION_ID,
-      inboxGuideSystemRole: mockInboxGuideContent,
-    });
-
-    const messages = [{ id: 'u1', role: 'user', content: 'Regular question' }];
-
-    const ctx = createContext(messages);
-    const result = await provider.process(ctx);
-
-    // Should not have system message
-    const systemMessage = result.messages.find((msg) => msg.role === 'system');
-    expect(systemMessage).toBeUndefined();
-
-    // Should not have metadata
-    expect(result.metadata.inboxGuide).toBeUndefined();
-  });
-
-  it('should skip injection when not in inbox session', async () => {
-    const provider = new InboxGuideProvider({
-      isWelcomeQuestion: true,
-      sessionId: 'other-session',
-      inboxSessionId: INBOX_SESSION_ID,
-      inboxGuideSystemRole: mockInboxGuideContent,
-    });
-
-    const messages = [{ id: 'u1', role: 'user', content: 'Hello' }];
-
-    const ctx = createContext(messages);
-    const result = await provider.process(ctx);
-
-    // Should not have system message
-    const systemMessage = result.messages.find((msg) => msg.role === 'system');
-    expect(systemMessage).toBeUndefined();
-  });
-
-  it('should skip injection when no inbox guide content', async () => {
-    const provider = new InboxGuideProvider({
-      isWelcomeQuestion: true,
-      sessionId: INBOX_SESSION_ID,
-      inboxSessionId: INBOX_SESSION_ID,
-      inboxGuideSystemRole: '',
-    });
-
-    const messages = [{ id: 'u1', role: 'user', content: 'Hello' }];
-
-    const ctx = createContext(messages);
-    const result = await provider.process(ctx);
-
-    // Should not have system message
-    const systemMessage = result.messages.find((msg) => msg.role === 'system');
-    expect(systemMessage).toBeUndefined();
-  });
-});

package/src/services/chat/__snapshots__/chat.test.ts.snap

@@ -1,110 +0,0 @@
-// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
-
-exports[`ChatService > createAssistantMessage > with tools messages > work with dalle3 1`] = `
-{
-  "enabledSearch": undefined,
-  "messages": [
-    {
-      "content": "<plugins description="The plugins you can use below">
-<collection name="DALL·E 3">
-<collection.instructions>Whenever a description of an image is given, use lobe-image-designer to create the images and then summarize the prompts used to generate the images in plain text. If the user does not ask for a specific number of images, default to creating four captions to send to lobe-image-designer that are written to be as diverse as possible.
-
-  All captions sent to lobe-image-designer must abide by the following policies:
-
-1. If the description is not in English, then translate it.
-2. Do not create more than 4 images, even if the user requests more.
-3. Don't create images of politicians or other public figures. Recommend other ideas instead.
-5. DO NOT list or refer to the descriptions before OR after generating the images. They should ONLY ever be written out ONCE, in the \`prompts\` field of the request. You do not need to ask for permission to generate, just do it!
-6. Always mention the image type (photo, oil painting, watercolor painting, illustration, cartoon, drawing, vector, render, etc.) at the beginning of the caption. Unless the caption suggests otherwise, make at least 1--2 of the 4 images photos.
-7. Diversify depictions of ALL images with people to include DESCENT and GENDER for EACH person using direct terms. Adjust only human descriptions.
-  - EXPLICITLY specify these attributes, not abstractly reference them.  The attributes should be specified in a minimal way and should directly describe their physical form.
-  - Your choices should be grounded in reality. For example, all of a given OCCUPATION should not be the same gender or race. Additionally, focus on creating diverse, inclusive, and exploratory scenes via the properties you choose during rewrites.  Make choices that may be insightful or unique sometimes.
-  - Use "various" or "diverse" ONLY IF the description refers to groups of more than 3 people. Do not change the number of people requested in the original description.
-  - Don't alter memes, fictional character origins, or unseen people. Maintain the original prompt's intent and prioritize quality.
-  - Do not create any imagery that would be offensive.
-
-8. Silently modify descriptions that include names or hints or references of specific people or celebrities by carefully selecting a few minimal modifications to substitute references to the people with generic descriptions that don't divulge any information about their identities, except for their genders and physiques. Do this EVEN WHEN the instructions ask for the prompt to not be changed. Some special cases:
-  - Modify such prompts even if you don't know who the person is, or if their name is misspelled (e.g. "Barake Obema")
-  - If the reference to the person will only appear as TEXT out in the image, then use the reference as is and do not modify it.
-  - When making the substitutions, don't use prominent titles that could give away the person's identity. E.g., instead of saying "president", "prime minister", or "chancellor", say "politician"; instead of saying "king", "queen", "emperor", or "empress", say "public figure"; instead of saying "Pope", say "religious figure"; and so on.
-  - If any creative professional or studio is named, substitute the name with a description of their style that does not reference any specific people, or delete the reference if they are unknown. DO NOT refer to the artist or studio's style.
-
-The prompt must intricately describe every part of the image in concrete, objective detail. THINK about what the end goal of the description is, and extrapolate that to what would make satisfying images.
-All descriptions sent to lobe-image-designer should be a paragraph of text that is extremely descriptive and detailed. Each should be more than 3 sentences long.</collection.instructions>
-<api identifier="lobe-image-designer____text2image____builtin">Create images from a text-only prompt.</api>
-</collection>
-</plugins>",
-      "role": "system",
-    },
-    {
-      "content": "https://vercel.com/ 请分析 chatGPT 关键词
-
-",
-      "role": "user",
-    },
-  ],
-  "model": "gpt-3.5-turbo-1106",
-  "tools": [
-    {
-      "function": {
-        "description": "Create images from a text-only prompt.",
-        "name": "lobe-image-designer____text2image____builtin",
-        "parameters": {
-          "properties": {
-            "prompts": {
-              "description": "The user's original image description, potentially modified to abide by the lobe-image-designer policies. If the user does not suggest a number of captions to create, create four of them. If creating multiple captions, make them as diverse as possible. If the user requested modifications to previous images, the captions should not simply be longer, but rather it should be refactored to integrate the suggestions into each of the captions. Generate no more than 4 images, even if the user requests more.",
-              "items": {
-                "type": "string",
-              },
-              "maxItems": 4,
-              "minItems": 1,
-              "type": "array",
-            },
-            "quality": {
-              "default": "standard",
-              "description": "The quality of the image that will be generated. hd creates images with finer details and greater consistency across the image.",
-              "enum": [
-                "standard",
-                "hd",
-              ],
-              "type": "string",
-            },
-            "seeds": {
-              "description": "A list of seeds to use for each prompt. If the user asks to modify a previous image, populate this field with the seed used to generate that image from the image lobe-image-designer metadata.",
-              "items": {
-                "type": "integer",
-              },
-              "type": "array",
-            },
-            "size": {
-              "default": "1024x1024",
-              "description": "The resolution of the requested image, which can be wide, square, or tall. Use 1024x1024 (square) as the default unless the prompt suggests a wide image, 1792x1024, or a full-body portrait, in which case 1024x1792 (tall) should be used instead. Always include this parameter in the request.",
-              "enum": [
-                "1792x1024",
-                "1024x1024",
-                "1024x1792",
-              ],
-              "type": "string",
-            },
-            "style": {
-              "default": "vivid",
-              "description": "The style of the generated images. Must be one of vivid or natural. Vivid causes the model to lean towards generating hyper-real and dramatic images. Natural causes the model to produce more natural, less hyper-real looking images.",
-              "enum": [
-                "vivid",
-                "natural",
-              ],
-              "type": "string",
-            },
-          },
-          "required": [
-            "prompts",
-          ],
-          "type": "object",
-        },
-      },
-      "type": "function",
-    },
-  ],
-  "top_p": 1,
-}
-`;

package/src/store/chat/slices/builtinTool/actions/__tests__/dalle.test.ts

@@ -1,121 +0,0 @@
-import { UIChatMessage } from '@lobechat/types';
-import { act, renderHook } from '@testing-library/react';
-import { describe, expect, it, vi } from 'vitest';
-
-import { messageService } from '@/services/message';
-import { imageGenerationService } from '@/services/textToImage';
-import { uploadService } from '@/services/upload';
-import { useChatStore } from '@/store/chat';
-import { chatSelectors } from '@/store/chat/selectors';
-import { useFileStore } from '@/store/file';
-import { DallEImageItem } from '@/types/tool/dalle';
-
-describe('chatToolSlice - dalle', () => {
-  describe('generateImageFromPrompts', () => {
-    it('should generate images from prompts, update items, and upload images', async () => {
-      const { result } = renderHook(() => useChatStore());
-
-      const initialMessageContent = JSON.stringify([
-        { prompt: 'test prompt', previewUrl: 'old-url', imageId: 'old-id' },
-      ]);
-
-      vi.spyOn(chatSelectors, 'getMessageById').mockImplementationOnce(
-        (id) => () =>
-          ({
-            id,
-            content: initialMessageContent,
-          }) as UIChatMessage,
-      );
-
-      const messageId = 'message-id';
-      const prompts = [
-        { prompt: 'test prompt 1' },
-        { prompt: 'test prompt 2' },
-      ] as DallEImageItem[];
-      const mockUrl = 'https://example.com/image.png';
-      const mockId = 'image-id';
-
-      vi.spyOn(imageGenerationService, 'generateImage').mockResolvedValue(mockUrl);
-      vi.spyOn(uploadService, 'getImageFileByUrlWithCORS').mockResolvedValue(
-        new File(['1'], 'file.png', { type: 'image/png' }),
-      );
-
-      // Mock the new uploadWithProgress method from useFileStore
-      vi.spyOn(useFileStore, 'getState').mockReturnValue({
-        uploadWithProgress: vi.fn().mockResolvedValue({
-          id: mockId,
-          url: '',
-          dimensions: { width: 512, height: 512 },
-          filename: 'file.png',
-        }),
-      } as any);
-
-      // Mock store methods that are called in the implementation
-      vi.spyOn(result.current, 'toggleDallEImageLoading');
-      vi.spyOn(result.current, 'updatePluginState').mockResolvedValue(undefined);
-      vi.spyOn(result.current, 'internal_updateMessageContent').mockResolvedValue(undefined);
-
-      await act(async () => {
-        await result.current.generateImageFromPrompts(prompts, messageId);
-      });
-      // For each prompt, loading is toggled on and then off
-      expect(imageGenerationService.generateImage).toHaveBeenCalledTimes(prompts.length);
-      expect(useFileStore.getState().uploadWithProgress).toHaveBeenCalledTimes(prompts.length);
-      expect(result.current.toggleDallEImageLoading).toHaveBeenCalledTimes(prompts.length * 2);
-    });
-  });
-
-  describe('updateImageItem', () => {
-    it('should update image item correctly', async () => {
-      const { result } = renderHook(() => useChatStore());
-      const messageId = 'message-id';
-      const initialMessageContent = JSON.stringify([
-        { prompt: 'test prompt', previewUrl: 'old-url', imageId: 'old-id' },
-      ]);
-      const updateFunction = (draft: any) => {
-        draft[0].previewUrl = 'new-url';
-        draft[0].imageId = 'new-id';
-      };
-      vi.spyOn(result.current, 'internal_updateMessageContent').mockResolvedValue(undefined);
-
-      // 模拟 getMessageById 返回消息内容
-      vi.spyOn(chatSelectors, 'getMessageById').mockImplementationOnce(
-        (id) => () =>
-          ({
-            id,
-            content: initialMessageContent,
-          }) as UIChatMessage,
-      );
-      vi.spyOn(messageService, 'updateMessage').mockResolvedValueOnce(undefined);
-
-      await act(async () => {
-        await result.current.updateImageItem(messageId, updateFunction);
-      });
-
-      // 验证 internal_updateMessageContent 是否被正确调用以更新内容
-      expect(result.current.internal_updateMessageContent).toHaveBeenCalledWith(
-        messageId,
-        JSON.stringify([{ prompt: 'test prompt', previewUrl: 'new-url', imageId: 'new-id' }]),
-      );
-    });
-  });
-
-  describe('text2image', () => {
-    it('should call generateImageFromPrompts with provided data', async () => {
-      const { result } = renderHook(() => useChatStore());
-      const id = 'message-id';
-      const data = [{ prompt: 'prompt 1' }, { prompt: 'prompt 2' }] as DallEImageItem[];
-
-      // Mock generateImageFromPrompts
-      const generateImageFromPromptsMock = vi
-        .spyOn(result.current, 'generateImageFromPrompts')
-        .mockResolvedValue(undefined);
-
-      await act(async () => {
-        await result.current.text2image(id, data);
-      });
-
-      expect(generateImageFromPromptsMock).toHaveBeenCalledWith(data, id);
-    });
-  });
-});