From e0cedac1c932df3055431a3f4f6c4ea6b0f56ad0 Mon Sep 17 00:00:00 2001 From: Lei Zhang Date: Wed, 16 Jul 2025 01:04:28 +0000 Subject: [PATCH] fix mle memory query --- mle/agents/chat.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) 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(