Hardcode all models for now

Arize-ai · Oct 11, 2024 · a3d2a4e · a3d2a4e
1 parent 2da89ef
commit a3d2a4e
Showing 1 changed file with 17 additions and 2 deletions.
diff --git 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",