@lobehub/chat 1.86.0 → 1.87.0

package/src/app/[variants]/(main)/settings/system-agent/features/createForm.tsx

@@ -1,11 +1,12 @@
 'use client';
 
-import { Button, Form, type FormGroupItemType, type FormItemProps } from '@lobehub/ui';
-import { Form as AntForm, Switch } from 'antd';
+import { Button, Form, type FormGroupItemType, type FormItemProps, Icon } from '@lobehub/ui';
+import { Form as AntForm, Skeleton, Switch } from 'antd';
 import isEqual from 'fast-deep-equal';
-import { PencilIcon } from 'lucide-react';
-import { memo } from 'react';
+import { Loader2Icon, PencilIcon } from 'lucide-react';
+import { memo, useState } from 'react';
 import { useTranslation } from 'react-i18next';
+import { Flexbox } from 'react-layout-kit';
 
 import TextArea from '@/components/TextArea';
 import { FORM_STYLE } from '@/const/layoutTokens';
@@ -14,8 +15,6 @@ import { useUserStore } from '@/store/user';
 import { settingsSelectors } from '@/store/user/selectors';
 import { type UserSystemAgentConfigKey } from '@/types/user/settings';
 
-import { useSyncSystemAgent } from './useSync';
-
 interface SystemAgentFormProps {
   allowCustomPrompt?: boolean;
   allowDisable?: boolean;
@@ -26,11 +25,16 @@ interface SystemAgentFormProps {
 const SystemAgentForm = memo(
   ({ systemAgentKey, allowDisable, allowCustomPrompt, defaultPrompt }: SystemAgentFormProps) => {
     const { t } = useTranslation('setting');
-
+    const [form] = AntForm.useForm();
     const settings = useUserStore(settingsSelectors.currentSystemAgent, isEqual);
-    const [updateSystemAgent] = useUserStore((s) => [s.updateSystemAgent]);
+    const [updateSystemAgent, isUserStateInit] = useUserStore((s) => [
+      s.updateSystemAgent,
+      s.isUserStateInit,
+    ]);
+    const [loading, setLoading] = useState(false);
+
+    if (!isUserStateInit) return <Skeleton active paragraph={{ rows: 5 }} title={false} />;
 
-    const [form] = AntForm.useForm();
     const value = settings[systemAgentKey];
 
     const systemAgentSettings: FormGroupItemType = {
@@ -38,8 +42,10 @@ const SystemAgentForm = memo(
         {
           children: (
             <ModelSelect
-              onChange={(props) => {
-                updateSystemAgent(systemAgentKey, props);
+              onChange={async (props) => {
+                setLoading(true);
+                await updateSystemAgent(systemAgentKey, props);
+                setLoading(false);
               }}
               showAbility={false}
               // value={value}
@@ -52,8 +58,10 @@ const SystemAgentForm = memo(
         (!!allowCustomPrompt && {
           children: !!value.customPrompt ? (
             <TextArea
-              onChange={(e) => {
-                updateSystemAgent(systemAgentKey, { customPrompt: e });
+              onBlur={async (e) => {
+                setLoading(true);
+                await updateSystemAgent(systemAgentKey, { customPrompt: e.target.value });
+                setLoading(false);
               }}
               placeholder={t('systemAgent.customPrompt.placeholder')}
               style={{ minHeight: 160 }}
@@ -64,7 +72,9 @@ const SystemAgentForm = memo(
               block
               icon={PencilIcon}
               onClick={async () => {
+                setLoading(true);
                 await updateSystemAgent(systemAgentKey, { customPrompt: defaultPrompt });
+                setLoading(false);
               }}
             >
               {t('systemAgent.customPrompt.addPrompt')}
@@ -75,13 +85,20 @@ const SystemAgentForm = memo(
           name: [systemAgentKey, 'customPrompt'],
         }) as FormItemProps,
       ].filter(Boolean),
-      extra: allowDisable && (
-        <Switch
-          onChange={(enabled) => {
-            updateSystemAgent(systemAgentKey, { enabled });
-          }}
-          value={value.enabled}
-        />
+      extra: (
+        <Flexbox>
+          {loading && <Icon icon={Loader2Icon} size={16} spin style={{ opacity: 0.5 }} />}
+          {allowDisable && (
+            <Switch
+              onChange={async (enabled) => {
+                setLoading(true);
+                await updateSystemAgent(systemAgentKey, { enabled });
+                setLoading(false);
+              }}
+              value={value.enabled}
+            />
+          )}
+        </Flexbox>
       ),
       title: (
         <span
@@ -94,8 +111,6 @@ const SystemAgentForm = memo(
       ),
     };
 
-    useSyncSystemAgent(form, settings);
-
     return (
       <Form form={form} initialValues={settings} items={[systemAgentSettings]} {...FORM_STYLE} />
     );

package/src/app/[variants]/(main)/settings/tts/features/OpenAI.tsx

@@ -1,9 +1,11 @@
 'use client';
 
-import { Form, type FormGroupItemType } from '@lobehub/ui';
+import { Form, type FormGroupItemType, Icon } from '@lobehub/ui';
 import { Select } from '@lobehub/ui';
+import { Skeleton } from 'antd';
 import isEqual from 'fast-deep-equal';
-import { memo } from 'react';
+import { Loader2Icon } from 'lucide-react';
+import { memo, useState } from 'react';
 import { useTranslation } from 'react-i18next';
 
 import { FORM_STYLE } from '@/const/layoutTokens';
@@ -12,37 +14,45 @@ import { settingsSelectors } from '@/store/user/selectors';
 
 import { opeanaiSTTOptions, opeanaiTTSOptions } from './const';
 
-const TTS_SETTING_KEY = 'tts';
-
 const OpenAI = memo(() => {
   const { t } = useTranslation('setting');
   const [form] = Form.useForm();
-  const settings = useUserStore(settingsSelectors.currentSettings, isEqual);
-  const [setSettings] = useUserStore((s) => [s.setSettings]);
+  const { tts } = useUserStore(settingsSelectors.currentSettings, isEqual);
+  const [setSettings, isUserStateInit] = useUserStore((s) => [s.setSettings, s.isUserStateInit]);
+  const [loading, setLoading] = useState(false);
+
+  if (!isUserStateInit) return <Skeleton active paragraph={{ rows: 5 }} title={false} />;
 
   const openai: FormGroupItemType = {
     children: [
       {
         children: <Select options={opeanaiTTSOptions} />,
         label: t('settingTTS.openai.ttsModel'),
-        name: [TTS_SETTING_KEY, 'openAI', 'ttsModel'],
+        name: ['openAI', 'ttsModel'],
       },
       {
         children: <Select options={opeanaiSTTOptions} />,
         label: t('settingTTS.openai.sttModel'),
-        name: [TTS_SETTING_KEY, 'openAI', 'sttModel'],
+        name: ['openAI', 'sttModel'],
       },
     ],
+    extra: loading && <Icon icon={Loader2Icon} size={16} spin style={{ opacity: 0.5 }} />,
     title: t('settingTTS.openai.title'),
   };
 
   return (
     <Form
       form={form}
-      initialValues={settings}
+      initialValues={tts}
       items={[openai]}
       itemsType={'group'}
-      onValuesChange={setSettings}
+      onValuesChange={async (values) => {
+        setLoading(true);
+        await setSettings({
+          tts: values,
+        });
+        setLoading(false);
+      }}
       variant={'borderless'}
       {...FORM_STYLE}
     />

package/src/app/[variants]/(main)/settings/tts/features/STT.tsx

@@ -1,9 +1,10 @@
 'use client';
 
-import { Form, type FormGroupItemType, Select } from '@lobehub/ui';
-import { Switch } from 'antd';
+import { Form, type FormGroupItemType, Icon, Select } from '@lobehub/ui';
+import { Skeleton, Switch } from 'antd';
 import isEqual from 'fast-deep-equal';
-import { memo } from 'react';
+import { Loader2Icon } from 'lucide-react';
+import { memo, useState } from 'react';
 import { useTranslation } from 'react-i18next';
 
 import { FORM_STYLE } from '@/const/layoutTokens';
@@ -12,13 +13,14 @@ import { settingsSelectors } from '@/store/user/selectors';
 
 import { sttOptions } from './const';
 
-const TTS_SETTING_KEY = 'tts';
-
 const STT = memo(() => {
   const { t } = useTranslation('setting');
   const [form] = Form.useForm();
-  const settings = useUserStore(settingsSelectors.currentSettings, isEqual);
-  const [setSettings] = useUserStore((s) => [s.setSettings]);
+  const { tts } = useUserStore(settingsSelectors.currentSettings, isEqual);
+  const [setSettings, isUserStateInit] = useUserStore((s) => [s.setSettings, s.isUserStateInit]);
+  const [loading, setLoading] = useState(false);
+
+  if (!isUserStateInit) return <Skeleton active paragraph={{ rows: 5 }} title={false} />;
 
   const stt: FormGroupItemType = {
     children: [
@@ -26,7 +28,7 @@ const STT = memo(() => {
         children: <Select options={sttOptions} />,
         desc: t('settingTTS.sttService.desc'),
         label: t('settingTTS.sttService.title'),
-        name: [TTS_SETTING_KEY, 'sttServer'],
+        name: 'sttServer',
       },
       {
         children: <Switch />,
@@ -34,20 +36,27 @@ const STT = memo(() => {
         label: t('settingTTS.sttAutoStop.title'),
         layout: 'horizontal',
         minWidth: undefined,
-        name: [TTS_SETTING_KEY, 'sttAutoStop'],
+        name: 'sttAutoStop',
         valuePropName: 'checked',
       },
     ],
+    extra: loading && <Icon icon={Loader2Icon} size={16} spin style={{ opacity: 0.5 }} />,
     title: t('settingTTS.stt'),
   };
 
   return (
     <Form
       form={form}
-      initialValues={settings}
+      initialValues={tts}
       items={[stt]}
       itemsType={'group'}
-      onValuesChange={setSettings}
+      onValuesChange={async (values) => {
+        setLoading(true);
+        await setSettings({
+          tts: values,
+        });
+        setLoading(false);
+      }}
       variant={'borderless'}
       {...FORM_STYLE}
     />

package/src/const/settings/common.ts

@@ -2,4 +2,6 @@ import { UserGeneralConfig } from '@/types/user/settings';
 
 export const DEFAULT_COMMON_SETTINGS: UserGeneralConfig = {
   fontSize: 14,
+  highlighterTheme: 'lobe-theme',
+  mermaidTheme: 'lobe-theme',
 };

package/src/const/settings/llm.ts

@@ -1,4 +1,4 @@
-import { ModelProvider } from '@/libs/agent-runtime';
+import { ModelProvider } from '@/libs/model-runtime';
 import { genUserLLMConfig } from '@/utils/genUserLLMConfig';
 
 export const DEFAULT_LLM_CONFIG = genUserLLMConfig({

package/src/database/models/__tests__/aiProvider.test.ts

@@ -3,7 +3,7 @@ import { eq } from 'drizzle-orm/expressions';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { LobeChatDatabase } from '@/database/type';
-import { ModelProvider } from '@/libs/agent-runtime';
+import { ModelProvider } from '@/libs/model-runtime';
 import { sleep } from '@/utils/sleep';
 
 import { aiProviders, users } from '../../schemas';

package/src/database/models/aiProvider.ts

@@ -3,7 +3,7 @@ import { isEmpty } from 'lodash-es';
 
 import { DEFAULT_MODEL_PROVIDER_LIST } from '@/config/modelProviders';
 import { LobeChatDatabase } from '@/database/type';
-import { ModelProvider } from '@/libs/agent-runtime';
+import { ModelProvider } from '@/libs/model-runtime';
 import {
   AiProviderDetailItem,
   AiProviderListItem,

package/src/features/AgentSetting/AgentTTS/index.tsx

@@ -79,7 +79,7 @@ const AgentTTS = memo(() => {
         children: (
           <Select
             options={[
-              { label: t('settingTheme.lang.autoMode'), value: 'auto' },
+              { label: t('settingCommon.lang.autoMode'), value: 'auto' },
               ...(localeOptions || []),
             ]}
           />

package/src/features/ChatItem/index.tsx

@@ -0,0 +1,35 @@
+'use client';
+
+import { ChatItemProps, ChatItem as ChatItemRaw } from '@lobehub/ui/chat';
+import isEqual from 'fast-deep-equal';
+import { memo } from 'react';
+
+import { useUserStore } from '@/store/user';
+import { settingsSelectors } from '@/store/user/selectors';
+
+const ChatItem = memo<ChatItemProps>(({ markdownProps = {}, ...rest }) => {
+  const { componentProps, ...restMarkdown } = markdownProps;
+  const { general } = useUserStore(settingsSelectors.currentSettings, isEqual);
+  return (
+    <ChatItemRaw
+      fontSize={general.fontSize}
+      markdownProps={{
+        ...restMarkdown,
+        componentProps: {
+          ...componentProps,
+          highlight: {
+            theme: general.highlighterTheme,
+            ...componentProps?.highlight,
+          },
+          mermaid: {
+            theme: general.mermaidTheme,
+            ...componentProps?.mermaid,
+          },
+        },
+      }}
+      {...rest}
+    />
+  );
+});
+
+export default ChatItem;

package/src/features/Conversation/Error/APIKeyForm/Bedrock.tsx

@@ -5,7 +5,7 @@ import { Network, ShieldPlus } from 'lucide-react';
 import { memo, useState } from 'react';
 import { useTranslation } from 'react-i18next';
 
-import { ModelProvider } from '@/libs/agent-runtime';
+import { ModelProvider } from '@/libs/model-runtime';
 import { useUserStore } from '@/store/user';
 import { keyVaultsConfigSelectors } from '@/store/user/selectors';
 

package/src/features/Conversation/Error/APIKeyForm/index.tsx

@@ -4,7 +4,7 @@ import { memo, useMemo, useState } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Center, Flexbox } from 'react-layout-kit';
 
-import { ModelProvider } from '@/libs/agent-runtime';
+import { ModelProvider } from '@/libs/model-runtime';
 import { useChatStore } from '@/store/chat';
 import { GlobalLLMProviderKey } from '@/types/user/settings';
 

package/src/features/Conversation/Error/index.tsx

@@ -6,7 +6,7 @@ import { Suspense, memo, useMemo } from 'react';
 import { useTranslation } from 'react-i18next';
 
 import { useProviderName } from '@/hooks/useProviderName';
-import { AgentRuntimeErrorType, ILobeAgentRuntimeErrorType } from '@/libs/agent-runtime';
+import { AgentRuntimeErrorType, ILobeAgentRuntimeErrorType } from '@/libs/model-runtime';
 import { ChatErrorType, ErrorType } from '@/types/fetch';
 import { ChatMessage, ChatMessageError } from '@/types/message';
 

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

@@ -1,19 +1,17 @@
 'use client';
 
-import { ChatItem } from '@lobehub/ui/chat';
 import { createStyles } from 'antd-style';
 import isEqual from 'fast-deep-equal';
 import { MouseEventHandler, ReactNode, memo, use, useCallback, useMemo } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
 
+import ChatItem from '@/features/ChatItem';
 import { VirtuosoContext } from '@/features/Conversation/components/VirtualizedList/VirtuosoContext';
 import { useAgentStore } from '@/store/agent';
 import { agentChatConfigSelectors } from '@/store/agent/selectors';
 import { useChatStore } from '@/store/chat';
 import { chatSelectors } from '@/store/chat/selectors';
-import { useUserStore } from '@/store/user';
-import { userGeneralSettingsSelectors } from '@/store/user/selectors';
 import { ChatMessage } from '@/types/message';
 
 import ErrorMessageExtra, { useErrorContent } from '../../Error';
@@ -72,7 +70,6 @@ const Item = memo<ChatListItemProps>(
 
     const type = useAgentStore(agentChatConfigSelectors.displayMode);
     const item = useChatStore(chatSelectors.getMessageById(id), isEqual);
-    const fontSize = useUserStore(userGeneralSettingsSelectors.fontSize);
 
     const [
       isMessageLoading,
@@ -178,7 +175,7 @@ const Item = memo<ChatListItemProps>(
         enableCustomFootnotes: item?.role === 'assistant',
         rehypePlugins: item?.role === 'user' ? undefined : rehypePlugins,
         remarkPlugins: item?.role === 'user' ? undefined : remarkPlugins,
-        showCitations:
+        showFootnotes:
           item?.role === 'user'
             ? undefined
             : item?.search?.citations &&
@@ -235,7 +232,6 @@ const Item = memo<ChatListItemProps>(
               editing={editing}
               error={error}
               errorMessage={errorMessage}
-              fontSize={fontSize}
               loading={isProcessing}
               markdownProps={markdownProps}
               message={message}

package/src/features/User/UserPanel/LangButton.tsx

@@ -26,7 +26,7 @@ const LangButton = memo<{ placement?: PopoverProps['placement'] }>(({ placement
     () => [
       {
         key: 'auto',
-        label: t('settingTheme.lang.autoMode'),
+        label: t('settingCommon.lang.autoMode'),
         onClick: () => handleLangChange('auto'),
       },
       ...localeOptions.map((item) => ({

package/src/features/User/UserPanel/ThemeButton.tsx

@@ -27,19 +27,19 @@ const ThemeButton = memo<{ placement?: PopoverProps['placement'] }>(({ placement
       {
         icon: <Icon icon={themeIcons.auto} />,
         key: 'auto',
-        label: t('settingTheme.themeMode.auto'),
+        label: t('settingCommon.themeMode.auto'),
         onClick: () => switchThemeMode('auto'),
       },
       {
         icon: <Icon icon={themeIcons.light} />,
         key: 'light',
-        label: t('settingTheme.themeMode.light'),
+        label: t('settingCommon.themeMode.light'),
         onClick: () => switchThemeMode('light'),
       },
       {
         icon: <Icon icon={themeIcons.dark} />,
         key: 'dark',
-        label: t('settingTheme.themeMode.dark'),
+        label: t('settingCommon.themeMode.dark'),
         onClick: () => switchThemeMode('dark'),
       },
     ],

package/src/libs/{agent-runtime/AgentRuntime.test.ts → model-runtime/ModelRuntime.test.ts}

@@ -6,11 +6,11 @@ import { beforeEach, describe, expect, it, vi } from 'vitest';
 import * as langfuseCfg from '@/config/langfuse';
 import { JWTPayload } from '@/const/auth';
 import { TraceNameMap } from '@/const/trace';
-import { AgentRuntime, ChatStreamPayload, LobeOpenAI, ModelProvider } from '@/libs/agent-runtime';
-import { providerRuntimeMap } from '@/libs/agent-runtime/runtimeMap';
+import { AgentRuntime, ChatStreamPayload, LobeOpenAI, ModelProvider } from '@/libs/model-runtime';
+import { providerRuntimeMap } from '@/libs/model-runtime/runtimeMap';
 import { createTraceOptions } from '@/server/modules/AgentRuntime';
 
-import { AgentChatOptions } from './AgentRuntime';
+import { AgentChatOptions } from './ModelRuntime';
 
 const specialProviders = [
   { id: 'openai', payload: { apiKey: 'user-openai-key', baseURL: 'user-endpoint' } },

package/src/libs/{agent-runtime/AgentRuntime.ts → model-runtime/ModelRuntime.ts}

@@ -24,7 +24,7 @@ export interface AgentChatOptions {
   trace?: TracePayload;
 }
 
-class AgentRuntime {
+class ModelRuntime {
   private _runtime: LobeRuntimeAI;
 
   constructor(runtime: LobeRuntimeAI) {
@@ -110,8 +110,8 @@ class AgentRuntime {
     const providerAI = providerRuntimeMap[provider] ?? LobeOpenAI;
     const runtimeModel: LobeRuntimeAI = new providerAI(params);
 
-    return new AgentRuntime(runtimeModel);
+    return new ModelRuntime(runtimeModel);
   }
 }
 
-export default AgentRuntime;
+export default ModelRuntime;

package/src/libs/{agent-runtime → model-runtime}/ai21/index.test.ts

@@ -1,6 +1,6 @@
 // @vitest-environment node
-import { ModelProvider } from '@/libs/agent-runtime';
-import { testProvider } from '@/libs/agent-runtime/providerTestUtils';
+import { ModelProvider } from '@/libs/model-runtime';
+import { testProvider } from '@/libs/model-runtime/providerTestUtils';
 
 import { LobeAi21AI } from './index';
 

package/src/libs/{agent-runtime → model-runtime}/ai360/index.test.ts

@@ -1,6 +1,6 @@
 // @vitest-environment node
-import { ModelProvider } from '@/libs/agent-runtime';
-import { testProvider } from '@/libs/agent-runtime/providerTestUtils';
+import { ModelProvider } from '@/libs/model-runtime';
+import { testProvider } from '@/libs/model-runtime/providerTestUtils';
 
 import { LobeAi360AI } from './index';
 

package/src/libs/{agent-runtime → model-runtime}/anthropic/index.test.ts

@@ -1,7 +1,7 @@
 // @vitest-environment node
 import { Mock, afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { ChatCompletionTool, ChatStreamPayload } from '@/libs/agent-runtime';
+import { ChatCompletionTool, ChatStreamPayload } from '@/libs/model-runtime';
 
 import * as anthropicHelpers from '../utils/anthropicHelpers';
 import * as debugStreamModule from '../utils/debugStream';

package/src/libs/{agent-runtime → model-runtime}/baichuan/index.test.ts

@@ -1,7 +1,7 @@
 // @vitest-environment node
 import { Mock } from 'vitest';
 
-import { LobeOpenAICompatibleRuntime, ModelProvider } from '@/libs/agent-runtime';
+import { LobeOpenAICompatibleRuntime, ModelProvider } from '@/libs/model-runtime';
 
 import { testProvider } from '../providerTestUtils';
 import { LobeBaichuanAI } from './index';

package/src/libs/{agent-runtime → model-runtime}/bedrock/index.test.ts

@@ -2,7 +2,7 @@
 import { InvokeModelWithResponseStreamCommand } from '@aws-sdk/client-bedrock-runtime';
 import { Mock, afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { AgentRuntimeErrorType, ModelProvider } from '@/libs/agent-runtime';
+import { AgentRuntimeErrorType, ModelProvider } from '@/libs/model-runtime';
 
 import * as debugStreamModule from '../utils/debugStream';
 import { LobeBedrockAI } from './index';

package/src/libs/{agent-runtime → model-runtime}/cloudflare/index.test.ts

@@ -1,7 +1,7 @@
 // @vitest-environment node
 import { Mock, afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { ChatCompletionTool } from '@/libs/agent-runtime';
+import { ChatCompletionTool } from '@/libs/model-runtime';
 
 import * as debugStreamModule from '../utils/debugStream';
 import { LobeCloudflareAI } from './index';

package/src/libs/{agent-runtime → model-runtime}/deepseek/index.test.ts

@@ -6,8 +6,8 @@ import {
   LLMRoleType,
   LobeOpenAICompatibleRuntime,
   ModelProvider,
-} from '@/libs/agent-runtime';
-import { testProvider } from '@/libs/agent-runtime/providerTestUtils';
+} from '@/libs/model-runtime';
+import { testProvider } from '@/libs/model-runtime/providerTestUtils';
 
 import { LobeDeepSeekAI } from './index';
 

package/src/libs/{agent-runtime → model-runtime}/fireworksai/index.test.ts

@@ -1,6 +1,6 @@
 // @vitest-environment node
-import { ModelProvider } from '@/libs/agent-runtime';
-import { testProvider } from '@/libs/agent-runtime/providerTestUtils';
+import { ModelProvider } from '@/libs/model-runtime';
+import { testProvider } from '@/libs/model-runtime/providerTestUtils';
 
 import { LobeFireworksAI } from './index';
 

package/src/libs/{agent-runtime → model-runtime}/giteeai/index.test.ts

@@ -1,6 +1,6 @@
 // @vitest-environment node
-import { ModelProvider } from '@/libs/agent-runtime';
-import { testProvider } from '@/libs/agent-runtime/providerTestUtils';
+import { ModelProvider } from '@/libs/model-runtime';
+import { testProvider } from '@/libs/model-runtime/providerTestUtils';
 
 import { LobeGiteeAI } from './index';
 

package/src/libs/{agent-runtime → model-runtime}/github/index.test.ts

@@ -1,6 +1,6 @@
 // @vitest-environment node
-import { ModelProvider } from '@/libs/agent-runtime';
-import { testProvider } from '@/libs/agent-runtime/providerTestUtils';
+import { ModelProvider } from '@/libs/model-runtime';
+import { testProvider } from '@/libs/model-runtime/providerTestUtils';
 
 import { LobeGithubAI } from './index';
 

package/src/libs/{agent-runtime → model-runtime}/google/index.test.ts

@@ -3,7 +3,7 @@ import { FunctionDeclarationsTool } from '@google/generative-ai';
 import OpenAI from 'openai';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { OpenAIChatMessage } from '@/libs/agent-runtime';
+import { OpenAIChatMessage } from '@/libs/model-runtime';
 import * as imageToBase64Module from '@/utils/imageToBase64';
 
 import * as debugStreamModule from '../utils/debugStream';

package/src/libs/{agent-runtime → model-runtime}/groq/index.test.ts

@@ -1,8 +1,8 @@
 // @vitest-environment node
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { LobeOpenAICompatibleRuntime } from '@/libs/agent-runtime';
-import { testProvider } from '@/libs/agent-runtime/providerTestUtils';
+import { LobeOpenAICompatibleRuntime } from '@/libs/model-runtime';
+import { testProvider } from '@/libs/model-runtime/providerTestUtils';
 
 import { LobeGroq } from './index';
 

package/src/libs/{agent-runtime → model-runtime}/hunyuan/index.test.ts

@@ -1,8 +1,8 @@
 // @vitest-environment node
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { LobeOpenAICompatibleRuntime, ModelProvider } from '@/libs/agent-runtime';
-import { testProvider } from '@/libs/agent-runtime/providerTestUtils';
+import { LobeOpenAICompatibleRuntime, ModelProvider } from '@/libs/model-runtime';
+import { testProvider } from '@/libs/model-runtime/providerTestUtils';
 
 import { LobeHunyuanAI } from './index';
 

package/src/libs/{agent-runtime → model-runtime}/index.ts

@@ -1,4 +1,3 @@
-export { default as AgentRuntime } from './AgentRuntime';
 export { LobeAnthropicAI } from './anthropic';
 export { LobeAzureAI } from './azureai';
 export { LobeAzureOpenAI } from './azureOpenai';
@@ -11,6 +10,7 @@ export { LobeGroq } from './groq';
 export * from './helpers';
 export { LobeMinimaxAI } from './minimax';
 export { LobeMistralAI } from './mistral';
+export { default as AgentRuntime } from './ModelRuntime';
 export { LobeMoonshotAI } from './moonshot';
 export { LobeOllamaAI } from './ollama';
 export { LobeOpenAI } from './openai';