Browse Source

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

limbo 2 years ago
parent
commit
85008ff153
1 changed files with 5 additions and 2 deletions
  1. 5 2
      embedchain/embedchain.py

+ 5 - 2
embedchain/embedchain.py

@@ -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):