Update backend_api.py

backend_api.py

@@ -743,29 +743,34 @@ def extract_reasoning(code: str, language: str) -> str:
 
 @app.post("/api/generate")
 async def generate_code(request: CodeGenerationRequest, authorization: Optional[str] = Header(None)):
+    # 1. Lấy dữ liệu từ request
     query = request.query
     language = request.language
-    model_id = request.model_id # Ví dụ: "gemini-1.5-flash"
+    model_id = request.model_id # Ví dụ: "gemini-3.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():
+    # 2. Định nghĩa system prompt ở đây để nó khả dụng trong scope của event_stream
+    system_prompt = SYSTEM_PROMPT_CACHE.get(language, GENERIC_SYSTEM_PROMPT.format(language=language))
+    full_prompt = f"{system_prompt}\n\nTask: {query}"
+
+    async def event_stream() -> AsyncGenerator[str, None]:
         try:
-            # Dùng client mới để gọi model
-            # Lưu ý: Thư viện mới dùng model_id trực tiếp, không cần tiền tố 'models/' nếu gọi qua client
+            # 3. Lấy client từ backend_models
+            client = get_genai_client()
+            
+            # 4. Sử dụng client mới để tạo stream
+            # Lưu ý: google-genai dùng 'contents', không phải 'full_prompt'
             response = client.models.generate_content_stream(
-                model=request.model_id, 
+                model=model_id,
                 contents=full_prompt
             )
             
             for chunk in response:
                 if chunk.text:
                     yield f"data: {json.dumps({'type': 'chunk', 'content': chunk.text})}\n\n"
-            
+                    await asyncio.sleep(0.001)
+
             yield f"data: {json.dumps({'type': 'complete'})}\n\n"
-            
+
         except Exception as e:
             yield f"data: {json.dumps({'type': 'error', 'message': str(e)})}\n\n"