htmlgen-mcp 0.3.2__tar.gz → 0.3.3__tar.gz

{htmlgen_mcp-0.3.2 → htmlgen_mcp-0.3.3}/PKG-INFO

@@ -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 → htmlgen_mcp-0.3.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "htmlgen-mcp"
-version = "0.3.2"
+version = "0.3.3"
 description = "AI-powered HTML website generator with auto-upload functionality via Model Context Protocol"
 readme = "README.md"
 requires-python = ">=3.10"

{htmlgen_mcp-0.3.2 → htmlgen_mcp-0.3.3}/src/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-0.3.2 → htmlgen_mcp-0.3.3}/src/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 → htmlgen_mcp-0.3.3}/src/htmlgen_mcp.egg-info/PKG-INFO

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