Update main.py

main.py

@@ -5,17 +5,8 @@ import requests
 
 app = FastAPI()
 
-# Valid API keys and model aliases
+# Valid API keys
 valid_api_keys = ['PARTH-SADARIA-NI-API-CHAWI', 'HEET-NI-CHEESEWADI-API-KEY']
-model_aliases = {
-    'Llama-3.1-Nemotron-70B-Instruct': 'nvidia/Llama-3.1-Nemotron-70B-Instruct',
-    'Meta-Llama-3.1-8B-Instruct': 'meta-llama/Meta-Llama-3.1-8B-Instruct',
-    'Meta-Llama-3.1-70B-Instruct': 'meta-llama/Meta-Llama-3.1-70B-Instruct',
-    'Meta-Llama-3.1-70B-Instruct-Turbo': 'meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo',
-    'Meta-Llama-3.1-405B-Instruct': 'meta-llama/Meta-Llama-3.1-405B-Instruct',
-    'Llama-3.2-11B-Vision-Instruct': 'meta-llama/Llama-3.2-11B-Vision-Instruct',
-    'Meta-Llama-3-8B-Instruct': 'meta-llama/Meta-Llama-3-8B-Instruct',
-}
 
 class Payload(BaseModel):
     model: str
@@ -42,20 +33,11 @@ async def get_completion(payload: Payload, request: Request):
     if api_key not in valid_api_keys:
         raise HTTPException(status_code=403, detail="Forbidden: Invalid API key. Join dsc.gg/chadgang and DM @mr_leaderyt on Discord for a free API key :)")
     
-    # Resolve model alias or set default service
-    user_model = payload.model
-    full_model_name = model_aliases.get(user_model, user_model)
+    # Set the URL for all requests
+    url = "https://gpt.tiptopuni.com/api/openai/v1/chat/completions"
     
-    # If model alias is not found, default to tiptopuni
-    is_deepinfra_model = full_model_name in model_aliases.values()
-    url = (
-        "https://api.deepinfra.com/v1/openai/chat/completions"
-        if is_deepinfra_model else
-        "https://gpt.tiptopuni.com/api/openai/v1/chat/completions"
-    )
-    
-    # Set up payload for streaming
-    payload_dict = {**payload.dict(), "model": full_model_name, "stream": True}
+    # Prepare the payload for streaming
+    payload_dict = {**payload.dict(), "stream": True}
 
     # Define a generator to stream the response
     def stream_generator():