From de447a6ae529375f936dd7a3f0e9805c66cf547a Mon Sep 17 00:00:00 2001 From: Lucas Oliveira Date: Fri, 26 Sep 2025 14:21:20 -0300 Subject: [PATCH] hard-coded openai models --- src/services/models_service.py | 25 +++++++++++++++++++++++-- 1 file changed, 23 insertions(+), 2 deletions(-) diff --git a/src/services/models_service.py b/src/services/models_service.py index 35a54895..47a89da9 100644 --- a/src/services/models_service.py +++ b/src/services/models_service.py @@ -20,6 +20,27 @@ class ModelsService: "jina-embeddings-v2-base-en", ] + OPENAI_TOOL_CALLING_MODELS = [ + "gpt-5", + "gpt-5-mini", + "gpt-5-nano", + "gpt-4o-mini", + "gpt-4o", + "gpt-4.1", + "gpt-4.1-mini", + "gpt-4.1-nano", + "gpt-4-turbo", + "gpt-4-turbo-preview", + "gpt-4", + "gpt-3.5-turbo", + "o1", + "o3-mini", + "o3", + "o3-pro", + "o4-mini", + "o4-mini-high", + ] + def __init__(self): self.session_manager = None @@ -48,12 +69,12 @@ class ModelsService: model_id = model.get("id", "") # Language models (GPT models) - if any(prefix in model_id for prefix in ["gpt-4", "gpt-3.5"]): + if model_id in self.OPENAI_TOOL_CALLING_MODELS: language_models.append( { "value": model_id, "label": model_id, - "default": model_id == "gpt-4o-mini", + "default": model_id == "gpt-5", } )