|
11 | 11 | namespace TelegramSearchBot.LLMAgent.Service { |
12 | 12 | public sealed class LlmServiceProxy : IAgentTaskExecutor { |
13 | 13 | private readonly IServiceProvider _serviceProvider; |
| 14 | + private readonly LlmProviderRegistry _registry; |
14 | 15 | private readonly ILogger<LlmServiceProxy> _logger; |
15 | 16 |
|
16 | 17 | public LlmServiceProxy(IServiceProvider serviceProvider, ILogger<LlmServiceProxy> logger) { |
17 | 18 | _serviceProvider = serviceProvider; |
| 19 | + _registry = serviceProvider.GetRequiredService<LlmProviderRegistry>(); |
18 | 20 | _logger = logger; |
19 | 21 | } |
20 | 22 |
|
@@ -77,25 +79,21 @@ public async IAsyncEnumerable<string> CallAsync( |
77 | 79 | } |
78 | 80 |
|
79 | 81 | private ILlmProvider ResolveService(LLMProvider provider) { |
80 | | - return provider switch { |
81 | | - LLMProvider.Ollama => _serviceProvider.GetRequiredService<OllamaService>(), |
82 | | - LLMProvider.Gemini => _serviceProvider.GetRequiredService<GeminiService>(), |
83 | | - LLMProvider.Anthropic => _serviceProvider.GetRequiredService<AnthropicService>(), |
84 | | - LLMProvider.ResponsesAPI => _serviceProvider.GetRequiredService<OpenAIResponsesService>(), |
85 | | - _ => _serviceProvider.GetRequiredService<OpenAIService>() |
86 | | - }; |
| 82 | + try { |
| 83 | + return _registry.GetProvider(provider); |
| 84 | + } catch (KeyNotFoundException) { |
| 85 | + // 保持原有降级语义:未知 provider 回退到 OpenAI 兼容路径 |
| 86 | + return _registry.GetProvider(LLMProvider.OpenAI); |
| 87 | + } |
87 | 88 | } |
88 | 89 |
|
89 | | - /// <summary>按 binding 线协议解析 client(与 LlmProviderRegistry.GetProvider(LlmProtocol) 同构)。</summary> |
| 90 | + /// <summary>按 binding 线协议解析 client,统一委托给 LlmProviderRegistry;未知协议回退 OpenAI。</summary> |
90 | 91 | private ILlmProvider ResolveService(LlmProtocol protocol) { |
91 | | - return protocol switch { |
92 | | - LlmProtocol.OpenAIChat => _serviceProvider.GetRequiredService<OpenAIService>(), |
93 | | - LlmProtocol.OpenAIResponses => _serviceProvider.GetRequiredService<OpenAIResponsesService>(), |
94 | | - LlmProtocol.AnthropicMessages => _serviceProvider.GetRequiredService<AnthropicService>(), |
95 | | - LlmProtocol.Ollama => _serviceProvider.GetRequiredService<OllamaService>(), |
96 | | - LlmProtocol.Gemini => _serviceProvider.GetRequiredService<GeminiService>(), |
97 | | - _ => _serviceProvider.GetRequiredService<OpenAIService>() |
98 | | - }; |
| 92 | + try { |
| 93 | + return _registry.GetProvider(protocol); |
| 94 | + } catch (KeyNotFoundException) { |
| 95 | + return _registry.GetProvider(LlmProtocol.OpenAIChat); |
| 96 | + } |
99 | 97 | } |
100 | 98 |
|
101 | 99 | private void ApplyBotIdentity(string botName, long botUserId) { |
|
0 commit comments