Update vector_store_retriever.py
Browse files
vector_store_retriever.py
CHANGED
@@ -157,6 +157,7 @@ def predict(message, temperature=0.9, max_new_tokens=512, top_p=0.6, repetition_
|
|
157 |
for token_entry in token_list:
|
158 |
if token_entry:
|
159 |
try:
|
|
|
160 |
token_id = token_entry.get('id', None)
|
161 |
token_text = token_entry.get('text', None)
|
162 |
|
@@ -170,9 +171,10 @@ def predict(message, temperature=0.9, max_new_tokens=512, top_p=0.6, repetition_
|
|
170 |
pass
|
171 |
|
172 |
except KeyError as e:
|
173 |
-
|
174 |
continue
|
175 |
|
|
|
176 |
class TextGeneratorTool(Tool):
|
177 |
name = "vector_retriever"
|
178 |
description = "This tool searches in a vector store based on a given prompt."
|
|
|
157 |
for token_entry in token_list:
|
158 |
if token_entry:
|
159 |
try:
|
160 |
+
# Handle missing 'id' key gracefully
|
161 |
token_id = token_entry.get('id', None)
|
162 |
token_text = token_entry.get('text', None)
|
163 |
|
|
|
171 |
pass
|
172 |
|
173 |
except KeyError as e:
|
174 |
+
print(f"KeyError: {e} occurred for token entry: {token_entry}")
|
175 |
continue
|
176 |
|
177 |
+
|
178 |
class TextGeneratorTool(Tool):
|
179 |
name = "vector_retriever"
|
180 |
description = "This tool searches in a vector store based on a given prompt."
|