Update main.py

main.py

@@ -8,6 +8,7 @@ import uuid
 import logging
 import asyncio
 from typing import List, Dict, Optional, Union, Generator, Any, AsyncGenerator
+from contextlib import asynccontextmanager
 
 from fastapi import FastAPI, HTTPException, Depends, Request, status
 from fastapi.middleware.cors import CORSMiddleware
@@ -149,8 +150,7 @@ class QodoAI:
         self.url = "https://api.cli.qodo.ai/v2/agentic/start-task"
         self.info_url = "https://api.cli.qodo.ai/v2/info/get-things"
         self.timeout = timeout
-        self.
-api_key = api_key or "sk-dS7U-extxMWUxc8SbYYOuncqGUIE8-y2OY8oMCpu0eI-qnSUyH9CYWO_eAMpqwfMo7pXU3QNrclfZYMO0M6BJTM"
+        self.api_key = api_key
         
         # Generate fingerprint
         self.fingerprint = {"user_agent": "axios/1.10.0", "browser_type": "chrome"}
@@ -188,7 +188,7 @@ api_key = api_key or "sk-dS7U-extxMWUxc8SbYYOuncqGUIE8-y2OY8oMCpu0eI-qnSUyH9CYWO
                 "User-Agent": "axios/1.10.0",
             }
             
-temp_session.headers.update(temp_headers)
+            temp_session.headers.update(temp_headers)
             
             response = temp_session.get(self.info_url, timeout=self.timeout, impersonate="chrome110")
             
@@ -234,8 +234,7 @@ temp_session.headers.update(temp_headers)
         elif isinstance(chunk, str):
             try:
                 parsed = json.loads(chunk)
-                return QodoAI._qodo_ext
-ractor(parsed)
+                return QodoAI._qodo_extractor(parsed)
             except json.JSONDecodeError:
                 if chunk.strip():
                     return chunk.strip()
@@ -288,7 +287,7 @@ ractor(parsed)
             if message.role == "user":
                 user_prompt = message.content
                 
-break
+                break
 
         if not user_prompt:
             raise HTTPException(status_code=400, detail="No user message found in messages")
@@ -333,7 +332,7 @@ break
                     )
 
                     for content_chunk in processed_stream:
-if content_chunk:
+                        if content_chunk:
                             chunk_data = {
                                 "id": request_id,
                                 "object": "chat.completion.chunk",
@@ -382,8 +381,7 @@ if content_chunk:
 
         except Exception as e:
             logger.error(f"Stream creation error: {e}")
-            raise HTTPException(status_code=
-500, detail=str(e))
+            raise HTTPException(status_code=500, detail=str(e))
 
     async def _create_non_stream_response(self, request_id: str, created_time: int, model: str, payload: Dict[str, Any], user_prompt: str) -> ChatCompletionResponse:
         """Create non-streaming response."""
@@ -430,7 +428,7 @@ if content_chunk:
                         json_obj = json.loads(json_str)
                         content = QodoAI._qodo_extractor(json_obj)
                         if content:
-full_response += content
+                            full_response += content
                     except json.JSONDecodeError:
                         pass
 
@@ -489,8 +487,7 @@ app.add_middleware(
     allow_origins=["*"],
     allow_credentials=True,
     allow_methods=["*"],
-    
-allow_headers=["*"],
+    allow_headers=["*"],
 )
 
 # Authentication dependency
@@ -546,8 +543,7 @@ async def list_models(api_key: str = Depends(verify_api_key)):
 
 @app.post("/v1/chat/completions")
 async def create_chat_completion(
-    request: ChatCompletion
-Request,
+    request: ChatCompletionRequest,
     api_key: str = Depends(verify_api_key)
 ):
     """Create a chat completion."""
@@ -600,8 +596,7 @@ async def log_requests(request: Request, call_next):
     # Log request
     logger.info(f"{request.method} {request.url.path} - Start")
     
-    response
- = await call_next(request)
+    response = await call_next(request)
     
     # Log response
     process_time = time.time() - start_time