mirror of
https://github.com/awslabs/amazon-bedrock-agentcore-samples.git
synced 2025-09-08 20:50:46 +00:00
102 lines
3.3 KiB
Python
102 lines
3.3 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Simple cleanup script for BedrockAgentCore resources.
|
|
Run this periodically to clean up AWS resources and avoid costs.
|
|
"""
|
|
|
|
import boto3
|
|
import sys
|
|
from datetime import datetime, timedelta
|
|
|
|
def cleanup_browsers(region='us-west-2'):
|
|
"""Delete all BedrockAgentCore browsers to stop charges."""
|
|
print("🧹 Cleaning up browsers...")
|
|
|
|
try:
|
|
# This is a placeholder - the actual API endpoint might differ
|
|
# You need to use the correct BedrockAgentCore control plane API
|
|
from bedrock_agentcore._utils.endpoints import get_control_plane_endpoint
|
|
|
|
control_client = boto3.client(
|
|
"bedrock-agentcore-control",
|
|
region_name=region,
|
|
endpoint_url=get_control_plane_endpoint(region)
|
|
)
|
|
|
|
response = control_client.list_browsers()
|
|
browsers = response.get('browsers', [])
|
|
|
|
for browser in browsers:
|
|
try:
|
|
control_client.delete_browser(browserId=browser['browserId'])
|
|
print(f" ✅ Deleted browser: {browser['browserId']}")
|
|
except Exception as e:
|
|
print(f" ❌ Failed to delete {browser['browserId']}: {e}")
|
|
|
|
if not browsers:
|
|
print(" ✓ No browsers to clean up")
|
|
|
|
except Exception as e:
|
|
print(f" ⚠️ Could not list browsers: {e}")
|
|
print(" Note: This might mean no browsers exist or API has changed")
|
|
|
|
def cleanup_old_s3_recordings(bucket_name, days_to_keep=7):
|
|
"""Delete S3 recordings older than specified days."""
|
|
print(f"🧹 Cleaning S3 recordings older than {days_to_keep} days...")
|
|
|
|
if not bucket_name:
|
|
print(" ⚠️ No S3 bucket specified")
|
|
return
|
|
|
|
try:
|
|
s3 = boto3.client('s3')
|
|
cutoff = datetime.now() - timedelta(days=days_to_keep)
|
|
|
|
response = s3.list_objects_v2(
|
|
Bucket=bucket_name,
|
|
Prefix='competitive_intel/'
|
|
)
|
|
|
|
if 'Contents' not in response:
|
|
print(" ✓ No recordings found")
|
|
return
|
|
|
|
old_objects = []
|
|
for obj in response['Contents']:
|
|
if obj['LastModified'].replace(tzinfo=None) < cutoff:
|
|
old_objects.append({'Key': obj['Key']})
|
|
|
|
if old_objects:
|
|
s3.delete_objects(
|
|
Bucket=bucket_name,
|
|
Delete={'Objects': old_objects[:1000]} # Max 1000 at a time
|
|
)
|
|
print(f" ✅ Deleted {len(old_objects)} old recordings")
|
|
else:
|
|
print(" ✓ No old recordings to delete")
|
|
|
|
except Exception as e:
|
|
print(f" ❌ Error: {e}")
|
|
|
|
if __name__ == "__main__":
|
|
import os
|
|
|
|
print("=" * 50)
|
|
print("BedrockAgentCore Resource Cleanup")
|
|
print("=" * 50)
|
|
|
|
# Get config from environment
|
|
region = os.environ.get('AWS_REGION', 'us-west-2')
|
|
bucket = os.environ.get('S3_RECORDING_BUCKET', '')
|
|
|
|
# Clean browsers (main cost driver)
|
|
cleanup_browsers(region)
|
|
|
|
# Clean old S3 recordings
|
|
if '--delete-old-recordings' in sys.argv:
|
|
cleanup_old_s3_recordings(bucket)
|
|
else:
|
|
print("\n💡 Tip: Add --delete-old-recordings to also clean S3")
|
|
|
|
print("\n✅ Cleanup complete")
|
|
print("=" * 50) |