From a3d2a4e0cbe556b7b5de3d3924882dd7c532242d Mon Sep 17 00:00:00 2001 From: Dustin Ngo Date: Thu, 10 Oct 2024 17:03:04 -0400 Subject: [PATCH] Hardcode all models for now --- src/phoenix/server/api/queries.py | 19 +++++++++++++++++-- 1 file changed, 17 insertions(+), 2 deletions(-) diff --git a/src/phoenix/server/api/queries.py b/src/phoenix/server/api/queries.py index d1243fb81f..6ca3e5c848 100644 --- a/src/phoenix/server/api/queries.py +++ b/src/phoenix/server/api/queries.py @@ -84,18 +84,33 @@ async def model_providers( self, vendors: List[str], info: Info[Context, None] ) -> List[ModelProvider]: all_vendors = { - "OpenAI": ModelProvider( + "OpenAI": ModelProvider( # https://platform.openai.com/docs/models name="OpenAI", # currently only models using the chat completions API model_names=[ "o1-preview", + "o1-preview-2024-09-12", "o1-mini", + "o1-mini-2024-09-12", "gpt-4o", + "gpt-4o-2024-08-06", + "gpt-4o-2024-05-13", + "chatgpt-4o-latest", "gpt-4o-mini", + "gpt-4o-mini-2024-07-18", + "gpt-4-turbo", + "gpt-4-turbo-2024-04-09", + "gpt-4-turbo-preview", + "gpt-4-0125-preview", + "gpt-4-1106-preview", "gpt-4", + "gpt-4-0613", + "gpt-3.5-turbo-0125", "gpt-3.5-turbo", + "gpt-3.5-turbo-1106", + "gpt-3.5-turbo-instruct", ], ), - "Anthropic": ModelProvider( + "Anthropic": ModelProvider( # https://docs.anthropic.com/en/docs/about-claude/models#model-comparison name="Anthropic", # currently only models using the messages API model_names=[ "claude-3-5-sonnet-20240620",