Initial open-source release

src/ipc/utils/get_model_client.ts

@@ -0,0 +1,65 @@
+import { createOpenAI } from "@ai-sdk/openai";
+import { createGoogleGenerativeAI as createGoogle } from "@ai-sdk/google";
+import { createAnthropic } from "@ai-sdk/anthropic";
+import { createOpenRouter } from "@openrouter/ai-sdk-provider";
+import type { LargeLanguageModel, UserSettings } from "../../lib/schemas";
+import { PROVIDER_TO_ENV_VAR, AUTO_MODELS } from "../../constants/models";
+
+export function getModelClient(
+  model: LargeLanguageModel,
+  settings: UserSettings
+) {
+  // Handle 'auto' provider by trying each model in AUTO_MODELS until one works
+  if (model.provider === "auto") {
+    // Try each model in AUTO_MODELS in order until finding one with an API key
+    for (const autoModel of AUTO_MODELS) {
+      const apiKey =
+        settings.providerSettings?.[autoModel.provider]?.apiKey ||
+        process.env[PROVIDER_TO_ENV_VAR[autoModel.provider]];
+
+      if (apiKey) {
+        console.log(
+          `Using provider: ${autoModel.provider} model: ${autoModel.name}`
+        );
+        // Use the first model that has an API key
+        return getModelClient(
+          {
+            provider: autoModel.provider,
+            name: autoModel.name,
+          } as LargeLanguageModel,
+          settings
+        );
+      }
+    }
+
+    // If no models have API keys, throw an error
+    throw new Error("No API keys available for any model in AUTO_MODELS");
+  }
+
+  const apiKey =
+    settings.providerSettings?.[model.provider]?.apiKey ||
+    process.env[PROVIDER_TO_ENV_VAR[model.provider]];
+  switch (model.provider) {
+    case "openai": {
+      const provider = createOpenAI({ apiKey });
+      return provider(model.name);
+    }
+    case "anthropic": {
+      const provider = createAnthropic({ apiKey });
+      return provider(model.name);
+    }
+    case "google": {
+      const provider = createGoogle({ apiKey });
+      return provider(model.name);
+    }
+    case "openrouter": {
+      const provider = createOpenRouter({ apiKey });
+      return provider(model.name);
+    }
+    default: {
+      // Ensure exhaustive check if more providers are added
+      const _exhaustiveCheck: never = model.provider;
+      throw new Error(`Unsupported model provider: ${model.provider}`);
+    }
+  }
+}