Merge pull request #271 from gitautoai/wes

Enable create_header() to set media type
gitautoai · Aug 19, 2024 · a7f577b · a7f577b
2 parents b9d32ec + 4fc9469
commit a7f577b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/services/github/github_manager.py b/services/github/github_manager.py
@@ -295,10 +295,10 @@ def create_comment_on_issue_with_gitauto_button(payload: GitHubLabeledPayload) -
     return response.json()
 
 
-def create_headers(token: str) -> dict[str, str]:
+def create_headers(token: str, media_type: Optional[str] = "v3") -> dict[str, str]:
     """https://docs.github.com/en/rest/using-the-rest-api/getting-started-with-the-rest-api?apiVersion=2022-11-28#headers"""
     return {
-        "Accept": "application/vnd.github.v3+json",
+        "Accept": f"application/vnd.github.{media_type}+json",
         "Authorization": f"Bearer {token}",
         "User-Agent": GITHUB_APP_NAME,
         "X-GitHub-Api-Version": GITHUB_API_VERSION,