Herman/Yiqun
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

remove unused import

Browse Source
This commit is contained in:
Laura Abro
2025-04-25 07:28:02 -03:00
parent 44ba266ab5
commit 2a31027ad8
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
worker/orca-agent/src/workflows/repoSummarizerAudit/workflow.py
View File

@ -12,7 +12,6 @@ from prometheus_swarm.workflows.utils import (
cleanup_repository,
get_current_files,
)
from git import Repo
class Task:
@ -70,9 +69,15 @@ class repoSummarizerAuditWorkflow(Workflow):
# Check required environment variables and validate GitHub auth
check_required_env_vars(["GITHUB_TOKEN", "GITHUB_USERNAME"])
validate_github_auth(os.getenv("GITHUB_TOKEN"), os.getenv("GITHUB_USERNAME"))
self.context["repo_url"] = f"https://github.com/{self.context['repo_owner']}/{self.context['repo_name']}"
self.context["repo_url"] = (
f"https://github.com/{self.context['repo_owner']}/{self.context['repo_name']}"
)
# Set up repository directory
setup_result = setup_repository(self.context["repo_url"], github_token=os.getenv("GITHUB_TOKEN"), github_username=os.getenv("GITHUB_USERNAME"))
setup_result = setup_repository(
self.context["repo_url"],
github_token=os.getenv("GITHUB_TOKEN"),
github_username=os.getenv("GITHUB_USERNAME"),
)
if not setup_result["success"]:
raise Exception(f"Failed to set up repository: {setup_result['message']}")
@ -85,9 +90,7 @@ class repoSummarizerAuditWorkflow(Workflow):
# Enter repo directory
os.chdir(self.context["repo_path"])
gh = Github(self.context["github_token"])
repo = gh.get_repo(
f"{self.context['repo_owner']}/{self.context['repo_name']}"
)
repo = gh.get_repo(f"{self.context['repo_owner']}/{self.context['repo_name']}")
pr = repo.get_pull(self.context["pr_number"])
self.context["pr"] = pr
# Add remote for PR's repository and fetch the branch