update code

code_engine.py

@@ -6,6 +6,7 @@ from prompt_builder import build_prompt
 from model_client import model_client
 from response_formatter import build_response, build_error_response
 from code_retriever import retrieve_code_evidence
+from context_manager import context_manager
 from config import settings
 
 
@@ -33,28 +34,33 @@ def process_codex_request(request: CodeXRequest) -> CodeXResponse:
     start_time = time.perf_counter()
 
     try:
+        enriched_request = context_manager.enrich_request_with_context(
+            request=request,
+            session_id=request.session_id,
+        )
+
         task_type = detect_task_type(
-            message=request.message,
-            code=request.code,
-            error_message=request.error_message,
-            mode_hint=request.mode,
+            message=enriched_request.message,
+            code=enriched_request.code,
+            error_message=enriched_request.error_message,
+            mode_hint=enriched_request.mode,
         )
 
         evidence_list = []
         retrieval_used = False
 
-        if should_use_retrieval(task_type, request):
-            evidence_list = get_retrieved_evidence(task_type, request)
+        if should_use_retrieval(task_type, enriched_request):
+            evidence_list = get_retrieved_evidence(task_type, enriched_request)
             retrieval_used = len(evidence_list) > 0
 
         prompt = build_prompt(
             task_type=task_type,
-            message=request.message,
-            code=request.code,
-            error_message=request.error_message,
-            language=request.language,
-            framework=request.framework,
-            previous_context=request.previous_context,
+            message=enriched_request.message,
+            code=enriched_request.code,
+            error_message=enriched_request.error_message,
+            language=enriched_request.language,
+            framework=enriched_request.framework,
+            previous_context=enriched_request.previous_context,
             evidence_list=evidence_list,
         )
 
@@ -62,7 +68,7 @@ def process_codex_request(request: CodeXRequest) -> CodeXResponse:
 
         processing_time_ms = int((time.perf_counter() - start_time) * 1000)
 
-        return build_response(
+        response = build_response(
             task_type=task_type,
             model_output=model_output,
             model_used=model_used,
@@ -70,10 +76,22 @@ def process_codex_request(request: CodeXRequest) -> CodeXResponse:
             retrieval_used=retrieval_used,
             source_count=len(evidence_list),
             processing_time_ms=processing_time_ms,
-            original_code=request.code,
+            original_code=enriched_request.code,
             sources=evidence_list,
         )
 
+        if response.code_output and response.code_output.strip():
+            context_manager.save_artifact(
+                session_id=enriched_request.session_id,
+                code=response.code_output,
+                task_type=task_type,
+                language=enriched_request.language,
+                framework=enriched_request.framework,
+                last_user_goal=enriched_request.message,
+            )
+
+        return response
+
     except Exception as e:
         processing_time_ms = int((time.perf_counter() - start_time) * 1000)
         fallback_task = request.mode if request.mode else CodeTaskType.UNKNOWN