add sanity
This commit is contained in:
@@ -2,11 +2,19 @@ import tyro
|
|||||||
|
|
||||||
from lang_agent.mcp_server import MCPServerConfig, MCPServer
|
from lang_agent.mcp_server import MCPServerConfig, MCPServer
|
||||||
|
|
||||||
|
# NOTE: some sanity check
|
||||||
|
# async def main(conf:MCPServerConfig):
|
||||||
|
# server: MCPServer = conf.setup()
|
||||||
|
# u = await server.mcp._mcp_call_tool("retrieve", {"query":"test"})
|
||||||
|
# print(u)
|
||||||
|
|
||||||
|
# import asyncio
|
||||||
|
# asyncio.run(main(tyro.cli(MCPServerConfig)))
|
||||||
|
|
||||||
|
|
||||||
def main(conf:MCPServerConfig):
|
def main(conf:MCPServerConfig):
|
||||||
server: MCPServer = conf.setup()
|
server: MCPServer = conf.setup()
|
||||||
server.run()
|
server.run()
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main(tyro.cli(MCPServerConfig))
|
main(tyro.cli(MCPServerConfig))
|
||||||
Reference in New Issue
Block a user