Create app/model_router.py

app/model_router.py

@@ -0,0 +1,24 @@
+import random
+import httpx
+from .config import OPENROUTER_API_KEY, FREE_MODELS
+
+async def query_model(prompt: str, model: str = None):
+    if not model:
+        model = random.choice(FREE_MODELS)
+
+    headers = {
+        "Authorization": f"Bearer {OPENROUTER_API_KEY}",
+        "Content-Type": "application/json"
+    }
+
+    payload = {
+        "model": model,
+        "messages": [
+            {"role": "user", "content": prompt}
+        ]
+    }
+
+    async with httpx.AsyncClient() as client:
+        response = await client.post("https://openrouter.ai/api/v1/chat/completions", json=payload, headers=headers)
+        response.raise_for_status()
+        return response.json()["choices"][0]["message"]["content"]