From 68c5914a197e492525ceada57abaad559e544643 Mon Sep 17 00:00:00 2001 From: Ryan Date: Thu, 3 Apr 2025 18:11:07 -0400 Subject: [PATCH] add support for gpt-4 and gpt-4o --- .../src/aoai_api_simulator/generator/model_catalogue.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/aoai-api-simulator/src/aoai_api_simulator/generator/model_catalogue.py b/src/aoai-api-simulator/src/aoai_api_simulator/generator/model_catalogue.py index ad5de51..2da1713 100644 --- a/src/aoai-api-simulator/src/aoai_api_simulator/generator/model_catalogue.py +++ b/src/aoai-api-simulator/src/aoai_api_simulator/generator/model_catalogue.py @@ -3,6 +3,8 @@ model_catalogue = { "gpt-3.5-turbo": OpenAIChatModel(name="gpt-3.5-turbo"), "gpt-3.5-turbo-0613": OpenAIChatModel(name="gpt-3.5-turbo-0613"), + "gpt-4": OpenAIChatModel(name="gpt-4"), + "gpt-4o-mini": OpenAIChatModel(name="gpt-4o"), "text-embedding-ada-001": OpenAIEmbeddingModel(name="text-embedding-ada-001", supports_custom_dimensions=False), "text-embedding-ada-002": OpenAIEmbeddingModel(name="text-embedding-ada-002", supports_custom_dimensions=False), "text-embedding-3-small": OpenAIEmbeddingModel(name="text-embedding-3-small", supports_custom_dimensions=True),