Update app.py

app.py

@@ -57,7 +57,7 @@ IMAGE_STORAGE = {}
 
 # Simple API key verification (demo purposes only)
 # In production, you'd want a more secure authentication system
-API_KEYS = {"sk-demo-key": "demo"}
+# API_KEYS = {"sk-demo-key": "demo"}
 
 # Provider mapping
 PROVIDER_MAP = {
@@ -294,54 +294,54 @@ class APIError(Exception):
         self.param = param
         self.type = type
 
-# Authentication dependency
-async def verify_api_key(authorization: Optional[str] = Header(None)):
-    if authorization is None:
-        raise HTTPException(
-            status_code=401,
-            detail={
-                "error": {
-                    "message": "No API key provided",
-                    "type": "authentication_error",
-                    "param": None,
-                    "code": "no_api_key"
-                }
-            }
-        )
+# # Authentication dependency
+# async def verify_api_key(authorization: Optional[str] = Header(None)):
+#     if authorization is None:
+#         raise HTTPException(
+#             status_code=401,
+#             detail={
+#                 "error": {
+#                     "message": "No API key provided",
+#                     "type": "authentication_error",
+#                     "param": None,
+#                     "code": "no_api_key"
+#                 }
+#             }
+#         )
     
-    # Extract the key from the Authorization header
-    parts = authorization.split()
-    if len(parts) != 2 or parts[0].lower() != "bearer":
-        raise HTTPException(
-            status_code=401,
-            detail={
-                "error": {
-                    "message": "Invalid authentication format. Use 'Bearer YOUR_API_KEY'",
-                    "type": "authentication_error",
-                    "param": None,
-                    "code": "invalid_auth_format"
-                }
-            }
-        )
+#     # Extract the key from the Authorization header
+#     parts = authorization.split()
+#     if len(parts) != 2 or parts[0].lower() != "bearer":
+#         raise HTTPException(
+#             status_code=401,
+#             detail={
+#                 "error": {
+#                     "message": "Invalid authentication format. Use 'Bearer YOUR_API_KEY'",
+#                     "type": "authentication_error",
+#                     "param": None,
+#                     "code": "invalid_auth_format"
+#                 }
+#             }
+#         )
         
-    api_key = parts[1]
+#     api_key = parts[1]
     
-    # Check if the API key is valid
-    # In production, you'd want to use a more secure method
-    if api_key not in API_KEYS:
-        raise HTTPException(
-            status_code=401,
-            detail={
-                "error": {
-                    "message": "Invalid API key",
-                    "type": "authentication_error",
-                    "param": None,
-                    "code": "invalid_api_key"
-                }
-            }
-        )
+#     # Check if the API key is valid
+#     # In production, you'd want to use a more secure method
+#     if api_key not in API_KEYS:
+#         raise HTTPException(
+#             status_code=401,
+#             detail={
+#                 "error": {
+#                     "message": "Invalid API key",
+#                     "type": "authentication_error",
+#                     "param": None,
+#                     "code": "invalid_api_key"
+#                 }
+#             }
+#         )
     
-    return api_key
+#     return api_key
 
 # Find provider from model ID - updating this function to support provider/model format
 def get_provider_for_model(model: str):
@@ -395,7 +395,7 @@ async def health_check():
 # OpenAI-compatible endpoints
 
 # List available models
-@app.get("/v1/models", response_model=ModelsListResponse, dependencies=[Depends(verify_api_key)])
+@app.get("/v1/models", response_model=ModelsListResponse)
 async def list_models():
     models_data = []
     
@@ -422,7 +422,7 @@ async def list_models():
     }
 
 # Get model information
-@app.get("/v1/models/{model_id}", dependencies=[Depends(verify_api_key)])
+@app.get("/v1/models/{model_id}")
 async def get_model(model_id: str):
     try:
         provider_name, model = get_provider_for_model(model_id)
@@ -445,7 +445,7 @@ async def get_model(model_id: str):
         )
 
 # Generate images
-@app.post("/v1/images/generations", response_model=ImageGenerationResponse, dependencies=[Depends(verify_api_key)])
+@app.post("/v1/images/generations", response_model=ImageGenerationResponse)
 async def create_image(request: ImageGenerationRequest, background_tasks: BackgroundTasks):
     try:
         # Get provider for the requested model
@@ -662,7 +662,7 @@ async def create_image(request: ImageGenerationRequest, background_tasks: Backgr
         )
 
 # Image retrieval endpoint
-@app.get("/v1/images/{task_id}/{image_id}", dependencies=[Depends(verify_api_key)])
+@app.get("/v1/images/{task_id}/{image_id}")
 async def get_image(task_id: str, image_id: str):
     if task_id not in IMAGE_STORAGE:
         return JSONResponse(