chore: improve pr-agent maintenance path

tests/e2e_tests/e2e_utils.py

@@ -1,12 +1,12 @@
-FILE_PATH = "pr_agent/cli_pip.py"
+FILE_PATH: str = "pr_agent/cli_pip.py"
 
 PR_HEADER_START_WITH = '### **User description**\nupdate cli_pip.py\n\n\n___\n\n### **PR Type**'
 REVIEW_START_WITH = '## PR Reviewer Guide 🔍\n\n<table>\n<tr><td>⏱️&nbsp;<strong>Estimated effort to review</strong>:'
 IMPROVE_START_WITH_REGEX_PATTERN = r'^## PR Code Suggestions ✨\n\n<!-- [a-z0-9]+ -->\n\n<table><thead><tr><td>Category</td>'
 
-NUM_MINUTES = 5
+NUM_MINUTES: int = 5
 
-NEW_FILE_CONTENT = """\
+NEW_FILE_CONTENT: str = """\
 from pr_agent import cli
 from pr_agent.config_loader import get_settings
 

tests/health_test/main.py

@@ -16,7 +16,7 @@
 setup_logger(log_level)
 
 
-async def run_async():
+async def run_async() -> None:
     pr_url = os.getenv('TEST_PR_URL', 'https://github.com/Codium-ai/pr-agent/pull/1385')
 
     get_settings().set("config.git_provider", "github")
@@ -30,7 +30,9 @@ async def run_async():
         original_settings = copy.deepcopy(get_settings())
         await agent.handle_request(pr_url, ['describe'])
         pr_header_body = dict(get_settings().data)['artifact']
-        assert pr_header_body.startswith('###') and 'PR Type' in pr_header_body and 'Description' in pr_header_body
+        assert isinstance(pr_header_body, str), f"Expected artifact to be str, got {type(pr_header_body).__name__}"
+        assert pr_header_body.startswith("###") and "PR Type" in pr_header_body and "Description" in pr_header_body, \
+            "PR description artifact missing expected sections"
         context['settings'] = copy.deepcopy(original_settings) # Restore settings state after each test to prevent test interference
         get_logger().info("PR description generated successfully\n")
 
@@ -39,7 +41,9 @@ async def run_async():
         original_settings = copy.deepcopy(get_settings())
         await agent.handle_request(pr_url, ['review'])
         pr_review_body = dict(get_settings().data)['artifact']
-        assert pr_review_body.startswith('##') and 'PR Reviewer Guide' in pr_review_body
+        assert isinstance(pr_review_body, str), f"Expected artifact to be str, got {type(pr_review_body).__name__}"
+        assert pr_review_body.startswith("##") and "PR Reviewer Guide" in pr_review_body, \
+            "PR review artifact missing expected header"
         context['settings'] = copy.deepcopy(original_settings)  # Restore settings state after each test to prevent test interference
         get_logger().info("PR review generated successfully\n")
 
@@ -48,7 +52,9 @@ async def run_async():
         original_settings = copy.deepcopy(get_settings())
         await agent.handle_request(pr_url, ['improve'])
         pr_improve_body = dict(get_settings().data)['artifact']
-        assert pr_improve_body.startswith('##') and 'PR Code Suggestions' in pr_improve_body
+        assert isinstance(pr_improve_body, str), f"Expected artifact to be str, got {type(pr_improve_body).__name__}"
+        assert pr_improve_body.startswith("##") and "PR Code Suggestions" in pr_improve_body, \
+            "PR improve artifact missing expected header"
         context['settings'] = copy.deepcopy(original_settings)  # Restore settings state after each test to prevent test interference
         get_logger().info("PR improvements generated successfully\n")
 
@@ -59,7 +65,7 @@ async def run_async():
         raise e
 
 
-def run():
+def run() -> None:
     with request_cycle_context({}):
         context['settings'] = copy.deepcopy(global_settings)
         asyncio.run(run_async())