cold-msg 0.1.0__py3-none-any.whl

cold_msg/core/llm.py

@@ -0,0 +1,78 @@
+"""LLM 调用模块
+
+封装 llmdog 库的调用逻辑，提供重试、响应清理、JSON 解析等功能。
+"""
+
+import json
+import logging
+from typing import Optional
+
+from llmdog import chat as llmdog_chat
+from llmdog.config import load_config as llmdog_load_config
+
+from cold_msg.core.config import ColdMsgConfig
+
+logger = logging.getLogger(__name__)
+
+
+def _clean_llm_response(response: str) -> str:
+    """清理 LLM 返回的响应，移除 markdown 代码块标记"""
+    cleaned = response.strip()
+    if cleaned.startswith("```json"):
+        cleaned = cleaned[7:].strip()
+    if cleaned.startswith("```"):
+        cleaned = cleaned[3:].strip()
+    if cleaned.endswith("```"):
+        cleaned = cleaned[:-3].strip()
+    return cleaned
+
+
+def _parse_json_safely(json_str: str) -> dict:
+    """安全解析 JSON 字符串"""
+    try:
+        return json.loads(json_str)
+    except json.JSONDecodeError as e:
+        logger.error("JSON 解析失败: %s", e)
+        logger.debug("原始内容: %s...", json_str[:200])
+        return {}
+
+
+def call_llm(
+    prompt: str,
+    cfg: ColdMsgConfig,
+    model: Optional[str] = None,
+) -> dict:
+    """调用 LLM 生成邮件内容
+
+    Args:
+        prompt: 完整的 Prompt 字符串
+        cfg: ColdMsg 配置实例
+        model: 指定模型名称，为 None 时使用配置中的模型
+
+    Returns:
+        解析后的字典，包含 match_job_title, recommend_title, email_content 等字段。
+        调用失败时返回空字典。
+    """
+    model_name = model or cfg.llm_model
+
+    logger.info("正在调用 LLM (model=%s)...", model_name)
+
+    try:
+        raw_response = llmdog_chat(prompt)
+    except Exception as e:
+        logger.error("LLM 调用失败: %s", e)
+        return {}
+
+    if not raw_response or not isinstance(raw_response, str):
+        logger.warning("LLM 返回无效响应")
+        return {}
+
+    logger.info("LLM 调用成功，返回长度: %d", len(raw_response))
+
+    cleaned = _clean_llm_response(raw_response)
+    parsed = _parse_json_safely(cleaned)
+
+    if not parsed:
+        logger.error("无法解析 LLM 返回的 JSON")
+
+    return parsed

cold_msg/core/prompt.py

@@ -0,0 +1,128 @@
+"""提示词构建模块
+
+负责将候选人简历文本、职位列表、邮件模板组合为完整的 LLM Prompt。
+"""
+
+from cold_msg.core.config import ColdMsgConfig, JobPosition, EmailTemplate
+
+
+DEFAULT_SYSTEM_PROMPT = """你是招聘邮件撰写资深专家，以上是候选人的信息，请仔细阅读，先根据末尾的职位列表，评估他适合哪个岗位，然后写一份Email的Sales Pitch给他，如果评估不适合列表中的任何职位，match_job_title返回false，job_role和email_content都为空，如果评估适合列表中的职位，则给出具体的Email内容，Email的示例如下：JSON格式返回，key包括：{"match_job_title": true/false, "recommend_title": ""，"email_content": ""}"""
+
+DEFAULT_EMAIL_EXAMPLE = """您好{FullName}，
+我来自{TeamName}，很冒昧的给您发邮件。通过Linkedin了解到您，了解到您在XXX, XXX等方向有丰富经验。
+
+我负责{CompanyDesc}招聘，非常希望邀请到您这样的人才在此向您推荐"{JobTitle}"的职位。负责带领技术团队攻克更多XXX，XXX技术难题，进一步提升您在行业内的影响力。
+
+职位地点： {Location}
+
+如果您有兴趣具体了解的话，可以留个电话或者微信，给您详细介绍一下，期待与您的沟通~
+
+祝颂商祺
+{SenderName}"""
+
+DEFAULT_CONTENT_REQUIREMENTS = """# 请根据候选人技术背景，撰写一封直接可用的中文销售推介邮件，要求如下：
+
+1. 匹配评估：结合候选人技术方向与职位列表中的具体岗位（含JobTitle和Location），评估匹配度, 必须要写Location。
+
+2. 内容要求：
+   - 结构清晰：开头简要说明来意，主体突出技术匹配点，结尾自然引导联系。
+   - 简洁流畅：语言自然、口语化，避免冗余和套话。
+   - 态度诚恳：不浮夸，不使用"让我印象深刻""很欣赏您"等表达。
+   - 不含占位符：输出为完整、可直接发送的邮件正文。
+
+3. 语言风格：
+   - 使用中文，简洁自然，符合职场沟通习惯。
+   - 避免"体现了技术实力""由于网络原因无法解析链接"等无效或技术性表述。
+
+4. 结尾要求：
+   - 在邮件末尾自然添加一句联系方式索要语，例如："如果您有兴趣进一步了解，欢迎留下电话或微信，我将为您详细介绍。"
+
+5. 禁用词项（确保不出现）：
+   - "让我印象深刻"
+   - "很欣赏您"
+   - "体现了您技术实力"
+   - "由于网络原因，我无法成功解析您提供的个人主页链接"
+   - 其他主观性、冗余性表达
+
+请输出一封完整、可直接发送的中文销售推介邮件。"""
+
+
+def _format_jobs_section(jobs: list[JobPosition]) -> str:
+    """将职位列表格式化为 Prompt 中的文本段落"""
+    lines = ["职位列表:\n"]
+    for i, job in enumerate(jobs, 1):
+        lines.append(
+            f"{i}. JobTitle：{job.job_title}，"
+            f"Qualification：{job.qualification}，"
+            f"Location：{job.location}"
+        )
+    return "\n".join(lines)
+
+
+def _format_email_example(template: EmailTemplate) -> str:
+    """格式化邮件示例模板
+
+    使用 EmailTemplate 的各字段拼接出完整的邮件示例，
+    保留 {FullName}, {JobTitle}, {Location} 作为 LLM 需要填充的占位符。
+    """
+    parts = [
+        template.greeting,
+        template.intro,
+        "",
+        template.body,
+        "",
+        template.location_line,
+        "",
+        template.closing,
+        "",
+        template.sign_off,
+        template.signature,
+    ]
+    raw = "\n".join(parts)
+
+    return raw.format(
+        FullName="{FullName}",
+        TeamName=template.team_name,
+        CompanyDesc=template.company_desc,
+        JobTitle="{JobTitle}",
+        Location="{Location}",
+        SenderName=template.sender_name,
+    )
+
+
+def build_prompt(
+    resume_text: str,
+    jobs: list[JobPosition],
+    email_template: EmailTemplate | None = None,
+    custom_prompt: str | None = None,
+) -> str:
+    """构建完整的 LLM Prompt
+
+    Args:
+        resume_text: 候选人简历文本
+        jobs: 职位列表
+        email_template: 邮件模板配置，为 None 时使用默认值
+        custom_prompt: 自定义提示词，为 None 时使用默认值
+
+    Returns:
+        完整的 Prompt 字符串
+    """
+    if email_template is None:
+        email_template = EmailTemplate()
+
+    parts = []
+
+    parts.append(f"#### 候选人信息\n\n{resume_text}\n")
+
+    if custom_prompt:
+        parts.append(f"####\n{custom_prompt}\n")
+    else:
+        parts.append(f"####\n{DEFAULT_SYSTEM_PROMPT}\n")
+
+    parts.append(f"####\n{_format_email_example(email_template)}\n")
+
+    parts.append(DEFAULT_CONTENT_REQUIREMENTS)
+
+    parts.append(f"\n# 职位列表如下: \n\n```\n{_format_jobs_section(jobs)}\n```\n")
+
+    return "\n".join(parts)

cold_msg/web/__init__.py

@@ -0,0 +1 @@
+"""Web 模块"""

cold_msg/web/app.py

@@ -0,0 +1,279 @@
+"""Flask Web 应用
+
+提供 Web 界面用于：
+- 输入候选人简历文本
+- 配置职位列表
+- 配置邮件模板
+- 一键生成冷邮件
+"""
+
+import json
+import yaml
+from flask import Flask, render_template, request, jsonify
+
+from cold_msg.core.config import ColdMsgConfig, JobPosition, EmailTemplate, load_config, save_config
+from cold_msg.core.generator import generate_cold_email
+
+
+def create_app(cfg: ColdMsgConfig | None = None) -> Flask:
+    """创建 Flask 应用
+
+    Args:
+        cfg: 配置实例，为 None 时自动加载
+
+    Returns:
+        Flask 应用实例
+    """
+    app = Flask(
+        __name__,
+        template_folder="templates",
+        static_folder="static",
+    )
+
+    if cfg is None:
+        cfg = load_config()
+
+    app.config["COLDMSG_CFG"] = cfg
+
+    @app.route("/")
+    def index():
+        """主页 - 邮件生成"""
+        cfg: ColdMsgConfig = app.config["COLDMSG_CFG"]
+        return render_template("index.html", cfg=cfg)
+
+    @app.route("/config")
+    def config_page():
+        """配置页面"""
+        cfg: ColdMsgConfig = app.config["COLDMSG_CFG"]
+        return render_template("config.html", cfg=cfg)
+
+    @app.route("/jobs")
+    def jobs_page():
+        """职位管理页面"""
+        cfg: ColdMsgConfig = app.config["COLDMSG_CFG"]
+        return render_template("jobs.html", cfg=cfg)
+
+    @app.route("/api/generate", methods=["POST"])
+    def api_generate():
+        """生成冷邮件 API"""
+        data = request.get_json()
+        if not data:
+            return jsonify({"error": "请求数据为空"}), 400
+
+        resume_text = data.get("resume_text", "").strip()
+        if not resume_text:
+            return jsonify({"error": "请输入候选人简历文本"}), 400
+
+        cfg: ColdMsgConfig = app.config["COLDMSG_CFG"]
+
+        jobs_data = data.get("jobs")
+        jobs = None
+        if jobs_data:
+            jobs = [
+                JobPosition(
+                    job_title=j.get("job_title", ""),
+                    qualification=j.get("qualification", ""),
+                    location=j.get("location", ""),
+                )
+                for j in jobs_data
+            ]
+
+        tpl_data = data.get("email_template")
+        email_template = None
+        if tpl_data:
+            email_template = EmailTemplate(**tpl_data)
+
+        model = data.get("model")
+
+        try:
+            result = generate_cold_email(
+                resume_text=resume_text,
+                cfg=cfg,
+                jobs=jobs,
+                email_template=email_template,
+                model=model,
+            )
+            return jsonify(result.to_dict())
+        except Exception as e:
+            return jsonify({"error": f"生成失败: {str(e)}"}), 500
+
+    @app.route("/api/config", methods=["GET"])
+    def api_get_config():
+        """获取当前配置"""
+        cfg: ColdMsgConfig = app.config["COLDMSG_CFG"]
+        return jsonify(cfg.to_dict())
+
+    @app.route("/api/config", methods=["POST"])
+    def api_save_config():
+        """保存配置"""
+        data = request.get_json()
+        if not data:
+            return jsonify({"error": "请求数据为空"}), 400
+
+        cfg: ColdMsgConfig = app.config["COLDMSG_CFG"]
+
+        for key in ["llm_model"]:
+            if key in data:
+                setattr(cfg, key, data[key])
+
+        if "jobs" in data:
+            cfg.jobs = [
+                JobPosition(
+                    job_title=j.get("job_title", ""),
+                    qualification=j.get("qualification", ""),
+                    location=j.get("location", ""),
+                )
+                for j in data["jobs"]
+            ]
+
+        if "email_template" in data:
+            tpl_data = data["email_template"]
+            current = cfg.email_template.to_dict()
+            current.update({k: v for k, v in tpl_data.items() if v is not None})
+            cfg.email_template = EmailTemplate(**current)
+
+        try:
+            path = save_config(cfg)
+            return jsonify({"message": f"配置已保存: {path}"})
+        except Exception as e:
+            return jsonify({"error": f"保存失败: {str(e)}"}), 500
+
+    @app.route("/api/jobs", methods=["GET"])
+    def api_get_jobs():
+        """获取职位列表"""
+        cfg: ColdMsgConfig = app.config["COLDMSG_CFG"]
+        return jsonify({"jobs": [j.to_dict() for j in cfg.jobs]})
+
+    @app.route("/api/jobs", methods=["POST"])
+    def api_update_jobs():
+        """更新职位列表"""
+        data = request.get_json()
+        if not data or "jobs" not in data:
+            return jsonify({"error": "请求数据格式错误"}), 400
+
+        cfg: ColdMsgConfig = app.config["COLDMSG_CFG"]
+        cfg.jobs = [
+            JobPosition(
+                job_title=j.get("job_title", ""),
+                qualification=j.get("qualification", ""),
+                location=j.get("location", ""),
+            )
+            for j in data["jobs"]
+        ]
+
+        try:
+            save_config(cfg)
+            return jsonify({"message": f"已更新 {len(cfg.jobs)} 个职位"})
+        except Exception as e:
+            return jsonify({"error": f"保存失败: {str(e)}"}), 500
+
+    @app.route("/api/jobs/import-yaml", methods=["POST"])
+    def api_import_yaml():
+        """从 YAML 文本导入职位列表（追加模式）"""
+        data = request.get_json()
+        if not data or "yaml_text" not in data:
+            return jsonify({"error": "请求数据格式错误，需要 yaml_text 字段"}), 400
+
+        yaml_text = data["yaml_text"]
+        try:
+            parsed = yaml.safe_load(yaml_text)
+        except yaml.YAMLError as e:
+            return jsonify({"error": f"YAML 解析失败: {str(e)}"}), 400
+
+        if isinstance(parsed, dict):
+            parsed = parsed.get("jobs", parsed.get("positions", []))
+        if not isinstance(parsed, list):
+            return jsonify({"error": "YAML 内容应为列表格式"}), 400
+
+        new_jobs = []
+        for j in parsed:
+            title = j.get("job_title", j.get("JobTitle", ""))
+            if not title or not str(title).strip():
+                continue
+            new_jobs.append(JobPosition(
+                job_title=str(title).strip(),
+                qualification=str(j.get("qualification", j.get("Qualification", ""))).strip(),
+                location=str(j.get("location", j.get("Location", ""))).strip(),
+            ))
+
+        if not new_jobs:
+            return jsonify({"error": "未解析到有效职位（职位名称不能为空）"}), 400
+
+        cfg: ColdMsgConfig = app.config["COLDMSG_CFG"]
+        cfg.jobs.extend(new_jobs)
+
+        try:
+            save_config(cfg)
+            return jsonify({
+                "message": f"已导入 {len(new_jobs)} 个职位",
+                "imported": len(new_jobs),
+                "total": len(cfg.jobs),
+            })
+        except Exception as e:
+            return jsonify({"error": f"保存失败: {str(e)}"}), 500
+
+    @app.route("/api/email-template", methods=["GET"])
+    def api_get_email_template():
+        """获取邮件模板"""
+        cfg: ColdMsgConfig = app.config["COLDMSG_CFG"]
+        return jsonify({"email_template": cfg.email_template.to_dict()})
+
+    @app.route("/api/email-template", methods=["POST"])
+    def api_update_email_template():
+        """更新邮件模板"""
+        data = request.get_json()
+        if not data or "email_template" not in data:
+            return jsonify({"error": "请求数据格式错误，需要 email_template 字段"}), 400
+
+        cfg: ColdMsgConfig = app.config["COLDMSG_CFG"]
+        tpl_data = data["email_template"]
+
+        current = cfg.email_template.to_dict()
+        current.update({k: v for k, v in tpl_data.items() if v is not None})
+        cfg.email_template = EmailTemplate(**current)
+
+        try:
+            save_config(cfg)
+            return jsonify({
+                "message": "邮件模板已更新",
+                "email_template": cfg.email_template.to_dict(),
+            })
+        except Exception as e:
+            return jsonify({"error": f"保存失败: {str(e)}"}), 500
+
+    @app.route("/api/email-template/preview", methods=["POST"])
+    def api_preview_email_template():
+        """预览邮件模板（不保存）"""
+        data = request.get_json()
+        if not data:
+            return jsonify({"error": "请求数据为空"}), 400
+
+        tpl_data = data.get("email_template", {})
+        if not tpl_data:
+            cfg: ColdMsgConfig = app.config["COLDMSG_CFG"]
+            tpl_data = cfg.email_template.to_dict()
+
+        tpl = EmailTemplate(**tpl_data)
+
+        parts = [
+            tpl.greeting,
+            tpl.intro,
+            "",
+            tpl.body,
+            "",
+            tpl.location_line,
+            "",
+            tpl.closing,
+            "",
+            tpl.sign_off,
+            tpl.signature,
+        ]
+        text = "\n".join(parts)
+
+        text = text.replace("{TeamName}", tpl.team_name)
+        text = text.replace("{CompanyDesc}", tpl.company_desc)
+        text = text.replace("{SenderName}", tpl.sender_name)
+
+        return jsonify({"preview": text})
+
+    return app