Add OpenRouter-Auto

llm_client.py

@@ -28,6 +28,19 @@ class LLMClient:
         """Enable or disable test mode for simulating errors"""
         self.test_mode = enabled
 
+    def chat_auto(self, messages: List[Dict[str, str]]) -> str:
+        """Send chat completion request to Auto via OpenRouter"""
+        if not self.openrouter_client:
+            raise ValueError("OpenRouter client not initialized. Please check your API key.")
+
+        try:
+            return self.openrouter_client.create(
+                messages=messages,
+                model=Config.AUTO_MODEL
+            )
+        except Exception as e:
+            raise Exception(f"OpenRouter AUTO API error: {str(e)}")
+
     def chat_openai(self, messages: List[Dict[str, str]]) -> str:
         """Send chat completion request to OpenAI API"""
         if not self.openai_client:
@@ -52,31 +65,31 @@ class LLMClient:
         except openai.APIError as e:
             raise Exception(f"OpenAI API error: {str(e)}")
 
-    def chat_gemini(self, messages: List[Dict[str, str]]) -> str:
-        """Send chat completion request to Gemini via OpenRouter"""
+    def chat_claude(self, messages: List[Dict[str, str]]) -> str:
+        """Send chat completion request to Claude via OpenRouter"""
         if not self.openrouter_client:
             raise ValueError("OpenRouter client not initialized. Please check your API key.")
 
         try:
             return self.openrouter_client.create(
                 messages=messages,
-                model=Config.GEMINI_MODEL
+                model=Config.CLAUDE_MODEL
             )
         except Exception as e:
-            raise Exception(f"Gemini API error: {str(e)}")
+            raise Exception(f"Claude API error: {str(e)}")
 
-    def chat_claude(self, messages: List[Dict[str, str]]) -> str:
-        """Send chat completion request to Claude via OpenRouter"""
+    def chat_gemini(self, messages: List[Dict[str, str]]) -> str:
+        """Send chat completion request to Gemini via OpenRouter"""
         if not self.openrouter_client:
             raise ValueError("OpenRouter client not initialized. Please check your API key.")
 
         try:
             return self.openrouter_client.create(
                 messages=messages,
-                model=Config.CLAUDE_MODEL
+                model=Config.GEMINI_MODEL
             )
         except Exception as e:
-            raise Exception(f"Claude API error: {str(e)}")
+            raise Exception(f"Gemini API error: {str(e)}")
 
     def generate_context_summary(self, messages: List[Dict[str, str]]) -> str:
         """Generate a summary of the conversation context."""

main.py

@@ -205,9 +205,10 @@ def main():
 
             # Model selection logic
             model_map = {
+                "OpenRouter-Auto": llm_client.chat_auto,
                 "GPT-4": llm_client.chat_openai,
-                "Gemini-2.0": llm_client.chat_gemini,
-                "Claude-3.5": llm_client.chat_claude
+                "Claude-3.5": llm_client.chat_claude,
+                "Gemini-2.0": llm_client.chat_gemini
             }
 
             if model in model_map:

openrouter_client.py

@@ -30,7 +30,7 @@ class OpenRouterClient:
                 }
 
                 data = {
-                    "model": model or Config.GEMINI_MODEL,
+                    "model": model or Config.AUTO_MODEL,
                     "messages": messages,
                     "temperature": 0.7,
                     "max_tokens": 1000

ui_components.py

@@ -49,11 +49,11 @@ def render_sidebar(i18n, chat_manager):
             key="language_selector"
         )
 
-        # Model selection - デフォルトをGemini-2.0に設定
+        # Model selection - デフォルトをOpenRouter-Autoに設定
         model = st.selectbox(
             i18n.get_text("model_selection"),
-            ["GPT-4", "Gemini-2.0", "Claude-3.5"],
-            index=1,
+            ["OpenRouter-Auto", "GPT-4", "Claude-3.5", "Gemini-2.0"],
+            index=0,
             key="model_selection"
         )