MCP RAG

"""CLI interface for MCP-RAG service.""" import typer from pathlib import Path from .main import run_server, run_http_server_sync, run_stdio_server_sync from .config import settings app = typer.Typer() @app.command() def serve(): """启动MCP-RAG MCP stdio服务器。""" # 注意：不要使用 typer.echo，因为 stdout 必须只用于 MCP 协议消息 run_stdio_server_sync() @app.command() def web(): """启动MCP-RAG HTTP服务器。""" typer.echo("启动MCP-RAG HTTP服务器...") run_http_server_sync() @app.command() def init( data_dir: str = typer.Option("./data", "--data-dir", help="数据目录") ): """初始化MCP-RAG服务。""" data_path = Path(data_dir) chroma_dir = data_path / "chroma" # 创建目录 data_path.mkdir(parents=True, exist_ok=True) chroma_dir.mkdir(parents=True, exist_ok=True) typer.echo(f"在 {data_path} 初始化了MCP-RAG数据目录") @app.callback() def callback(): """MCP-RAG: 基于MCP协议的低延迟RAG服务。""" pass def main(): """CLI入口点。""" app() if __name__ == "__main__": main()