Update code

response_formatter.py

@@ -1,7 +1,7 @@
 import re
-from typing import Optional
+from typing import Optional, List
 
-from schemas import CodeTaskType, CodeXResponse, ResponseMeta
+from schemas import CodeTaskType, CodeXResponse, ResponseMeta, RetrievedEvidence
 
 
 def clean_text(text: Optional[str]) -> str:
@@ -271,6 +271,7 @@ def build_response(
     source_count: int = 0,
     processing_time_ms: Optional[int] = None,
     original_code: Optional[str] = None,
+    sources: Optional[List[RetrievedEvidence]] = None,
 ) -> CodeXResponse:
     cleaned_output = clean_text(model_output)
 
@@ -299,7 +300,7 @@ def build_response(
         code_output=code_output,
         explanation=explanation,
         warnings=warnings,
-        sources=[],
+        sources=sources or [],
         needs_clarification=False,
         meta=ResponseMeta(
             used_model=model_used,