Create a Collection

TaskingAI also allows for the creation of collections programmatically through its client SDK. This method is particularly useful for multi-tenant applications where collections are created dynamically for different users.

import taskingai

collection = taskingai.retrieval.create_collection(
    embedding_model_id="YOUR_EMBEDDING_MODEL_ID",
    capacity=1000,
)

The method will return a Collection object if the creation is successful.