Python: limit the number of retrieved documents when using AzureCosmosDBNoSQLCollection

[vslepakov]

Hi,

I am using AzureCosmosDBNoSQLCollection to store and retrieve chat history from CosmosDB.
The whole chat history should be persisted BUT to rehydrate the chat history, I'd like to limit the amount of documents retrieved using collection.get(). Is there a way to achieve this?

Thanks!

[hellodhana]

You can restrict using Cosmos Query with LIMIT statement ordered by date and time.

Use container.query_items methods

[vslepakov]

Thanks, not sure I am following. Do you have an example of how to do this using AzureCosmosDBNoSQLCollection?

[westey-m]

@vslepakov, are you doing a vector search? If so, see here for top and skip support:
https://learn.microsoft.com/en-us/semantic-kernel/concepts/vector-store-connectors/vector-search?pivots=programming-language-python#top-and-skip-1

If you are doing a get, then you have to supply the set of record ids, so this limits the result set already to only include the records you request.

[vslepakov]

Thanks @westey-m, that makes sense!