htmlgen-mcp 0.3.7__py3-none-any.whl → 0.3.8__py3-none-any.whl

htmlgen_mcp/web_agent_server.py

@@ -82,6 +82,7 @@ _PROGRESS_LOG_BY_ID: dict[str, str] = {}
 _PROGRESS_LOG_BY_JOB: dict[str, str] = {}
 _JOB_REGISTRY: dict[str, Dict[str, Any]] = {}
 _CONTEXT_CACHE_BY_ID: dict[str, Dict[str, Any]] = {}
+_CONTEXT_ID_BY_PLAN: dict[str, str] = {}
 
 
 def _job_state_path(job_id: str) -> Path:
@@ -140,6 +141,18 @@ def _load_job_states() -> None:
 _load_job_states()
 
 
+def _load_job_state_from_disk(job_id: str) -> Optional[Dict[str, Any]]:
+    path = _job_state_path(job_id)
+    if not path.exists():
+        return None
+    try:
+        data = json.loads(path.read_text(encoding="utf-8"))
+        data.setdefault("job_id", job_id)
+        return data
+    except Exception:
+        return None
+
+
 def _context_cache_path(context_id: str) -> Path:
     return CONTEXT_CACHE_DIR / f"{context_id}.json"
 
@@ -453,8 +466,6 @@ async def execute_plan(
     # verbose: bool = False,  # 后台执行时详细日志意义不大
     save_output: bool = False,
     progress_log: Optional[str] = None,
-    auto_upload: bool = False,
-    upload_url: Optional[str] = None,
 ) -> Dict[str, Any]:
     """执行网页构建计划，始终以后台模式运行。
 
@@ -479,21 +490,11 @@ async def execute_plan(
         如果都未指定：不记录进度日志
         路径可以是绝对路径或相对于 project_root 的相对路径
 
-    - auto_upload: 是否在构建完成后自动上传到MCP服务器
-        True: 构建完成后自动打包为ZIP并上传，返回访问URL
-        False: 仅构建，不上传
-        默认为 False
-
-    - upload_url: 文件上传API地址（可选）
-        默认为 "https://www.mcpcn.cc/api/fileUploadAndDownload/uploadMcpFile"
-        只在 auto_upload=True 时生效，一般不需要修改
-
     执行流程：
     - 任务始终在后台异步执行，立即返回 job_id 和 progress_log 路径
     - 使用 get_progress(job_id=..., limit=...) 可以实时查询执行状态和进度
     - 支持通过进度日志文件追踪详细的执行步骤和结果
-    - 执行完成后可以获取完整的执行报告和生成的文件列表
-    - 🎯 新增：auto_upload=True时，构建完成(100%)后自动上传并返回访问URL
+    - 执行完成后可以获取完整的执行报告、生成文件列表，以及自动上传结果
     """
     try:
         # 如果没有提供project_root，尝试从缓存中获取
@@ -638,38 +639,26 @@ async def execute_plan(
 
         _persist_job_state(job_id)
 
-        # 使用默认URL如果没有提供
-        effective_upload_url = upload_url or DEFAULT_UPLOAD_URL
-        
         asyncio.create_task(
             _run_execution_job(
                 job_id,
                 agent,
                 plan_dict,
                 progress_log_path=progress_log_path,
-                auto_upload=auto_upload,
-                upload_url=effective_upload_url,
             )
         )
 
-        if auto_upload:
-            message = (
-                "执行已在后台启动（含自动上传）：调用 get_progress(job_id='{}', limit=20) "
-                "可获取构建和上传进度。完成后将返回网站访问URL。"
-            ).format(job_id)
-        else:
-            message = (
-                "执行已在后台启动：调用 get_progress(job_id='{}', limit=20) "
-                "或传入 progress_log='{}' 可获取实时进度"
-            ).format(job_id, progress_log_path or "<未启用进度日志>")
+        message = (
+            "执行已在后台启动（含自动上传）：调用 get_progress(job_id='{}', limit=20) "
+            "或传入 progress_log='{}' 可获取实时进度与上传结果"
+        ).format(job_id, progress_log_path or "<未启用进度日志>")
 
         return {
             "status": "started",
             "job_id": job_id,
             "plan_id": plan_id,
             "progress_log": progress_log_path,
-            "auto_upload": auto_upload,
-            "upload_url": upload_url if auto_upload else None,
+            "upload_url": DEFAULT_UPLOAD_URL,
             "message": message,
         }
     except Exception as exc:
@@ -686,8 +675,6 @@ async def _run_execution_job(
     plan_dict: Dict[str, Any],
     *,
     progress_log_path: Optional[str],
-    auto_upload: bool = False,
-    upload_url: str = None,
 ) -> None:
     job_info = _JOB_REGISTRY.get(job_id)
     if not job_info:
@@ -705,52 +692,7 @@ async def _run_execution_job(
         job_info["completed_at"] = time.time()
         _persist_job_state(job_id)
 
-        # 🎯 关键：任务完成后自动上传
-        if auto_upload and job_info.get("project_directory"):
-            try:
-                # 记录上传开始
-                job_info["upload_status"] = "uploading"
-                _persist_job_state(job_id)
-
-                # 只在不是默认值时传递upload_url
-                upload_params = {"folder_path": job_info["project_directory"]}
-                if upload_url and upload_url != DEFAULT_UPLOAD_URL:
-                    upload_params["upload_url"] = upload_url
-                
-                upload_result = await upload_project_to_mcp_server(**upload_params)
-
-                # 记录上传结果
-                job_info["upload_result"] = upload_result
-                job_info["upload_status"] = upload_result["status"]
-                if upload_result["status"] == "success":
-                    job_info["website_url"] = upload_result["url"]
-                    job_info["upload_completed_at"] = time.time()
-
-                    # 添加到进度日志
-                    if progress_log_path:
-                        upload_event = {
-                            "timestamp": time.time(),
-                            "type": "upload_completed",
-                            "status": "success",
-                            "website_url": upload_result["url"],
-                            "message": upload_result["message"],
-                        }
-                        try:
-                            with open(
-                                progress_log_path, "a", encoding="utf-8"
-                            ) as log_file:
-                                log_file.write(
-                                    json.dumps(upload_event, ensure_ascii=False)
-                                )
-                                log_file.write("\n")
-                        except Exception:
-                            pass
-
-                _persist_job_state(job_id)
-            except Exception as upload_exc:
-                job_info["upload_status"] = "failed"
-                job_info["upload_error"] = str(upload_exc)
-                _persist_job_state(job_id)
+        await _handle_auto_upload(job_id, job_info, progress_log_path)
 
     except Exception as exc:
         job_info["status"] = "failed"
@@ -764,13 +706,57 @@ async def _run_execution_job(
         _persist_job_state(job_id)
 
 
+async def _handle_auto_upload(
+    job_id: str,
+    job_info: Dict[str, Any],
+    progress_log_path: Optional[str],
+) -> None:
+    project_dir = job_info.get("project_directory")
+    if not project_dir:
+        return
+
+    job_info["upload_status"] = "uploading"
+    _persist_job_state(job_id)
+
+    try:
+        upload_result = await upload_project_to_mcp_server(
+            folder_path=project_dir,
+            upload_url=DEFAULT_UPLOAD_URL,
+        )
+
+        job_info["upload_result"] = upload_result
+        job_info["upload_status"] = upload_result.get("status")
+
+        if upload_result.get("status") == "success":
+            job_info["website_url"] = upload_result.get("url")
+            job_info["upload_completed_at"] = time.time()
+
+            if progress_log_path:
+                upload_event = {
+                    "timestamp": time.time(),
+                    "type": "upload_completed",
+                    "status": "success",
+                    "website_url": upload_result.get("url"),
+                    "message": upload_result.get("message"),
+                }
+                try:
+                    with open(progress_log_path, "a", encoding="utf-8") as log_file:
+                        log_file.write(json.dumps(upload_event, ensure_ascii=False))
+                        log_file.write("\n")
+                except Exception:
+                    pass
+
+    except Exception as exc:
+        job_info["upload_status"] = "failed"
+        job_info["upload_error"] = str(exc)
+    finally:
+        _persist_job_state(job_id)
+
+
 @mcp.tool()
 async def create_simple_site(
     description: str,
     site_title: str = "我的网站",
-    project_root: Optional[str] = None,
-    model: Optional[str] = None,
-    context_id: Optional[str] = None,
     context_content: Optional[str] = None,
 ) -> Dict[str, Any]:
     """使用AI分析需求，生成简单但美观的网站计划。
@@ -784,11 +770,6 @@ async def create_simple_site(
       * 如果有地图查询结果、API返回数据等，都应该放在这个参数中
       * 格式可以是文本、JSON字符串或结构化数据的字符串表示
 
-    其他参数：
-    - project_root: 项目根目录，缺省使用默认目录
-    - model: 使用的AI模型，缺省使用默认模型
-    - context_id: 可选，引用已缓存的上下文快照以复用历史资料
-
     图片配置参数（可选）：
     - image_style: 图片风格 (professional|artistic|minimal|vibrant|luxury)
     - image_topics: 用户自定义的图片主题列表，如 ["modern office", "team collaboration"]
@@ -813,22 +794,17 @@ async def create_simple_site(
     这样AI就能基于真实数据来生成个性化的网站内容。
     """
     try:
-        # 使用指定模型或默认模型
-        used_model = model or DEFAULT_MODEL
+        # 使用默认模型
+        used_model = DEFAULT_MODEL
 
-        # 使用新的路径解析逻辑
-        # 如果没有提供 project_root，使用 site_title 作为项目名
-        project_directory = _resolve_project_directory(project_root, site_title)
+        # 使用新的路径解析逻辑，默认在共享目录下按标题创建项目
+        project_directory = _resolve_project_directory(None, site_title)
 
         # 处理上下文
         context_data = ""
-        actual_context_id = context_id
+        actual_context_id: Optional[str] = None
 
-        if context_id and context_id in _CONTEXT_CACHE_BY_ID:
-            # 使用缓存的上下文
-            cached_context = _CONTEXT_CACHE_BY_ID[context_id]
-            context_data = cached_context.get("content", "")
-        elif context_content:
+        if context_content:
             # 使用新提供的上下文内容
             context_data = context_content
             # 生成新的上下文ID并缓存
@@ -919,13 +895,16 @@ async def create_simple_site(
             "plan_id": plan_id,
         }
 
+        if actual_context_id:
+            cached_entry["context_id"] = actual_context_id
+
         _PLAN_CACHE_BY_ID[plan_id] = cached_entry
         cache_key = (project_directory, description)
         _PLAN_CACHE[cache_key] = cached_entry
 
         # 将上下文信息关联到计划
         if actual_context_id:
-            _PROGRESS_LOG_BY_ID[plan_id] = actual_context_id
+            _CONTEXT_ID_BY_PLAN[plan_id] = actual_context_id
 
         # 保存计划到文件
         plan_filename = f"simple_site_plan_{plan_id}.json"
@@ -1029,6 +1008,20 @@ async def get_progress(
 
         if job_id:
             job_info = _JOB_REGISTRY.get(job_id)
+            if not job_info:
+                disk_state = _load_job_state_from_disk(job_id)
+                if disk_state:
+                    _JOB_REGISTRY[job_id] = disk_state
+                    job_info = disk_state
+                    progress_log_from_state = job_info.get("progress_log")
+                    if progress_log_from_state:
+                        progress_log_str = str(progress_log_from_state)
+                        _PROGRESS_LOG_BY_JOB[job_id] = progress_log_str
+                        plan_in_state = job_info.get("plan_id")
+                        if plan_in_state:
+                            _PROGRESS_LOG_BY_ID.setdefault(
+                                plan_in_state, progress_log_str
+                            )
             if job_info and not plan_id:
                 plan_id = job_info.get("plan_id")
             if job_id in _PROGRESS_LOG_BY_JOB:
@@ -1122,14 +1115,12 @@ async def get_progress(
                     "report": job_info.get("result", {}).get("report"),
                     "created_files": job_info.get("result", {}).get("created_files"),
                 }
-                # 添加上传结果信息
                 if job_info.get("upload_result"):
                     job_snapshot["upload_result"] = job_info.get("upload_result")
 
             if job_info.get("status") == "failed":
                 job_snapshot["error"] = job_info.get("error")
 
-            # 添加上传错误信息
             if job_info.get("upload_error"):
                 job_snapshot["upload_error"] = job_info.get("upload_error")
 
@@ -1147,15 +1138,11 @@ async def get_progress(
 @mcp.tool()
 async def upload_project_to_mcp_server(
     folder_path: str,
-    upload_url: Optional[str] = None,
 ) -> Dict[str, Any]:
     """将项目文件夹打包成ZIP并上传到MCP服务器。
 
     参数说明：
     - folder_path: 项目文件夹的绝对路径
-    - upload_url: 上传API地址（可选）
-        默认为 "https://www.mcpcn.cc/api/fileUploadAndDownload/uploadMcpFile"
-        一般不需要修改
 
     返回值：
     - status: 上传状态 ("success" 或 "error")
@@ -1164,9 +1151,6 @@ async def upload_project_to_mcp_server(
     - message: 状态信息
     """
     try:
-        # 使用默认URL如果没有提供
-        effective_upload_url = upload_url or DEFAULT_UPLOAD_URL
-        
         # 验证文件夹路径
         if not os.path.exists(folder_path):
             return {"status": "error", "message": f"项目文件夹不存在: {folder_path}"}
@@ -1206,7 +1190,7 @@ async def upload_project_to_mcp_server(
                     "file", f, filename=zip_filename, content_type="application/zip"
                 )
 
-                async with session.post(effective_upload_url, data=data) as response:
+                async with session.post(DEFAULT_UPLOAD_URL, data=data) as response:
                     response_text = await response.text()
 
                     if response.status != 200:

{htmlgen_mcp-0.3.7.dist-info → htmlgen_mcp-0.3.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: htmlgen-mcp
-Version: 0.3.7
+Version: 0.3.8
 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.7.dist-info → htmlgen_mcp-0.3.8.dist-info}/RECORD

@@ -8,7 +8,7 @@ htmlgen_mcp/progress_tools.py,sha256=SOScPSr3hEv4rvGzqvwUcomEFiPhhNxJ7CbWURlFpBs
 htmlgen_mcp/progress_tracker.py,sha256=2TVduWNJJH08EQ7Vf9EpiwPjtp61JX7muSCdbGHZAfM,12210
 htmlgen_mcp/prompt_enhancer.py,sha256=8UIxt45vSNarS5uqfxC5thOfnGY7luDs2YjHZRx1GAM,7976
 htmlgen_mcp/sse_optimizations.py,sha256=_UTpgLtxgNAiiEkO5lPihOi1-eEQk6R4ejNParufrrc,6932
-htmlgen_mcp/web_agent_server.py,sha256=Uqz4xBhTaBqjAdouNbf4mCHpeluqyM8PAZZ2WA2LsPA,52617
+htmlgen_mcp/web_agent_server.py,sha256=kw1ryQHZjcm1Rz99kZir8byylABvfOb0hkfUmnf54CQ,51145
 htmlgen_mcp/agents/__init__.py,sha256=Xydfjzw9s9O6I5Ixx6EmsTdXu26136NDPUAqt9B1hzE,121
 htmlgen_mcp/agents/ai_content_generator.py,sha256=tWGC9cY6Wp7MB1P9J7uCv8LUdiS02rgS6vxaNHD7KQk,10311
 htmlgen_mcp/agents/quick_generator.py,sha256=2wV4PCRugV0suTedLDV91_etHy_2Fiw4J0MraT7MQjw,34201
@@ -31,8 +31,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.7.dist-info/METADATA,sha256=XVni-ah0fkx0HvGkRo5tiHM0xAdOvQMpZGfmQGBKbkw,5161
-htmlgen_mcp-0.3.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-htmlgen_mcp-0.3.7.dist-info/entry_points.txt,sha256=w7ufTQJobIxT3FYI24yKsCEwEQvBOWhNjckUd9Amu_k,66
-htmlgen_mcp-0.3.7.dist-info/top_level.txt,sha256=KnglzX4ekV8SQkHTsJg2_nTBXz2TxaYLdvoMMovHLNk,12
-htmlgen_mcp-0.3.7.dist-info/RECORD,,
+htmlgen_mcp-0.3.8.dist-info/METADATA,sha256=RQNybh9_YO-SB8uwqVyYlkFVXgs4u6K086J_g5bJNzo,5161
+htmlgen_mcp-0.3.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+htmlgen_mcp-0.3.8.dist-info/entry_points.txt,sha256=w7ufTQJobIxT3FYI24yKsCEwEQvBOWhNjckUd9Amu_k,66
+htmlgen_mcp-0.3.8.dist-info/top_level.txt,sha256=KnglzX4ekV8SQkHTsJg2_nTBXz2TxaYLdvoMMovHLNk,12
+htmlgen_mcp-0.3.8.dist-info/RECORD,,