replyme 0.1.0__py3-none-any.whl

replyme/__init__.py

@@ -0,0 +1,8 @@
+"""replyme：有礼貌地生成一系列不重样的跟进消息，坚持不懈直到对方回复。"""
+
+from __future__ import annotations
+
+from replyme.core import generate_one, generate_series
+
+__all__ = ["generate_one", "generate_series"]
+__version__ = "0.1.0"

replyme/cli.py

@@ -0,0 +1,251 @@
+"""replyme 命令行接口，基于 Typer 实现。"""
+
+from __future__ import annotations
+
+import os
+import sys
+from pathlib import Path
+from typing import Annotated, Optional
+
+# Windows 下确保 stdout 使用 UTF-8 编码，避免 emoji 等字符输出报错
+if sys.platform == "win32":
+    os.environ.setdefault("PYTHONIOENCODING", "utf-8")
+    if hasattr(sys.stdout, "reconfigure"):
+        try:
+            sys.stdout.reconfigure(encoding="utf-8", errors="replace")
+        except Exception:
+            pass
+    if hasattr(sys.stderr, "reconfigure"):
+        try:
+            sys.stderr.reconfigure(encoding="utf-8", errors="replace")
+        except Exception:
+            pass
+
+import typer
+from rich.console import Console
+from rich.panel import Panel
+from rich.text import Text
+from rich.table import Table
+
+from replyme.config import (
+    EMAIL_STYLES,
+    IM_STYLES,
+    ReplymeConfig,
+    load_config,
+    save_config,
+    init_config,
+    get_config_dir,
+)
+from replyme.core import generate_series
+from replyme.reader import read_files
+from replyme.logger import save_follow_up_log, make_stem_from_topic
+
+app = typer.Typer(
+    name="replyme",
+    help="有礼貌地生成一系列不重样的跟进消息，坚持不懈直到对方回复。",
+    add_completion=False,
+    no_args_is_help=True,
+)
+
+console = Console()
+
+
+# ── init 命令 ───────────────────────────────────────────────
+
+@app.command()
+def init() -> None:
+    """初始化本地配置目录和默认提示词文件。"""
+    config_dir = get_config_dir()
+
+    if config_dir.exists():
+        console.print(
+            f"[yellow]配置目录已存在:[/] {config_dir}\n"
+            f"[dim]如需重置，请先删除 {config_dir} 后重新运行 init。[/dim]"
+        )
+        cfg = load_config()
+        _show_config(cfg)
+        return
+
+    paths = init_config()
+
+    console.print(
+        Panel(
+            f"[bold green]初始化完成！[/bold green]\n\n"
+            f"配置文件: {paths['config']}\n"
+            f"提示词文件: {paths['prompts']}\n\n"
+            f"[bold]修改提示词:[/]\n"
+            f"  编辑 {paths['prompts']} 即可自定义各场景/风格的系统提示词。\n"
+            f"  文件结构为 JSON，按 scene > style > prompt_text 层级组织。\n"
+            f"  修改后立即生效，无需重启。\n\n"
+            f"[bold]修改默认配置:[/]\n"
+            f"  编辑 {paths['config']} 可修改默认场景、风格等参数。",
+            title="replyme init",
+            border_style="green",
+        )
+    )
+
+
+# ── send 命令 ───────────────────────────────────────────────
+
+@app.command()
+def send(
+    topic: Annotated[
+        Optional[str],
+        typer.Argument(help="想要讨论的主题或问题（与 --input 二选一）"),
+    ] = None,
+    input_files: Annotated[
+        Optional[list[Path]],
+        typer.Option(
+            "--input", "-i",
+            help="输入文件路径（支持 txt/json/md），可指定多个",
+            exists=True,
+        ),
+    ] = None,
+    scene: Annotated[
+        str,
+        typer.Option("--scene", "-s", help="场景: email(邮件) 或 im(即时通讯)"),
+    ] = "im",
+    style: Annotated[
+        Optional[str],
+        typer.Option("--style", "-t", help="风格（不指定则使用配置文件或默认值）"),
+    ] = None,
+    count: Annotated[
+        int,
+        typer.Option("--count", "-n", help="生成消息的数量", min=1, max=20),
+    ] = 5,
+    model: Annotated[
+        Optional[str],
+        typer.Option("--model", "-m", help="指定 LLM 模型名称"),
+    ] = None,
+    interval: Annotated[
+        int,
+        typer.Option("--interval", "--wait", help="每条消息之间的间隔秒数", min=0),
+    ] = 0,
+    output_dir: Annotated[
+        Optional[Path],
+        typer.Option("--output", "-o", help="日志输出目录（默认 replyme/logs/）"),
+    ] = None,
+) -> None:
+    """生成一系列有礼貌、不重样的跟进消息。"""
+    import time
+
+    # 验证输入
+    if not topic and not input_files:
+        console.print(
+            "[red]错误:[/] 请提供主题（topic）或输入文件（--input），至少需要一个。"
+        )
+        raise typer.Exit(code=1)
+
+    # 加载本地配置
+    cfg = load_config()
+
+    # 合并参数：命令行参数优先于配置文件
+    effective_scene = scene or cfg.scene
+    effective_style = style or cfg.style or _default_style(effective_scene)
+    effective_model = model or cfg.model
+    effective_count = count
+    effective_interval = interval or cfg.interval
+
+    # 验证场景
+    if effective_scene not in ("email", "im"):
+        console.print(f"[red]错误:[/] 不支持的场景 '{effective_scene}'，请使用 email 或 im")
+        raise typer.Exit(code=1)
+
+    # 读取输入文件内容
+    original_content: str | None = None
+    stem: str = "topic"
+
+    if input_files:
+        try:
+            original_content = read_files(input_files)
+            stem = input_files[0].stem
+        except (FileNotFoundError, ValueError) as e:
+            console.print(f"[red]读取文件失败:[/] {e}")
+            raise typer.Exit(code=1)
+    elif topic:
+        stem = make_stem_from_topic(topic)
+
+    # 显示信息面板
+    style_label = _get_style_label(effective_scene, effective_style)
+    info_lines = [
+        f"[bold blue]场景:[/] {effective_scene}  [bold blue]风格:[/] {style_label}",
+        f"[bold blue]数量:[/] {effective_count}",
+    ]
+    if topic:
+        info_lines.insert(0, f"[bold blue]主题:[/] {topic}")
+    if input_files:
+        files_str = ", ".join(f.name for f in input_files)
+        info_lines.insert(0, f"[bold blue]输入文件:[/] {files_str}")
+
+    console.print(Panel("\n".join(info_lines), title="replyme", border_style="blue"))
+
+    # 生成消息
+    def on_generated(index: int, message: str) -> None:
+        _print_message(index, effective_count, message)
+        if effective_interval > 0 and index < effective_count - 1:
+            console.print(f"\n[dim]等待 {effective_interval} 秒后生成下一条...[/dim]")
+            time.sleep(effective_interval)
+
+    messages = generate_series(
+        topic=topic,
+        count=effective_count,
+        original_content=original_content,
+        scene=effective_scene,
+        style=effective_style,
+        model=effective_model,
+        on_generated=on_generated,
+    )
+
+    # 保存日志
+    log_path = save_follow_up_log(
+        stem=stem,
+        scene=effective_scene,
+        style=effective_style,
+        topic=topic,
+        original_content=original_content,
+        generated_messages=messages,
+        output_dir=output_dir,
+    )
+
+    # 打印汇总
+    success_count = sum(1 for m in messages if m)
+    console.print(
+        f"\n[bold green]完成！[/bold green] "
+        f"成功生成 {success_count}/{effective_count} 条消息。"
+    )
+    console.print(f"[dim]日志已保存: {log_path}[/dim]")
+
+
+# ── 辅助函数 ────────────────────────────────────────────────
+
+def _default_style(scene: str) -> str:
+    if scene == "email":
+        return "formal"
+    return "polite"
+
+
+def _get_style_label(scene: str, style: str) -> str:
+    styles_map = EMAIL_STYLES if scene == "email" else IM_STYLES
+    return styles_map.get(style, style)
+
+
+def _print_message(index: int, total: int, message: str) -> None:
+    num = Text(f"#{index + 1}/{total}", style="bold cyan")
+    panel = Panel(message, title=num, border_style="green", padding=(0, 1))
+    console.print(panel)
+
+
+def _show_config(cfg: ReplymeConfig) -> None:
+    table = Table(title="当前配置", show_header=False)
+    table.add_column("键", style="bold")
+    table.add_column("值")
+    table.add_row("场景", cfg.scene)
+    table.add_row("风格", cfg.style)
+    table.add_row("模型", cfg.model or "(llmdog 默认)")
+    table.add_row("数量", str(cfg.count))
+    table.add_row("间隔", f"{cfg.interval} 秒")
+    console.print(table)
+
+
+if __name__ == "__main__":
+    app()

replyme/config.py

@@ -0,0 +1,299 @@
+"""replyme 本地配置管理：init 命令、配置文件读写、自定义提示词。"""
+
+from __future__ import annotations
+
+import json
+from pathlib import Path
+from dataclasses import dataclass, field, asdict
+
+# 默认配置目录名
+_CONFIG_DIR = "replyme"
+_CONFIG_FILE = "config.json"
+_PROMPTS_FILE = "prompts.json"
+
+# ── 场景与风格 ──────────────────────────────────────────────
+
+# 邮件风格
+EMAIL_STYLES = {
+    "formal": "正式商务",
+    "friendly": "友好亲切",
+    "persistent": "坚持不懈",
+    "angry": "愤怒警告",
+    "poor": "可怜求助",
+}
+
+# IM 风格
+IM_STYLES = {
+    "casual": "轻松随意",
+    "polite": "礼貌得体",
+    "persistent": "坚持不懈",
+    "angry": "愤怒警告",
+    "poor": "可怜求助",
+}
+
+# ── 内置提示词模板 ──────────────────────────────────────────
+
+_BUILTIN_PROMPTS: dict = {
+    "email": {
+        "formal": (
+            "你是一个跟进邮件撰写助手。\n\n"
+            "场景：我之前给对方发过邮件，但对方一直没有回复。我需要继续发跟进邮件，"
+            "礼貌地推动对方回复。\n\n"
+            "你的工作流程：\n"
+            "1. 先理解我之前发了什么邮件、核心诉求是什么\n"
+            "2. 基于这个理解，写一封新的跟进邮件，换一个角度或切入点来推动回复\n\n"
+            "绝对规则（必须遵守）：\n"
+            "- 只输出一封邮件的正文纯文本，不要输出其他任何内容\n"
+            "- 绝对不要输出多封邮件供选择，只输出一封\n"
+            "- 不要加任何说明、解释、注释、编号、选项、分析过程\n"
+            "- 不要使用markdown格式（不要加粗、列表、代码块等）\n"
+            "- 语气正式、专业、礼貌，体现商务礼仪\n"
+            "- 字数控制在80-200字\n"
+            "- 新邮件必须与之前发过的所有邮件（包括原始邮件和跟进邮件）明显不同\n"
+            "- 每次跟进换一个角度：可以补充信息、提供帮助、说明紧迫性、提出具体问题等"
+        ),
+        "friendly": (
+            "你是一个跟进邮件撰写助手。\n\n"
+            "场景：我之前给对方发过邮件，但对方一直没有回复。我需要继续发跟进邮件，"
+            "友好地推动对方回复。\n\n"
+            "你的工作流程：\n"
+            "1. 先理解我之前发了什么邮件、核心诉求是什么\n"
+            "2. 基于这个理解，写一封新的跟进邮件，换一个角度或切入点来推动回复\n\n"
+            "绝对规则（必须遵守）：\n"
+            "- 只输出一封邮件的正文纯文本，不要输出其他任何内容\n"
+            "- 绝对不要输出多封邮件供选择，只输出一封\n"
+            "- 不要加任何说明、解释、注释、编号、选项、分析过程\n"
+            "- 不要使用markdown格式（不要加粗、列表、代码块等）\n"
+            "- 语气友好、亲切、自然，像同事之间沟通\n"
+            "- 字数控制在80-200字\n"
+            "- 新邮件必须与之前发过的所有邮件明显不同\n"
+            "- 每次跟进换一个角度：可以表达关心、分享相关进展、询问是否需要帮助等"
+        ),
+        "persistent": (
+            "你是一个跟进邮件撰写助手。\n\n"
+            "场景：我之前给对方发过邮件，但对方一直没有回复。我需要继续发跟进邮件，"
+            "礼貌但坚定地推动对方回复。\n\n"
+            "你的工作流程：\n"
+            "1. 先理解我之前发了什么邮件、核心诉求是什么\n"
+            "2. 基于这个理解，写一封新的跟进邮件，换一个角度或切入点来推动回复\n\n"
+            "绝对规则（必须遵守）：\n"
+            "- 只输出一封邮件的正文纯文本，不要输出其他任何内容\n"
+            "- 绝对不要输出多封邮件供选择，只输出一封\n"
+            "- 不要加任何说明、解释、注释、编号、选项、分析过程\n"
+            "- 不要使用markdown格式（不要加粗、列表、代码块等）\n"
+            "- 语气礼貌但坚定，让对方感受到这件事确实需要推进\n"
+            "- 字数控制在80-200字\n"
+            "- 新邮件必须与之前发过的所有邮件明显不同\n"
+            "- 每次跟进换一个角度：可以说明影响和风险、设定时间节点、请求具体行动等"
+        ),
+        "angry": (
+            "你是一个跟进邮件撰写助手。\n\n"
+            "场景：我之前给对方发过邮件，但对方一直没有回复。我需要继续发跟进邮件，"
+            "用愤怒但仍保持基本礼貌的语气表达不满，推动对方回复。\n\n"
+            "你的工作流程：\n"
+            "1. 先理解我之前发了什么邮件、核心诉求是什么\n"
+            "2. 基于这个理解，写一封新的跟进邮件，换一个角度或切入点来推动回复\n\n"
+            "绝对规则（必须遵守）：\n"
+            "- 只输出一封邮件的正文纯文本，不要输出其他任何内容\n"
+            "- 绝对不要输出多封邮件供选择，只输出一封\n"
+            "- 不要加任何说明、解释、注释、编号、选项、分析过程\n"
+            "- 不要使用markdown格式（不要加粗、列表、代码块等）\n"
+            "- 语气表达强烈不满但仍保持基本商务礼仪，不能人身攻击或使用粗俗语言\n"
+            "- 字数控制在80-200字\n"
+            "- 新邮件必须与之前发过的所有邮件明显不同\n"
+            "- 每次跟进换一个角度：可以质问拖延原因、说明已严重影响你的计划、暗示后果等"
+        ),
+        "poor": (
+            "你是一个跟进邮件撰写助手。\n\n"
+            "场景：我之前给对方发过邮件，但对方一直没有回复。我需要继续发跟进邮件，"
+            "用示弱、博取同情的语气推动对方回复。\n\n"
+            "你的工作流程：\n"
+            "1. 先理解我之前发了什么邮件、核心诉求是什么\n"
+            "2. 基于这个理解，写一封新的跟进邮件，换一个角度或切入点来推动回复\n\n"
+            "绝对规则（必须遵守）：\n"
+            "- 只输出一封邮件的正文纯文本，不要输出其他任何内容\n"
+            "- 绝对不要输出多封邮件供选择，只输出一封\n"
+            "- 不要加任何说明、解释、注释、编号、选项、分析过程\n"
+            "- 不要使用markdown格式（不要加粗、列表、代码块等）\n"
+            "- 语气真诚示弱但不要卑微到失去尊严，不要道德绑架或过分诉苦\n"
+            "- 字数控制在80-200字\n"
+            "- 新邮件必须与之前发过的所有邮件明显不同\n"
+            "- 每次跟进换一个角度：可以说明自己的困境、表达无奈、请求帮助等"
+        ),
+    },
+    "im": {
+        "casual": (
+            "你是一个跟进消息撰写助手。\n\n"
+            "场景：我之前给对方发过消息，但对方一直没有回复。我需要继续发跟进消息，"
+            "轻松地推动对方回复。\n\n"
+            "你的工作流程：\n"
+            "1. 先理解我之前发了什么消息、核心诉求是什么\n"
+            "2. 基于这个理解，写一条新的跟进消息，换一个角度或切入点来推动回复\n\n"
+            "绝对规则（必须遵守）：\n"
+            "- 只输出一条消息的纯文本内容，不要输出其他任何内容\n"
+            "- 绝对不要输出多条消息供选择，只输出一条\n"
+            "- 不要加任何说明、解释、注释、编号、引号、分析过程\n"
+            "- 不要使用markdown格式\n"
+            "- 语气轻松随意，像朋友之间发微信一样自然\n"
+            "- 字数控制在30-80字\n"
+            "- 新消息必须与之前发过的所有消息明显不同\n"
+            "- 每次跟进换一个角度：可以换个说法、加点新信息、用幽默感等"
+        ),
+        "polite": (
+            "你是一个跟进消息撰写助手。\n\n"
+            "场景：我之前给对方发过消息，但对方一直没有回复。我需要继续发跟进消息，"
+            "礼貌地推动对方回复。\n\n"
+            "你的工作流程：\n"
+            "1. 先理解我之前发了什么消息、核心诉求是什么\n"
+            "2. 基于这个理解，写一条新的跟进消息，换一个角度或切入点来推动回复\n\n"
+            "绝对规则（必须遵守）：\n"
+            "- 只输出一条消息的纯文本内容，不要输出其他任何内容\n"
+            "- 绝对不要输出多条消息供选择，只输出一条\n"
+            "- 不要加任何说明、解释、注释、编号、引号、分析过程\n"
+            "- 不要使用markdown格式\n"
+            "- 语气礼貌温和，得体但不失亲切\n"
+            "- 字数控制在40-100字\n"
+            "- 新消息必须与之前发过的所有消息明显不同\n"
+            "- 每次跟进换一个角度：可以补充背景、提供选项、表达理解对方忙碌等"
+        ),
+        "persistent": (
+            "你是一个跟进消息撰写助手。\n\n"
+            "场景：我之前给对方发过消息，但对方一直没有回复。我需要继续发跟进消息，"
+            "礼貌但明确地推动对方回复。\n\n"
+            "你的工作流程：\n"
+            "1. 先理解我之前发了什么消息、核心诉求是什么\n"
+            "2. 基于这个理解，写一条新的跟进消息，换一个角度或切入点来推动回复\n\n"
+            "绝对规则（必须遵守）：\n"
+            "- 只输出一条消息的纯文本内容，不要输出其他任何内容\n"
+            "- 绝对不要输出多条消息供选择，只输出一条\n"
+            "- 不要加任何说明、解释、注释、编号、引号、分析过程\n"
+            "- 不要使用markdown格式\n"
+            "- 语气礼貌但坚定，让对方感受到你确实需要回复\n"
+            "- 字数控制在40-100字\n"
+            "- 新消息必须与之前发过的所有消息明显不同\n"
+            "- 每次跟进换一个角度：可以说明事情的影响、请求具体时间、直接提问等"
+        ),
+        "angry": (
+            "你是一个跟进消息撰写助手。\n\n"
+            "场景：我之前给对方发过消息，但对方一直没有回复。我需要继续发跟进消息，"
+            "用愤怒但仍保持基本礼貌的语气表达不满，推动对方回复。\n\n"
+            "你的工作流程：\n"
+            "1. 先理解我之前发了什么消息、核心诉求是什么\n"
+            "2. 基于这个理解，写一条新的跟进消息，换一个角度或切入点来推动回复\n\n"
+            "绝对规则（必须遵守）：\n"
+            "- 只输出一条消息的纯文本内容，不要输出其他任何内容\n"
+            "- 绝对不要输出多条消息供选择，只输出一条\n"
+            "- 不要加任何说明、解释、注释、编号、引号、分析过程\n"
+            "- 不要使用markdown格式\n"
+            "- 语气表达强烈不满但仍保持基本礼貌，不能人身攻击或使用粗俗语言\n"
+            "- 字数控制在40-100字\n"
+            "- 新消息必须与之前发过的所有消息明显不同\n"
+            "- 每次跟进换一个角度：可以质问为什么已读不回、说明对你的影响、暗示后果等"
+        ),
+        "poor": (
+            "你是一个跟进消息撰写助手。\n\n"
+            "场景：我之前给对方发过消息，但对方一直没有回复。我需要继续发跟进消息，"
+            "用示弱、博取同情的语气推动对方回复。\n\n"
+            "你的工作流程：\n"
+            "1. 先理解我之前发了什么消息、核心诉求是什么\n"
+            "2. 基于这个理解，写一条新的跟进消息，换一个角度或切入点来推动回复\n\n"
+            "绝对规则（必须遵守）：\n"
+            "- 只输出一条消息的纯文本内容，不要输出其他任何内容\n"
+            "- 绝对不要输出多条消息供选择，只输出一条\n"
+            "- 不要加任何说明、解释、注释、编号、引号、分析过程\n"
+            "- 不要使用markdown格式\n"
+            "- 语气真诚示弱但不要卑微到失去尊严，不要道德绑架或过分诉苦\n"
+            "- 字数控制在40-100字\n"
+            "- 新消息必须与之前发过的所有消息明显不同\n"
+            "- 每次跟进换一个角度：可以表达无奈、说自己的困境、求帮忙等"
+        ),
+    },
+}
+
+
+@dataclass
+class ReplymeConfig:
+    """replyme 运行时配置。"""
+
+    scene: str = "im"          # "email" 或 "im"
+    style: str = "polite"      # 风格名称
+    model: str | None = None   # LLM 模型
+    count: int = 5             # 生成消息数量
+    interval: int = 0          # 消息间隔秒数
+
+    def to_dict(self) -> dict:
+        return asdict(self)
+
+    @classmethod
+    def from_dict(cls, d: dict) -> ReplymeConfig:
+        return cls(**{k: v for k, v in d.items() if k in cls.__dataclass_fields__})
+
+
+def get_config_dir() -> Path:
+    """获取配置目录路径（当前目录下的 replyme）。"""
+    return Path.cwd() / _CONFIG_DIR
+
+
+def load_config() -> ReplymeConfig:
+    """从本地配置文件加载配置，不存在则返回默认。"""
+    config_path = get_config_dir() / _CONFIG_FILE
+    if config_path.exists():
+        with open(config_path, "r", encoding="utf-8") as f:
+            data = json.load(f)
+        return ReplymeConfig.from_dict(data)
+    return ReplymeConfig()
+
+
+def save_config(cfg: ReplymeConfig) -> Path:
+    """保存配置到本地文件。"""
+    config_dir = get_config_dir()
+    config_dir.mkdir(exist_ok=True)
+    config_path = config_dir / _CONFIG_FILE
+    with open(config_path, "w", encoding="utf-8") as f:
+        json.dump(cfg.to_dict(), f, ensure_ascii=False, indent=2)
+    return config_path
+
+
+def load_prompts() -> dict:
+    """加载提示词配置，本地文件优先，否则用内置默认。"""
+    prompts_path = get_config_dir() / _PROMPTS_FILE
+    if prompts_path.exists():
+        with open(prompts_path, "r", encoding="utf-8") as f:
+            return json.load(f)
+    return _BUILTIN_PROMPTS
+
+
+def save_prompts(prompts: dict | None = None) -> Path:
+    """保存提示词到本地文件。默认保存内置提示词。"""
+    config_dir = get_config_dir()
+    config_dir.mkdir(exist_ok=True)
+    prompts_path = config_dir / _PROMPTS_FILE
+    data = prompts if prompts is not None else _BUILTIN_PROMPTS
+    with open(prompts_path, "w", encoding="utf-8") as f:
+        json.dump(data, f, ensure_ascii=False, indent=2)
+    return prompts_path
+
+
+def get_system_prompt(scene: str, style: str) -> str:
+    """获取指定场景和风格的系统提示词。"""
+    prompts = load_prompts()
+    scene_prompts = prompts.get(scene, {})
+    prompt = scene_prompts.get(style)
+    if prompt:
+        return prompt
+    # 回退到该场景的第一个风格
+    if scene_prompts:
+        return next(iter(scene_prompts.values()))
+    # 最终回退到 IM polite
+    return _BUILTIN_PROMPTS["im"]["polite"]
+
+
+def init_config() -> dict[str, Path]:
+    """初始化本地配置目录和文件，返回创建的文件路径。"""
+    config_dir = get_config_dir()
+    config_dir.mkdir(exist_ok=True)
+
+    config_path = save_config(ReplymeConfig())
+    prompts_path = save_prompts()
+
+    return {"config": config_path, "prompts": prompts_path}