From a4b9ae94711abcab7a9855ce3375d33430099742 Mon Sep 17 00:00:00 2001
From: Game_Time <108236317+RayBytes@users.noreply.github.com>
Date: Wed, 19 Nov 2025 14:59:34 +0500
Subject: [PATCH] Implement latest GPT-5.1 models

---
 chatmock/routes_openai.py | 5 ++++-
 chatmock/upstream.py      | 3 +++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/chatmock/routes_openai.py b/chatmock/routes_openai.py
index 13dc314..76fc9a8 100644
--- a/chatmock/routes_openai.py
+++ b/chatmock/routes_openai.py
@@ -24,7 +24,7 @@ openai_bp = Blueprint("openai", __name__)
 
 def _instructions_for_model(model: str) -> str:
     base = current_app.config.get("BASE_INSTRUCTIONS", BASE_INSTRUCTIONS)
-    if model == "gpt-5-codex":
+    if model == "gpt-5-codex" or model == "gpt-5.1-codex":
         codex = current_app.config.get("GPT5_CODEX_INSTRUCTIONS") or GPT5_CODEX_INSTRUCTIONS
         if isinstance(codex, str) and codex.strip():
             return codex
@@ -441,7 +441,10 @@ def list_models() -> Response:
     expose_variants = bool(current_app.config.get("EXPOSE_REASONING_MODELS"))
     model_groups = [
         ("gpt-5", ["high", "medium", "low", "minimal"]),
+        ("gpt-5.1", ["high", "medium", "low", "minimal"]),
         ("gpt-5-codex", ["high", "medium", "low"]),
+        ("gpt-5.1-codex", ["high", "medium", "low"]),
+        ("gpt-5.1-codex-mini", []),
         ("codex-mini", []),
     ]
     model_ids: List[str] = []
diff --git a/chatmock/upstream.py b/chatmock/upstream.py
index 56c9739..bc4051d 100644
--- a/chatmock/upstream.py
+++ b/chatmock/upstream.py
@@ -31,12 +31,15 @@ def normalize_model_name(name: str | None, debug_model: str | None = None) -> st
         "gpt5": "gpt-5",
         "gpt-5-latest": "gpt-5",
         "gpt-5": "gpt-5",
+        "gpt-5.1": "gpt-5.1",
         "gpt5-codex": "gpt-5-codex",
         "gpt-5-codex": "gpt-5-codex",
         "gpt-5-codex-latest": "gpt-5-codex",
+        "gpt-5.1-codex": "gpt-5.1-codex",
         "codex": "codex-mini-latest",
         "codex-mini": "codex-mini-latest",
         "codex-mini-latest": "codex-mini-latest",
+        "gpt-5.1-codex-mini": "gpt-5.1-codex-mini",
     }
     return mapping.get(base, base)