19、用 Python + OpenAI 构建一个命令行 AI 问答助手

🎯 教你用最少的代码,快速打造一个能在命令行对话的"ChatGPT 助手",支持多轮问答、聊天记录、持久化保存,既是工具也是项目展示。


✅ 本文目标

  • 通过 openai API 与 GPT 模型对话
  • 支持命令行输入 & 结构化输出
  • 加入历史上下文(记忆)
  • 可扩展成 CLI 工具或本地知识助手

🧰 环境准备

1. 安装依赖

复制代码
pip install openai rich

2. 获取 OpenAI API Key


🧠 基本结构(chat.py

python 复制代码
import openai
from rich import print
from rich.console import Console

# ✅ 替换成你的 OpenAI API 密钥
openai.api_key = "sk-xxxxx"

console = Console()
messages = [{"role": "system", "content": "你是一个有趣又专业的编程助手"}]

def chat():
    while True:
        try:
            user_input = console.input("[bold green]你:[/] ")
            if user_input.lower() in ["exit", "quit"]:
                break

            messages.append({"role": "user", "content": user_input})

            res = openai.ChatCompletion.create(
                model="gpt-3.5-turbo",  # 或 gpt-4
                messages=messages
            )

            reply = res.choices[0].message["content"]
            messages.append({"role": "assistant", "content": reply})
            console.print(f"[bold blue]助手:[/] {reply}\n")

        except Exception as e:
            print(f"[red]❌ 错误:{e}[/]")

if __name__ == "__main__":
    chat()

✨ 效果演示

ruby 复制代码
$ python chat.py
你:帮我写个快速排序
助手:
当然可以!这是 Python 版本的快速排序代码:
...

🧠 加分功能建议

功能 说明
保存对话记录 写入 chat_log.txt
使用 argparse 支持 --model gpt-4--prompt 等参数
本地嵌入文档 实现 Retrieval-Augmented Generation(进阶)
响应多行 markdown 使用 rich.markdown.Markdown 美化输出

📁 示例目录结构

bash 复制代码
chat-assistant/
├── chat.py
├── .env               # 可存放 API_KEY
└── README.md

✅ 项目实用价值

  • CLI 工具原型,能快速回答技术问题
  • 可扩展成"Chat with PDF"、Chatbot 网站
  • 可部署到终端、桌面、网页任意形态

📦 扩展挑战

  • ✅ 用 langchain + Chroma 本地知识库
  • ✅ 封装为 todoai CLI 工具(支持 chat / code / explain)
  • ✅ 用 Textual 打造 TUI UI(超好看)
相关推荐
烛阴几秒前
简单入门Python装饰器
前端·python
好开心啊没烦恼20 分钟前
Python 数据分析:numpy,说人话,说说数组维度。听故事学知识点怎么这么容易?
开发语言·人工智能·python·数据挖掘·数据分析·numpy
面朝大海,春不暖,花不开33 分钟前
使用 Python 实现 ETL 流程:从文本文件提取到数据处理的全面指南
python·etl·原型模式
ai小鬼头35 分钟前
AIStarter如何助力用户与创作者?Stable Diffusion一键管理教程!
后端·架构·github
简佐义的博客1 小时前
破解非模式物种GO/KEGG注释难题
开发语言·数据库·后端·oracle·golang
Code blocks1 小时前
使用Jenkins完成springboot项目快速更新
java·运维·spring boot·后端·jenkins
2301_805054562 小时前
Python训练营打卡Day59(2025.7.3)
开发语言·python
追逐时光者2 小时前
一款开源免费、通用的 WPF 主题控件包
后端·.net