Use LLMApiRegistry for API keys in IferPage Program.cs

Remove the old BackendType extension and centralize API key metadata in LLMApiRegistry (moved to MaIN.Domain.Models.Concrete). Program.cs now looks up the registry entry for each BackendType to read ApiKeyEnvName instead of calling GetApiKeyVariable. Updated numerous LLM and image service files (and McpService) to reference the new namespace. This change consolidates API key configuration and removes the duplicated extension method.

Author: Madzionator

src/MaIN.Core.UnitTests/ChatContextTests.cs

@@ -88,7 +88,7 @@ public async Task CompleteAsync_ShouldCallChatService()
         };
 
 
-        _mockChatService.Setup(s => s.Completions(It.IsAny<Chat>(), It.IsAny<bool>(), It.IsAny<bool>(), null))
+        _mockChatService.Setup(s => s.Completions(It.IsAny<Chat>(), It.IsAny<bool>(), It.IsAny<bool>(), null, It.IsAny<CancellationToken>()))
             .ReturnsAsync(chatResult);
 
         _chatContext.WithMessage("User message");
@@ -98,7 +98,7 @@ public async Task CompleteAsync_ShouldCallChatService()
         var result = await _chatContext.CompleteAsync();
 
         // Assert
-        _mockChatService.Verify(s => s.Completions(It.IsAny<Chat>(), false, false, null), Times.Once);
+        _mockChatService.Verify(s => s.Completions(It.IsAny<Chat>(), false, false, null, It.IsAny<CancellationToken>()), Times.Once);
         Assert.Equal(chatResult, result);
     }
 
@@ -128,6 +128,6 @@ await _chatContext.WithModel(model)
             .CompleteAsync();
 
         // Assert
-        _mockChatService.Verify(s => s.Completions(It.Is<Chat>(c => c.ModelId == _testModelId && c.ModelInstance == model), It.IsAny<bool>(), It.IsAny<bool>(), null), Times.Once);
+        _mockChatService.Verify(s => s.Completions(It.Is<Chat>(c => c.ModelId == _testModelId && c.ModelInstance == model), It.IsAny<bool>(), It.IsAny<bool>(), null, It.IsAny<CancellationToken>()), Times.Once);
     }
 }

src/MaIN.Domain/Extensions.cs

@@ -1,4 +1,6 @@
-namespace MaIN.Services.Services.LLMService.Utils;
+using MaIN.Domain.Configuration;
+
+namespace MaIN.Domain.Models.Concrete;
 
 public static class LLMApiRegistry
 {
@@ -9,6 +11,18 @@ public static class LLMApiRegistry
     public static readonly LLMApiRegistryEntry Anthropic = new("Anthropic", "ANTHROPIC_API_KEY");
     public static readonly LLMApiRegistryEntry Xai = new("Xai", "XAI_API_KEY");
     public static readonly LLMApiRegistryEntry Ollama = new("Ollama", "OLLAMA_API_KEY");
+
+    public static LLMApiRegistryEntry? GetEntry(BackendType backendType) => backendType switch
+    {
+        BackendType.OpenAi => OpenAi,
+        BackendType.Gemini => Gemini,
+        BackendType.DeepSeek => Deepseek,
+        BackendType.GroqCloud => Groq,
+        BackendType.Anthropic => Anthropic,
+        BackendType.Xai => Xai,
+        BackendType.Ollama => Ollama,
+        _ => null
+    };
 }
 
 public record LLMApiRegistryEntry(string ApiName, string ApiKeyEnvName);

…vices/LLMService/Utils/LLMApiRegistry.cs …Domain/Models/Concrete/LLMApiRegistry.cssrc/MaIN.Services/Services/LLMService/Utils/LLMApiRegistry.cs renamed to src/MaIN.Domain/Models/Concrete/LLMApiRegistry.cs src/MaIN.Services/Services/LLMService/Utils/LLMApiRegistry.cs renamed to src/MaIN.Domain/Models/Concrete/LLMApiRegistry.cs

@@ -1,6 +1,6 @@
 using MaIN.Core;
-using MaIN.Domain;
 using MaIN.Domain.Configuration;
+using MaIN.Domain.Models.Concrete;
 using MaIN.Domain.Models.Abstract;
 using Microsoft.FluentUI.AspNetCore.Components;
 using MaIN.InferPage.Components;
@@ -34,7 +34,7 @@
 
         if (Utils.BackendType != BackendType.Self)
         {
-            var apiKeyVariable = Utils.BackendType.GetApiKeyVariable();
+            var apiKeyVariable = LLMApiRegistry.GetEntry(Utils.BackendType)?.ApiKeyEnvName ?? string.Empty;
             var key = Environment.GetEnvironmentVariable(apiKeyVariable);
 
             if (string.IsNullOrEmpty(key) && !string.IsNullOrEmpty(apiKeyVariable))

src/MaIN.InferPage/Program.cs

@@ -7,6 +7,7 @@
 using System.Net.Http.Json;
 using System.Text.Json.Serialization;
 using MaIN.Domain.Exceptions;
+using MaIN.Domain.Models.Concrete;
 using MaIN.Services.Services.LLMService.Utils;
 
 namespace MaIN.Services.Services.ImageGenServices;

src/MaIN.Services/Services/ImageGenServices/GeminiImageGenService.cs

@@ -3,6 +3,7 @@
 using MaIN.Domain.Configuration;
 using MaIN.Domain.Entities;
 using MaIN.Domain.Exceptions;
+using MaIN.Domain.Models.Concrete;
 using MaIN.Services.Constants;
 using MaIN.Services.Services.Abstract;
 using MaIN.Services.Services.LLMService.Utils;

src/MaIN.Services/Services/ImageGenServices/ImageGenDalleService.cs

@@ -7,6 +7,7 @@
 using System.Net.Http.Json;
 using System.Text.Json;
 using MaIN.Domain.Exceptions;
+using MaIN.Domain.Models.Concrete;
 using MaIN.Services.Services.LLMService.Utils;
 
 namespace MaIN.Services.Services.ImageGenServices;

src/MaIN.Services/Services/ImageGenServices/XaiImageGenService.cs

@@ -12,6 +12,7 @@
 using MaIN.Domain.Configuration;
 using MaIN.Domain.Entities.Tools;
 using MaIN.Domain.Exceptions;
+using MaIN.Domain.Models.Concrete;
 using MaIN.Services.Services.LLMService.Utils;
 
 namespace MaIN.Services.Services.LLMService;

src/MaIN.Services/Services/LLMService/AnthropicService.cs

@@ -10,6 +10,7 @@
 using System.Text.Json;
 using System.Text.Json.Serialization;
 using MaIN.Domain.Exceptions;
+using MaIN.Domain.Models.Concrete;
 using MaIN.Services.Services.LLMService.Utils;
 
 namespace MaIN.Services.Services.LLMService;

src/MaIN.Services/Services/LLMService/DeepSeekService.cs

@@ -11,6 +11,7 @@
 using MaIN.Domain.Entities;
 using MaIN.Domain.Exceptions;
 using MaIN.Domain.Models;
+using MaIN.Domain.Models.Concrete;
 using MaIN.Services.Services.LLMService.Utils;
 using MaIN.Services.Utils;