From ee0c47ac1e4047fdd5aa2982c7080abf0c5c74b4 Mon Sep 17 00:00:00 2001 From: zhayujie Date: Thu, 12 Mar 2026 00:11:34 +0800 Subject: [PATCH] feat: file send prompt --- agent/prompt/builder.py | 19 ++----------------- common/cloud_client.py | 35 +++++++++++++++++++++++++++++++++++ 2 files changed, 37 insertions(+), 17 deletions(-) diff --git a/agent/prompt/builder.py b/agent/prompt/builder.py index ab8d44e..1ff91b8 100644 --- a/agent/prompt/builder.py +++ b/agent/prompt/builder.py @@ -398,26 +398,11 @@ def _build_workspace_section(workspace_dir: str, language: str) -> List[str]: def _build_cloud_website_section(workspace_dir: str) -> List[str]: """Build cloud website access prompt when cloud deployment is configured.""" try: - from common.cloud_client import get_website_base_url + from common.cloud_client import build_website_prompt + return build_website_prompt(workspace_dir) except Exception: return [] - base_url = get_website_base_url() - if not base_url: - return [] - - return [ - "**网页/网站生成规则** (非常重要):", - "", - f"- 当需要编写网页、网站、H5页面等前端代码时，**必须**将所有文件统一放到工作空间的 `websites/` 目录中（路径: `{workspace_dir}/websites/`）", - f"- 云端已为该目录配置好路由映射，访问地址为: `{base_url}`", - f" - 例如: 你在 `websites/index.html` 创建了页面，访问地址就是 `{base_url}/index.html`", - f" - 例如: 你在 `websites/my-app/index.html` 创建了页面，访问地址就是 `{base_url}/my-app/index.html`", - "- **生成网页后，必须将完整的访问链接直接发送给用户**，让用户可以直接点击访问", - "- 建议为每个独立项目在 `websites/` 下创建子目录，保持结构清晰", - "", - ] - def _build_context_files_section(context_files: List[ContextFile], language: str) -> List[str]: """构建项目上下文文件section""" diff --git a/common/cloud_client.py b/common/cloud_client.py index 6be83c3..0107da2 100644 --- a/common/cloud_client.py +++ b/common/cloud_client.py @@ -548,12 +548,47 @@ def get_website_base_url() -> str: deployment_id = get_deployment_id() if not deployment_id: return "" + + websites_domain = os.environ.get("CLOUD_WEBSITES_DOMAIN") or conf().get("cloud_websites_domain", "") + if websites_domain: + websites_domain = websites_domain.strip().rstrip("/") + return f"https://{websites_domain}/{deployment_id}" + domain = get_root_domain() if not domain: return "" return f"https://app.{domain}/{deployment_id}" +def build_website_prompt(workspace_dir: str) -> list: + """Build system prompt lines for cloud website/file sharing rules. + + Returns an empty list when cloud deployment is not configured, + so callers can safely do ``lines.extend(build_website_prompt(...))``. + """ + base_url = get_website_base_url() + if not base_url: + return [] + + return [ + "**文件分享与网页生成规则** (非常重要 — 当前为云部署模式):", + "", + f"云端已为工作空间的 `websites/` 目录配置好公网路由映射，访问地址前缀为: `{base_url}`", + "", + "1. **网页/网站**: 编写网页、H5页面等前端代码时，**必须**将文件放到 `websites/` 目录中", + f" - 例如: `websites/index.html` → `{base_url}/index.html`", + f" - 例如: `websites/my-app/index.html` → `{base_url}/my-app/index.html`", + "", + "2. **生成文件分享** (PPT、PDF、图片、音视频等): 当你为用户生成了需要下载或查看的文件时，**可以**将文件保存到 `websites/` 目录中", + f" - 例如: 生成的PPT保存到 `websites/files/report.pptx` → 下载链接为 `{base_url}/files/report.pptx`", + " - 你仍然可以同时使用 `send` 工具发送文件（在飞书、钉钉等IM渠道中有效），但**必须同时在回复文本中提供下载链接**作为兜底，因为部分渠道（如网页端）无法通过 send 接收本地文件", + "", + "3. **必须发送链接**: 无论是网页还是文件，生成后**必须将完整的访问/下载链接直接写在回复文本中发送给用户**", + "", + "4. 建议为每个独立项目在 `websites/` 下创建子目录，保持结构清晰", + "", + ] + def start(channel, channel_mgr=None): global chat_client chat_client = CloudClient(api_key=conf().get("linkai_api_key"), host=conf().get("cloud_host", ""), channel=channel)