Spaces:

Oldmangrizzz
/

Agent-Node

Paused

Agent-Node / agent_zero_framework /python /api /mcp_servers_apply.py

Upload folder using huggingface_hub

dff1e71 verified 10 months ago

952 Bytes

import time
from python.helpers.api import ApiHandler, Request, Response

from typing import Any

from python.helpers.mcp_handler import MCPConfig
from python.helpers.settings import set_settings_delta


class McpServersApply(ApiHandler):
    async def process(self, input: dict[Any, Any], request: Request) -> dict[Any, Any] | Response:
        mcp_servers = input["mcp_servers"]
        try:
            # MCPConfig.update(mcp_servers) # done in settings automatically
            set_settings_delta({"mcp_servers": "[]"}) # to force reinitialization
            set_settings_delta({"mcp_servers": mcp_servers})

            time.sleep(1) # wait at least a second
            # MCPConfig.wait_for_lock() # wait until config lock is released
            status = MCPConfig.get_instance().get_servers_status()
            return {"success": True, "status": status}

        except Exception as e:
            return {"success": False, "error": str(e)}

	import time
	from python.helpers.api import ApiHandler, Request, Response

	from typing import Any

	from python.helpers.mcp_handler import MCPConfig
	from python.helpers.settings import set_settings_delta


	class McpServersApply(ApiHandler):
	async def process(self, input: dict[Any, Any], request: Request) -> dict[Any, Any] \| Response:
	mcp_servers = input["mcp_servers"]
	try:
	# MCPConfig.update(mcp_servers) # done in settings automatically
	set_settings_delta({"mcp_servers": "[]"}) # to force reinitialization
	set_settings_delta({"mcp_servers": mcp_servers})

	time.sleep(1) # wait at least a second
	# MCPConfig.wait_for_lock() # wait until config lock is released
	status = MCPConfig.get_instance().get_servers_status()
	return {"success": True, "status": status}

	except Exception as e:
	return {"success": False, "error": str(e)}