goodbye modes 1

This commit is contained in:
lda
2026-05-16 18:30:22 +07:00 Verified
parent 8987aceb9d
commit 5369fa1bae
11 changed files with 47 additions and 124 deletions
+1 -7
View File
@@ -5,11 +5,7 @@ from .discovery import (
specs_from_discovered_tools,
)
from .events import McpEvent, make_event
from .server import (
create_broker_server,
run_broker_server,
run_transparent_proxy_server,
)
from .server import create_broker_server
from .config import build_service_from_config, load_broker_config
from .transport import normalize_transport
from .service import WfMcpService
@@ -24,8 +20,6 @@ __all__ = [
"discover_connection_capabilities",
"load_broker_config",
"make_event",
"run_broker_server",
"run_transparent_proxy_server",
"snapshot_from_specs",
"specs_from_discovered_tools",
"normalize_transport",
-41
View File
@@ -1,18 +1,12 @@
from __future__ import annotations
import os
from pathlib import Path
from mcp.server.fastmcp import FastMCP
from ..transparent_proxy import create_transparent_proxy_server
from .artifact_tools import register_artifact_tools
from .config import build_service_from_config, load_broker_config
from .prompts import register_broker_prompts
from .resources import register_broker_resources
from .service import WfMcpService
from .tools import register_broker_tools
from .transport import normalize_transport
def create_broker_server(service: WfMcpService) -> FastMCP:
@@ -31,38 +25,3 @@ def create_broker_server(service: WfMcpService) -> FastMCP:
register_broker_prompts(server, service)
return server
def main() -> None:
config_path = os.environ.get("WF_MCP_CONFIG", "wf_mcp.config.json")
transport_env = os.environ.get("WF_MCP_TRANSPORT", "stdio")
run_broker_server(config_path, transport_env)
def run_broker_server(config_path: str | Path, transport: str = "stdio") -> None:
config = load_broker_config(config_path)
service = build_service_from_config(config)
server = create_broker_server(service)
server.run(transport=normalize_transport(transport))
def run_transparent_proxy_server(
config_path: str | Path,
transport: str = "stdio",
*,
resources_as_tools: bool = False,
prompts_as_tools: bool = False,
search_tools: bool = False,
) -> None:
config = load_broker_config(config_path)
server = create_transparent_proxy_server(
config,
config_path=config_path,
resources_as_tools=resources_as_tools,
prompts_as_tools=prompts_as_tools,
search_tools=search_tools,
)
server.run(transport=normalize_transport(transport), show_banner=False)
if __name__ == "__main__":
main()