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(超好看)
相关推荐
Ice__Cai11 分钟前
Django + Celery 详细解析:构建高效的异步任务队列
分布式·后端·python·django
阿华的代码王国15 分钟前
【Android】卡片式布局 && 滚动容器ScrollView
android·xml·java·前端·后端·卡片布局·滚动容器
MediaTea20 分钟前
Python 库手册:doctest 文档测试模块
开发语言·python·log4j
2025年一定要上岸21 分钟前
【pytest高阶】源码的走读方法及插件hook
运维·前端·python·pytest
angushine34 分钟前
Python将Word转换为Excel
python·word·excel
抠头专注python环境配置1 小时前
Anaconda创建环境报错:CondaHTTPEFTOT: HTTP 403 FORBIDDEN for url
python·conda
云边散步1 小时前
《校园生活平台从 0 到 1 的搭建》第四篇:微信授权登录前端
前端·javascript·后端
架构师沉默1 小时前
让我们一起用 DDD,构建更美好的软件世界!
java·后端·架构
王者鳜錸1 小时前
PYTHON从入门到实践-15数据可视化
开发语言·python·信息可视化
杨航 AI2 小时前
ADB+Python控制(有线/无线) Scrcpy+按键映射(推荐)
开发语言·python·adb