diff --git a/scripts/start_mcp_server.py b/scripts/start_mcp_server.py index 98c5832..ee83003 100644 --- a/scripts/start_mcp_server.py +++ b/scripts/start_mcp_server.py @@ -2,19 +2,19 @@ import tyro 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): +# ### NOTE: some sanity check +# async def main(conf:MCPServerConfig): # server: MCPServer = conf.setup() -# server.run() +# u = await server.mcp._mcp_call_tool("retrieve", {"query":"test"}) +# print(u) -# if __name__ == "__main__": -# main(tyro.cli(MCPServerConfig)) \ No newline at end of file +# import asyncio +# asyncio.run(main(tyro.cli(MCPServerConfig))) + + +def main(conf:MCPServerConfig): + server: MCPServer = conf.setup() + server.run() + +if __name__ == "__main__": + main(tyro.cli(MCPServerConfig)) \ No newline at end of file