Fix: Solve IndexError when no relevant documents are found in the vector db. (#59)

This commit is contained in:
limbo
2023-07-05 00:48:56 +08:00
committed by GitHub
parent 8987a0c8d9
commit 85008ff153

View File

@@ -28,7 +28,7 @@ DB_DIR = os.path.join(ABS_PATH, "db")
class EmbedChain:
def __init__(self, db=None):
"""
Initializes the EmbedChain instance, sets up a vector DB client and
Initializes the EmbedChain instance, sets up a vector DB client and
creates a collection.
:param db: The instance of the VectorDB subclass.
@@ -181,7 +181,10 @@ class EmbedChain:
n_results=1,
)
result_formatted = self._format_result(result)
content = result_formatted[0][0].page_content
if result_formatted:
content = result_formatted[0][0].page_content
else:
content = ""
return content
def generate_prompt(self, input_query, context):