diff --git a/app/plugins/builtins/ai_assistant/routes.py b/app/plugins/builtins/ai_assistant/routes.py
index 1a91234..d367355 100644
--- a/app/plugins/builtins/ai_assistant/routes.py
+++ b/app/plugins/builtins/ai_assistant/routes.py
@@ -184,8 +184,9 @@ async def create_model(
         tenant_id=tenant_id,
     )
     db.add(model)
-    await db.commit()
+    await db.flush()
     await db.refresh(model)
+    await db.commit()
     return model_to_response(model)