Skip to content

Commit 07212b6

Browse files
committed
Fix worker runtime retention compatibility
1 parent 0d040c4 commit 07212b6

2 files changed

Lines changed: 7 additions & 1 deletion

File tree

‎src/select_ai/agent/a2a/session_process.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -349,7 +349,7 @@ async def _run_session_process(
349349
session_id: str,
350350
owner: str,
351351
team_name: str,
352-
conversation_retention_days: int,
352+
conversation_retention_days: int = 7,
353353
) -> None:
354354
"""Open one async connection and execute A2A operations."""
355355
import select_ai

‎tests/a2a/test_worker_runtime.py‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -347,6 +347,9 @@ async def fetch(team_name):
347347
initialized.append("team")
348348

349349
class Store:
350+
def __init__(self, retention_days=7):
351+
self.retention_days = retention_days
352+
350353
async def initialize(self):
351354
initialized.append(self)
352355

@@ -393,6 +396,9 @@ async def fetch(team_name):
393396
raise RuntimeError("team does not exist")
394397

395398
class Store:
399+
def __init__(self, retention_days=7):
400+
self.retention_days = retention_days
401+
396402
async def initialize(self):
397403
initialized.append(self)
398404

0 commit comments

Comments
 (0)