diff --git a/src/renderer/src/aiCore/middleware/feat/ThinkingTagExtractionMiddleware.ts b/src/renderer/src/aiCore/middleware/feat/ThinkingTagExtractionMiddleware.ts
index 5ab19a6175..447b9d2f23 100644
--- a/src/renderer/src/aiCore/middleware/feat/ThinkingTagExtractionMiddleware.ts
+++ b/src/renderer/src/aiCore/middleware/feat/ThinkingTagExtractionMiddleware.ts
@@ -7,6 +7,7 @@ import {
ThinkingDeltaChunk,
ThinkingStartChunk
} from '@renderer/types/chunk'
+import { getLowerBaseModelName } from '@renderer/utils'
import { TagConfig, TagExtractor } from '@renderer/utils/tagExtraction'
import { CompletionsParams, CompletionsResult, GenericChunk } from '../schemas'
@@ -22,13 +23,16 @@ const reasoningTags: TagConfig[] = [
{ openingTag: '', closingTag: '', separator: '\n' },
{ openingTag: '###Thinking', closingTag: '###Response', separator: '\n' },
{ openingTag: '◁think▷', closingTag: '◁/think▷', separator: '\n' },
- { openingTag: '', closingTag: '', separator: '\n' }
+ { openingTag: '', closingTag: '', separator: '\n' },
+ { openingTag: '', closingTag: '', separator: '\n' }
]
const getAppropriateTag = (model?: Model): TagConfig => {
- if (model?.id?.includes('qwen3')) return reasoningTags[0]
- if (model?.id?.includes('gemini-2.5')) return reasoningTags[1]
- if (model?.id?.includes('kimi-vl-a3b-thinking')) return reasoningTags[3]
+ const modelId = model?.id ? getLowerBaseModelName(model.id) : undefined
+ if (modelId?.includes('qwen3')) return reasoningTags[0]
+ if (modelId?.includes('gemini-2.5')) return reasoningTags[1]
+ if (modelId?.includes('kimi-vl-a3b-thinking')) return reasoningTags[3]
+ if (modelId?.includes('seed-oss-36b')) return reasoningTags[5]
// 可以在这里添加更多模型特定的标签配置
return reasoningTags[0] // 默认使用 标签
}