This commit is contained in:
2025-10-13 21:58:52 +08:00
parent 25393acff6
commit ec72d25c4a

View File

@@ -2,19 +2,19 @@ import tyro
from lang_agent.mcp_server import MCPServerConfig, MCPServer from lang_agent.mcp_server import MCPServerConfig, MCPServer
### NOTE: some sanity check # ### NOTE: some sanity check
async def main(conf:MCPServerConfig): # 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):
# server: MCPServer = conf.setup() # server: MCPServer = conf.setup()
# server.run() # u = await server.mcp._mcp_call_tool("retrieve", {"query":"test"})
# print(u)
# if __name__ == "__main__": # import asyncio
# main(tyro.cli(MCPServerConfig)) # asyncio.run(main(tyro.cli(MCPServerConfig)))
def main(conf:MCPServerConfig):
server: MCPServer = conf.setup()
server.run()
if __name__ == "__main__":
main(tyro.cli(MCPServerConfig))