Upload classifier.py

backend/ml/classifier.py

@@ -0,0 +1,42 @@
+import logging
+from backend.config import settings
+
+logger = logging.getLogger(__name__)
+
+class RouterClassifier:
+    def __init__(self):
+        self.tokenizer = None
+        self.model = None
+        logger.info("Initializing Classifier (Heuristic Mode)")
+
+    def is_loaded(self):
+        return True
+
+    async def classify(self, prompt: str):
+        # Heuristic Logic
+        prompt_lower = prompt.lower()
+        
+        # Complex Keywords (imply Large LLM)
+        complex_keywords = [
+            "code", "python", "javascript", "react", "sql", "algorithm",
+            "explain", "analyze", "compare", "difference", "history",
+            "physics", "quantum", "essay", "poem", "generate", "write"
+        ]
+        
+        has_keyword = any(kw in prompt_lower for kw in complex_keywords)
+        is_long = len(prompt) > 80
+        
+        if has_keyword or is_long:
+            return {
+                "selected_tier": settings.LARGE_LLM_TIER,
+                "confidence": 0.92 if has_keyword else 0.85,
+                "routing_reason": "Complex keywords detected" if has_keyword else "Prompt length > 80"
+            }
+        else:
+            return {
+                "selected_tier": settings.SMALL_LLM_TIER,
+                "confidence": 0.88,
+                "routing_reason": "Simple query detected"
+            }
+
+