Update main.py

main.py

@@ -87,18 +87,12 @@ def update_db(user_id, conversation_id, message, response):
 
 
 ModelID = Literal[
-    "openai/gpt-4o-mini",
-    "openai/chatgpt-4o-latest",
-    "deepseek/deepseek-chat",
-    "anthropic/claude-3.5-sonnet",
-    "anthropic/claude-3-haiku",
-    "meta-llama/llama-3.3-70b-instruct",
-    "google/gemini-2.0-flash-exp:free",
-    "google/gemini-flash-1.5",
-    "meta-llama/llama-3-70b-instruct",
-    "meta-llama/llama-3-8b-instruct",
-
+    "anthropic.claude-3-sonnet-20240229-v1:0",
+    "anthropic.claude-3-haiku-20240307-v1:0",
+    "meta.llama3-8b-instruct-v1:0",
+    "meta.llama3-70b-instruct-v1:0"
 ]
+
 class FollowupQueryModel(BaseModel):
     query: str = Field(..., description="User's query for the followup agent")
     model_id: ModelID = Field(