20 lines
487 B
Python
20 lines
487 B
Python
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):
|
|
server: MCPServer = conf.setup()
|
|
server.run()
|
|
|
|
if __name__ == "__main__":
|
|
main(tyro.cli(MCPServerConfig)) |