Explorar el Código

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

limbo hace 2 años
padre
commit
85008ff153
Se han modificado 1 ficheros con 5 adiciones y 2 borrados
  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:
     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):