fix
This commit is contained in:
@@ -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))
|
||||||
Reference in New Issue
Block a user