diff --git a/README.md b/README.md index 4636f841..d9d7c7da 100644 --- a/README.md +++ b/README.md @@ -332,6 +332,14 @@ Resets the database and deletes all embeddings. Irreversible. Requires reinitial app.reset() ``` +### Count +Counts the number of embeddings (chunks) in the database. + +``` +print(app.count()) +# returns: 481 +``` + # How does it work? Creating a chat bot over any dataset needs the following steps to happen diff --git a/embedchain/embedchain.py b/embedchain/embedchain.py index 2835171e..d3cf0ae4 100644 --- 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): """