From b9013b7b2a878dd09816caefdb956a7f8630cdb7 Mon Sep 17 00:00:00 2001
From: Jon Michael Aanes <jonjmaa@gmail.com>
Date: Wed, 23 Apr 2025 22:24:41 +0200
Subject: [PATCH] Fixing types

---
 aider_gitea/__init__.py     | 10 +++++++---
 aider_gitea/gitea_client.py |  3 ++-
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/aider_gitea/__init__.py b/aider_gitea/__init__.py
index 4f356a5..2df0e46 100644
--- a/aider_gitea/__init__.py
+++ b/aider_gitea/__init__.py
@@ -103,7 +103,11 @@ class RepositoryConfig:
 class IssueResolution:
     success: bool
     pull_request_url: str | None = None
-    pull_request_id: str | None = None
+    pull_request_id: int | None = None
+
+    def __post_init__(self):
+        assert self.pull_request_id is None or isinstance(self.pull_request_id, int)
+        assert self.pull_request_url is None or isinstance(self.pull_request_url, str)
 
 
 def generate_branch_name(issue_number: str, issue_title: str) -> str:
@@ -283,8 +287,8 @@ def push_changes(
     # Extract PR number and URL if available
     return IssueResolution(
         True,
-        str(pr_response.get('number')),
         pr_response.get('html_url'),
+        int(pr_response.get('number')),
     )
 
 
@@ -542,7 +546,7 @@ def solve_issues_in_repository(
 
 def handle_pr_comments(
     repository_config,
-    pr_number,
+    pr_number: int,
     branch_name,
     repository_path,
     client,
diff --git a/aider_gitea/gitea_client.py b/aider_gitea/gitea_client.py
index 4133096..89e9498 100644
--- a/aider_gitea/gitea_client.py
+++ b/aider_gitea/gitea_client.py
@@ -187,11 +187,12 @@ class GiteaClient:
         self,
         owner: str,
         repo: str,
-        pr_number: str,
+        pr_number: int,
     ) -> list[dict]:
         """
         Fetch comments for a pull request.
         """
+        assert isinstance(pr_number, int)
         url = f'{self.gitea_url}/repos/{owner}/{repo}/pulls/{pr_number}/comments'
         response = self.session.get(url)
         response.raise_for_status()