diff --git a/mle/agents/chat.py b/mle/agents/chat.py
index b544101..a838c6d 100644
--- a/mle/agents/chat.py
+++ b/mle/agents/chat.py
@@ -124,10 +124,10 @@ def chat(self, user_prompt):
             user_prompt += f"""
             \nThese reference files and their snippets may be useful for the question:\n\n
             """
-
-            for t in query[0]:
-                snippet, metadata = t.get('text'), t.get('metadata')
-                user_prompt += f"**File**: {metadata.get('file')}\n**Snippet**: {snippet}\n"
+            if query:
+                for t in query[0]:
+                    snippet, metadata = t.get('text'), t.get('metadata')
+                    user_prompt += f"**File**: {metadata.get('file')}\n**Snippet**: {snippet}\n"
         self.chat_history.append({"role": "user", "content": user_prompt})
 
         for content in self.model.stream(