include short model names in the response (#439)

* include short model names in the response

Signed-off-by: Peter Jausovec <peter.jausovec@solo.io>

* lint

Signed-off-by: Peter Jausovec <peter.jausovec@solo.io>

---------

Signed-off-by: Peter Jausovec <peter.jausovec@solo.io>

Author: peterj

python/src/autogenstudio/web/routes/models.py

@@ -3,44 +3,50 @@
 from autogen_ext.models.anthropic._model_info import _MODEL_INFO as anthropic_models
 from autogen_ext.models.ollama._model_info import _MODEL_INFO as ollama_models
 from autogen_ext.models.openai._model_info import _MODEL_INFO as openai_models
+from autogen_ext.models.openai._model_info import _MODEL_POINTERS
 from fastapi import APIRouter
 
 router = APIRouter()
 
 
 @router.get("/")
 async def list_models() -> Dict[str, List[Dict[str, Any]]]:
-    # Get the model names from the model info, also get the  "function_calling" value for each model
-
-    response = {
-        "anthropic": [
-            {
-                "name": model,
-                "function_calling": anthropic_models[model]["function_calling"],
-            }
-            for model in anthropic_models.keys()
-        ],
-        "ollama": [
-            {
-                "name": model,
-                "function_calling": ollama_models[model]["function_calling"],
-            }
-            for model in ollama_models.keys()
-        ],
-        "openAI": [
-            {
-                "name": model,
-                "function_calling": openai_models[model]["function_calling"],
-            }
-            for model in openai_models.keys()
-        ],
-        "azureOpenAI": [
+    response_ollama = []
+    for model_name, model_data in ollama_models.items():
+        response_ollama.append(
             {
-                "name": model,
-                "function_calling": openai_models[model]["function_calling"],
+                "name": model_name,
+                "function_calling": model_data["function_calling"],
             }
-            for model in openai_models.keys()
-        ],
-    }
+        )
 
-    return response
+    final_anthropic_models_map = {}
+    for model_name, model_data in anthropic_models.items():
+        final_anthropic_models_map[model_name] = {"function_calling": model_data["function_calling"]}
+
+    for short_name, long_name_target in _MODEL_POINTERS.items():
+        if short_name.startswith("claude-"):
+            if long_name_target in anthropic_models:
+                properties = anthropic_models[long_name_target]
+                final_anthropic_models_map[short_name] = {"function_calling": properties["function_calling"]}
+
+    response_anthropic = [{"name": name, **props} for name, props in final_anthropic_models_map.items()]
+
+    final_openai_models_map = {}
+    for model_name, model_data in openai_models.items():
+        final_openai_models_map[model_name] = {"function_calling": model_data["function_calling"]}
+
+    for short_name, long_name_target in _MODEL_POINTERS.items():
+        if not short_name.startswith("claude-"):
+            if long_name_target in openai_models:
+                properties = openai_models[long_name_target]
+                final_openai_models_map[short_name] = {"function_calling": properties["function_calling"]}
+
+    response_openai = [{"name": name, **props} for name, props in final_openai_models_map.items()]
+
+    return {
+        "anthropic": response_anthropic,
+        "ollama": response_ollama,
+        "openAI": response_openai,
+        "azureOpenAI": response_openai,
+    }