update pipeline

This commit is contained in:
sarbjitgrewal
2023-09-15 12:38:34 +05:30
parent 4043dfff9e
commit 978c56c128

View File

@ -1,6 +1,6 @@
import os import os
from pr_agent.agent.pr_agent import PRAgent from pr_agent.agent.pr_agent import PRAgent
import os from pr_agent.config_loader import get_settings
from pr_agent.tools.pr_reviewer import PRReviewer from pr_agent.tools.pr_reviewer import PRReviewer
import asyncio import asyncio
@ -8,9 +8,19 @@ async def run_action():
pull_request_id = os.environ.get("BITBUCKET_PR_ID", '') pull_request_id = os.environ.get("BITBUCKET_PR_ID", '')
slug = os.environ.get("BITBUCKET_REPO_SLUG", '') slug = os.environ.get("BITBUCKET_REPO_SLUG", '')
workspace = os.environ.get("BITBUCKET_WORKSPACE", '') workspace = os.environ.get("BITBUCKET_WORKSPACE", '')
bearer_token = os.environ.get('BITBUCKET_BEARER_TOKEN', None)
# Check if required environment variables are set
if not bearer_token:
print("BITBUCKET_BEARER_TOKEN not set")
# Set the environment variables in the settings
get_settings().set("BITBUCKET.BEARER_TOKEN", bearer_token)
if pull_request_id and slug and workspace: if pull_request_id and slug and workspace:
pr_url = f"https://bitbucket.org/{workspace}/{slug}/pull-requests/{pull_request_id}" pr_url = f"https://bitbucket.org/{workspace}/{slug}/pull-requests/{pull_request_id}"
await PRReviewer(pr_url).run() await PRReviewer(pr_url).run()
if __name__ == "__main__": if __name__ == "__main__":
asyncio.run(run_action()) asyncio.run(run_action())