Max retries per step configuration per org (#168)

This commit is contained in:
Kerem Yilmaz
2024-04-08 16:58:45 -07:00
committed by GitHub
parent 8e2aaa95d2
commit ffe917f2b5
7 changed files with 47 additions and 6 deletions

View File

@@ -418,12 +418,14 @@ class AgentDB:
organization_name: str,
webhook_callback_url: str | None = None,
max_steps_per_run: int | None = None,
max_retries_per_step: int | None = None,
) -> Organization:
async with self.Session() as session:
org = OrganizationModel(
organization_name=organization_name,
webhook_callback_url=webhook_callback_url,
max_steps_per_run=max_steps_per_run,
max_retries_per_step=max_retries_per_step,
)
session.add(org)
await session.commit()