Max
30 lines
959 B
Python
30 lines
959 B
Python
from mcp.server import MCPServer
|
|
from mcp.types import Completion, CompletionArgument, CompletionContext, PromptReference, ResourceTemplateReference
|
|
|
|
mcp = MCPServer("GitHub Explorer")
|
|
|
|
LANGUAGES = ["go", "javascript", "python", "rust", "typescript"]
|
|
|
|
|
|
@mcp.resource("github://repos/{owner}/{repo}")
|
|
def github_repo(owner: str, repo: str) -> str:
|
|
"""A GitHub repository."""
|
|
return f"Repository: {owner}/{repo}"
|
|
|
|
|
|
@mcp.prompt()
|
|
def review_code(language: str, code: str) -> str:
|
|
"""Review a snippet of code."""
|
|
return f"Review this {language} code:\n{code}"
|
|
|
|
|
|
@mcp.completion()
|
|
async def handle_completion(
|
|
ref: PromptReference | ResourceTemplateReference,
|
|
argument: CompletionArgument,
|
|
context: CompletionContext | None,
|
|
) -> Completion | None:
|
|
if isinstance(ref, PromptReference) and argument.name == "language":
|
|
return Completion(values=[lang for lang in LANGUAGES if lang.startswith(argument.value)])
|
|
return None
|