Add AI Chat

Author: mythz

MyApp.ServiceInterface/Data/CustomUserSession.cs

@@ -21,6 +21,7 @@ public override void PopulateFromClaims(IRequest httpReq, ClaimsPrincipal princi
 public class AdditionalUserClaimsPrincipalFactory(
     UserManager<ApplicationUser> userManager,
     RoleManager<IdentityRole> roleManager,
+    IApiKeySource apiKeySource,
     IOptions<IdentityOptions> optionsAccessor)
     : UserClaimsPrincipalFactory<ApplicationUser,IdentityRole>(userManager, roleManager, optionsAccessor)
 {
@@ -35,6 +36,15 @@ public override async Task<ClaimsPrincipal> CreateAsync(ApplicationUser user)
         {
             claims.Add(new Claim(JwtClaimTypes.Picture, user.ProfileUrl));
         }
+        
+        // Add Users latest API Key to Auth Cookie (Allows [ValidateApiKey] with User Auth)
+        var latestApiKey = (await apiKeySource.GetApiKeysByUserIdAsync(user.Id))
+            .OrderByDescending(x => x.CreatedDate)
+            .FirstOrDefault();
+        if (latestApiKey != null)
+        {
+            claims.Add(new Claim(JwtClaimTypes.ApiKey, latestApiKey.Key));
+        }
 
         identity.AddClaims(claims);
         return principal;

MyApp/Configure.AI.Chat.cs

@@ -0,0 +1,41 @@
+using ServiceStack.AI;
+
+[assembly: HostingStartup(typeof(MyApp.ConfigureAiChat))]
+
+namespace MyApp;
+
+public class ConfigureAiChat : IHostingStartup
+{
+    public void Configure(IWebHostBuilder builder) => builder
+        .ConfigureServices((context, services) => {
+
+            // Docs: https://docs.servicestack.net/ai-chat-api
+            services.AddPlugin(new ChatFeature {
+                EnableProviders = [
+                    "servicestack",
+                    // "groq",
+                    // "google_free",
+                    // "openrouter_free",
+                    // "ollama",
+                    // "google",
+                    // "anthropic",
+                    // "openai",
+                    // "grok",
+                    // "qwen",
+                    // "z.ai",
+                    // "mistral",
+                    // "openrouter",
+                ]
+            });
+
+            // Persist AI Chat History, enables analytics at /admin-ui/chat
+            services.AddSingleton<IChatStore, DbChatStore>();
+            // Or store history in monthly partitioned tables in PostgreSQL:
+            // services.AddSingleton<IChatStore, PostgresChatStore>();
+            
+            // Add AI Chat link to /metadata
+            services.ConfigurePlugin<MetadataFeature>(feature => {
+                feature.AddPluginLink("/chat", "AI Chat");
+            });
+        });
+}

MyApp/Configure.ApiKeys.cs

@@ -0,0 +1,60 @@
+using MyApp.Data;
+using ServiceStack;
+using ServiceStack.Data;
+using ServiceStack.OrmLite;
+using ServiceStack.Configuration;
+
+[assembly: HostingStartup(typeof(MyApp.ConfigureApiKeys))]
+
+namespace MyApp;
+
+public class ConfigureApiKeys : IHostingStartup
+{
+    public void Configure(IWebHostBuilder builder) => builder
+        .ConfigureServices(services =>
+        {
+            services.AddPlugin(new ApiKeysFeature
+            {
+                // Optional: Available Scopes Admin Users can assign to any API Key
+                // Features = [
+                //     "Paid",
+                //     "Tracking",
+                // ],
+                // Optional: Available Features Admin Users can assign to any API Key
+                // Scopes = [
+                //     "todo:read",
+                //     "todo:write",
+                // ],
+                
+                // Optional: Limit available Scopes Users can assign to their own API Keys
+                // UserScopes = [
+                //     "todo:read",
+                // ],
+                // Optional: Limit available Features Users can assign to their own API Keys
+                // UserFeatures = [
+                //     "Tracking",
+                // ],
+            });
+        })
+        .ConfigureAppHost(appHost =>
+        {
+            using var db = appHost.Resolve<IDbConnectionFactory>().Open();
+            var feature = appHost.GetPlugin<ApiKeysFeature>();
+            feature.InitSchema(db);
+            
+            // Optional: Create API Key for specified Users on Startup
+            if (feature.ApiKeyCount(db) == 0 && db.TableExists(IdentityUsers.TableName))
+            {
+                var createApiKeysFor = new [] { "[email protected]", "[email protected]", "[email protected]" };
+                var users = IdentityUsers.GetByUserNames(db, createApiKeysFor);
+                foreach (var user in users)
+                {
+                    List<string> scopes = user.UserName == "[email protected]"
+                        ? [RoleNames.Admin] 
+                        : [];
+                    feature.Insert(db, 
+                        new() { Name = "Seed API Key", UserId = user.Id, UserName = user.UserName, Scopes = scopes });
+                }
+            }
+        });
+}