Delete directory 'mcp'
This commit is contained in:
@@ -1,58 +0,0 @@
|
||||
import asyncio
|
||||
|
||||
from mcp.client.session import ClientSession
|
||||
from mcp.client.streamable_http import streamablehttp_client
|
||||
|
||||
|
||||
class MCPClient:
|
||||
|
||||
def __init__(self, base_url="http://localhost:8081"):
|
||||
self.base_url = base_url
|
||||
|
||||
async def _run(self, tool, arguments=None):
|
||||
|
||||
if arguments is None:
|
||||
arguments = {}
|
||||
|
||||
async with streamablehttp_client(f"{self.base_url}/mcp") as (
|
||||
read,
|
||||
write,
|
||||
_,
|
||||
):
|
||||
|
||||
async with ClientSession(read, write) as session:
|
||||
|
||||
await session.initialize()
|
||||
|
||||
result = await session.call_tool(
|
||||
tool,
|
||||
arguments=arguments,
|
||||
)
|
||||
|
||||
return result
|
||||
|
||||
def call_tool(self, tool, arguments=None):
|
||||
|
||||
return asyncio.run(
|
||||
self._run(tool, arguments)
|
||||
)
|
||||
async def _list_tools(self):
|
||||
|
||||
async with streamablehttp_client(f"{self.base_url}/mcp") as (
|
||||
read,
|
||||
write,
|
||||
_,
|
||||
):
|
||||
|
||||
async with ClientSession(read, write) as session:
|
||||
|
||||
await session.initialize()
|
||||
|
||||
return await session.list_tools()
|
||||
|
||||
|
||||
def list_tools(self):
|
||||
|
||||
return asyncio.run(
|
||||
self._list_tools()
|
||||
)
|
||||
@@ -1,7 +0,0 @@
|
||||
from mcp.client import MCPClient
|
||||
|
||||
mcp = MCPClient()
|
||||
|
||||
tools = mcp.list_tools()
|
||||
|
||||
print(tools)
|
||||
Reference in New Issue
Block a user