Prep for custom models: support reading custom providers (#131)

src/ipc/handlers/app_handlers.ts

@@ -22,7 +22,6 @@ import {
   killProcess,
   removeAppIfCurrentProcess,
 } from "../utils/process_manager";
-import { ALLOWED_ENV_VARS } from "../../constants/models";
 import { getEnvVar } from "../utils/read_env";
 import { readSettings } from "../../main/settings";
 
@@ -33,6 +32,7 @@ import util from "util";
 import log from "electron-log";
 import { getSupabaseProjectName } from "../../supabase_admin/supabase_management_client";
 import { createLoggedHandler } from "./safe_handle";
+import { getLanguageModelProviders } from "../shared/language_model_helpers";
 
 const logger = log.scope("app_handlers");
 const handle = createLoggedHandler(logger);
@@ -291,8 +291,11 @@ export function registerAppHandlers() {
   // Do NOT use handle for this, it contains sensitive information.
   ipcMain.handle("get-env-vars", async () => {
     const envVars: Record<string, string | undefined> = {};
-    for (const key of ALLOWED_ENV_VARS) {
-      envVars[key] = getEnvVar(key);
+    const providers = await getLanguageModelProviders();
+    for (const provider of providers) {
+      if (provider.envVarName) {
+        envVars[provider.envVarName] = getEnvVar(provider.envVarName);
+      }
     }
     return envVars;
   });

src/ipc/handlers/chat_stream_handlers.ts

@@ -212,7 +212,10 @@ export function registerChatStreamHandlers() {
       } else {
         // Normal AI processing for non-test prompts
         const settings = readSettings();
-        const modelClient = getModelClient(settings.selectedModel, settings);
+        const modelClient = await getModelClient(
+          settings.selectedModel,
+          settings,
+        );
 
         // Extract codebase information if app is associated with the chat
         let codebaseInfo = "";

src/ipc/handlers/language_model_handlers.ts

@@ -0,0 +1,16 @@
+import type { LanguageModelProvider } from "@/ipc/ipc_types";
+import { createLoggedHandler } from "./safe_handle";
+import log from "electron-log";
+import { getLanguageModelProviders } from "../shared/language_model_helpers";
+
+const logger = log.scope("language_model_handlers");
+const handle = createLoggedHandler(logger);
+
+export function registerLanguageModelHandlers() {
+  handle(
+    "get-language-model-providers",
+    async (): Promise<LanguageModelProvider[]> => {
+      return getLanguageModelProviders();
+    },
+  );
+}