chore: Tweak agent runner to be more general-purpose (#326)

Two changes:

  1. Change the variable name from TYPESCRIPT_SESSIONS_BUCKET -> AGENT_SESSIONS_BUCKET; this makes the agent code more general purpose than *just* TS
  2. Use a session prefix that is equal to the GH repo name - this enables different repositories (and forks) to not conflict with eachother

Co-authored-by: Mackenzie Zastrow <[email protected]>

Author: zastrowm

.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")

.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 }}