Update main.py

main.py

@@ -184,7 +184,21 @@ async def followup_agent(query: FollowupQueryModel, background_tasks: Background
         logger.error(f"Error in followup_agent: {str(e)}")
         raise HTTPException(status_code=500, detail="An error occurred while processing the followup agent request.")
 
-       
+class ModelsResponse(BaseModel):
+    models: List[str]
+
+@app.get("/models", response_model=ModelsResponse)
+def get_models():
+    """
+    Endpoint to return a list of available model IDs dynamically extracted from the ModelID Literal.
+    """
+    try:
+        # Dynamically extract the literal values of ModelID
+        model_list = list(get_args(ModelID))
+        return ModelsResponse(models=model_list)
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"An error occurred while fetching the models: {str(e)}")
+
 from fastapi.middleware.cors import CORSMiddleware
 
 app.add_middleware(