mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
890796cc9d
Co-authored-by: openhands <openhands@all-hands.dev> Co-authored-by: Xingyao Wang <xingyao@all-hands.dev> Co-authored-by: Robert Brennan <accounts@rbren.io>
142 lines
4.7 KiB
Python
142 lines
4.7 KiB
Python
import re
|
|
from typing import Annotated
|
|
from pydantic import Field
|
|
from fastmcp import FastMCP
|
|
from fastmcp.server.dependencies import get_http_request
|
|
from openhands.integrations.github.github_service import GithubServiceImpl
|
|
from openhands.integrations.gitlab.gitlab_service import GitLabServiceImpl
|
|
from openhands.integrations.provider import ProviderToken
|
|
from openhands.integrations.service_types import ProviderType
|
|
from openhands.server.shared import ConversationStoreImpl, config
|
|
from openhands.server.user_auth import get_access_token, get_provider_tokens, get_user_id
|
|
from openhands.core.logger import openhands_logger as logger
|
|
from openhands.storage.data_models.conversation_metadata import ConversationMetadata
|
|
|
|
mcp_server = FastMCP('mcp')
|
|
|
|
|
|
async def save_pr_metadata(user_id: str, conversation_id: str, tool_result: str) -> None:
|
|
conversation_store = await ConversationStoreImpl.get_instance(config, user_id)
|
|
conversation: ConversationMetadata = await conversation_store.get_metadata(conversation_id)
|
|
|
|
|
|
pull_pattern = r"pull/(\d+)"
|
|
merge_request_pattern = r"merge_requests/(\d+)"
|
|
|
|
# Check if the tool_result contains the PR number
|
|
pr_number = None
|
|
match_pull = re.search(pull_pattern, tool_result)
|
|
match_merge_request = re.search(merge_request_pattern, tool_result)
|
|
|
|
if match_pull:
|
|
pr_number = int(match_pull.group(1))
|
|
elif match_merge_request:
|
|
pr_number = int(match_merge_request.group(1))
|
|
|
|
|
|
if pr_number:
|
|
conversation.pr_number.append(pr_number)
|
|
await conversation_store.save_metadata(conversation)
|
|
|
|
@mcp_server.tool()
|
|
async def create_pr(
|
|
repo_name: Annotated[
|
|
str, Field(description='GitHub repository ({{owner}}/{{repo}})')
|
|
],
|
|
source_branch: Annotated[str, Field(description='Source branch on repo')],
|
|
target_branch: Annotated[str, Field(description='Target branch on repo')],
|
|
title: Annotated[str, Field(description='PR Title')],
|
|
body: Annotated[str | None, Field(description='PR body')]
|
|
) -> str:
|
|
"""Open a draft PR in GitHub"""
|
|
|
|
logger.info('Calling OpenHands MCP create_pr')
|
|
|
|
request = get_http_request()
|
|
headers = request.headers
|
|
conversation_id = headers.get('X-OpenHands-Conversation-ID', None)
|
|
|
|
provider_tokens = await get_provider_tokens(request)
|
|
access_token = await get_access_token(request)
|
|
user_id = await get_user_id(request)
|
|
|
|
github_token = provider_tokens.get(ProviderType.GITHUB, ProviderToken()) if provider_tokens else ProviderToken()
|
|
|
|
github_service = GithubServiceImpl(
|
|
user_id=github_token.user_id,
|
|
external_auth_id=user_id,
|
|
external_auth_token=access_token,
|
|
token=github_token.token,
|
|
base_domain=github_token.host
|
|
)
|
|
|
|
try:
|
|
response = await github_service.create_pr(
|
|
repo_name=repo_name,
|
|
source_branch=source_branch,
|
|
target_branch=target_branch,
|
|
title=title,
|
|
body=body
|
|
)
|
|
|
|
if conversation_id and user_id:
|
|
await save_pr_metadata(user_id, conversation_id, response)
|
|
|
|
except Exception as e:
|
|
response = str(e)
|
|
|
|
return response
|
|
|
|
|
|
|
|
@mcp_server.tool()
|
|
async def create_mr(
|
|
id: Annotated[
|
|
int | str, Field(description='GitLab repository (ID or URL-encoded path of the project)')
|
|
],
|
|
source_branch: Annotated[str, Field(description='Source branch on repo')],
|
|
target_branch: Annotated[str, Field(description='Target branch on repo')],
|
|
title: Annotated[str, Field(description='MR Title')],
|
|
description: Annotated[str | None, Field(description='MR description')]
|
|
) -> str:
|
|
"""Open a draft MR in GitLab"""
|
|
|
|
logger.info('Calling OpenHands MCP create_mr')
|
|
|
|
request = get_http_request()
|
|
headers = request.headers
|
|
conversation_id = headers.get('X-OpenHands-Conversation-ID', None)
|
|
|
|
provider_tokens = await get_provider_tokens(request)
|
|
access_token = await get_access_token(request)
|
|
user_id = await get_user_id(request)
|
|
|
|
github_token = provider_tokens.get(ProviderType.GITLAB, ProviderToken()) if provider_tokens else ProviderToken()
|
|
|
|
github_service = GitLabServiceImpl(
|
|
user_id=github_token.user_id,
|
|
external_auth_id=user_id,
|
|
external_auth_token=access_token,
|
|
token=github_token.token,
|
|
base_domain=github_token.host
|
|
)
|
|
|
|
try:
|
|
response = await github_service.create_mr(
|
|
id=id,
|
|
source_branch=source_branch,
|
|
target_branch=target_branch,
|
|
title=title,
|
|
description=description,
|
|
)
|
|
|
|
if conversation_id and user_id:
|
|
await save_pr_metadata(user_id, conversation_id, response)
|
|
|
|
except Exception as e:
|
|
response = str(e)
|
|
|
|
return response
|
|
|
|
|