From edba1f049e00789cdbd8652e5406580e5fdf6af8 Mon Sep 17 00:00:00 2001
From: "guangfei.zhao" <guangfei.zhao@t-systems.com>
Date: Fri, 24 Oct 2025 15:40:34 +0800
Subject: [PATCH] feat(models): implement model configuration dialogs for
 Azure, Bedrock and Ollama

---
 src/interfaces/request/llm.ts                 |   6 +-
 .../components/Dialog/AzureOpenAIDialog.tsx   | 183 +++++++++---
 .../components/Dialog/BedrockDialog.tsx       | 227 ++++++++++++---
 .../components/Dialog/OllamaDialog.tsx        | 270 ++++++++++++++++--
 src/pages/setting/hooks/useModelDialogs.ts    |  43 +--
 src/pages/setting/models.tsx                  |  18 +-
 6 files changed, 603 insertions(+), 144 deletions(-)

diff --git a/src/interfaces/request/llm.ts b/src/interfaces/request/llm.ts
index 9178a5a..2daa24e 100644
--- a/src/interfaces/request/llm.ts
+++ b/src/interfaces/request/llm.ts
@@ -9,9 +9,9 @@ export interface ISetApiKeyRequestBody {
 export interface IAddLlmRequestBody {
   llm_factory: string; // Ollama
   llm_name: string;
-  model_type: string;
-  api_base?: string; // chat|embedding|speech2text|image2text
-  api_key: string;
+  model_type: string;// chat|embedding|speech2text|image2text
+  api_base: string; 
+  api_key?: string;
   max_tokens: number;
 }
 
diff --git a/src/pages/setting/components/Dialog/AzureOpenAIDialog.tsx b/src/pages/setting/components/Dialog/AzureOpenAIDialog.tsx
index 385288b..1dbd98e 100644
--- a/src/pages/setting/components/Dialog/AzureOpenAIDialog.tsx
+++ b/src/pages/setting/components/Dialog/AzureOpenAIDialog.tsx
@@ -11,14 +11,26 @@ import {
   IconButton,
   InputAdornment,
   CircularProgress,
+  MenuItem,
+  Select,
+  FormControl,
+  InputLabel,
+  FormHelperText,
+  Link,
 } from '@mui/material';
 import { Visibility, VisibilityOff } from '@mui/icons-material';
 import { Controller, useForm } from 'react-hook-form';
+import type { IAddLlmRequestBody } from '@/interfaces/request/llm';
+
+// 模型类型选项
+const MODEL_TYPE_OPTIONS = [
+  { value: 'chat', label: 'Chat' },
+  { value: 'embedding', label: 'Embedding' },
+  { value: 'image2text', label: 'Image2Text' },
+];
 
 // 表单数据接口
-export interface AzureOpenAIFormData {
-  api_key: string;
-  endpoint: string;
+export interface AzureOpenAIFormData extends IAddLlmRequestBody {
   api_version: string;
 }
 
@@ -35,7 +47,7 @@ export interface AzureOpenAIDialogProps {
 /**
  * Azure OpenAI 配置对话框
  */
-function AzureOpenAIDialog ({
+function AzureOpenAIDialog({
   open,
   onClose,
   onSubmit,
@@ -52,16 +64,28 @@ function AzureOpenAIDialog ({
     formState: { errors },
   } = useForm<AzureOpenAIFormData>({
     defaultValues: {
+      model_type: 'embedding',
+      llm_name: 'gpt-3.5-turbo',
+      api_base: '',
       api_key: '',
-      endpoint: '',
       api_version: '2024-02-01',
+      max_tokens: 4096,
+      llm_factory: 'Azure-OpenAI',
     },
   });
 
   // 当对话框打开或初始数据变化时重置表单
   useEffect(() => {
     if (open) {
-      reset(initialData || { api_key: '', endpoint: '', api_version: '2024-02-01' });
+      reset({
+        model_type: 'embedding',
+        llm_name: 'gpt-3.5-turbo',
+        api_base: '',
+        api_key: '',
+        api_version: '2024-02-01',
+        max_tokens: 4096,
+        llm_factory: initialData?.llm_factory || 'Azure-OpenAI',
+      });
     }
   }, [open, initialData, reset]);
 
@@ -80,19 +104,86 @@ function AzureOpenAIDialog ({
       </DialogTitle>
       <DialogContent>
         <Box component="form" sx={{ mt: 2 }}>
+          {/* 模型类型选择 */}
           <Controller
-            name="api_key"
+            name="model_type"
             control={control}
-            rules={{ required: 'API Key 是必填项' }}
+            rules={{ required: '模型类型是必填项' }}
+            render={({ field }) => (
+              <FormControl fullWidth margin="normal" error={!!errors.model_type}>
+                <InputLabel>模型类型</InputLabel>
+                <Select
+                  {...field}
+                  label="模型类型"
+                >
+                  {MODEL_TYPE_OPTIONS.map((option) => (
+                    <MenuItem key={option.value} value={option.value}>
+                      {option.label}
+                    </MenuItem>
+                  ))}
+                </Select>
+                {errors.model_type && (
+                  <FormHelperText>{errors.model_type.message}</FormHelperText>
+                )}
+              </FormControl>
+            )}
+          />
+
+          {/* 模型名称 */}
+          <Controller
+            name="llm_name"
+            control={control}
+            rules={{ required: '模型名称是必填项' }}
             render={({ field }) => (
               <TextField
                 {...field}
                 fullWidth
-                label="API Key"
+                label="模型名称"
+                margin="normal"
+                error={!!errors.llm_name}
+                helperText={errors.llm_name?.message || '请输入模型名称'}
+                placeholder="gpt-3.5-turbo"
+              />
+            )}
+          />
+
+          {/* 基础 URL */}
+          <Controller
+            name="api_base"
+            control={control}
+            rules={{
+              required: '基础 URL 是必填项',
+              pattern: {
+                value: /^https?:\/\/.+/,
+                message: '基础 URL 必须是有效的 URL'
+              }
+            }}
+            render={({ field }) => (
+              <TextField
+                {...field}
+                fullWidth
+                label="基础 Url"
+                margin="normal"
+                error={!!errors.api_base}
+                helperText={errors.api_base?.message || 'Azure OpenAI 服务的端点 URL'}
+                placeholder="https://your-resource.openai.azure.com/"
+              />
+            )}
+          />
+
+          {/* API Key */}
+          <Controller
+            name="api_key"
+            control={control}
+            render={({ field }) => (
+              <TextField
+                {...field}
+                fullWidth
+                label="API-Key"
                 type={showApiKey ? 'text' : 'password'}
                 margin="normal"
                 error={!!errors.api_key}
-                helperText={errors.api_key?.message}
+                helperText={errors.api_key?.message || '输入api key（如果是本地部署的模型，请忽略）'}
                 InputProps={{
                   endAdornment: (
                     <InputAdornment position="end">
@@ -110,29 +201,7 @@ function AzureOpenAIDialog ({
             )}
           />
 
-          <Controller
-            name="endpoint"
-            control={control}
-            rules={{ 
-              required: 'Endpoint 是必填项',
-              pattern: {
-                value: /^https?:\/\/.+/,
-                message: 'Endpoint 必须是有效的 URL'
-              }
-            }}
-            render={({ field }) => (
-              <TextField
-                {...field}
-                fullWidth
-                label="Endpoint"
-                margin="normal"
-                error={!!errors.endpoint}
-                helperText={errors.endpoint?.message || 'Azure OpenAI 服务的端点 URL'}
-                placeholder="https://your-resource.openai.azure.com/"
-              />
-            )}
-          />
-
+          {/* API Version */}
           <Controller
             name="api_version"
             control={control}
@@ -149,20 +218,46 @@ function AzureOpenAIDialog ({
               />
             )}
           />
+
+          {/* 最大token数 */}
+          <Controller
+            name="max_tokens"
+            control={control}
+            rules={{
+              required: '最大token数是必填项',
+              min: { value: 1, message: '最大token数必须大于0' },
+              max: { value: 100000, message: '最大token数不能超过100000' }
+            }}
+            render={({ field }) => (
+              <TextField
+                {...field}
+                fullWidth
+                label="最大token数"
+                type="number"
+                margin="normal"
+                error={!!errors.max_tokens}
+                helperText={errors.max_tokens?.message || '设置了模型输出的最大长度，以token（单词片段）的数量表示'}
+                onChange={(e) => field.onChange(parseInt(e.target.value) || 0)}
+              />
+            )}
+          />
         </Box>
       </DialogContent>
       <DialogActions>
-        <Button onClick={onClose} disabled={loading}>
-          取消
-        </Button>
-        <Button
-          onClick={handleSubmit(handleFormSubmit)}
-          variant="contained"
-          disabled={loading}
-          startIcon={loading ? <CircularProgress size={20} /> : null}
-        >
-          确定
-        </Button>
+          {/* 右侧按钮组 */}
+          <Box sx={{ display: 'flex', gap: 1 }}>
+            <Button onClick={onClose} disabled={loading}>
+              取消
+            </Button>
+            <Button
+              onClick={handleSubmit(handleFormSubmit)}
+              variant="contained"
+              disabled={loading}
+              startIcon={loading ? <CircularProgress size={20} /> : null}
+            >
+              确定
+            </Button>
+          </Box>
       </DialogActions>
     </Dialog>
   );
diff --git a/src/pages/setting/components/Dialog/BedrockDialog.tsx b/src/pages/setting/components/Dialog/BedrockDialog.tsx
index f85a886..6a66ceb 100644
--- a/src/pages/setting/components/Dialog/BedrockDialog.tsx
+++ b/src/pages/setting/components/Dialog/BedrockDialog.tsx
@@ -15,25 +15,63 @@ import {
   Select,
   MenuItem,
   CircularProgress,
+  FormHelperText,
+  Link,
 } from '@mui/material';
 import { Visibility, VisibilityOff } from '@mui/icons-material';
 import { Controller, useForm } from 'react-hook-form';
+import type { IAddLlmRequestBody } from '@/interfaces/request/llm';
 
 // AWS Bedrock 支持的区域列表
 export const BEDROCK_REGIONS = [
-  { value: 'us-east-1', label: 'US East (N. Virginia)' },
-  { value: 'us-west-2', label: 'US West (Oregon)' },
-  { value: 'ap-southeast-2', label: 'Asia Pacific (Sydney)' },
-  { value: 'ap-northeast-1', label: 'Asia Pacific (Tokyo)' },
-  { value: 'eu-central-1', label: 'Europe (Frankfurt)' },
-  { value: 'eu-west-3', label: 'Europe (Paris)' },
+  'us-east-2',
+  'us-east-1',
+  'us-west-1',
+  'us-west-2',
+  'af-south-1',
+  'ap-east-1',
+  'ap-south-2',
+  'ap-southeast-3',
+  'ap-southeast-5',
+  'ap-southeast-4',
+  'ap-south-1',
+  'ap-northeast-3',
+  'ap-northeast-2',
+  'ap-southeast-1',
+  'ap-southeast-2',
+  'ap-east-2',
+  'ap-southeast-7',
+  'ap-northeast-1',
+  'ca-central-1',
+  'ca-west-1',
+  'eu-central-1',
+  'eu-west-1',
+  'eu-west-2',
+  'eu-south-1',
+  'eu-west-3',
+  'eu-south-2',
+  'eu-north-1',
+  'eu-central-2',
+  'il-central-1',
+  'mx-central-1',
+  'me-south-1',
+  'me-central-1',
+  'sa-east-1',
+  'us-gov-east-1',
+  'us-gov-west-1',
+];
+
+// 模型类型选项
+const MODEL_TYPE_OPTIONS = [
+  { value: 'chat', label: 'Chat' },
+  { value: 'embedding', label: 'Embedding' },
 ];
 
 // 表单数据接口
-export interface BedrockFormData {
-  access_key_id: string;
-  secret_access_key: string;
-  region: string;
+export interface BedrockFormData extends IAddLlmRequestBody {
+  bedrock_ak: string;
+  bedrock_sk: string;
+  bedrock_region: string;
 }
 
 // 对话框 Props 接口
@@ -67,16 +105,28 @@ function BedrockDialog ({
     formState: { errors },
   } = useForm<BedrockFormData>({
     defaultValues: {
-      access_key_id: '',
-      secret_access_key: '',
-      region: 'us-east-1',
+      model_type: 'chat',
+      llm_name: '',
+      bedrock_ak: '',
+      bedrock_sk: '',
+      bedrock_region: 'us-east-1',
+      max_tokens: 4096,
+      llm_factory: 'Bedrock',
     },
   });
 
   // 当对话框打开或初始数据变化时重置表单
   useEffect(() => {
     if (open) {
-      reset(initialData || { access_key_id: '', secret_access_key: '', region: 'us-east-1' });
+      reset({
+        model_type: 'chat',
+        llm_name: '',
+        bedrock_ak: '',
+        bedrock_sk: '',
+        bedrock_region: 'us-east-1',
+        max_tokens: 4096,
+        llm_factory: initialData?.llm_factory || 'Bedrock',
+      });
     }
   }, [open, initialData, reset]);
 
@@ -92,26 +142,73 @@ function BedrockDialog ({
     setShowSecretKey(!showSecretKey);
   };
 
+  const docInfo = {
+    url: 'https://console.aws.amazon.com/',
+    text: '如何集成 Bedrock',
+  };
+
   return (
     <Dialog open={open} onClose={onClose} maxWidth="sm" fullWidth>
       <DialogTitle>
-        {editMode ? '编辑' : '配置'} AWS Bedrock
+        {editMode ? '编辑' : '添加'} LLM
       </DialogTitle>
       <DialogContent>
         <Box component="form" sx={{ mt: 2 }}>
+          {/* 模型类型 */}
           <Controller
-            name="access_key_id"
+            name="model_type"
             control={control}
-            rules={{ required: 'Access Key ID 是必填项' }}
+            rules={{ required: '模型类型是必填项' }}
+            render={({ field }) => (
+              <FormControl fullWidth margin="normal" error={!!errors.model_type}>
+                <InputLabel>* 模型类型</InputLabel>
+                <Select {...field} label="* 模型类型">
+                  {MODEL_TYPE_OPTIONS.map((option) => (
+                    <MenuItem key={option.value} value={option.value}>
+                      {option.label}
+                    </MenuItem>
+                  ))}
+                </Select>
+                {errors.model_type && (
+                  <FormHelperText>{errors.model_type.message}</FormHelperText>
+                )}
+              </FormControl>
+            )}
+          />
+
+          {/* 模型名称 */}
+          <Controller
+            name="llm_name"
+            control={control}
+            rules={{ required: '模型名称是必填项' }}
             render={({ field }) => (
               <TextField
                 {...field}
                 fullWidth
-                label="Access Key ID"
+                label="* 模型名称"
+                margin="normal"
+                placeholder="请输入模型名称"
+                error={!!errors.llm_name}
+                helperText={errors.llm_name?.message}
+              />
+            )}
+          />
+
+          {/* ACCESS KEY */}
+          <Controller
+            name="bedrock_ak"
+            control={control}
+            rules={{ required: 'ACCESS KEY 是必填项' }}
+            render={({ field }) => (
+              <TextField
+                {...field}
+                fullWidth
+                label="* ACCESS KEY"
                 type={showAccessKey ? 'text' : 'password'}
                 margin="normal"
-                error={!!errors.access_key_id}
-                helperText={errors.access_key_id?.message}
+                placeholder="请输入 ACCESS KEY"
+                error={!!errors.bedrock_ak}
+                helperText={errors.bedrock_ak?.message}
                 InputProps={{
                   endAdornment: (
                     <InputAdornment position="end">
@@ -129,19 +226,21 @@ function BedrockDialog ({
             )}
           />
 
+          {/* SECRET KEY */}
           <Controller
-            name="secret_access_key"
+            name="bedrock_sk"
             control={control}
-            rules={{ required: 'Secret Access Key 是必填项' }}
+            rules={{ required: 'SECRET KEY 是必填项' }}
             render={({ field }) => (
               <TextField
                 {...field}
                 fullWidth
-                label="Secret Access Key"
+                label="* SECRET KEY"
                 type={showSecretKey ? 'text' : 'password'}
                 margin="normal"
-                error={!!errors.secret_access_key}
-                helperText={errors.secret_access_key?.message}
+                placeholder="请输入 SECRET KEY"
+                error={!!errors.bedrock_sk}
+                helperText={errors.bedrock_sk?.message}
                 InputProps={{
                   endAdornment: (
                     <InputAdornment position="end">
@@ -159,42 +258,76 @@ function BedrockDialog ({
             )}
           />
 
+          {/* AWS Region */}
           <Controller
-            name="region"
+            name="bedrock_region"
             control={control}
-            rules={{ required: 'Region 是必填项' }}
+            rules={{ required: 'AWS Region 是必填项' }}
             render={({ field }) => (
-              <FormControl fullWidth margin="normal" error={!!errors.region}>
-                <InputLabel>Region</InputLabel>
-                <Select {...field} label="Region">
+              <FormControl fullWidth margin="normal" error={!!errors.bedrock_region}>
+                <InputLabel>* AWS Region</InputLabel>
+                <Select {...field} label="* AWS Region">
                   {BEDROCK_REGIONS.map((region) => (
-                    <MenuItem key={region.value} value={region.value}>
-                      {region.label}
+                    <MenuItem key={region} value={region}>
+                      {region}
                     </MenuItem>
                   ))}
                 </Select>
-                {errors.region && (
-                  <Typography variant="caption" color="error" sx={{ mt: 1, ml: 2 }}>
-                    {errors.region.message}
-                  </Typography>
+                {errors.bedrock_region && (
+                  <FormHelperText>{errors.bedrock_region.message}</FormHelperText>
                 )}
               </FormControl>
             )}
           />
+
+          {/* 最大token数 */}
+          <Controller
+            name="max_tokens"
+            control={control}
+            rules={{
+              required: '最大token数是必填项',
+              min: { value: 1, message: '最大token数必须大于0' },
+            }}
+            render={({ field }) => (
+              <TextField
+                {...field}
+                fullWidth
+                label="* 最大token数"
+                type="number"
+                margin="normal"
+                placeholder="这设置了模型输出的最大长度，以token（单词或词片段）的数量来衡量"
+                error={!!errors.max_tokens}
+                helperText={errors.max_tokens?.message}
+                onChange={(e) => field.onChange(Number(e.target.value))}
+              />
+            )}
+          />
         </Box>
       </DialogContent>
       <DialogActions>
-        <Button onClick={onClose} disabled={loading}>
-          取消
-        </Button>
-        <Button
-          onClick={handleSubmit(handleFormSubmit)}
-          variant="contained"
-          disabled={loading}
-          startIcon={loading ? <CircularProgress size={20} /> : null}
-        >
-          确定
-        </Button>
+        <Box sx={{ display: 'flex', justifyContent: 'space-between', width: '100%' }}>
+          <Link
+            href={docInfo.url}
+            target="_blank"
+            rel="noopener noreferrer"
+            sx={{ alignSelf: 'center', textDecoration: 'none', ml:2 }}
+          >
+            {docInfo.text}
+          </Link>
+          <Box>
+            <Button onClick={onClose} disabled={loading} sx={{ mr: 1 }}>
+              取消
+            </Button>
+            <Button
+              onClick={handleSubmit(handleFormSubmit)}
+              variant="contained"
+              disabled={loading}
+              startIcon={loading ? <CircularProgress size={20} /> : null}
+            >
+              确定
+            </Button>
+          </Box>
+        </Box>
       </DialogActions>
     </Dialog>
   );
diff --git a/src/pages/setting/components/Dialog/OllamaDialog.tsx b/src/pages/setting/components/Dialog/OllamaDialog.tsx
index 88e183b..d574c10 100644
--- a/src/pages/setting/components/Dialog/OllamaDialog.tsx
+++ b/src/pages/setting/components/Dialog/OllamaDialog.tsx
@@ -1,4 +1,4 @@
-import React, { useEffect } from 'react';
+import React, { useEffect, useMemo } from 'react';
 import {
   Dialog,
   DialogTitle,
@@ -9,14 +9,28 @@ import {
   Box,
   Typography,
   CircularProgress,
+  MenuItem,
+  Select,
+  FormControl,
+  InputLabel,
+  FormHelperText,
+  Link,
 } from '@mui/material';
 import { Controller, useForm } from 'react-hook-form';
+import logger from '@/utils/logger';
+import { LLM_FACTORY_LIST, type LLMFactory } from '@/constants/llm';
 
 // 表单数据接口
 export interface OllamaFormData {
-  base_url: string;
+  model_type: string;
+  llm_name: string;
+  api_base: string;
+  api_key?: string;
+  max_tokens: number;
+  llm_factory: string;
 }
 
+
 // 对话框 Props 接口
 export interface OllamaDialogProps {
   open: boolean;
@@ -27,10 +41,49 @@ export interface OllamaDialogProps {
   editMode?: boolean;
 }
 
+const llmFactoryToUrlMap: { [x: string]: string } = {
+  [LLM_FACTORY_LIST.Ollama]:
+    'https://github.com/infiniflow/ragflow/blob/main/docs/guides/models/deploy_local_llm.mdx',
+  [LLM_FACTORY_LIST.Xinference]:
+    'https://inference.readthedocs.io/en/latest/user_guide',
+  [LLM_FACTORY_LIST.ModelScope]:
+    'https://www.modelscope.cn/docs/model-service/API-Inference/intro',
+  [LLM_FACTORY_LIST.LocalAI]: 'https://localai.io/docs/getting-started/models/',
+  [LLM_FACTORY_LIST.LMStudio]: 'https://lmstudio.ai/docs/basics',
+  [LLM_FACTORY_LIST.OpenAiAPICompatible]:
+    'https://platform.openai.com/docs/models/gpt-4',
+  [LLM_FACTORY_LIST.TogetherAI]: 'https://docs.together.ai/docs/deployment-options',
+  [LLM_FACTORY_LIST.Replicate]: 'https://replicate.com/docs/topics/deployments',
+  [LLM_FACTORY_LIST.OpenRouter]: 'https://openrouter.ai/docs',
+  [LLM_FACTORY_LIST.HuggingFace]:
+    'https://huggingface.co/docs/text-embeddings-inference/quick_tour',
+  [LLM_FACTORY_LIST.GPUStack]: 'https://docs.gpustack.ai/latest/quickstart',
+  [LLM_FACTORY_LIST.VLLM]: 'https://docs.vllm.ai/en/latest/',
+} as const;
+
+function getURLByFactory(factory: LLMFactory) {
+  const url = llmFactoryToUrlMap[factory];
+  return {
+    textTip: `如何集成 ${factory}`,
+    url,
+  }
+}
+
+
+// 模型类型选项
+const MODEL_TYPE_OPTIONS = [
+  { value: 'chat', label: 'Chat' },
+  { value: 'embedding', label: 'Embedding' },
+  { value: 'rerank', label: 'Rerank' },
+  { value: 'image2text', label: 'Image2Text' },
+  { value: 'speech2text', label: 'Speech2Text' },
+];
+
+
 /**
- * Ollama 配置对话框
+ * Ollama / local llm 配置对话框
  */
-function OllamaDialog ({
+function OllamaDialog({
   open,
   onClose,
   onSubmit,
@@ -45,14 +98,60 @@ function OllamaDialog ({
     formState: { errors },
   } = useForm<OllamaFormData>({
     defaultValues: {
-      base_url: 'http://localhost:11434',
+      model_type: 'chat',
+      llm_name: '',
+      api_base: 'http://localhost:11434',
+      api_key: '',
+      max_tokens: 4096,
+      llm_factory: 'Ollama',
     },
   });
 
+  const modelTypeOptions = useMemo(() => {
+    const factory = initialData?.llm_factory || LLM_FACTORY_LIST.Ollama;
+    if (factory == LLM_FACTORY_LIST.HuggingFace) {
+      return [
+        { value: 'embedding', label: 'Embedding' },
+        { value: 'chat', label: 'Chat' },
+        { value: 'rerank', label: 'Rerank' },
+      ]
+    } else if (factory == LLM_FACTORY_LIST.Xinference) {
+      return [
+        { value: 'chat', label: 'Chat' },
+        { value: 'embedding', label: 'Embedding' },
+        { value: 'rerank', label: 'Rerank' },
+        { value: 'image2text', label: 'Image2Text' },
+        { value: 'speech2text', label: 'Speech2Text' },
+        { value: 'tts', label: 'TTS' },
+      ]
+    } else if (factory == LLM_FACTORY_LIST.ModelScope) {
+      return [
+        { value: 'chat', label: 'Chat' },
+      ]
+    } else if (factory == LLM_FACTORY_LIST.GPUStack) {
+      return [
+        { value: 'chat', label: 'Chat' },
+        { value: 'embedding', label: 'Embedding' },
+        { value: 'rerank', label: 'Rerank' },
+        { value: 'image2text', label: 'Image2Text' },
+      ]
+    }
+    return MODEL_TYPE_OPTIONS;
+  }, [initialData])
+
+  logger.debug('OllamaDialog', { open, initialData, editMode });
+
   // 当对话框打开或初始数据变化时重置表单
   useEffect(() => {
     if (open) {
-      reset(initialData || { base_url: 'http://localhost:11434' });
+      reset({
+        model_type: 'chat',
+        llm_name: '',
+        api_base: initialData?.api_base,
+        api_key: initialData?.api_key,
+        max_tokens: initialData?.max_tokens,
+        llm_factory: initialData?.llm_factory || 'Ollama',
+      });
     }
   }, [open, initialData, reset]);
 
@@ -60,49 +159,168 @@ function OllamaDialog ({
     onSubmit(data);
   };
 
+  // 获取文档链接信息
+  const docInfo = getURLByFactory((initialData?.llm_factory || LLM_FACTORY_LIST.Ollama) as LLMFactory);
+
   return (
     <Dialog open={open} onClose={onClose} maxWidth="sm" fullWidth>
       <DialogTitle>
-        {editMode ? '编辑' : '配置'} Ollama
+        {editMode ? `编辑 ${initialData?.llm_factory || LLM_FACTORY_LIST.Ollama}` : `配置 ${initialData?.llm_factory || LLM_FACTORY_LIST.Ollama}`}
       </DialogTitle>
       <DialogContent>
         <Box component="form" sx={{ mt: 2 }}>
+          {/* 模型类型选择 */}
           <Controller
-            name="base_url"
+            name="model_type"
             control={control}
-            rules={{ 
-              required: 'Base URL 是必填项',
+            rules={{ required: '模型类型是必填项' }}
+            render={({ field }) => (
+              <FormControl fullWidth margin="normal" error={!!errors.model_type}>
+                <InputLabel>模型类型 *</InputLabel>
+                <Select
+                  {...field}
+                  label="模型类型 *"
+                >
+                  {modelTypeOptions.map((option) => (
+                    <MenuItem key={option.value} value={option.value}>
+                      {option.label}
+                    </MenuItem>
+                  ))}
+                </Select>
+                {errors.model_type && (
+                  <FormHelperText>{errors.model_type.message}</FormHelperText>
+                )}
+              </FormControl>
+            )}
+          />
+
+          {/* 模型名称 */}
+          <Controller
+            name="llm_name"
+            control={control}
+            rules={{ required: '模型名称是必填项' }}
+            render={({ field }) => (
+              <TextField
+                {...field}
+                fullWidth
+                label="模型名称"
+                margin="normal"
+                required
+                error={!!errors.llm_name}
+                helperText={errors.llm_name?.message || '请输入模型名称'}
+                placeholder="例如: llama2, mistral"
+              />
+            )}
+          />
+
+          {/* 基础 URL */}
+          <Controller
+            name="api_base"
+            control={control}
+            rules={{
+              required: '基础 URL 是必填项',
               pattern: {
                 value: /^https?:\/\/.+/,
-                message: 'Base URL 必须是有效的 URL'
+                message: '基础 URL 必须是有效的 URL'
               }
             }}
             render={({ field }) => (
               <TextField
                 {...field}
                 fullWidth
-                label="Base URL"
+                label="基础 URL"
                 margin="normal"
-                error={!!errors.base_url}
-                helperText={errors.base_url?.message || 'Ollama 服务的基础 URL'}
-                placeholder="http://localhost:11434"
+                required
+                error={!!errors.api_base}
+                helperText={errors.api_base?.message || '基础 URL'}
+                placeholder="http://localhost:8888"
+              />
+            )}
+          />
+
+          {/* API Key (可选) */}
+          <Controller
+            name="api_key"
+            control={control}
+            render={({ field }) => (
+              <TextField
+                {...field}
+                fullWidth
+                label="API Key"
+                margin="normal"
+                error={!!errors.api_key}
+                helperText={errors.api_key?.message || 'API Key (可选)'}
+                placeholder="如果需要认证，请输入 API Key"
+              />
+            )}
+          />
+
+          {/* 最大 Token 数 */}
+          <Controller
+            name="max_tokens"
+            control={control}
+            rules={{
+              required: '最大 Token 数是必填项',
+              min: {
+                value: 1,
+                message: '最大 Token 数必须大于 0'
+              },
+              max: {
+                value: 100000,
+                message: '最大 Token 数不能超过 100000'
+              }
+            }}
+            render={({ field }) => (
+              <TextField
+                {...field}
+                fullWidth
+                label="最大 Token 数"
+                margin="normal"
+                type="number"
+                required
+                error={!!errors.max_tokens}
+                helperText={errors.max_tokens?.message || '模型支持的最大 Token 数'}
+                placeholder="4096"
+                onChange={(e) => field.onChange(parseInt(e.target.value) || 0)}
               />
             )}
           />
         </Box>
       </DialogContent>
       <DialogActions>
-        <Button onClick={onClose} disabled={loading}>
-          取消
-        </Button>
-        <Button
-          onClick={handleSubmit(handleFormSubmit)}
-          variant="contained"
-          disabled={loading}
-          startIcon={loading ? <CircularProgress size={20} /> : null}
-        >
-          确定
-        </Button>
+        <Box sx={{ display: 'flex', justifyContent: 'space-between', width: '100%', alignItems: 'center' }}>
+          {/* 左侧文档链接 */}
+          <Link
+            href={docInfo.url}
+            target="_blank"
+            rel="noopener noreferrer"
+            sx={{
+              ml: 2,
+              fontSize: '16',
+              textDecoration: 'none',
+              '&:hover': {
+                textDecoration: 'underline'
+              }
+            }}
+          >
+            {docInfo.textTip}
+          </Link>
+
+          {/* 右侧按钮组 */}
+          <Box sx={{ display: 'flex', gap: 1 }}>
+            <Button onClick={onClose} disabled={loading}>
+              取消
+            </Button>
+            <Button
+              onClick={handleSubmit(handleFormSubmit)}
+              variant="contained"
+              disabled={loading}
+              startIcon={loading ? <CircularProgress size={20} /> : null}
+            >
+              确定
+            </Button>
+          </Box>
+        </Box>
       </DialogActions>
     </Dialog>
   );
diff --git a/src/pages/setting/hooks/useModelDialogs.ts b/src/pages/setting/hooks/useModelDialogs.ts
index 43f81dd..0a8c300 100644
--- a/src/pages/setting/hooks/useModelDialogs.ts
+++ b/src/pages/setting/hooks/useModelDialogs.ts
@@ -107,23 +107,25 @@ export const useAzureOpenAIDialog = () => {
     dialogState.setLoading(true);
     try {
       // 调用 Azure OpenAI 特定的 API
-      await userService.set_api_key({
-        llm_factory: 'AzureOpenAI',
-        // llm_name: data.deployment_name,
+      await userService.add_llm({
+        llm_factory: data.llm_factory,
+        llm_name: data.llm_name,
+        model_type: data.model_type,
+        api_base: data.api_base,
         api_key: data.api_key,
-        // azure_endpoint: data.azure_endpoint,
-        // api_version: data.api_version,
+        // @ts-ignore
+        api_version: data.api_version,
+        max_tokens: data.max_tokens,
       });
       showMessage.success('Azure OpenAI 配置成功');
       dialogState.closeDialog();
     } catch (error) {
       logger.error('Azure OpenAI 配置失败:', error);
-      showMessage.error('Azure OpenAI 配置失败');
       throw error;
     } finally {
       dialogState.setLoading(false);
     }
-  }, [dialogState]);
+  }, [dialogState, showMessage]);
 
   return {
     ...dialogState,
@@ -140,13 +142,15 @@ export const useBedrockDialog = () => {
     dialogState.setLoading(true);
     try {
       // 调用 Bedrock 特定的 API
-      await userService.set_api_key({
-        llm_factory: 'Bedrock',
-        llm_name: '',
-        api_key: '', // Bedrock 使用 access key
-        // access_key_id: data.access_key_id,
-        // secret_access_key: data.secret_access_key,
-        // region: data.region,
+      await userService.add_llm({
+        llm_factory: data.llm_factory,
+        llm_name: data.llm_name,
+        model_type: data.model_type,
+        // @ts-ignore
+        bedrock_ak: data.bedrock_ak,
+        bedrock_sk: data.bedrock_sk,
+        bedrock_region: data.bedrock_region,
+        max_tokens: data.max_tokens,
       });
       showMessage.success('AWS Bedrock 配置成功');
       dialogState.closeDialog();
@@ -175,9 +179,12 @@ export const useOllamaDialog = () => {
     try {
       // 调用添加 LLM 的 API
       await userService.add_llm({
-        llm_factory: 'Ollama',
-        // llm_name: data.model_name,
-        // base_url: data.base_url,
+        llm_factory: data.llm_factory,
+        llm_name: data.llm_name,
+        model_type: data.model_type,
+        api_base: data.api_base,
+        api_key: data.api_key || '',
+        max_tokens: data.max_tokens,
       });
       showMessage.success('Ollama 模型添加成功');
       dialogState.closeDialog();
@@ -188,7 +195,7 @@ export const useOllamaDialog = () => {
     } finally {
       dialogState.setLoading(false);
     }
-  }, [dialogState]);
+  }, [dialogState, showMessage]);
 
   return {
     ...dialogState,
diff --git a/src/pages/setting/models.tsx b/src/pages/setting/models.tsx
index b89a174..b5cceda 100644
--- a/src/pages/setting/models.tsx
+++ b/src/pages/setting/models.tsx
@@ -69,7 +69,7 @@ function ModelsPage() {
   const { llmFactory, myLlm, refreshLlmModel } = useLlmModelSetting();
   const modelDialogs = useModelDialogs(refreshLlmModel);
 
-  // 折叠状态管理 - 使用 Map 来管理每个工厂的折叠状态
+  // 折叠状态管理 - 使用 Map 来管理每个工厂的折叠状态，默认所有工厂都是折叠的
   const [collapsedFactories, setCollapsedFactories] = useState<Record<string, boolean>>({});
 
   // 切换工厂折叠状态
@@ -120,11 +120,17 @@ function ModelsPage() {
     // 然后有很多自定义的配置项，需要单独用 dialog 来配置
     const factoryName = factory.name as LLMFactory;
     if (LocalLlmFactories.includes(factoryName)) {
-      // modelDialogs.localLlmDialog.openLocalLlmDialog(factoryName);
+      modelDialogs.ollamaDialog.openDialog({
+        llm_factory: factory.name,
+      });
     } else if (factoryName == LLM_FACTORY_LIST.AzureOpenAI) {
-
+      modelDialogs.azureDialog.openDialog({
+        llm_factory: factory.name,
+      });
     } else if (factoryName == LLM_FACTORY_LIST.Bedrock) {
-
+      modelDialogs.bedrockDialog.openDialog({
+        llm_factory: factory.name,
+      });
     } else if (factoryName == LLM_FACTORY_LIST.BaiduYiYan) {
 
     } else if (factoryName == LLM_FACTORY_LIST.GoogleCloud) {
@@ -241,7 +247,7 @@ function ModelsPage() {
                           <Box sx={{ display: 'flex', alignItems: 'center', gap: 1 }}>
                             {/* 折叠/展开图标 */}
                             <IconButton size="small">
-                              {collapsedFactories[factoryName] ? <ExpandMoreIcon /> : <ExpandLessIcon />}
+                              {collapsedFactories[factoryName] ? <ExpandLessIcon />: <ExpandMoreIcon /> }
                             </IconButton>
                             <Box>
                               {/* 模型工厂名称 */}
@@ -281,7 +287,7 @@ function ModelsPage() {
                           </Box>
                         </Box>
                         {/* 模型列表 - 使用 Collapse 组件包装 */}
-                        <Collapse in={!collapsedFactories[factoryName]} timeout="auto" unmountOnExit>
+                        <Collapse in={collapsedFactories[factoryName]} timeout="auto" unmountOnExit>
                           <Box sx={{ mt: 2 }}>
                             <Grid container spacing={2}>
                               {group.llm.map((model) => (