71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
# jarvis_mcp/client.py
|
|
import asyncio
|
|
from mcp import ClientSession
|
|
from mcp.client.streamable_http import streamablehttp_client
|
|
from util.logger import get_logger
|
|
|
|
logger = get_logger("MCP")
|
|
|
|
|
|
class MCPClient:
|
|
def __init__(self, url: str):
|
|
# Entferne alles nach /mcp und füge exakt /mcp/ hinzu
|
|
base = url.rstrip('/').split('/mcp')[0]
|
|
self.url = base + "/mcp/" # ← WICHTIG: Slash am Ende
|
|
|
|
self.session = None
|
|
self._client_context = None
|
|
|
|
async def connect(self):
|
|
if self.session is not None:
|
|
return
|
|
|
|
logger.info(f"Verbinde mit OpenHAB MCP Server: {self.url}")
|
|
|
|
try:
|
|
self._client_context = streamablehttp_client(
|
|
self.url,
|
|
terminate_on_close=True
|
|
)
|
|
|
|
streams = await self._client_context.__aenter__()
|
|
|
|
# Flexibles Unpacking
|
|
if isinstance(streams, tuple):
|
|
read, write = streams[:2] # nimm nur die ersten zwei
|
|
else:
|
|
raise ValueError("Unerwartetes Response-Format vom Client")
|
|
|
|
self.session = ClientSession(read, write)
|
|
await asyncio.wait_for(self.session.initialize(), timeout=12.0)
|
|
|
|
logger.info("✅ OpenHAB MCP Verbindung erfolgreich hergestellt!")
|
|
|
|
except asyncio.TimeoutError:
|
|
logger.error("❌ Timeout - MCP Server antwortet nicht rechtzeitig")
|
|
raise
|
|
except Exception as e:
|
|
logger.error(f"❌ MCP Verbindungsfehler: {e}")
|
|
raise
|
|
|
|
async def list_tools(self):
|
|
await self.connect()
|
|
return await self.session.list_tools()
|
|
|
|
async def call_tool(self, name: str, arguments: dict):
|
|
await self.connect()
|
|
return await self.session.call_tool(name, arguments)
|
|
|
|
async def close(self):
|
|
for obj in [self.session, self._client_context]:
|
|
if obj:
|
|
try:
|
|
if hasattr(obj, "close"):
|
|
await obj.close()
|
|
elif hasattr(obj, "__aexit__"):
|
|
await obj.__aexit__(None, None, None)
|
|
except Exception as e:
|
|
logger.warning(f"Close Warning: {e}")
|
|
|
|
self.session = None
|
|
self._client_context = None |