Create teacher.py

teacher.py

@@ -0,0 +1,125 @@
+"""Teacher Model - Calls OpenRouter API for knowledge distillation"""
+
+import requests
+import json
+from typing import Optional, Dict, Any, List
+from config import (
+    OPENROUTER_API_KEY,
+    OPENROUTER_BASE_URL,
+    TEACHER_MODEL,
+    TEACHER_TEMPERATURE,
+    TEACHER_MAX_TOKENS,
+)
+
+
+class TeacherModel:
+    """Wrapper for the OpenRouter teacher model (Dolphin Mistral)"""
+
+    def __init__(self):
+        self.api_key = OPENROUTER_API_KEY
+        self.base_url = OPENROUTER_BASE_URL
+        self.model = TEACHER_MODEL
+        self.headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {self.api_key}",
+            "HTTP-Referer": "https://huggingface.co/spaces/vedaco/veda-programming",
+            "X-Title": "Veda Programming Assistant",
+        }
+
+    def ask(
+        self,
+        user_message: str,
+        conversation_history: List[Dict[str, str]] = None,
+        temperature: float = None,
+        max_tokens: int = None,
+    ) -> Optional[str]:
+        """
+        Ask the teacher model a question.
+        
+        Args:
+            user_message: The user's question
+            conversation_history: Previous messages [{"role": "user/assistant", "content": "..."}]
+            temperature: Sampling temperature
+            max_tokens: Maximum tokens to generate
+            
+        Returns:
+            Teacher's response as string, or None if error
+        """
+        if not self.api_key or self.api_key == "":
+            print("Warning: No OpenRouter API key configured")
+            return None
+
+        temperature = temperature or TEACHER_TEMPERATURE
+        max_tokens = max_tokens or TEACHER_MAX_TOKENS
+
+        # Build messages
+        messages = []
+        
+        # System prompt to make teacher respond like a programming assistant
+        messages.append({
+            "role": "system",
+            "content": """You are a helpful programming assistant. You help users with:
+- Writing Python code
+- Explaining programming concepts
+- Debugging code
+- Answering questions about algorithms and data structures
+
+Be clear, concise, and provide code examples when helpful.
+Format code blocks with ```python and ```.
+"""
+        })
+
+        # Add conversation history
+        if conversation_history:
+            for msg in conversation_history[-6:]:  # Last 6 messages for context
+                messages.append({
+                    "role": msg.get("role", "user"),
+                    "content": msg.get("content", "")
+                })
+
+        # Add current user message
+        messages.append({
+            "role": "user",
+            "content": user_message
+        })
+
+        payload = {
+            "model": self.model,
+            "messages": messages,
+            "temperature": temperature,
+            "max_tokens": max_tokens,
+        }
+
+        try:
+            response = requests.post(
+                self.base_url,
+                headers=self.headers,
+                json=payload,
+                timeout=60
+            )
+
+            if response.status_code == 200:
+                data = response.json()
+                if "choices" in data and len(data["choices"]) > 0:
+                    return data["choices"][0]["message"]["content"]
+                else:
+                    print(f"Teacher API unexpected response: {data}")
+                    return None
+            else:
+                print(f"Teacher API error: {response.status_code} - {response.text}")
+                return None
+
+        except requests.exceptions.Timeout:
+            print("Teacher API timeout")
+            return None
+        except Exception as e:
+            print(f"Teacher API exception: {e}")
+            return None
+
+    def is_available(self) -> bool:
+        """Check if teacher API is available"""
+        return bool(self.api_key and self.api_key != "")
+
+
+# Global teacher instance
+teacher = TeacherModel()