diff --git a/api/services/prompt_builder.py b/api/services/prompt_builder.py
index 81ac11c..8dbd850 100644
--- a/api/services/prompt_builder.py
+++ b/api/services/prompt_builder.py
@@ -65,6 +65,8 @@ def build_case_prompt(context: dict) -> str:
     recent_notes = context.get("recentNotes", [])
     templates = context.get("availableTemplates", [])
     documents = context.get("documents", {})
+    if isinstance(documents, list):
+        documents = {"totalFiles": len(documents)}
     recent_calls = context.get("recentCalls", [])
     case_memory = context.get("caseMemory", {})
     assistant_rules = context.get("assistantRules", [])