feat: add count to get document count in vector db (#164)
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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):
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user