Update backend_api.py

backend_api.py

@@ -729,13 +729,16 @@ def extract_reasoning(code: str, language: str) -> str:
 async def generate_code(request: CodeGenerationRequest, authorization: Optional[str] = Header(None)):
     query = request.query
     language = request.language
-    model_id = request.model_id # Ví dụ: "gemini-1.5-flash" hoặc "gemini-1.5-pro"
+    model_id = request.model_id # Ví dụ: "gemini-1.5-flash"
+    
+    # Đảm bảo model_id có tiền tố 'models/' để tránh lỗi 404
+    if not model_id.startswith("models/"):
+        model_id = f"models/{model_id}"
     
     async def event_stream() -> AsyncGenerator[str, None]:
         try:
-            # 1. Khởi tạo model Gemini trực tiếp
-            # Không cần qua get_inference_client hay get_real_model_id cũ
-            model = genai.GenerativeModel(model_id)
+            # 1. Khởi tạo model với ID đã chuẩn hóa
+            model = genai.GenerativeModel(model_name=model_id)
             
             # 2. Định nghĩa System Prompt
             system_prompt = SYSTEM_PROMPT_CACHE.get(language, GENERIC_SYSTEM_PROMPT.format(language=language))
@@ -747,7 +750,6 @@ async def generate_code(request: CodeGenerationRequest, authorization: Optional[
             # 4. Stream phản hồi về frontend
             for chunk in response:
                 if chunk.text:
-                    # Giữ nguyên định dạng JSON để frontend không cần sửa code
                     yield f"data: {json.dumps({'type': 'chunk', 'content': chunk.text})}\n\n"
                     await asyncio.sleep(0.001)
 
@@ -755,11 +757,11 @@ async def generate_code(request: CodeGenerationRequest, authorization: Optional[
             yield f"data: {json.dumps({'type': 'complete'})}\n\n"
 
         except Exception as e:
-            # Gửi thông báo lỗi chi tiết nếu có
+            # Gửi thông báo lỗi chi tiết để debug
             yield f"data: {json.dumps({'type': 'error', 'message': str(e)})}\n\n"
             
     return StreamingResponse(event_stream(), media_type="text/event-stream")
-    
+
 @app.post("/api/deploy")
 async def deploy(
     request: DeploymentRequest,