【AI大模型】DeepSeek + Nodejs 实现AI对话，打造强大聊天工具

然后他就拿棍子来打我

已于 2025-04-29 15:04:52 修改

阅读量717

点赞数 21

CC 4.0 BY-SA版权

分类专栏： API Nodejs DB 文章标签： node.js sqlite ai

于 2025-04-28 13:39:04 首次发布

本文链接：https://blog.csdn.net/DianliuKekedou/article/details/147580794

API 同时被 3 个专栏收录

22 篇文章

订阅专栏

2 篇文章

订阅专栏

Nodejs

1 篇文章

订阅专栏

文章目录

最终效果

AI聊天展示效果

项目目录结构

chatgpt-web-service/
├── client/                # 前端页面
│   └── index.html
├── server/                # Node.js 后端
│   ├── app.js
│   ├── routes/chat.js
│   ├── controllers/chatController.js
│   ├── services/openaiService.js
│   ├── db/
│   │   ├── chat.db
│   │   └── init.js
├── .env
├── package.json
└── README.md

开发环境

系统： Window10
nvm: 1.1.2 （切换node 版本）
node: v18.20.4
yarn: 1.22.22
vsCode 插件：SQLite Viewer （查看sqlite数据）

依赖版本

better-sqlite3: ^11.9.1 （嵌入式存储，保存对话历史）
cors: ^2.8.5
dotenv: ^16.5.0
express: ^5.1.0
node-fetch: ^3.3.2
openai: ^4.95.1
uuid: ^11.1.0

一、初始化项目和安装依赖

mkdir chatgpt-web-service && cd chatgpt-web-service
npm init -y
npm install express openai dotenv sqlite3 cors uuid

二、本地配置文件 `.env`

openai-key 去官网申请，openai 有代理限制。建议申请 deepseek api ，最下方有api-key申请地址

OPENAI_API_KEY=sk-xxx-your-openai-key
PORT=3000

三、SQLite 初始化脚本 `server/db/init.js`

const sqlite3 = require('sqlite3').verbose();
const db = new sqlite3.Database('./server/db/chat.db');

db.serialize(() => {
  db.run(`CREATE TABLE IF NOT EXISTS messages (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    session_id TEXT,
    role TEXT,
    content TEXT,
    timestamp DATETIME DEFAULT CURRENT_TIMESTAMP
  )`);
});

db.close();

运行一次：

node server/db/init.js

四、OpenAI 封装 `server/services/openaiService.js`

const { Configuration, OpenAIApi } = require('openai');

const configuration = new Configuration({
  apiKey: process.env.OPENAI_API_KEY,
});

const openai = new OpenAIApi(configuration);

async function getChatResponse(messages) {
  const res = await openai.createChatCompletion({
    model: 'gpt-3.5-turbo',
    messages,
  });
  return res.data.choices[0].message.content;
}

module.exports = { getChatResponse };

五、Controller 逻辑 `server/controllers/chatController.js`

const sqlite3 = require('sqlite3').verbose();
const db = new sqlite3.Database('./server/db/chat.db');
const { getChatResponse } = require('../services/openaiService');

const chatWithGPT = async (req, res) => {
  const { message, session_id } = req.body;

  if (!message || !session_id) {
    return res.status(400).json({ error: 'message and session_id are required' });
  }

  // 获取最近的上下文记录
  db.all(
    `SELECT role, content FROM messages WHERE session_id = ? ORDER BY timestamp DESC LIMIT 6`,
    [session_id],
    async (err, rows) => {
      if (err) return res.status(500).json({ error: err.message });

      const context = rows.reverse(); // 旧的在前
      context.push({ role: 'user', content: message });

      try {
        const reply = await getChatResponse(context);

        // 存入数据库
        const stmt = db.prepare(`INSERT INTO messages (session_id, role, content) VALUES (?, ?, ?)`);
        stmt.run(session_id, 'user', message);
        stmt.run(session_id, 'assistant', reply);
        stmt.finalize();

        res.json({ reply });
      } catch (error) {
        res.status(500).json({ error: error.message });
      }
    }
  );
};

module.exports = { chatWithGPT };

七、路由文件 `server/routes/chat.js`

const express = require('express');
const router = express.Router();
const { chatWithGPT } = require('../controllers/chatController');

router.post('/', chatWithGPT);

module.exports = router;

八、启动服务器 `server/app.js`

const express = require('express');
const dotenv = require('dotenv');
const cors = require('cors');
const path = require('path');

dotenv.config();
const app = express();
const port = process.env.PORT || 3000;

app.use(cors());
app.use(express.json());
app.use(express.static(path.join(__dirname, '../client'))); // 静态前端页面

const chatRoutes = require('./routes/chat');
app.use('/api/chat', chatRoutes);

app.listen(port, () => {
  console.log(`Server running at http://localhost:${port}`);
});

九、前端页面 `client/index.html`

用于验证是否有效

<!DOCTYPE html>
<html lang="en">
<head>
  <meta charset="UTF-8" />
  <title>ChatGPT 多轮对话</title>
  <style>
    body { font-family: sans-serif; padding: 20px; }
    #chat { border: 1px solid #ccc; padding: 10px; max-width: 600px; height: 400px; overflow-y: auto; }
    .msg { margin: 8px 0; }
    .user { color: blue; }
    .ai { color: green; }
  </style>
</head>
<body>
  <h2>ChatGPT 多轮对话演示</h2>
  <div id="chat"></div>
  <input id="input" type="text" placeholder="输入问题..." style="width: 400px;" />
  <button onclick="send()">发送</button>

  <script>
    const session_id = localStorage.getItem('session_id') || (crypto.randomUUID());
    localStorage.setItem('session_id', session_id);

    async function send() {
      const input = document.getElementById('input');
      const message = input.value.trim();
      if (!message) return;

      appendMessage('user', message);
      input.value = '';

      const res = await fetch('/api/chat', {
        method: 'POST',
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ message, session_id })
      });

      const data = await res.json();
      appendMessage('ai', data.reply);
    }

    function appendMessage(role, content) {
      const chat = document.getElementById('chat');
      const div = document.createElement('div');
      div.className = 'msg ' + role;
      div.textContent = `${role === 'user' ? '你' : 'AI'}：${content}`;
      chat.appendChild(div);
      chat.scrollTop = chat.scrollHeight;
    }
  </script>
</body>
</html>