Fix to also prepend openrouter/ when muxing

Moved inside a function and called parent

Author: aponcedeleonch

src/codegate/providers/openrouter/provider.py

@@ -2,6 +2,7 @@
 
 from fastapi import Header, HTTPException, Request
 
+from codegate.clients.clients import ClientType
 from codegate.clients.detector import DetectClient
 from codegate.pipeline.factory import PipelineFactory
 from codegate.providers.openai import OpenAIProvider
@@ -15,6 +16,21 @@ def __init__(self, pipeline_factory: PipelineFactory):
     def provider_route_name(self) -> str:
         return "openrouter"
 
+    async def process_request(
+        self,
+        data: dict,
+        api_key: str,
+        request_url_path: str,
+        client_type: ClientType,
+    ):
+        # litellm workaround - add openrouter/ prefix to model name to make it openai-compatible
+        # once we get rid of litellm, this can simply be removed
+        original_model = data.get("model", "")
+        if not original_model.startswith("openrouter/"):
+            data["model"] = f"openrouter/{original_model}"
+
+        return await super().process_request(data, api_key, is_fim_request, client_type)
+
     def _setup_routes(self):
         @self.router.post(f"/{self.provider_route_name}/api/v1/chat/completions")
         @self.router.post(f"/{self.provider_route_name}/chat/completions")
@@ -33,12 +49,6 @@ async def create_completion(
             base_url = self._get_base_url()
             data["base_url"] = base_url
 
-            # litellm workaround - add openrouter/ prefix to model name to make it openai-compatible
-            # once we get rid of litellm, this can simply be removed
-            original_model = data.get("model", "")
-            if not original_model.startswith("openrouter/"):
-                data["model"] = f"openrouter/{original_model}"
-
             return await self.process_request(
                 data,
                 api_key,