ok

app/api/v1/chat.py

@@ -5,9 +5,9 @@ from typing import Any, Dict, List, Optional
 from fastapi import APIRouter, Depends, HTTPException, Request
 
 from app.api.deps import require_auth
-from app.services.chat_service import MODEL_MAP, chat_completion
+from app.services.chat_service import chat_completion
 
-VALID_MODELS = list(MODEL_MAP.keys())
+VALID_MODELS = ["agentdeck-1.0", "agentdeck-flash", "agentdeck-0.1"]
 VALID_PROVIDERS = ["openprovider", "meganova", "aionlabs"]
 
 router = APIRouter()

app/services/chat_service.py

@@ -31,12 +31,6 @@ PREFIX = "ai_lb"
 AION_PREFIX = "ai_lb:aion"
 DEFAULT_JSON_PROMPT = "Return your response as a valid JSON object inside a JSON code block (```json)."
 
-MODEL_MAP: Dict[str, str] = {
-    "agentdeck-flash": "aionlabs",
-    "agentdeck-1.0": "meganova",
-    "agentdeck-0.1": "openprovider",
-}
-
 KEY_IDS: List[str] = []
 KEY_MAP: Dict[str, str] = {}
 
@@ -439,7 +433,7 @@ async def chat_completion(
     messages = inject_system_identity(messages, model)
 
     if not provider:
-        provider = MODEL_MAP.get(model)
+        provider = "meganova"
 
     logger.info(
         "Chat completion: model=%s provider=%s messages=%d",
@@ -448,13 +442,7 @@ async def chat_completion(
         len(messages),
     )
 
-    if provider == "openprovider":
-        result = await call_openrouter_mimika(
-            messages, response_format, max_tokens, temperature, top_p,
-        )
-        if result:
-            return result
-        raise RuntimeError("OpenRouter Mimika request failed")
+    target = model if model in MODELS else None
 
     if provider == "aionlabs":
         if redis and scripts:
@@ -462,56 +450,55 @@ async def chat_completion(
                 redis, messages, response_format, model or AION_LABS_DEFAULT_MODEL,
                 max_tokens, temperature, top_p,
             )
-            if result:
-                return result
-            raise RuntimeError("AionLabs request failed")
-        # degraded fallback without Redis
-        result = await call_aion_labs_no_redis(
-            messages, response_format, model or AION_LABS_DEFAULT_MODEL,
-            max_tokens, temperature, top_p,
-        )
+        else:
+            result = await call_aion_labs_no_redis(
+                messages, response_format, model or AION_LABS_DEFAULT_MODEL,
+                max_tokens, temperature, top_p,
+            )
         if result:
             return result
         raise RuntimeError("AionLabs request failed")
 
-    if provider != "meganova":
-        mimika_result = await call_openrouter_mimika(
+    if provider == "openprovider":
+        result = await call_openrouter_mimika(
             messages, response_format, max_tokens, temperature, top_p,
         )
-        if mimika_result is not None:
-            logger.info("OpenRouter Mimika handled request, skipping meganova")
-            return mimika_result
+        if result:
+            return result
+        raise RuntimeError("OpenRouter Mimika request failed")
 
-    target = model if model in MODELS else None
     if redis and scripts:
-        meganova_result = await call_meganova(
+        result = await call_meganova(
             redis, scripts, messages, response_format,
             max_tokens, temperature, top_p, target,
         )
-        if meganova_result is not None:
-            return meganova_result
     else:
-        meganova_result = await call_meganova_no_redis(
+        result = await call_meganova_no_redis(
             messages, response_format, max_tokens, temperature, top_p, target,
         )
-        if meganova_result is not None:
-            return meganova_result
+    if result:
+        return result
 
     logger.info("MegaNova failed, falling back to AionLabs")
     if redis and scripts:
-        aion_result = await call_aion_labs(
+        result = await call_aion_labs(
             redis, messages, response_format, model or AION_LABS_DEFAULT_MODEL,
             max_tokens, temperature, top_p,
         )
-        if aion_result is not None:
-            return aion_result
     else:
-        aion_result = await call_aion_labs_no_redis(
+        result = await call_aion_labs_no_redis(
             messages, response_format, model or AION_LABS_DEFAULT_MODEL,
             max_tokens, temperature, top_p,
         )
-        if aion_result is not None:
-            return aion_result
+    if result:
+        return result
+
+    logger.info("AionLabs failed, falling back to OpenRouter Mimika")
+    result = await call_openrouter_mimika(
+        messages, response_format, max_tokens, temperature, top_p,
+    )
+    if result:
+        return result
 
     raise RuntimeError("All AI providers exhausted")