Upload main.py

main.py

@@ -224,7 +224,7 @@ class ToneStyle(str, Enum):
     FRIENDLY = "friendly"
     BULLET   = "bullet"     # convert to bullet points
 
-DEFAULT_MODEL = AIModel.NEURONES_SELF
+DEFAULT_MODEL = "neurones-pro-1.0"
 
 BLOCKED_PATTERNS = [
     r"(?i)\b(nude|sex|porn|erotic|18\+|naked|rape|fetish|incest|adult content|horny)\b"
@@ -1478,16 +1478,19 @@ class ChatMessage(BaseModel):
     instructions:    str  = ""
     autoPersonality: bool = False
     additionalInfor: str  = ""
-    model:           AIModel = DEFAULT_MODEL      # legacy enum — kept for compat
-    model_id:        str     = ""                 # NEW: registry model id e.g. "neurones-r1-1.0"
-    force_groq:      bool    = False
+    # IMPORTANT: must be str, NOT AIModel enum.
+    # Pydantic rejects any value not in the enum (e.g. "neurones-pro-1.0")
+    # causing a 422 before the endpoint even runs → "failed to fetch" on frontend.
+    model:           str  = "neurones-pro-1.0"
+    model_id:        str  = ""                 # registry model id — preferred field
+    force_groq:      bool = False
     persona:         Optional[str] = None
-    deep_think:      bool    = False
-    deep_search:     bool    = False
+    deep_think:      bool = False
+    deep_search:     bool = False
     response_length: ResponseLength = ResponseLength.BALANCED
     tone:            ToneStyle      = ToneStyle.DEFAULT
-    json_mode:       bool    = False
-    image_session_id: str    = ""                 # NEW: reference an uploaded image by session ID
+    json_mode:       bool = False
+    image_session_id: str = ""
 
 class TranslateRequest(BaseModel):
     user_id:  str
@@ -1576,8 +1579,8 @@ async def chat(payload: ChatMessage, request: Request):
         return {"response": "😏 Sorry, I can't respond to that type of message."}
 
     # ── Resolve model from registry ─────────────────────────────
-    # Prefer model_id (new field) over legacy enum
-    raw_model_id = payload.model_id.strip() or payload.model.value
+    # model_id is the preferred field (new). model is legacy compat (plain string now).
+    raw_model_id = payload.model_id.strip() or payload.model.strip() or "neurones-pro-1.0"
     model_cfg    = model_registry.get(raw_model_id)
 
     # ── Persona ─────────────────────────────────────────────────
@@ -1724,7 +1727,7 @@ async def chat_stream(payload: ChatMessage, request: Request):
             yield "data: [DONE]\n\n"
         return StreamingResponse(rate_error(), media_type="text/event-stream")
 
-    raw_model_id = payload.model_id.strip() or payload.model.value
+    raw_model_id = payload.model_id.strip() or payload.model.strip() or "neurones-pro-1.0"
     model_cfg    = model_registry.get(raw_model_id)
     tz_str       = get_user_timezone(user_id, ip)