38 lines
944 B
Python
38 lines
944 B
Python
import contextlib
|
|
from mcp import ClientSession
|
|
from mcp.client.streamable_http import streamable_http_client
|
|
from config import MCP_URL
|
|
|
|
|
|
class MCPClient:
|
|
def __init__(self):
|
|
self.session = None
|
|
self.exit_stack = contextlib.AsyncExitStack()
|
|
|
|
async def connect(self):
|
|
"""
|
|
Verbindung zum OpenHAB MCP Server
|
|
"""
|
|
|
|
transport = await self.exit_stack.enter_async_context(
|
|
streamable_http_client(MCP_URL)
|
|
)
|
|
|
|
read, write, _ = transport
|
|
|
|
self.session = await self.exit_stack.enter_async_context(
|
|
ClientSession(read, write)
|
|
)
|
|
|
|
await self.session.initialize()
|
|
|
|
return self.session
|
|
|
|
async def list_tools(self):
|
|
return await self.session.list_tools()
|
|
|
|
async def call_tool(self, name, args):
|
|
return await self.session.call_tool(name, args)
|
|
|
|
async def close(self):
|
|
await self.exit_stack.aclose() |