From 7a007daf581aba08b7b1e3768aa4263b9f5cabf3 Mon Sep 17 00:00:00 2001
From: Logan Yang <logancyang@gmail.com>
Date: Tue, 24 Jun 2025 15:35:58 -0700
Subject: [PATCH] feat: Update ChatModels and add new OpenRouter models (#1571)

- Updated GEMINI_PRO and GEMINI_FLASH constants to their stable versions.
- Added new OpenRouter models: OPENROUTER_GEMINI_2_5_FLASH, OPENROUTER_GEMINI_2_5_PRO, and OPENROUTER_GEMINI_2_5_FLASH_LITE.
- Enhanced ChatModelManager with default headers for API requests.
---
 src/LLMProviders/chatModelManager.ts |  4 ++++
 src/constants.ts                     | 34 ++++++++++++++++++++++++++--
 2 files changed, 36 insertions(+), 2 deletions(-)

diff --git a/src/LLMProviders/chatModelManager.ts b/src/LLMProviders/chatModelManager.ts
index f3ce5483..6c2627a6 100644
--- a/src/LLMProviders/chatModelManager.ts
+++ b/src/LLMProviders/chatModelManager.ts
@@ -196,6 +196,10 @@ export default class ChatModelManager {
         configuration: {
           baseURL: customModel.baseUrl || "https://openrouter.ai/api/v1",
           fetch: customModel.enableCors ? safeFetch : undefined,
+          defaultHeaders: {
+            "HTTP-Referer": "https://obsidiancopilot.com",
+            "X-Title": "Obsidian Copilot",
+          },
         },
       },
       [ChatModelProviders.GROQ]: {
diff --git a/src/constants.ts b/src/constants.ts
index 354f29ce..3a7294fb 100644
--- a/src/constants.ts
+++ b/src/constants.ts
@@ -115,8 +115,8 @@ export enum ChatModels {
   GPT_41_nano = "gpt-4.1-nano",
   O4_mini = "o4-mini",
   AZURE_OPENAI = "azure-openai",
-  GEMINI_PRO = "gemini-2.5-pro-preview-06-05", // TODO(logan): update this when it's GA
-  GEMINI_FLASH = "gemini-2.5-flash-preview-04-17", // TODO(logan): update this when it's GA
+  GEMINI_PRO = "gemini-2.5-pro",
+  GEMINI_FLASH = "gemini-2.5-flash",
   CLAUDE_3_5_SONNET = "claude-3-5-sonnet-latest",
   CLAUDE_3_7_SONNET = "claude-3-7-sonnet-latest",
   CLAUDE_4_SONNET = "claude-sonnet-4-20250514",
@@ -130,6 +130,9 @@ export enum ChatModels {
   MISTRAL_TINY = "mistral-tiny-latest",
   DEEPSEEK_REASONER = "deepseek-reasoner",
   DEEPSEEK_CHAT = "deepseek-chat",
+  OPENROUTER_GEMINI_2_5_FLASH = "google/gemini-2.5-flash",
+  OPENROUTER_GEMINI_2_5_PRO = "google/gemini-2.5-pro",
+  OPENROUTER_GEMINI_2_5_FLASH_LITE = "google/gemini-2.5-flash-lite-preview-06-17",
 }
 
 // Model Providers
@@ -173,6 +176,33 @@ export const BUILTIN_CHAT_MODELS: CustomModel[] = [
     projectEnabled: false,
     capabilities: [ModelCapability.VISION],
   },
+  {
+    name: ChatModels.OPENROUTER_GEMINI_2_5_FLASH_LITE,
+    provider: ChatModelProviders.OPENROUTERAI,
+    enabled: true,
+    isBuiltIn: true,
+    core: true,
+    projectEnabled: true,
+    capabilities: [ModelCapability.VISION],
+  },
+  {
+    name: ChatModels.OPENROUTER_GEMINI_2_5_FLASH,
+    provider: ChatModelProviders.OPENROUTERAI,
+    enabled: true,
+    isBuiltIn: true,
+    core: true,
+    projectEnabled: true,
+    capabilities: [ModelCapability.VISION],
+  },
+  {
+    name: ChatModels.OPENROUTER_GEMINI_2_5_PRO,
+    provider: ChatModelProviders.OPENROUTERAI,
+    enabled: true,
+    isBuiltIn: true,
+    core: true,
+    projectEnabled: true,
+    capabilities: [ModelCapability.VISION],
+  },
   {
     name: ChatModels.GPT_41,
     provider: ChatModelProviders.OPENAI,