项目简介
本项目是基于Python和Hugging Face的人情世故AI系统,借助自然语言处理技术助力用户提升情商与社交能力。系统设有多个功能模块,如对话生成、知识库问答、模型微调等,可处理敬酒礼仪、送礼文化、化解尴尬等多种社交场景。
项目的主要特性和功能
- 多样化LLM应用
- Prompt工程:利用内置Prompt模板生成多场景对话。
- AI游戏:类似哄哄模拟器,助用户轻松学社交技巧。
- Agent应用:基于MetaGPT,生成更详细定制化回答。
- 知识库问答:通过LangChain和LlamaIndex构建对话应用,支持多种RAG框架。
- 模型微调
- 支持从数据收集到模型微调全流程,用户可按需微调模型生成特定场景对话。
- 支持Transformers和Xtuner框架的Lora与全量微调。
- 多语言支持:支持中文、英文和日文,用户可切换语言环境。
- 在线体验:提供在线体验地址,可在网页体验Prompt、知识库、Agent应用等功能。
安装使用步骤
1. 环境安装
确保已安装Python 3.7及以上版本,执行以下命令安装项目依赖:
bash
pip install -r requirements.txt
pip install .
2. 配置API密钥
在项目根目录下新建.env
文件,填入以下API密钥信息:
bash
OPENAI_API_KEY=
OPENAI_API_BASE=
ZHIPUAI_API_KEY=
BAIDU_API_KEY=
OPENAI_API_MODEL=
HF_HOME='./cache/'
HF_ENDPOINT='https://hf-mirror.com'
HF_TOKEN=
3. 运行项目
运行Prompt WebUI前端
bash
python3 run/tianji_prompt_webui.py
运行Agent前端
bash
streamlit run run/metagpt_webui.py
运行LangChain前端
bash
python run/demo_rag_langchain_onlinellm.py
4. 开发环境配置
提交代码前,建议安装pre-commit
进行代码格式检查:
bash
pip install pre-commit
pre-commit install
git add .
git commit -m "提交信息"
git push
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】