From 076ceee29d00eaece90e7720bce4b8d27e8bbe3a Mon Sep 17 00:00:00 2001
From: yujiangping <yujiangping@taofen8.com>
Date: Fri, 6 Mar 2026 15:30:30 +0800
Subject: [PATCH] fix(web): filter vision models for image2text and cleanup
 tool management

- Add vision capability filter for image2text model options in CreateModal
- Filter model options to only include models with 'vision' capability when type is 'image2text'
- Remove outdated file header comments from ToolManagement component
- Comment out 'market' tab from tabKeys array in ToolManagement
- Ensure image2text tool only displays compatible vision-capable models
---
 web/src/views/KnowledgeBase/components/CreateModal.tsx | 10 +++++++++-
 web/src/views/ToolManagement/index.tsx                 | 10 +---------
 2 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/web/src/views/KnowledgeBase/components/CreateModal.tsx b/web/src/views/KnowledgeBase/components/CreateModal.tsx
index d9727d18..35eb52d0 100644
--- a/web/src/views/KnowledgeBase/components/CreateModal.tsx
+++ b/web/src/views/KnowledgeBase/components/CreateModal.tsx
@@ -672,9 +672,17 @@ const CreateModal = forwardRef<CreateModalRef, CreateModalRefProps>(({
       {currentType !== 'Folder' && dynamicTypeList.map((tp) => {
         const fieldKey = typeToFieldKey(tp);
         // When tp is 'llm', merge llm and chat options
-        const options = tp.toLowerCase() === 'llm' || tp.toLowerCase() === 'image2text'
+        let options = tp.toLowerCase() === 'llm' || tp.toLowerCase() === 'image2text'
           ? [...(modelOptionsByType['llm'] || []), ...(modelOptionsByType['chat'] || [])]
           : modelOptionsByType[tp] || [];
+        
+        // When tp is 'image2text', filter to only include models with 'vision' capability
+        if (tp.toLowerCase() === 'image2text') {
+          options = options.filter((opt: any) => {
+            const model = models?.items?.find((m: any) => m.id === opt.value);
+            return model?.capability?.includes('vision');
+          });
+        }
         return (
           <Form.Item
             key={tp}
diff --git a/web/src/views/ToolManagement/index.tsx b/web/src/views/ToolManagement/index.tsx
index c9383258..45dcc942 100644
--- a/web/src/views/ToolManagement/index.tsx
+++ b/web/src/views/ToolManagement/index.tsx
@@ -1,11 +1,3 @@
-/*
- * @Description: 
- * @Version: 0.0.1
- * @Author: yujiangping
- * @Date: 2026-01-05 17:22:23
- * @LastEditors: yujiangping
- * @LastEditTime: 2026-03-04 15:12:48
- */
 import React, { useState } from 'react';
 import { Tabs } from 'antd';
 import { useTranslation } from 'react-i18next';
@@ -16,7 +8,7 @@ import Custom from './Custom';
 import Market from './Market';
 import Tag from '@/components/Tag'
 
-const tabKeys = ['mcp', 'inner', 'custom', 'market']
+const tabKeys = ['mcp', 'inner', 'custom'] // , 'market'
 const ToolManagement: React.FC = () => {
   const { t } = useTranslation();
   const [activeTab, setActiveTab] = useState('mcp');