feat: add count to get document count in vector db (#164)

2023-07-07 11:01:14 +02:00
parent f3888cc7a8
commit 41a1dbab0e
2 changed files with 16 additions and 0 deletions
--- a/embedchain/embedchain.py
+++ b/embedchain/embedchain.py
@@ -298,6 +298,14 @@ class EmbedChain:
        prompt = self.generate_prompt(input_query, context)
        return prompt

+    def count(self):
+        """
+        Count the number of embeddings.
+
+        :return: The number of embeddings.
+        """
+        return self.collection.count()
+

    def reset(self):
        """