Update code

code_engine.py

@@ -36,8 +36,11 @@ def process_codex_request(request: CodeXRequest) -> CodeXResponse:
         )
 
         evidence_list = []
+        retrieval_used = False
+
         if should_use_retrieval(task_type, request):
             evidence_list = get_retrieved_evidence(task_type, request)
+            retrieval_used = len(evidence_list) > 0
 
         prompt = build_prompt(
             task_type=task_type,
@@ -57,6 +60,8 @@ def process_codex_request(request: CodeXRequest) -> CodeXResponse:
             model_output=model_output,
             model_used=model_used,
             used_fallback=used_fallback,
+            retrieval_used=retrieval_used,
+            source_count=len(evidence_list),
         )
 
     except Exception as e: