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()
|
app.reset()
|
||||||
```
|
```
|
||||||
|
|
||||||
|
### Count
|
||||||
|
Counts the number of embeddings (chunks) in the database.
|
||||||
|
|
||||||
|
```
|
||||||
|
print(app.count())
|
||||||
|
# returns: 481
|
||||||
|
```
|
||||||
|
|
||||||
# How does it work?
|
# How does it work?
|
||||||
|
|
||||||
Creating a chat bot over any dataset needs the following steps to happen
|
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)
|
prompt = self.generate_prompt(input_query, context)
|
||||||
return prompt
|
return prompt
|
||||||
|
|
||||||
|
def count(self):
|
||||||
|
"""
|
||||||
|
Count the number of embeddings.
|
||||||
|
|
||||||
|
:return: The number of embeddings.
|
||||||
|
"""
|
||||||
|
return self.collection.count()
|
||||||
|
|
||||||
|
|
||||||
def reset(self):
|
def reset(self):
|
||||||
"""
|
"""
|
||||||
|
|||||||
Reference in New Issue
Block a user