Update vector_store_retriever.py

vector_store_retriever.py

@@ -157,6 +157,7 @@ def predict(message, temperature=0.9, max_new_tokens=512, top_p=0.6, repetition_
     for token_entry in token_list:
         if token_entry:
             try:
+                # Handle missing 'id' key gracefully
                 token_id = token_entry.get('id', None)
                 token_text = token_entry.get('text', None)
 
@@ -170,9 +171,10 @@ def predict(message, temperature=0.9, max_new_tokens=512, top_p=0.6, repetition_
                     pass
 
             except KeyError as e:
-                gr.Warning(f"KeyError: {e} occurred for token entry: {token_entry}")
+                print(f"KeyError: {e} occurred for token entry: {token_entry}")
                 continue
 
+
 class TextGeneratorTool(Tool):
     name = "vector_retriever"
     description = "This tool searches in a vector store based on a given prompt."