bakery-admin/bakery-ia
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix DB issue 2s

Browse Source
This commit is contained in:
Urtzi Alfaro
2025-09-30 21:58:10 +02:00
parent 147893015e
commit 7cc4b957a5
77 changed files with 4385 additions and 1211 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
services/tenant/app/repositories/tenant_repository.py
View File

@@ -55,8 +55,8 @@ class TenantRepository(TenantBaseRepository):
tenant_data["is_active"] = True
if "subscription_tier" not in tenant_data:
tenant_data["subscription_tier"] = "basic"
if "model_trained" not in tenant_data:
tenant_data["model_trained"] = False
if "ml_model_trained" not in tenant_data:
tenant_data["ml_model_trained"] = False
# Create tenant
tenant = await self.create(tenant_data)
@@ -159,26 +159,26 @@ class TenantRepository(TenantBaseRepository):
async def update_tenant_model_status(
self,
tenant_id: str,
model_trained: bool,
ml_model_trained: bool,
last_training_date: datetime = None
) -> Optional[Tenant]:
"""Update tenant model training status"""
try:
update_data = {
"model_trained": model_trained,
"ml_model_trained": ml_model_trained,
"updated_at": datetime.utcnow()
}
if last_training_date:
update_data["last_training_date"] = last_training_date
elif model_trained:
elif ml_model_trained:
update_data["last_training_date"] = datetime.utcnow()
updated_tenant = await self.update(tenant_id, update_data)
logger.info("Tenant model status updated",
tenant_id=tenant_id,
model_trained=model_trained,
ml_model_trained=ml_model_trained,
last_training_date=last_training_date)
return updated_tenant
@@ -306,8 +306,8 @@ class TenantRepository(TenantBaseRepository):
# Get model training statistics
model_query = text("""
SELECT
COUNT(CASE WHEN model_trained = true THEN 1 END) as trained_count,
COUNT(CASE WHEN model_trained = false THEN 1 END) as untrained_count,
COUNT(CASE WHEN ml_model_trained = true THEN 1 END) as trained_count,
COUNT(CASE WHEN ml_model_trained = false THEN 1 END) as untrained_count,
AVG(EXTRACT(EPOCH FROM (NOW() - last_training_date))/86400) as avg_days_since_training
FROM tenants
WHERE is_active = true
@@ -407,4 +407,4 @@ class TenantRepository(TenantBaseRepository):
async def activate_tenant(self, tenant_id: str) -> Optional[Tenant]:
"""Activate a tenant"""
return await self.activate_record(tenant_id)
return await self.activate_record(tenant_id)