klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
limbo
2023-07-05 00:48:56 +08:00
committed by GitHub
parent 8987a0c8d9
commit 85008ff153
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
embedchain/embedchain.py
View File

@@ -28,7 +28,7 @@ DB_DIR = os.path.join(ABS_PATH, "db")
class EmbedChain: class EmbedChain:
def __init__(self, db=None): 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. creates a collection.
:param db: The instance of the VectorDB subclass. :param db: The instance of the VectorDB subclass.
@@ -181,7 +181,10 @@ class EmbedChain:
n_results=1, n_results=1,
) )
result_formatted = self._format_result(result) 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 return content
def generate_prompt(self, input_query, context): def generate_prompt(self, input_query, context):