hard-coded openai models
This commit is contained in:
parent
e0015f35db
commit
de447a6ae5
1 changed files with 23 additions and 2 deletions
|
|
@ -20,6 +20,27 @@ class ModelsService:
|
||||||
"jina-embeddings-v2-base-en",
|
"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):
|
def __init__(self):
|
||||||
self.session_manager = None
|
self.session_manager = None
|
||||||
|
|
||||||
|
|
@ -48,12 +69,12 @@ class ModelsService:
|
||||||
model_id = model.get("id", "")
|
model_id = model.get("id", "")
|
||||||
|
|
||||||
# Language models (GPT models)
|
# 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(
|
language_models.append(
|
||||||
{
|
{
|
||||||
"value": model_id,
|
"value": model_id,
|
||||||
"label": model_id,
|
"label": model_id,
|
||||||
"default": model_id == "gpt-4o-mini",
|
"default": model_id == "gpt-5",
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue