mirror of
https://github.com/All-Hands-AI/OpenHands.git
synced 2026-04-29 03:00:45 -04:00
Enforce modern Python typing annotations with Ruff (#8296)
Co-authored-by: openhands <openhands@all-hands.dev>
This commit is contained in:
@@ -6,6 +6,11 @@ from typing import Any
|
||||
import httpx
|
||||
from pydantic import SecretStr
|
||||
|
||||
from openhands.core.logger import openhands_logger as logger
|
||||
from openhands.integrations.github.queries import (
|
||||
suggested_task_issue_graphql_query,
|
||||
suggested_task_pr_graphql_query,
|
||||
)
|
||||
from openhands.integrations.service_types import (
|
||||
BaseGitService,
|
||||
Branch,
|
||||
@@ -20,9 +25,6 @@ from openhands.integrations.service_types import (
|
||||
)
|
||||
from openhands.server.types import AppMode
|
||||
from openhands.utils.import_utils import get_impl
|
||||
from openhands.integrations.github.queries import suggested_task_pr_graphql_query, suggested_task_issue_graphql_query
|
||||
from datetime import datetime
|
||||
from openhands.core.logger import openhands_logger as logger
|
||||
|
||||
|
||||
class GitHubService(BaseGitService, GitService):
|
||||
@@ -291,7 +293,7 @@ class GitHubService(BaseGitService, GitService):
|
||||
Returns:
|
||||
- PRs authored by the user.
|
||||
- Issues assigned to the user.
|
||||
|
||||
|
||||
Note: Queries are split to avoid timeout issues.
|
||||
"""
|
||||
# Get user info to use in queries
|
||||
@@ -301,9 +303,11 @@ class GitHubService(BaseGitService, GitService):
|
||||
variables = {'login': login}
|
||||
|
||||
try:
|
||||
pr_response = await self.execute_graphql_query(suggested_task_pr_graphql_query, variables)
|
||||
pr_response = await self.execute_graphql_query(
|
||||
suggested_task_pr_graphql_query, variables
|
||||
)
|
||||
pr_data = pr_response['data']['user']
|
||||
|
||||
|
||||
# Process pull requests
|
||||
for pr in pr_data['pullRequests']['nodes']:
|
||||
repo_name = pr['repository']['nameWithOwner']
|
||||
@@ -341,16 +345,22 @@ class GitHubService(BaseGitService, GitService):
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
except Exception as e:
|
||||
logger.info(f"Error fetching suggested task for PRs: {e}",
|
||||
extra={'signal': 'github_suggested_tasks', 'user_id': self.external_auth_id})
|
||||
|
||||
logger.info(
|
||||
f'Error fetching suggested task for PRs: {e}',
|
||||
extra={
|
||||
'signal': 'github_suggested_tasks',
|
||||
'user_id': self.external_auth_id,
|
||||
},
|
||||
)
|
||||
|
||||
try:
|
||||
# Execute issue query
|
||||
issue_response = await self.execute_graphql_query(suggested_task_issue_graphql_query, variables)
|
||||
issue_response = await self.execute_graphql_query(
|
||||
suggested_task_issue_graphql_query, variables
|
||||
)
|
||||
issue_data = issue_response['data']['user']
|
||||
|
||||
|
||||
# Process issues
|
||||
for issue in issue_data['issues']['nodes']:
|
||||
repo_name = issue['repository']['nameWithOwner']
|
||||
@@ -365,10 +375,15 @@ class GitHubService(BaseGitService, GitService):
|
||||
)
|
||||
|
||||
return tasks
|
||||
|
||||
|
||||
except Exception as e:
|
||||
logger.info(f"Error fetching suggested task for issues: {e}",
|
||||
extra={'signal': 'github_suggested_tasks', 'user_id': self.external_auth_id})
|
||||
logger.info(
|
||||
f'Error fetching suggested task for issues: {e}',
|
||||
extra={
|
||||
'signal': 'github_suggested_tasks',
|
||||
'user_id': self.external_auth_id,
|
||||
},
|
||||
)
|
||||
|
||||
return tasks
|
||||
|
||||
|
||||
@@ -29,7 +29,7 @@ suggested_task_pr_graphql_query = """
|
||||
}
|
||||
"""
|
||||
|
||||
|
||||
|
||||
suggested_task_issue_graphql_query = """
|
||||
query GetUserIssues($login: String!) {
|
||||
user(login: $login) {
|
||||
|
||||
Reference in New Issue
Block a user