Update code

code_engine.py

@@ -1,12 +1,13 @@
 import time
 
-from schemas import CodeXRequest, CodeXResponse, CodeTaskType
+from schemas import CodeXRequest, CodeXResponse, CodeTaskType, ResponseMeta
 from code_router import detect_task_type
 from prompt_builder import build_prompt
 from model_client import model_client
 from response_formatter import build_response, build_error_response
 from code_retriever import retrieve_code_evidence
 from context_manager import context_manager
+from scope_router import decide_scope, ScopeRoute
 from config import settings
 
 
@@ -30,10 +31,55 @@ def get_retrieved_evidence(task_type: CodeTaskType, request: CodeXRequest):
     return retrieve_code_evidence(task_type, request)
 
 
+def build_scope_response(
+    route: ScopeRoute,
+    message: str,
+    processing_time_ms: int,
+) -> CodeXResponse:
+    warnings = []
+
+    if route == ScopeRoute.RESTRICTED:
+        warnings.append("Restricted or unsafe request detected.")
+    elif route == ScopeRoute.IMAGE:
+        warnings.append("Image-related request detected.")
+    elif route == ScopeRoute.NON_CODE:
+        warnings.append("Out-of-scope non-code request detected.")
+    elif route == ScopeRoute.GREETING:
+        warnings.append("Greeting request detected.")
+    else:
+        warnings.append("Unsupported or unclear request detected.")
+
+    return CodeXResponse(
+        answer=message,
+        task_type=CodeTaskType.UNKNOWN,
+        code_output=None,
+        explanation=message,
+        warnings=warnings,
+        sources=[],
+        needs_clarification=False,
+        meta=ResponseMeta(
+            used_model="scope_router",
+            fallback_used=False,
+            retrieval_used=False,
+            source_count=0,
+            processing_time_ms=processing_time_ms,
+        ),
+    )
+
+
 def process_codex_request(request: CodeXRequest) -> CodeXResponse:
     start_time = time.perf_counter()
 
     try:
+        scope_decision = decide_scope(request)
+        if not scope_decision.should_continue_to_codex:
+            processing_time_ms = int((time.perf_counter() - start_time) * 1000)
+            return build_scope_response(
+                route=scope_decision.route,
+                message=scope_decision.message,
+                processing_time_ms=processing_time_ms,
+            )
+
         enriched_request = context_manager.enrich_request_with_context(
             request=request,
             session_id=request.session_id,