feat: Vertex AI integration

Examples/Examples/Chat/ChatExampleVertex.cs

@@ -0,0 +1,24 @@
+using Examples.Utils;
+using MaIN.Core.Hub;
+using MaIN.Domain.Configuration.BackendInferenceParams;
+using MaIN.Domain.Models;
+
+namespace Examples.Chat;
+
+public class ChatExampleVertex : IExample
+{
+    public async Task Start()
+    {
+        VertexExample.Setup(); //We need to provide Google service account config 
+        Console.WriteLine("(Vertex AI) ChatExample is running!");
+
+        await AIHub.Chat()
+            .WithModel(Models.Vertex.Gemini2_5Pro)
+            .WithMessage("Is the killer whale the smartest animal?")
+            .WithInferenceParams(new VertexInferenceParams
+            {
+                Location = "europe-central2"
+            })
+            .CompleteAsync(interactive: true);
+    }
+}

Examples/Examples/Program.cs

@@ -73,6 +73,7 @@ static void RegisterExamples(IServiceCollection services)
     services.AddTransient<ChatGrammarExampleGemini>();
     services.AddTransient<ChatWithImageGenGeminiExample>();
     services.AddTransient<ChatWithFilesExampleGemini>();
+    services.AddTransient<ChatExampleVertex>();
     services.AddTransient<ChatWithReasoningDeepSeekExample>();
     services.AddTransient<ChatWithTextToSpeechExample>();
     services.AddTransient<ChatExampleGroqCloud>();
@@ -186,6 +187,7 @@ public class ExampleRegistry(IServiceProvider serviceProvider)
                 ("\u25a0 Gemini Chat with grammar", serviceProvider.GetRequiredService<ChatGrammarExampleGemini>()),
                 ("\u25a0 Gemini Chat with image", serviceProvider.GetRequiredService<ChatWithImageGenGeminiExample>()),
                 ("\u25a0 Gemini Chat with files", serviceProvider.GetRequiredService<ChatWithFilesExampleGemini>()),
+                ("\u25a0 Vertex Chat", serviceProvider.GetRequiredService<ChatExampleVertex>()),
                 ("\u25a0 DeepSeek Chat with reasoning", serviceProvider.GetRequiredService<ChatWithReasoningDeepSeekExample>()),
                 ("\u25a0 GroqCloud Chat", serviceProvider.GetRequiredService<ChatExampleGroqCloud>()),
                 ("\u25a0 Anthropic Chat", serviceProvider.GetRequiredService<ChatExampleAnthropic>()),

Examples/Examples/Utils/VertexExample.cs

@@ -0,0 +1,22 @@
+using MaIN.Core;
+using MaIN.Domain.Configuration;
+using MaIN.Domain.Configuration.Vertex;
+
+namespace Examples.Utils;
+
+public class VertexExample
+{
+    public static void Setup()
+    {
+        MaINBootstrapper.Initialize(configureSettings: options =>
+        {
+            options.BackendType = BackendType.Vertex;
+            options.GoogleServiceAccountAuth = new GoogleServiceAccountConfig
+            {
+                ProjectId = "<YOUR_GCP_PROJECT_ID>",
+                ClientEmail = "<YOUR_SERVICE_ACCOUNT_EMAIL>",
+                PrivateKey = @"<YOUR_PRIVATE_KEY>"
+            };
+        });
+    }
+}

Releases/0.10.4.md

@@ -0,0 +1,3 @@
+# 0.10.4 release
+
+Adds Google Vertex AI as a backend with authentication, MCP support, and new models including image generation, along with UI configuration and example usage.

src/MaIN.Core/.nuspec

@@ -2,7 +2,7 @@
 <package>
   <metadata>
     <id>MaIN.NET</id>
-    <version>0.10.3</version>
+    <version>0.10.4</version>
     <authors>Wisedev</authors>
     <owners>Wisedev</owners>
     <icon>favicon.png</icon>

src/MaIN.Domain/Configuration/BackendInferenceParams/BackendParamsFactory.cs

@@ -14,6 +14,7 @@ public static class BackendParamsFactory
         BackendType.Gemini => new GeminiInferenceParams(),
         BackendType.Anthropic => new AnthropicInferenceParams(),
         BackendType.Ollama => new OllamaInferenceParams(),
+        BackendType.Vertex => new VertexInferenceParams(),
         _ => new LocalInferenceParams()
     };
 }

src/MaIN.Domain/Configuration/BackendInferenceParams/VertexInferenceParams.cs

@@ -0,0 +1,18 @@
+using MaIN.Domain.Entities;
+using Grammar = MaIN.Domain.Models.Grammar;
+
+namespace MaIN.Domain.Configuration.BackendInferenceParams;
+
+public class VertexInferenceParams : IBackendInferenceParams
+{
+    public BackendType Backend => BackendType.Vertex;
+
+    public string Location { get; init; } = "us-central1";
+
+    public float? Temperature { get; init; }
+    public int? MaxTokens { get; init; }
+    public float? TopP { get; init; }
+    public string[]? StopSequences { get; init; }
+    public Grammar? Grammar { get; set; }
+    public Dictionary<string, object>? AdditionalParams { get; init; }
+}

src/MaIN.Domain/Configuration/MaINSettings.cs

@@ -1,3 +1,4 @@
+using MaIN.Domain.Configuration.Vertex;
 
 namespace MaIN.Domain.Configuration;
 
@@ -18,6 +19,7 @@ public class MaINSettings
     public SqliteSettings? SqliteSettings { get; set; }
     public SqlSettings? SqlSettings { get; set; }
     public string? VoicesPath { get; set; }
+    public GoogleServiceAccountConfig? GoogleServiceAccountAuth { get; set; }
 }
 
 public enum BackendType
@@ -30,4 +32,5 @@ public enum BackendType
     Anthropic = 5,
     Xai = 6,
     Ollama = 7,
+    Vertex = 8,
 }

src/MaIN.Domain/Configuration/Vertex/GoogleServiceAccountConfig.cs

@@ -0,0 +1,9 @@
+namespace MaIN.Domain.Configuration.Vertex;
+
+public class GoogleServiceAccountConfig
+{
+    public required string ProjectId { get; init; }
+    public required string ClientEmail { get; init; }
+    public required string PrivateKey { get; init; }
+    public string TokenUri { get; init; } = "https://oauth2.googleapis.com/token";
+}

src/MaIN.Domain/Entities/Mcp.cs

@@ -8,6 +8,7 @@ public class Mcp
     public required List<string> Arguments { get; init; }
     public required string Command { get; init; }
     public required string Model { get; init; }
+    public string Location { get; set; } = "us-central1";
     public Dictionary<string, string> Properties { get; set; } = [];
     public BackendType? Backend { get; set; }
     public Dictionary<string, string> EnvironmentVariables { get; set; } = [];

src/MaIN.Domain/Models/Concrete/CloudModels.cs

@@ -93,6 +93,59 @@ public sealed record Gemini2_0Flash() : CloudModel(
     public string? MMProjectName => null;
 }
 
+public sealed record Gemini2_5Pro() : CloudModel(
+    Models.Gemini.Gemini2_5Pro,
+    BackendType.Gemini,
+    "Gemini 2.5 Pro",
+    1000000,
+    "Google's most capable Gemini model"), IVisionModel
+{
+    public string? MMProjectName => null;
+}
+
+public sealed record GeminiImagen4_0FastGenerate() : CloudModel(
+    Models.Gemini.Imagen4_0_FastGenerate,
+    BackendType.Gemini,
+    "Imagen 4.0 Fast (Gemini)",
+    4000,
+    "Google's fast image generation model via Gemini API"), IImageGenerationModel;
+
+// ===== Vertex AI Models =====
+
+public sealed record VertexGemini2_5Pro() : CloudModel(
+    Models.Vertex.Gemini2_5Pro,
+    BackendType.Vertex,
+    "Gemini 2.5 Pro (Vertex)",
+    1000000,
+    "Fast and efficient Gemini model served via Vertex AI"), IVisionModel
+{
+    public string? MMProjectName => null;
+}
+
+public sealed record VertexGemini2_5Flash() : CloudModel(
+    Models.Vertex.Gemini2_5Flash,
+    BackendType.Vertex,
+    "Gemini 2.5 Flash (Vertex)",
+    1000000,
+    "Fast and efficient Gemini model served via Vertex AI"), IVisionModel
+{
+    public string? MMProjectName => null;
+}
+
+public sealed record VertexVeo2_0Generate() : CloudModel(
+    Models.Vertex.Veo2_0_Generate,
+    BackendType.Vertex,
+    "Veo 2.0 Generate",
+    4000,
+    "Google's video generation model available through Vertex AI"), IImageGenerationModel;
+
+public sealed record VertexImagen4_0Generate() : CloudModel(
+    Models.Vertex.Imagen4_0_Generate,
+    BackendType.Vertex,
+    "Imagen 4.0 (Vertex)",
+    4000,
+    "Google's latest image generation model available through Vertex AI"), IImageGenerationModel;
+
 // ===== xAI Models =====
 
 public sealed record Grok3Beta() : CloudModel(

src/MaIN.Domain/Models/Concrete/LLMApiRegistry.cs

@@ -11,6 +11,7 @@ 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 readonly LLMApiRegistryEntry Vertex = new("Vertex", "GOOGLE_APPLICATION_CREDENTIALS");
 
     public static LLMApiRegistryEntry? GetEntry(BackendType backendType) => backendType switch
     {
@@ -21,6 +22,7 @@ public static class LLMApiRegistry
         BackendType.Anthropic => Anthropic,
         BackendType.Xai => Xai,
         BackendType.Ollama => Ollama,
+        BackendType.Vertex => Vertex,
         _ => null
     };
 }

src/MaIN.Domain/Models/Concrete/LocalModels.cs

@@ -293,6 +293,16 @@ public sealed record Olmo2_7b() : LocalModel(
     8192,
     "Open-source 7B model for research, benchmarking, and academic studies");
 
+// ===== Image Generation =====
+
+public sealed record Flux1Shnell() : LocalModel(
+    Models.Local.Flux1Shnell,
+    "FLUX.1_Shnell",
+    null,
+    "FLUX.1 Schnell",
+    4096,
+    "Fast local image generation model"), IImageGenerationModel;
+
 // ===== Embedding Model =====
 
 public sealed record Mxbai_Embedding() : LocalModel(

src/MaIN.Domain/Models/Models.cs

@@ -23,8 +23,10 @@ public static class Anthropic
 
     public static class Gemini
     {
+        public const string Gemini2_5Pro = "gemini-2.5-pro";
         public const string Gemini2_5Flash = "gemini-2.5-flash";
         public const string Gemini2_0Flash = "gemini-2.0-flash";
+        public const string Imagen4_0_FastGenerate = "imagen-4.0-fast-generate-001";
     }
 
     public static class Xai
@@ -49,6 +51,14 @@ public static class Ollama
         public const string Gemma3_4b = "gemma3:4b";
     }
 
+    public static class Vertex
+    {
+        public const string Gemini2_5Pro = "google/gemini-2.5-pro";
+        public const string Gemini2_5Flash = "google/gemini-2.5-flash";
+        public const string Veo2_0_Generate = "google/veo-2.0-generate-001";
+        public const string Imagen4_0_Generate = "google/imagen-4.0-generate-001";
+    }
+
     public static class Local
     {
         // Gemma

src/MaIN.InferPage/Components/Pages/Home.razor

@@ -3,7 +3,7 @@
 @inject IJSRuntime JS
 @inject SettingsService SettingsStorage
 @inject SettingsStateService SettingsState
-@inject MaIN.Domain.Configuration.MaINSettings MaINSettings
+@inject MaINSettings MaINSettings
 @implements IDisposable
 @using MaIN.Core.Hub
 @using MaIN.Core.Hub.Contexts.Interfaces.ChatContext
@@ -427,12 +427,32 @@
         if (backendType != BackendType.Self)
         {
             var backendKey = backendType == BackendType.Ollama
-                ? (settings.IsOllamaCloud ? "OllamaCloud" : "OllamaLocal")
-                : backendType.ToString();
+                ? (settings.IsOllamaCloud 
+                    ? "OllamaCloud" 
+                    : "OllamaLocal")
+                : backendType == BackendType.Vertex 
+                    ? "Vertex"
+                    : backendType.ToString();
 
             apiKey = await SettingsStorage.GetApiKeyForBackendAsync(backendKey);
         }
 
+        // Load Vertex auth from localStorage if applicable
+        Domain.Configuration.Vertex.GoogleServiceAccountConfig? vertexAuth = null;
+        if (backendType == BackendType.Vertex)
+        {
+            var stored = await SettingsStorage.GetVertexAuthAsync();
+            if (stored != null)
+            {
+                vertexAuth = new Domain.Configuration.Vertex.GoogleServiceAccountConfig
+                {
+                    ProjectId = stored.ProjectId,
+                    ClientEmail = stored.ClientEmail,
+                    PrivateKey = stored.PrivateKey
+                };
+            }
+        }
+
         Utils.ApplySettings(
             backendType,
             settings.Model!,
@@ -442,7 +462,8 @@
             settings.HasImageGen,
             settings.MmProjName,
             MaINSettings,
-            apiKey);
+            apiKey,
+            vertexAuth);
     }
 
     private void ShowSettingsFromGear()