update models selected

src/settings/components/ModelProviderSettings.tsx

@@ -4,7 +4,7 @@ import { t } from '../../lang/helpers';
 import InfioPlugin from "../../main";
 import { ApiProvider } from '../../types/llm/model';
 import { InfioSettings } from '../../types/settings';
-import { GetAllProviders } from '../../utils/api';
+import { GetAllProviders, GetDefaultModelId, GetEmbeddingProviders } from '../../utils/api';
 import { getProviderApiUrl } from '../../utils/provider-urls';
 
 import { ApiKeyComponent, CustomUrlComponent } from './FormComponents';
@@ -57,7 +57,72 @@ const CustomProviderSettings: React.FC<CustomProviderSettingsProps> = ({ plugin,
 		onSettingsUpdate?.();
 	};
 
-	const providers = GetAllProviders();
+	const providers = GetAllProviders(); // 按照重要程度排序
+	const embeddingProviders = GetEmbeddingProviders(); // 按照重要程度排序
+
+	// 获取已设置API Key的提供商列表
+	const getSettedProviders = (): ApiProvider[] => {
+		return providers.filter(provider => {			
+			const providerSetting = getProviderSetting(provider);
+			return providerSetting.apiKey && providerSetting.apiKey.trim() !== '';
+		});
+	};
+
+	// 一键配置模型
+	const handleOneClickConfig = () => {
+		const settedProviders = getSettedProviders();
+		
+		if (settedProviders.length === 0) {
+			// 提示用户未设置任何key
+			alert("当前未设置任何key");
+			return;
+		}
+
+		// 选择chat和autocomplete的提供商（按providers排序选择最靠前的）
+		const selectedProvider = providers.find(provider => settedProviders.includes(provider));
+		
+		// 选择embedding的提供商（按embeddingProviders排序选择最靠前的）
+		const embeddingProvider = embeddingProviders.find(provider => settedProviders.includes(provider));
+
+		// 准备要更新的设置对象
+		const newSettings = { ...settings };
+		let hasUpdates = false;
+
+		if (selectedProvider) {
+			const defaultModels = GetDefaultModelId(selectedProvider);
+			
+			// 设置chat和autocomplete模型
+			if (defaultModels.chat) {
+				newSettings.chatModelProvider = selectedProvider;
+				newSettings.chatModelId = defaultModels.chat;
+				hasUpdates = true;
+				console.log(`已自动配置聊天模型：${selectedProvider}/${defaultModels.chat}`);
+			}
+			if (defaultModels.autoComplete) {
+				newSettings.applyModelProvider = selectedProvider;
+				newSettings.applyModelId = defaultModels.autoComplete;
+				hasUpdates = true;
+				console.log(`已自动配置自动补全模型：${selectedProvider}/${defaultModels.autoComplete}`);
+			}
+		}
+
+		if (embeddingProvider) {
+			const embeddingDefaultModels = GetDefaultModelId(embeddingProvider);
+			
+			// 设置embedding模型
+			if (embeddingDefaultModels.embedding) {
+				newSettings.embeddingModelProvider = embeddingProvider;
+				newSettings.embeddingModelId = embeddingDefaultModels.embedding;
+				hasUpdates = true;
+				console.log(`已自动配置嵌入模型：${embeddingProvider}/${embeddingDefaultModels.embedding}`);
+			}
+		}
+
+		// 一次性更新所有设置
+		if (hasUpdates) {
+			handleSettingsUpdate(newSettings);
+		}
+	};
 
 	const updateProviderApiKey = (provider: ApiProvider, value: string) => {
 		const providerKey = getProviderSettingKey(provider);
@@ -232,7 +297,16 @@ const CustomProviderSettings: React.FC<CustomProviderSettingsProps> = ({ plugin,
 
 			{/* 模型选择区域 */}
 			<div className="model-selection-section">
-				<h2 className="section-title">模型选择</h2>
+				<div className="model-selection-header">
+					<h2 className="section-title">模型选择</h2>
+					<button 
+						className="one-click-config-btn"
+						onClick={handleOneClickConfig}
+						title="自动配置模型为已设置API Key的提供商的推荐模型"
+					>
+						一键配置
+					</button>
+				</div>
 
 				<div className="model-selectors">
 					<ComboBoxComponent
@@ -293,6 +367,44 @@ const CustomProviderSettings: React.FC<CustomProviderSettingsProps> = ({ plugin,
 					border-bottom: 1px solid var(--background-modifier-border);
 				}
 
+				/* 模型选择区域头部样式 */
+				.model-selection-header {
+					display: flex;
+					align-items: center;
+					justify-content: space-between;
+					margin-bottom: var(--size-4-3);
+				}
+
+				.model-selection-header .section-title {
+					margin: 0;
+					padding-bottom: 0;
+					border-bottom: none;
+				}
+
+				/* 一键配置按钮样式 */
+				.one-click-config-btn {
+					background: var(--interactive-accent);
+					color: var(--text-on-accent);
+					border: none;
+					border-radius: var(--radius-s);
+					padding: var(--size-2-1) var(--size-4-2);
+					font-size: var(--font-ui-smaller);
+					font-weight: var(--font-weight-medium);
+					cursor: pointer;
+					transition: all 0.15s ease-in-out;
+					box-shadow: 0 1px 3px rgba(0, 0, 0, 0.1);
+				}
+
+				.one-click-config-btn:hover {
+					background: var(--interactive-accent-hover);
+					box-shadow: 0 2px 6px rgba(0, 0, 0, 0.15);
+				}
+
+				.one-click-config-btn:active {
+					transform: translateY(1px);
+					box-shadow: 0 1px 3px rgba(0, 0, 0, 0.1);
+				}
+
 				/* 提供商标签页容器 */
 				.provider-tabs {
 					display: flex;
@@ -380,6 +492,12 @@ const CustomProviderSettings: React.FC<CustomProviderSettingsProps> = ({ plugin,
 						padding: var(--size-2-1) var(--size-4-1);
 						font-size: var(--font-ui-smaller);
 					}
+
+					.model-selection-header {
+						flex-direction: column;
+						align-items: flex-start;
+						gap: var(--size-2-2);
+					}
 				}
 
 				/* 提供商名称高亮样式 */

src/utils/api.ts

@@ -35,6 +35,8 @@ export interface EmbeddingModelInfo {
 // https://docs.anthropic.com/en/docs/about-claude/models
 export type AnthropicModelId = keyof typeof anthropicModels
 export const anthropicDefaultModelId: AnthropicModelId = "claude-sonnet-4-20250514"
+export const anthropicDefaultAutoCompleteModelId: AnthropicModelId = "claude-3-5-haiku-20241022"
+export const anthropicDefaultEmbeddingModelId: AnthropicModelId = null // this is not supported embedding model
 export const anthropicModels = {
 	"claude-sonnet-4-20250514": {
 		maxTokens: 64_000, // Overridden to 8k if `enableReasoningEffort` is false.
@@ -128,7 +130,9 @@ export const anthropicModels = {
 } as const satisfies Record<string, ModelInfo> // as const assertion makes the object 
 
 // Infio
-export const infioDefaultModelId = "deepseek/deepseek-v3" // will always exist 
+export const infioDefaultModelId = "anthropic/claude-sonnet-4" // for chat
+export const infioDefaultAutoCompleteModelId = "groq/llama-3.3-70b-versatile" // for auto complete
+export const infioDefaultEmbeddingModelId = "openai/text-embedding-3-small" // for embedding
 export const infioDefaultModelInfo: ModelInfo = {
 	maxTokens: 8192,
 	contextWindow: 65_536,
@@ -210,7 +214,9 @@ export const infioEmbeddingModels = {
 
 // OpenRouter
 // https://openrouter.ai/models?order=newest&supported_parameters=tools
-export const openRouterDefaultModelId = "anthropic/claude-sonnet-4" // will always exist in openRouterModels
+export const openRouterDefaultModelId = "google/gemini-2.5-pro-preview" // for chat
+export const openRouterDefaultAutoCompleteModelId = "google/gemini-2.5-flash-preview-05-20" // for auto complete
+export const openRouterDefaultEmbeddingModelId = null // this is not supported embedding model
 export const openRouterDefaultModelInfo: ModelInfo = {
 	maxTokens: 8192,
 	contextWindow: 200_000,
@@ -262,7 +268,10 @@ async function fetchOpenRouterModels(): Promise<Record<string, ModelInfo>> {
 // Gemini
 // https://ai.google.dev/gemini-api/docs/models/gemini
 export type GeminiModelId = keyof typeof geminiModels
-export const geminiDefaultModelId: GeminiModelId = "gemini-2.5-flash-preview-05-20"
+export const geminiDefaultModelId: GeminiModelId = "gemini-2.5-pro-preview-05-06"
+export const geminiDefaultAutoCompleteModelId: GeminiModelId = "gemini-2.5-flash-preview-05-20"
+export const geminiDefaultEmbeddingModelId: keyof typeof geminiEmbeddingModels = "text-embedding-004"
+
 export const geminiModels = {
 	"gemini-2.5-flash-preview-05-20:thinking": {
 		maxTokens: 65_535,
@@ -489,6 +498,9 @@ export const geminiEmbeddingModels = {
 // https://openai.com/api/pricing/
 export type OpenAiNativeModelId = keyof typeof openAiNativeModels
 export const openAiNativeDefaultModelId: OpenAiNativeModelId = "gpt-4o"
+export const openAiNativeDefaultAutoCompleteModelId: OpenAiNativeModelId = "gpt-4o-mini"
+export const openAiNativeDefaultEmbeddingModelId: keyof typeof openAINativeEmbeddingModels = "text-embedding-3-small"
+
 export const openAiNativeModels = {
 	// don't support tool use yet
 	"o3-mini": {
@@ -594,6 +606,9 @@ export const openAINativeEmbeddingModels = {
 // https://api-docs.deepseek.com/quick_start/pricing
 export type DeepSeekModelId = keyof typeof deepSeekModels
 export const deepSeekDefaultModelId: DeepSeekModelId = "deepseek-chat"
+export const deepSeekDefaultAutoCompleteModelId: DeepSeekModelId = "deepseek-chat"
+export const deepSeekDefaultEmbeddingModelId = null // this is not supported embedding model
+
 export const deepSeekModels = {
 	"deepseek-chat": {
 		maxTokens: 8_000,
@@ -620,7 +635,10 @@ export const deepSeekModels = {
 // Qwen
 // https://help.aliyun.com/zh/model-studio/getting-started/
 export type QwenModelId = keyof typeof qwenModels
-export const qwenDefaultModelId: QwenModelId = "qwen-max-latest"
+export const qwenDefaultModelId: QwenModelId = "qwen3-235b-a22b"
+export const qwenDefaultAutoCompleteModelId: QwenModelId = "qwen3-32b"
+export const qwenDefaultEmbeddingModelId: keyof typeof qwenEmbeddingModels = "text-embedding-v3"
+
 export const qwenModels = {
 	"qwen3-235b-a22b": {
 		maxTokens: 129_024,
@@ -916,6 +934,9 @@ export const qwenEmbeddingModels = {
 // https://docs.siliconflow.cn/
 export type SiliconFlowModelId = keyof typeof siliconFlowModels
 export const siliconFlowDefaultModelId: SiliconFlowModelId = "deepseek-ai/DeepSeek-V3"
+export const siliconFlowDefaultAutoCompleteModelId: SiliconFlowModelId = "deepseek-ai/DeepSeek-V3"
+export const siliconFlowDefaultEmbeddingModelId: keyof typeof siliconFlowEmbeddingModels = "BAAI/bge-m3"
+
 export const siliconFlowModels = {
 	"01-ai/Yi-1.5-9B-Chat-16K": {
 		maxTokens: 8192,
@@ -1396,6 +1417,9 @@ export const siliconFlowEmbeddingModels = {
 // https://console.groq.com/docs/overview
 export type GroqModelId = keyof typeof groqModels
 export const groqDefaultModelId: GroqModelId = "llama-3.3-70b-versatile"
+export const groqDefaultAutoCompleteModelId: GroqModelId = "llama-3.3-70b-versatile"
+export const groqDefaultEmbeddingModelId = null // this is not supported embedding model
+
 export const groqModels = {
 	"meta-llama/llama-4-scout-17b-16e-instruct": {
 		maxTokens: 8192,
@@ -1554,6 +1578,9 @@ export const groqModels = {
 // https://docs.x.ai/docs/models
 export type GrokModelId = keyof typeof grokModels
 export const grokDefaultModelId: GrokModelId = "grok-3"
+export const grokDefaultAutoCompleteModelId: GrokModelId = "grok-3-mini-fast"
+export const grokDefaultEmbeddingModelId = null // this is not supported embedding model
+
 export const grokModels = {
 	"grok-3": {
 		maxTokens: 8192,
@@ -1606,18 +1633,31 @@ export const grokModels = {
 } as const satisfies Record<string, ModelInfo>
 
 /// helper functions
-// get all providers
+// get all providers, used for the provider dropdown
 export const GetAllProviders = (): ApiProvider[] => {
-	return Object.values(ApiProvider)
+	return [
+		ApiProvider.Infio,
+		ApiProvider.OpenRouter,
+		ApiProvider.Anthropic,
+		ApiProvider.OpenAI,
+		ApiProvider.Google,
+		ApiProvider.Grok,
+		ApiProvider.AlibabaQwen,
+		ApiProvider.SiliconFlow,
+		ApiProvider.Deepseek,
+		ApiProvider.Groq,
+		ApiProvider.Ollama,
+		ApiProvider.OpenAICompatible,
+	]
 }
 
 export const GetEmbeddingProviders = (): ApiProvider[] => {
 	return [
 		ApiProvider.Infio,
 		ApiProvider.OpenAI,
-		ApiProvider.SiliconFlow,
 		ApiProvider.Google,
 		ApiProvider.AlibabaQwen,
+		ApiProvider.SiliconFlow,
 		ApiProvider.OpenAICompatible,
 		ApiProvider.Ollama,
 	]
@@ -1728,3 +1768,69 @@ export const GetEmbeddingModelInfo = (provider: ApiProvider, modelId: string): E
 	const models = GetEmbeddingProviderModels(provider)
 	return models[modelId]
 }
+
+// Get default model id for a provider
+export const GetDefaultModelId = (provider: ApiProvider): { chat: string, autoComplete: string, embedding: string } => {
+	switch (provider) {
+		case ApiProvider.Infio:
+			return {
+				"chat": infioDefaultModelId,
+				"autoComplete": infioDefaultAutoCompleteModelId,
+				"embedding": infioDefaultEmbeddingModelId,
+			}
+		case ApiProvider.OpenRouter:
+			return {
+				"chat": openRouterDefaultModelId,
+				"autoComplete": openRouterDefaultAutoCompleteModelId,
+				"embedding": openRouterDefaultEmbeddingModelId,
+			}
+		case ApiProvider.Anthropic:
+			return {
+				"chat": anthropicDefaultModelId,
+				"autoComplete": anthropicDefaultAutoCompleteModelId,
+				"embedding": anthropicDefaultEmbeddingModelId,
+			}
+		case ApiProvider.Deepseek:
+			return {
+				"chat": deepSeekDefaultModelId,
+				"autoComplete": deepSeekDefaultAutoCompleteModelId,
+				"embedding": deepSeekDefaultEmbeddingModelId,
+			}
+		case ApiProvider.Google:
+			return {
+				"chat": geminiDefaultModelId,
+				"autoComplete": geminiDefaultAutoCompleteModelId,
+				"embedding": geminiDefaultEmbeddingModelId,
+			}
+		case ApiProvider.AlibabaQwen:
+			return {
+				"chat": qwenDefaultModelId,
+				"autoComplete": qwenDefaultAutoCompleteModelId,
+				"embedding": qwenDefaultEmbeddingModelId,
+			}
+		case ApiProvider.SiliconFlow:
+			return {
+				"chat": siliconFlowDefaultModelId,
+				"autoComplete": siliconFlowDefaultAutoCompleteModelId,
+				"embedding": siliconFlowDefaultEmbeddingModelId,
+			}
+		case ApiProvider.Groq:
+			return {
+				"chat": groqDefaultModelId,
+				"autoComplete": groqDefaultAutoCompleteModelId,
+				"embedding": groqDefaultEmbeddingModelId,
+			}
+		case ApiProvider.Grok:
+			return {
+				"chat": grokDefaultModelId,
+				"autoComplete": grokDefaultAutoCompleteModelId,
+				"embedding": grokDefaultEmbeddingModelId,
+			}
+		default:
+			return {
+				"chat": null,
+				"autoComplete": null,
+				"embedding": null,
+			}
+	}
+}