diff --git a/.github/scripts/python/agent_runner.py b/.github/scripts/python/agent_runner.py index 425d2607..db10cead 100644 --- a/.github/scripts/python/agent_runner.py +++ b/.github/scripts/python/agent_runner.py @@ -109,13 +109,14 @@ def run_agent(query: str): system_prompt = os.getenv("INPUT_SYSTEM_PROMPT", DEFAULT_SYSTEM_PROMPT) session_id = os.getenv("SESSION_ID") s3_bucket = os.getenv("S3_SESSION_BUCKET") + s3_prefix = os.getenv("GITHUB_REPOSITORY", "") if s3_bucket and session_id: print(f"🤖 Using session manager with session ID: {session_id}") session_manager = S3SessionManager( session_id=session_id, bucket=s3_bucket, - prefix="", + prefix=s3_prefix, ) else: raise ValueError("Both SESSION_ID and S3_SESSION_BUCKET must be set") diff --git a/.github/workflows/strands-command.yml b/.github/workflows/strands-command.yml index 7ce644dc..803f19e4 100644 --- a/.github/workflows/strands-command.yml +++ b/.github/workflows/strands-command.yml @@ -133,7 +133,7 @@ jobs: session_id: ${{ needs.setup-and-process.outputs.session_id }} task_prompt: ${{ needs.setup-and-process.outputs.prompt }} aws_role_arn: ${{ secrets.AWS_ROLE_ARN }} - sessions_bucket: ${{ secrets.TYPESCRIPT_SESSIONS_BUCKET }} + sessions_bucket: ${{ secrets.AGENT_SESSIONS_BUCKET }} write_permission: 'false' ref: ${{ needs.setup-and-process.outputs.branch }}