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

2023-07-05 00:48:56 +08:00
parent 8987a0c8d9
commit 85008ff153
1 changed files with 5 additions and 2 deletions
--- a/embedchain/embedchain.py
+++ b/embedchain/embedchain.py
@@ -181,7 +181,10 @@ class EmbedChain:
            n_results=1,
        )
        result_formatted = self._format_result(result)
+        if result_formatted:
            content = result_formatted[0][0].page_content
+        else:
+            content = ""
        return content
    
    def generate_prompt(self, input_query, context):