UUpdate code

response_formatter.py

@@ -1,4 +1,4 @@
-from schemas import CodeTaskType, CodeXResponse
+from schemas import CodeTaskType, CodeXResponse, ResponseMeta
 
 
 def clean_text(text: str) -> str:
@@ -12,16 +12,26 @@ def build_response(
     model_output: str,
     model_used: str,
     used_fallback: bool,
+    retrieval_used: bool = False,
+    source_count: int = 0,
 ) -> CodeXResponse:
     cleaned_output = clean_text(model_output)
 
     return CodeXResponse(
         task_type=task_type,
         answer=cleaned_output,
-        model_used=model_used,
-        used_fallback=used_fallback,
-        success=True,
-        error=None,
+        code_output=None,
+        explanation=None,
+        warnings=[],
+        sources=[],
+        needs_clarification=False,
+        meta=ResponseMeta(
+            used_model=model_used,
+            fallback_used=used_fallback,
+            retrieval_used=retrieval_used,
+            source_count=source_count,
+            processing_time_ms=None,
+        ),
     )
 
 
@@ -32,8 +42,16 @@ def build_error_response(
     return CodeXResponse(
         task_type=task_type,
         answer="",
-        model_used="none",
-        used_fallback=False,
-        success=False,
-        error=clean_text(error_message),
+        code_output=None,
+        explanation=None,
+        warnings=[clean_text(error_message)],
+        sources=[],
+        needs_clarification=False,
+        meta=ResponseMeta(
+            used_model="none",
+            fallback_used=False,
+            retrieval_used=False,
+            source_count=0,
+            processing_time_ms=None,
+        ),
     )