fix

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