something like that. man anthropic is old

1 files changed, 54 insertions, 21 deletions

diff --git a/index.ts b/index.ts
index 83545f0..3bc3a34 100644
--- a/index.ts
+++ b/index.ts
@@ -2,35 +2,68 @@
 import Claude from "./src/claude";
 import Gemini from "./src/gemini";
 import Generic from "./src/generic";
-import type { AIModelAPI, AIModelChoice } from "./src/types";
+import type { AIModelAPI, LLMChoice } from "./src/types";
 
 export type * from "./src/types";
 export * as NLP from "./src/nlp";
 
-export default function (choice: AIModelChoice): AIModelAPI {
+export default function (choice: LLMChoice): AIModelAPI {
   const api =
-    "other" in choice
-      ? new Generic(choice.other)
-      : choice.name === "deepseek"
-        ? new Generic({
-            baseURL: "https://api.deepseek.com",
-            apiKey: Bun.env.DEEPSEEK_API_KEY!,
-            model: "deepseek-chat",
-          })
-        : choice.name === "grok"
+    "gemini" in choice
+      ? new Gemini(choice.gemini)
+      : "claude" in choice
+        ? new Claude(choice.claude)
+        : "chatgpt" in choice
           ? new Generic({
-              baseURL: "https://api.x.ai/v1",
-              apiKey: Bun.env.GROK_API_KEY!,
-              model: "grok-2-latest",
+              baseURL: "https://api.openai.com/v1",
+              apiKey: Bun.env.OPENAI_API_KEY!,
+              model: choice.chatgpt || "o4-mini",
             })
-          : choice.name === "chatgpt"
+          : "deepseek" in choice
             ? new Generic({
-                baseURL: "https://api.openai.com/v1",
-                apiKey: Bun.env.OPENAI_API_KEY!,
-                model: "gpt-4o",
+                baseURL: "https://api.deepseek.com",
+                apiKey: Bun.env.DEEPSEEK_API_KEY!,
+                model: "deepseek-reasoner",
               })
-            : choice.name === "claude"
-              ? new Claude()
-              : new Gemini();
+            : "kimi" in choice
+              ? new Generic({
+                  baseURL: "https://api.moonshot.ai/v1",
+                  apiKey: Bun.env.MOONSHOT_API_KEY!,
+                  model: "kimi-k2-0711-preview", // "kimi-latest"?
+                })
+              : "grok" in choice
+                ? new Generic({
+                    baseURL: "https://api.x.ai/v1",
+                    apiKey: Bun.env.XAI_API_KEY!,
+                    model: "grok-4", // "kimi-latest"?
+                  })
+                : new Generic({
+                    baseURL: choice.openai.url,
+                    apiKey: choice.openai.apiKey,
+                    model: choice.openai.model,
+                  });
+  // "" in choice
+  //   ? new Generic(choice.other)
+  //   : choice.name === "deepseek"
+  //     ? new Generic({
+  //         baseURL: "https://api.deepseek.com",
+  //         apiKey: Bun.env.DEEPSEEK_API_KEY!,
+  //         model: "deepseek-chat",
+  //       })
+  //     : choice.name === "grok"
+  //       ? new Generic({
+  //           baseURL: "https://api.x.ai/v1",
+  //           apiKey: Bun.env.GROK_API_KEY!,
+  //           model: "grok-2-latest",
+  //         })
+  //       : choice.name === "chatgpt"
+  //         ? new Generic({
+  //             baseURL: "https://api.openai.com/v1",
+  //             apiKey: Bun.env.OPENAI_API_KEY!,
+  //             model: "gpt-4o",
+  //           })
+  //         : choice.name === "claude"
+  //           ? new Claude()
+  //           : new Gemini();
   return api;
 }