feat: add more models in prompts (#718)

lunary-ai · Jan 20, 2025 · aa6a8a0 · aa6a8a0
1 parent 26d37e7
commit aa6a8a0
Showing 1 changed file with 26 additions and 1 deletion.
diff --git a/packages/shared/models.ts b/packages/shared/models.ts
@@ -88,7 +88,22 @@ export const MODELS = [
   { id: "gpt-3.5-turbo-16k", name: "gpt-3.5-turbo-16k", provider: "openai" },
   {
     id: "google/gemini-pro",
-    name: "gemini-pro",
+    name: "gemini-pro-1.0",
+    provider: "openrouter",
+  },
+  {
+    id: "google/gemini-pro-1.5",
+    name: "gemini-pro-1.5",
+    provider: "openrouter",
+  },
+  {
+    id: "google/gemini-flash-1.5",
+    name: "gemini-flash-1.5",
+    provider: "openrouter",
+  },
+  {
+    id: "google/gemini-flash-1.5-8b",
+    name: "gemini-flash-1.5-8b",
     provider: "openrouter",
   },
   { id: "claude-2", name: "claude-2", provider: "anthropic" },
@@ -173,4 +188,14 @@ export const MODELS = [
     name: "llama-3.3-70b-instruct",
     provider: "openrouter",
   },
+  {
+    id: "x-ai/grok-2",
+    name: "grok-2",
+    provider: "openrouter",
+  },
+  {
+    id: "x-ai/grok-2-1212",
+    name: "grok-2-1212",
+    provider: "openrouter",
+  },
 ];