基于 RAG 技术的智能心理咨询伴侣

基于范例引导，运用理情行为疗法（REBT），为用户提供专业的心理健康支持

PsyChat 是一个基于检索增强生成（RAG）技术的心理咨询 AI 助手。它结合了大语言模型的对话能力和心理咨询知识库的专业性，运用理情行为疗法（REBT）理论，帮助用户识别和调整非理性信念，缓解负面情绪。

• 🧠 智能 Agent 决策：自动判断是否需要检索知识库，动态调整回答策略
• 📚 专业心理知识库：涵盖情绪管理、人际关系、职场压力、家庭婚恋等多个主题
• 🔄 对话持续监控：连续多轮无 RAG 时自动触发强制检索，保证回答质量
• 💡 查询智能改写：基于上下文优化检索查询，提升检索准确性
• 🎨 现代 Web 界面：简洁美观的对话界面，支持移动端自适应

graph TD
    A[用户输入] --> B{Psychology Agent<br/>智能决策层}
    
    B --> C{是否需要RAG?}
    C -->|YES| D[主题分类]
    C -->|NO| E[直接生成回答]
    
    D --> F[轻检索获取锚点]
    F --> G{是否需要改写?}
    
    G -->|YES| H[引导式查询改写]
    G -->|NO| I[使用原始查询]
    
    H --> J[向量检索<br/>ChromaDB]
    I --> J
    
    J --> K{找到相关文档?}
    K -->|YES| L[构建上下文]
    K -->|NO| E
    
    L --> M[LLM生成回答<br/>DeepSeek]
    E --> M
    
    M --> N[更新对话历史]
    N --> O{对话监控Agent<br/>连续3轮无RAG?}
    
    O -->|YES| P[强制触发RAG检索]
    O -->|NO| Q[返回结果]
    
    P --> D
    Q --> R[输出回答]

git clone https://github.com/your-username/PsyChat.git
cd PsyChat

pip install -r requirements.txt

复制 env.example 为 .env 并填入你的 API 密钥：

cp env.example .env

编辑 .env 文件：

# DeepSeek API (https://platform.deepseek.com/)
DEEPSEEK_API_KEY=your_deepseek_api_key_here

# 阿里云百炼 API (https://bailian.console.aliyun.com/)
ALIBABA_API_KEY=your_alibaba_api_key_here

首次运行需要构建向量知识库：

python main.py --rebuild

# 启动 Web 界面（默认）
python main.py

# 或使用命令行模式
python main.py --cli

访问 http://localhost:8000 即可开始对话！

PsyChat/
├── main.py                 # 程序入口
├── requirements.txt        # 依赖列表
├── env.example            # 环境变量模板
├── src/
│   ├── config.py          # 配置文件
│   ├── agent/
│   │   └── psychology_agent.py  # 心理咨询 Agent
│   ├── core/
│   │   ├── rag_system.py       # RAG 系统核心
│   │   └── vector_store.py     # 向量存储
│   ├── data/
│   │   └── processor.py        # 数据处理
│   └── web/
│       └── interface.py        # Web 界面
├── resources/
│   ├── knowledge/         # 心理学知识库文档
│   └── prompts/           # 系统提示词
├── storage/
│   └── chroma_db/         # 向量数据库（自动生成）
└── docs/
    └── architecture.md    # 架构文档

本项目的知识库基于 PsyDTCorpus 数据集构建。

原始数据集经过以下预处理和增强：

• 按心理学主题（情绪、人际、职场、家庭、婚恋等）重新分类整理
• 优化问答对格式，适配 RAG 检索
• 添加主题标签，支持精准主题过滤

主要配置项（src/config.py）：

本项目仅供学习和研究使用。AI 咨询师的回复不构成专业医疗建议，如有严重心理健康问题，请及时寻求专业心理咨询师或医生的帮助。

本项目采用 MIT License 开源协议。

欢迎提交 Issue 和 Pull Request！

Made with 💝 by wink-wink-wink555