Update code

code_engine.py

@@ -1,3 +1,5 @@
+import time
+
 from schemas import CodeXRequest, CodeXResponse, CodeTaskType
 from code_router import detect_task_type
 from prompt_builder import build_prompt
@@ -27,6 +29,8 @@ def get_retrieved_evidence(task_type: CodeTaskType, request: CodeXRequest):
 
 
 def process_codex_request(request: CodeXRequest) -> CodeXResponse:
+    start_time = time.perf_counter()
+
     try:
         task_type = detect_task_type(
             message=request.message,
@@ -55,6 +59,8 @@ def process_codex_request(request: CodeXRequest) -> CodeXResponse:
 
         model_output, model_used, used_fallback = model_client.generate(prompt)
 
+        processing_time_ms = int((time.perf_counter() - start_time) * 1000)
+
         return build_response(
             task_type=task_type,
             model_output=model_output,
@@ -62,11 +68,15 @@ def process_codex_request(request: CodeXRequest) -> CodeXResponse:
             used_fallback=used_fallback,
             retrieval_used=retrieval_used,
             source_count=len(evidence_list),
+            processing_time_ms=processing_time_ms,
         )
 
     except Exception as e:
+        processing_time_ms = int((time.perf_counter() - start_time) * 1000)
         fallback_task = request.mode if request.mode else CodeTaskType.UNKNOWN
+
         return build_error_response(
             task_type=fallback_task,
             error_message=str(e),
+            processing_time_ms=processing_time_ms,
         )