Update mcp_output/start_mcp.py

mcp_output/start_mcp.py

@@ -1,44 +1,45 @@
-"""
-MatDeepLearn MCP Service Startup Entry
-"""
-import sys
-import os
-
-# Add project root to path
-project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
-matdeeplearn_root = os.path.dirname(project_root)
-
-# Add paths
-if matdeeplearn_root not in sys.path:
-    sys.path.insert(0, matdeeplearn_root)
-
-mcp_plugin_dir = os.path.join(project_root, "mcp_plugin")
-if mcp_plugin_dir not in sys.path:
-    sys.path.insert(0, mcp_plugin_dir)
-
-from mcp_service import create_app
-
-
-def main():
-    """Start FastMCP service"""
-    app = create_app()
-    
-    # Use environment variable to configure port, default 7860 (HuggingFace default)
-    port = int(os.environ.get("MCP_PORT", "7860"))
-    
-    # Choose transport mode based on environment variable
-    transport = os.environ.get("MCP_TRANSPORT", "stdio")
-    
-    print(f"Starting MatDeepLearn MCP Service...")
-    print(f"Transport: {transport}")
-    
-    if transport == "http":
-        print(f"Port: {port}")
-        app.run(transport="http", host="0.0.0.0", port=port)
-    else:
-        # Default to STDIO mode
-        app.run()
-
-
-if __name__ == "__main__":
-    main()
+"""
+MatDeepLearn MCP Service Startup Entry
+"""
+import sys
+import os
+
+# 获取当前脚本所在目录
+current_dir = os.path.dirname(os.path.abspath(__file__))
+
+# 添加 mcp_plugin 目录到路径
+mcp_plugin_dir = os.path.join(current_dir, "mcp_plugin")
+if mcp_plugin_dir not in sys.path:
+    sys.path.insert(0, mcp_plugin_dir)
+
+# 添加项目根目录到路径 (MatDeepLearn)
+project_root = os.path.dirname(current_dir)
+if project_root not in sys.path:
+    sys.path.insert(0, project_root)
+
+from mcp_service import create_app
+
+
+def main():
+    """Start FastMCP service"""
+    app = create_app()
+    
+    # Use environment variable to configure port, default 7860 (HuggingFace default)
+    port = int(os.environ.get("MCP_PORT", "7860"))
+    
+    # Choose transport mode based on environment variable
+    transport = os.environ.get("MCP_TRANSPORT", "stdio")
+    
+    print(f"Starting MatDeepLearn MCP Service...")
+    print(f"Transport: {transport}")
+    
+    if transport == "http":
+        print(f"Port: {port}")
+        app.run(transport="http", host="0.0.0.0", port=port)
+    else:
+        # Default to STDIO mode
+        app.run()
+
+
+if __name__ == "__main__":
+    main()