client.py hinzugefügt
This commit is contained in:
@@ -0,0 +1,30 @@
|
||||
import anyio
|
||||
|
||||
from mcp import ClientSession
|
||||
from mcp.client.streamable_http import streamable_http_client
|
||||
|
||||
|
||||
class MCPClient:
|
||||
def __init__(self, url: str):
|
||||
self.url = url
|
||||
self.session = None
|
||||
self._client = None
|
||||
|
||||
async def connect(self):
|
||||
self._client = streamable_http_client(self.url)
|
||||
read, write = await self._client.__aenter__()
|
||||
|
||||
self.session = ClientSession(read, write)
|
||||
await self.session.initialize()
|
||||
|
||||
print("[MCP] verbunden")
|
||||
|
||||
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):
|
||||
if self._client:
|
||||
await self._client.__aexit__(None, None, None)
|
||||
Reference in New Issue
Block a user