Spaces:
Sleeping
Sleeping
File size: 1,541 Bytes
106317a 2713f86 106317a 3a0b13e 3c2b0d5 6b2fdc4 0baf4dd ac1fe52 f78715f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | import motor.motor_asyncio
from .config import config
client = motor.motor_asyncio.AsyncIOMotorClient(
config.MONGO_URL,
maxPoolSize=100,
minPoolSize=5,
maxIdleTimeMS=60_000,
waitQueueTimeoutMS=10_000,
serverSelectionTimeoutMS=5_000,
)
database = client[config.KITTYKAT_AGENT_CLUSTER]
kittykat_agent_brands_collection = database["brands"]
kittykat_agent_users_collection = database["users"]
kittykat_agent_roles_collection = database["roles"]
kittykat_agent_gallery_collection = database["gallery"]
kittykat_agent_queue_collection = database["queue"]
kittykat_agent_product_categories_collection = database["product_categories"]
kittykat_agent_notification_queue_collection = database["notification_queue"]
kittykat_agent_notification_collection = database["notification"]
ask_kittykat_tasklists_collection = database["ask_kittykat_tasklists"]
models_collection = database["models"]
config_collection = database["config"]
credit_transactions_collection = database["credit_transactions"]
kittykat_agent_audit_logs_collection = database["audit_logs"]
kittykat_agent_teams_collection = database["teams"]
kittykat_agent_product_sku_collection = database["product_sku"]
kittykat_agent_campaign_concepts_collection = database["campaign_concepts"]
kittykat_agent_brand_prompts_collection = database["brand_prompts"]
kittykat_agent_shot_types_collection = database["shot_types"]
kittykat_agent_generation_parameters_collection = database["a2i_generation_parameters"]
a2i_generations_collection = database["generations"]
|