htmlgen-mcp 0.3.2__py3-none-any.whl → 0.3.3__py3-none-any.whl

htmlgen_mcp/config.py

@@ -1,4 +1,4 @@
-"""项目配置管理 - 跨平台支持"""
+"""项目配置管理 - 跨平台支持与集群部署优化"""
 import os
 import sys
 import platform
@@ -8,7 +8,7 @@ from datetime import datetime
 
 
 class ProjectConfig:
-    """项目配置管理器 - 支持 Windows/macOS/Linux"""
+    """项目配置管理器 - 支持 Windows/macOS/Linux/集群部署"""
     
     @staticmethod
     def get_system_info() -> dict:
@@ -22,17 +22,38 @@ class ProjectConfig:
     
     @staticmethod
     def get_default_output_dir() -> Path:
-        """获取默认的项目输出目录（跨平台）
+        """获取默认的项目输出目录（跨平台 + 集群部署）
         
-        优先级：
-        1. 环境变量 WEB_AGENT_OUTPUT_DIR / WEB_AGENT_PROJECT_ROOT
-        2. 系统特定的文档目录
-        3. 用户主目录下的隐藏目录
-        4. 系统临时目录
+        优先级（集群部署优化）：
+        1. 环境变量 WEB_AGENT_PROJECT_ROOT（集群共享存储，最高优先级）
+        2. 环境变量 WEB_AGENT_OUTPUT_DIR（兼容旧配置）
+        3. 系统特定的文档目录
+        4. 用户主目录下的隐藏目录
+        5. 系统临时目录
+        
+        集群部署说明：
+        - 在 SSE 集群中，建议配置 WEB_AGENT_PROJECT_ROOT 为 NAS 共享路径
+        - 示例：WEB_AGENT_PROJECT_ROOT=/app/mcp-servers/mcp-servers/html_agent/projects
         """
         
-        # 1. 检查环境变量（所有平台通用）
-        env_dir = os.environ.get('WEB_AGENT_OUTPUT_DIR') or os.environ.get('WEB_AGENT_PROJECT_ROOT')
+        # 1a. 检查集群共享存储环境变量（最高优先级）
+        project_root = os.environ.get('WEB_AGENT_PROJECT_ROOT')
+        if project_root:
+            output_dir = Path(project_root)
+            try:
+                output_dir.mkdir(parents=True, exist_ok=True)
+                # 验证目录可写
+                test_file = output_dir / '.write_test'
+                test_file.touch()
+                test_file.unlink()
+                print(f"✅ 使用集群共享存储: {output_dir}")
+                return output_dir
+            except Exception as e:
+                print(f"⚠️ 集群共享存储不可用 ({output_dir}): {e}")
+                print(f"💡 请检查 NAS 挂载状态和权限")
+        
+        # 1b. 检查兼容性环境变量
+        env_dir = os.environ.get('WEB_AGENT_OUTPUT_DIR')
         if env_dir:
             output_dir = Path(env_dir)
             try:
@@ -111,7 +132,7 @@ class ProjectConfig:
         try:
             temp_dir.mkdir(parents=True, exist_ok=True)
             print(f"⚠️ 使用临时目录: {temp_dir}")
-            print(f"💡 建议设置环境变量 WEB_AGENT_OUTPUT_DIR 或 WEB_AGENT_PROJECT_ROOT 到更合适的位置")
+            print(f"💡 建议设置环境变量 WEB_AGENT_OUTPUT_DIR 到更合适的位置")
             return temp_dir
         except Exception as e:
             # 如果连临时目录都无法创建，使用当前工作目录
@@ -280,7 +301,7 @@ def get_project_directory(project_name: str = None) -> str:
     project_dir = config.create_project_directory(project_name, use_timestamp=True)
     
     print(f"📁 项目将生成在: {project_dir}")
-    print(f"💡 提示: 可通过设置环境变量 WEB_AGENT_OUTPUT_DIR 或 WEB_AGENT_PROJECT_ROOT 来自定义输出目录")
+    print(f"💡 提示: 可通过设置环境变量 WEB_AGENT_OUTPUT_DIR 来自定义输出目录")
     
     return str(project_dir)
 
@@ -323,4 +344,4 @@ __all__ = [
     'get_project_directory',
     'list_recent_projects', 
     'clean_temp_projects'
-]
+]

htmlgen_mcp/web_agent_server.py

@@ -42,34 +42,13 @@ DEFAULT_BASE_URL = os.environ.get(
 
 mcp = FastMCP("smart-web-agent")
 
-# MCP 服务持久化目录：优先使用项目根目录下的 .mcp_state，可通过环境变量覆盖
+# MCP 服务持久化目录：默认为 ~/.mcp/make_web，可通过环境变量覆盖
 DEFAULT_MCP_STORAGE = Path.home() / ".mcp"
 MCP_SERVICE_NAME = os.environ.get("MCP_SERVICE_NAME", "make_web")
-
-
-def _resolve_mcp_data_root() -> Path:
-    """解析 MCP 数据目录，优先使用共享项目目录"""
-    env_dir = os.environ.get("MCP_DATA_DIR")
-    if env_dir:
-        path = Path(env_dir)
-        path.mkdir(parents=True, exist_ok=True)
-        return path
-
-    shared_dir = Path(DEFAULT_PROJECT_ROOT) / ".mcp_state"
-    try:
-        shared_dir.mkdir(parents=True, exist_ok=True)
-        return shared_dir
-    except Exception:
-        fallback = DEFAULT_MCP_STORAGE / MCP_SERVICE_NAME
-        fallback.mkdir(parents=True, exist_ok=True)
-        print(
-            f"⚠️ 无法在 {shared_dir} 创建 MCP 数据目录，改用 {fallback}",
-            file=sys.stderr,
-        )
-        return fallback
-
-
-MCP_DATA_ROOT = _resolve_mcp_data_root()
+MCP_DATA_ROOT = Path(
+    os.environ.get("MCP_DATA_DIR", DEFAULT_MCP_STORAGE / MCP_SERVICE_NAME)
+)
+MCP_DATA_ROOT.mkdir(parents=True, exist_ok=True)
 
 # 简单的缓存：记录最近一次生成的计划，避免“create_simple_site → execute_plan”时需手动传递
 PLAN_CACHE_DIR = MCP_DATA_ROOT / "plan_cache"

{htmlgen_mcp-0.3.2.dist-info → htmlgen_mcp-0.3.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: htmlgen-mcp
-Version: 0.3.2
+Version: 0.3.3
 Summary: AI-powered HTML website generator with auto-upload functionality via Model Context Protocol
 Author-email: HTML Generator Team <contact@htmlgen-mcp.com>
 License: MIT

{htmlgen_mcp-0.3.2.dist-info → htmlgen_mcp-0.3.3.dist-info}/RECORD

@@ -1,7 +1,7 @@
 htmlgen_mcp/__init__.py,sha256=8jambwGFxu8RNWes1BUEGDHErV-LcvDaABWqNja9GW0,114
-htmlgen_mcp/config.py,sha256=TXDAvNFmeCI4RIznrBhZhhvWLgXAw6MO6j83prJxFsE,11373
+htmlgen_mcp/config.py,sha256=U_itOO9bG5vdZRuijWlQgDaVBPTyQ-fN1MH621QN7Mk,12375
 htmlgen_mcp/sse_optimizations.py,sha256=_UTpgLtxgNAiiEkO5lPihOi1-eEQk6R4ejNParufrrc,6932
-htmlgen_mcp/web_agent_server.py,sha256=UfjO3DtiHekr61nJ0z-Ne9cMXTqzjRLuIfIr85wBhIY,48091
+htmlgen_mcp/web_agent_server.py,sha256=69Ht2qoQZ8uFIoov6md4MoJdQWm4cuWOZllBu_FFCJc,47463
 htmlgen_mcp/agents/__init__.py,sha256=Xydfjzw9s9O6I5Ixx6EmsTdXu26136NDPUAqt9B1hzE,121
 htmlgen_mcp/agents/ai_content_generator.py,sha256=tWGC9cY6Wp7MB1P9J7uCv8LUdiS02rgS6vxaNHD7KQk,10311
 htmlgen_mcp/agents/cluster_state.py,sha256=eDAqVm_A94_x7YfmhKKQ6T1FFt_4XQOyUiHdiTPF7Qg,14615
@@ -26,8 +26,8 @@ htmlgen_mcp/agents/web_tools/simple_css.py,sha256=kj9X3sHHhj1wGwBVL20j6w2qIHXRdx
 htmlgen_mcp/agents/web_tools/simple_js.py,sha256=xMiuF-u-h_IIkUONZIa4Xf8vKB5mcXxwQf5b_BIcpoE,12174
 htmlgen_mcp/agents/web_tools/simple_templates.py,sha256=-Rs-SsWpGZT2hiwa3jZNVDHOMZOo1vV2pWbmBdR30os,6471
 htmlgen_mcp/agents/web_tools/validation.py,sha256=bNA6aWXrCSi7sPqQw5bBR3XF69gRf85D5jSMi996CtI,2069
-htmlgen_mcp-0.3.2.dist-info/METADATA,sha256=ZbrJ1I9ijuFaPO0LDrRrVZS1r8d82_qLZ6xM-EP_dN0,5161
-htmlgen_mcp-0.3.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-htmlgen_mcp-0.3.2.dist-info/entry_points.txt,sha256=w7ufTQJobIxT3FYI24yKsCEwEQvBOWhNjckUd9Amu_k,66
-htmlgen_mcp-0.3.2.dist-info/top_level.txt,sha256=KnglzX4ekV8SQkHTsJg2_nTBXz2TxaYLdvoMMovHLNk,12
-htmlgen_mcp-0.3.2.dist-info/RECORD,,
+htmlgen_mcp-0.3.3.dist-info/METADATA,sha256=9kWd0XVCvjZfico2ZWuW5uxF3EcTjk7H21L-UqYNbkM,5161
+htmlgen_mcp-0.3.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+htmlgen_mcp-0.3.3.dist-info/entry_points.txt,sha256=w7ufTQJobIxT3FYI24yKsCEwEQvBOWhNjckUd9Amu_k,66
+htmlgen_mcp-0.3.3.dist-info/top_level.txt,sha256=KnglzX4ekV8SQkHTsJg2_nTBXz2TxaYLdvoMMovHLNk,12
+htmlgen_mcp-0.3.3.dist-info/RECORD,,