Update code

context_manager.py

@@ -13,8 +13,13 @@ FOLLOW_UP_PATTERNS = [
     "explain this",
     "review this",
     "add validation",
+    "input validation",
+    "add input validation",
     "add error handling",
     "add comments",
+    "add logging",
+    "add docstring",
+    "add type hints",
     "optimize this",
     "continue this",
     "continue from above",
@@ -24,6 +29,11 @@ FOLLOW_UP_PATTERNS = [
     "the above code",
     "use the previous code",
     "based on the previous code",
+    "now add",
+    "now update",
+    "now change",
+    "now remove",
+    "now use",
 ]
 
 
@@ -89,7 +99,26 @@ class ContextManager:
         if not message:
             return False
 
-        return any(pattern in message for pattern in FOLLOW_UP_PATTERNS)
+        if any(pattern in message for pattern in FOLLOW_UP_PATTERNS):
+            return True
+
+        loose_follow_up_signals = [
+            "add ",
+            "update ",
+            "change ",
+            "modify ",
+            "remove ",
+            "replace ",
+            "rename ",
+            "use ",
+            "include ",
+        ]
+
+        has_session = bool(request.session_id and str(request.session_id).strip())
+        if has_session and any(signal in message for signal in loose_follow_up_signals):
+            return True
+
+        return False
 
     def enrich_request_with_context(
         self,
@@ -115,10 +144,15 @@ class ContextManager:
         if request.previous_context and request.previous_context.strip():
             previous_context_parts.append(request.previous_context.strip())
 
-        merged_previous_context = "\n\n".join(previous_context_parts) if previous_context_parts else request.previous_context
+        merged_previous_context = (
+            "\n\n".join(previous_context_parts)
+            if previous_context_parts
+            else request.previous_context
+        )
 
         return CodeXRequest(
             message=request.message,
+            session_id=request.session_id,
             mode=request.mode,
             language=enriched_language,
             code=artifact.code,