mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
e074b2d36f
Co-authored-by: openhands <openhands@all-hands.dev> Co-authored-by: Engel Nyst <enyst@users.noreply.github.com> Co-authored-by: Rohit Malhotra <rohitvinodmalhotra@gmail.com>
111 lines
3.8 KiB
Python
111 lines
3.8 KiB
Python
from openhands.core.config import LLMConfig
|
|
from openhands.integrations.provider import ProviderType
|
|
from openhands.resolver.interfaces.bitbucket import (
|
|
BitbucketIssueHandler,
|
|
BitbucketPRHandler,
|
|
)
|
|
from openhands.resolver.interfaces.github import GithubIssueHandler, GithubPRHandler
|
|
from openhands.resolver.interfaces.gitlab import GitlabIssueHandler, GitlabPRHandler
|
|
from openhands.resolver.interfaces.issue_definitions import (
|
|
ServiceContextIssue,
|
|
ServiceContextPR,
|
|
)
|
|
|
|
|
|
class IssueHandlerFactory:
|
|
def __init__(
|
|
self,
|
|
owner: str,
|
|
repo: str,
|
|
token: str,
|
|
username: str,
|
|
platform: ProviderType,
|
|
base_domain: str,
|
|
issue_type: str,
|
|
llm_config: LLMConfig,
|
|
) -> None:
|
|
self.owner = owner
|
|
self.repo = repo
|
|
self.token = token
|
|
self.username = username
|
|
self.platform = platform
|
|
self.base_domain = base_domain
|
|
self.issue_type = issue_type
|
|
self.llm_config = llm_config
|
|
|
|
def create(self) -> ServiceContextIssue | ServiceContextPR:
|
|
if self.issue_type == 'issue':
|
|
if self.platform == ProviderType.GITHUB:
|
|
return ServiceContextIssue(
|
|
GithubIssueHandler(
|
|
self.owner,
|
|
self.repo,
|
|
self.token,
|
|
self.username,
|
|
self.base_domain,
|
|
),
|
|
self.llm_config,
|
|
)
|
|
elif self.platform == ProviderType.GITLAB:
|
|
return ServiceContextIssue(
|
|
GitlabIssueHandler(
|
|
self.owner,
|
|
self.repo,
|
|
self.token,
|
|
self.username,
|
|
self.base_domain,
|
|
),
|
|
self.llm_config,
|
|
)
|
|
elif self.platform == ProviderType.BITBUCKET:
|
|
return ServiceContextIssue(
|
|
BitbucketIssueHandler(
|
|
self.owner,
|
|
self.repo,
|
|
self.token,
|
|
self.username,
|
|
self.base_domain,
|
|
),
|
|
self.llm_config,
|
|
)
|
|
else:
|
|
raise ValueError(f'Unsupported platform: {self.platform}')
|
|
elif self.issue_type == 'pr':
|
|
if self.platform == ProviderType.GITHUB:
|
|
return ServiceContextPR(
|
|
GithubPRHandler(
|
|
self.owner,
|
|
self.repo,
|
|
self.token,
|
|
self.username,
|
|
self.base_domain,
|
|
),
|
|
self.llm_config,
|
|
)
|
|
elif self.platform == ProviderType.GITLAB:
|
|
return ServiceContextPR(
|
|
GitlabPRHandler(
|
|
self.owner,
|
|
self.repo,
|
|
self.token,
|
|
self.username,
|
|
self.base_domain,
|
|
),
|
|
self.llm_config,
|
|
)
|
|
elif self.platform == ProviderType.BITBUCKET:
|
|
return ServiceContextPR(
|
|
BitbucketPRHandler(
|
|
self.owner,
|
|
self.repo,
|
|
self.token,
|
|
self.username,
|
|
self.base_domain,
|
|
),
|
|
self.llm_config,
|
|
)
|
|
else:
|
|
raise ValueError(f'Unsupported platform: {self.platform}')
|
|
else:
|
|
raise ValueError(f'Invalid issue type: {self.issue_type}')
|