"""
MCP Client for API Server
Direct import of MCP server functionality
"""
import sys
import os
from pathlib import Path
# Add mcp_server to path (it's copied into the container)
mcp_server_path = Path(__file__).parent / "mcp_server"
if mcp_server_path.exists():
sys.path.insert(0, str(mcp_server_path))
else:
# Fallback: try parent directory (for local development)
sys.path.insert(0, str(Path(__file__).parent.parent))
from mcp_server.server import get_server, MCPServer
class MCPClient:
"""Client for MCP server"""
def __init__(self):
self.server: MCPServer = get_server()
def query_counter(self, counter_name: str, query_time: str, interface_name: str = None):
"""Query counter value"""
return self.server.query_counter(counter_name, query_time, interface_name)
def query_natural_language(self, query: str):
"""Execute natural language query"""
return self.server.query_natural_language(query)
def list_interfaces(self):
"""List all interfaces"""
return self.server.list_interfaces()
def list_counters(self):
"""List all counters"""
return self.server.list_counters()
def query_alerts(self, start_time: str = None, end_time: str = None):
"""Query alerts"""
return self.server.query_alerts(start_time, end_time)
