From 3dd909855b0d67cbfacb20633dae8e8e76d6e89f Mon Sep 17 00:00:00 2001
From: Renato Rosa <renatoxsr@gmail.com>
Date: Wed, 19 Aug 2026 19:57:40 -0300
Subject: [PATCH] Adicionar app/services/model_router.py

---
 app/services/model_router.py | 39 ++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 app/services/model_router.py

diff --git a/app/services/model_router.py b/app/services/model_router.py
new file mode 100644
index 0000000..5ab06d6
--- /dev/null
+++ b/app/services/model_router.py
@@ -0,0 +1,39 @@
+import json
+from services.ollama_service import OllamaService
+from services.external_model_service import ExternalModelService
+
+class ModelRouter:
+    def __init__(self):
+        self.ollama = OllamaService()
+        self.external = ExternalModelService()
+
+        with open("models.json", "r") as f:
+            self.models = json.load(f)
+
+    def pick_model(self, intent: str, mode: str):
+        # regras simples (pode ser expandido)
+        if intent == "document" and mode == "advogado":
+            return self.models["local"]["llama3.1-70b"]
+
+        if intent == "document" and mode == "corporativo":
+            return self.models["local"]["qwen2.5-7b"]
+
+        if intent == "search":
+            return self.models["external"]["claude-3.5-sonnet"]
+
+        if intent == "monitoring":
+            return self.models["external"]["mistral-large"]
+
+        # fallback
+        return self.models["local"]["llama3.1-8b"]
+
+    def generate(self, intent: str, mode: str, prompt: str):
+        model = self.pick_model(intent, mode)
+
+        if model["provider"] == "ollama":
+            return self.ollama.generate(prompt)
+
+        if model["provider"] in ["openai", "anthropic", "mistral"]:
+            return self.external.generate(model["provider"], model["model"], prompt)
+
+        return "Erro: modelo não encontrado."