update code

code_router.py

@@ -54,6 +54,38 @@ REVIEW_PATTERNS = [
     r"\bcode review\b",
 ]
 
+MODIFICATION_PATTERNS = [
+    r"\badd\b",
+    r"\bupdate\b",
+    r"\bmodify\b",
+    r"\bchange\b",
+    r"\brename\b",
+    r"\bconvert\b",
+    r"\bextend\b",
+    r"\binclude\b",
+    r"\buse\b",
+    r"\breplace\b",
+    r"\bremove\b",
+    r"\bdelete\b",
+    r"\bdocstring\b",
+    r"\btype hints?\b",
+    r"\bvalidation\b",
+    r"\berror handling\b",
+    r"\blogging\b",
+    r"\bcomments\b",
+    r"\bexception handling\b",
+    r"\bmake it\b",
+    r"\bturn this into\b",
+]
+
+DIRECT_EXPLAIN_ONLY_PATTERNS = [
+    r"\bexplain\b",
+    r"\bwhat does this do\b",
+    r"\bhow does this work\b",
+    r"\bdescribe this code\b",
+    r"\bhelp me understand\b",
+]
+
 
 def normalize_text(text: str) -> str:
     text = text or ""
@@ -66,6 +98,14 @@ def contains_pattern(text: str, patterns: list[str]) -> bool:
     return any(re.search(pattern, text) for pattern in patterns)
 
 
+def is_modification_request(text: str) -> bool:
+    return contains_pattern(text, MODIFICATION_PATTERNS)
+
+
+def is_direct_explanation_request(text: str) -> bool:
+    return contains_pattern(text, DIRECT_EXPLAIN_ONLY_PATTERNS)
+
+
 def detect_task_type(
     message: str,
     code: str | None = None,
@@ -85,19 +125,30 @@ def detect_task_type(
     if contains_pattern(normalized_message, FIX_PATTERNS):
         return CodeTaskType.FIX
 
-    if contains_pattern(normalized_message, EXPLAIN_PATTERNS):
-        return CodeTaskType.EXPLAIN
+    if contains_pattern(normalized_message, REVIEW_PATTERNS):
+        return CodeTaskType.REVIEW
 
     if contains_pattern(normalized_message, REFACTOR_PATTERNS):
         return CodeTaskType.REFACTOR
 
-    if contains_pattern(normalized_message, REVIEW_PATTERNS):
-        return CodeTaskType.REVIEW
+    if is_direct_explanation_request(normalized_message) and not is_modification_request(normalized_message):
+        return CodeTaskType.EXPLAIN
+
+    if has_code and is_modification_request(normalized_message):
+        return CodeTaskType.REFACTOR
 
     if contains_pattern(normalized_message, GENERATE_PATTERNS):
         return CodeTaskType.GENERATE
 
+    if contains_pattern(normalized_message, EXPLAIN_PATTERNS) and not is_modification_request(normalized_message):
+        return CodeTaskType.EXPLAIN
+
     if has_code and not has_error:
+        if is_modification_request(normalized_message):
+            return CodeTaskType.REFACTOR
         return CodeTaskType.EXPLAIN
 
+    if is_modification_request(normalized_message):
+        return CodeTaskType.GENERATE
+
     return CodeTaskType.UNKNOWN