From 613ccb4c34c4398a43f4f230b4ef55ac28f18218 Mon Sep 17 00:00:00 2001 From: Ori Kotek Date: Tue, 21 Nov 2023 16:48:36 +0200 Subject: [PATCH] Add support for base_url in GitHub SDK --- pr_agent/git_providers/github_provider.py | 4 ++-- pr_agent/settings/configuration.toml | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/pr_agent/git_providers/github_provider.py b/pr_agent/git_providers/github_provider.py index c9abe2b1..634b8694 100644 --- a/pr_agent/git_providers/github_provider.py +++ b/pr_agent/git_providers/github_provider.py @@ -405,7 +405,7 @@ class GithubProvider(GitProvider): raise ValueError("GitHub app installation ID is required when using GitHub app deployment") auth = AppAuthentication(app_id=app_id, private_key=private_key, installation_id=self.installation_id) - return Github(app_auth=auth) + return Github(app_auth=auth, base_url=get_settings().github.base_url) if deployment_type == 'user': try: @@ -414,7 +414,7 @@ class GithubProvider(GitProvider): raise ValueError( "GitHub token is required when using user deployment. See: " "https://github.com/Codium-ai/pr-agent#method-2-run-from-source") from e - return Github(auth=Auth.Token(token)) + return Github(auth=Auth.Token(token), base_url=get_settings().github.base_url) def _get_repo(self): if hasattr(self, 'repo_obj') and \ diff --git a/pr_agent/settings/configuration.toml b/pr_agent/settings/configuration.toml index da69ad8e..38e96fd1 100644 --- a/pr_agent/settings/configuration.toml +++ b/pr_agent/settings/configuration.toml @@ -79,6 +79,7 @@ extra_instructions = "" # The type of deployment to create. Valid values are 'app' or 'user'. deployment_type = "user" ratelimit_retries = 5 +base_url = "https://api.github.com" [github_action] # auto_review = true # set as env var in .github/workflows/pr-agent.yaml