some sht

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))
+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))