Max retries per step configuration per org (#168)
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user