Update code

code_retriever.py

@@ -54,36 +54,64 @@ def sort_evidence(evidence_list: List[RetrievedEvidence]) -> List[RetrievedEvide
 def trim_evidence(evidence_list: List[RetrievedEvidence]) -> List[RetrievedEvidence]:
     stack_items = [item for item in evidence_list if item.source_type == SourceType.STACKOVERFLOW]
     github_items = [item for item in evidence_list if item.source_type == SourceType.GITHUB]
+    other_items = [
+        item for item in evidence_list
+        if item.source_type not in {SourceType.STACKOVERFLOW, SourceType.GITHUB}
+    ]
 
     stack_items = stack_items[:3]
-    github_items = github_items[:2]
+    github_items = github_items[:3]
 
-    combined = stack_items + github_items
+    combined = stack_items + github_items + other_items
     combined = sort_evidence(combined)
 
     return combined[: settings.MAX_RETRIEVED_ITEMS]
 
 
-def should_use_github(request: CodeXRequest) -> bool:
+def should_use_stack(task_type: CodeTaskType) -> bool:
+    if not settings.ENABLE_STACK_SEARCH:
+        return False
+
+    return task_type in {
+        CodeTaskType.FIX,
+        CodeTaskType.REVIEW,
+        CodeTaskType.REFACTOR,
+    }
+
+
+def should_use_github(task_type: CodeTaskType, request: CodeXRequest) -> bool:
     if not settings.ENABLE_GITHUB_SEARCH:
         return False
 
-    if request.framework and request.framework.strip():
-        return True
+    if task_type == CodeTaskType.FIX:
+        return bool(
+            (request.framework and request.framework.strip())
+            or (request.error_message and request.error_message.strip())
+            or (request.language and request.language.strip())
+        )
 
-    if request.error_message and request.error_message.strip():
-        return True
+    if task_type in {CodeTaskType.REVIEW, CodeTaskType.REFACTOR}:
+        return bool(
+            (request.framework and request.framework.strip())
+            or (request.language and request.language.strip())
+        )
 
     return False
 
 
 def retrieve_code_evidence(task_type: CodeTaskType, request: CodeXRequest) -> List[RetrievedEvidence]:
-    if task_type != CodeTaskType.FIX:
+    supported_tasks = {
+        CodeTaskType.FIX,
+        CodeTaskType.REVIEW,
+        CodeTaskType.REFACTOR,
+    }
+
+    if task_type not in supported_tasks:
         return []
 
     collected: List[RetrievedEvidence] = []
 
-    if settings.ENABLE_STACK_SEARCH:
+    if should_use_stack(task_type):
         stack_results = search_stackoverflow(
             message=request.message,
             error_message=request.error_message,
@@ -94,7 +122,7 @@ def retrieve_code_evidence(task_type: CodeTaskType, request: CodeXRequest) -> Li
         )
         collected.extend(stack_results)
 
-    if should_use_github(request):
+    if should_use_github(task_type, request):
         github_results = search_github(
             message=request.message,
             error_message=request.error_message,