add Models.classification

self_hosted_env/voice_agent/core/enums.py

@@ -18,6 +18,7 @@ class StudentNationality(str, Enum):
 
 class Models(str, Enum):
     chat = "gpt-4o-mini"
+    classification = "gpt-4o-mini"
     tts = "gpt-4o-mini-tts"
     embedding = "text-embedding-3-small"
     transcription = "gpt-4o-transcribe"

self_hosted_env/voice_agent/services/agent_helpers/query_handlers.py

@@ -3,7 +3,7 @@ import sys
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../../')))
 
 from typing import Dict, Any
-from core import StudentNationality, StudyLanguage
+from core import StudentNationality, StudyLanguage, Models
 import logging
 
 logger = logging.getLogger(__name__)
@@ -274,7 +274,7 @@ class QueryHandler:
 
         try:
             response = self.openai_service.client.chat.completions.create(
-                model="gpt-4o-mini",
+                model=Models.classification,
                 messages=[{"role": "user", "content": classification_prompt}],
                 temperature=0,
                 max_tokens=20