78 lines
3.7 KiB
JavaScript
78 lines
3.7 KiB
JavaScript
const express = require('express');
|
||
const axios = require('axios');
|
||
const { execSync } = require('child_process');
|
||
const fs = require('fs');
|
||
const Database = require('better-sqlite3');
|
||
const { v4: uuidv4 } = require('uuid');
|
||
const DEEPSEEK_KEY = 'ARK_API_KEY_REDACTED';
|
||
|
||
const db = new Database('/opt/zhuyuan/agent/conversations3.db');
|
||
db.exec('CREATE TABLE IF NOT EXISTS convs (id TEXT PRIMARY KEY, title TEXT, ctime TEXT, utime TEXT)');
|
||
db.exec('CREATE TABLE IF NOT EXISTS msgs (id INTEGER PRIMARY KEY AUTOINCREMENT, cid TEXT, role TEXT, content TEXT, ctime TEXT)');
|
||
|
||
const app = express();
|
||
app.use(express.json());
|
||
app.use(express.static('/opt/zhuyuan/agent/public'));
|
||
|
||
function raw(cmd){ try{ return execSync(cmd,{timeout:5000}).toString().trim() }catch(e){ return '?' } }
|
||
|
||
function reality(){
|
||
return {
|
||
host: raw('hostname'),
|
||
cpu: raw('top -bn1 | head -5 | head -c 200'),
|
||
mem: raw('free -h'),
|
||
disk: raw('df -h /'),
|
||
pm2: raw('pm2 jlist 2>/dev/null | head -c 500')
|
||
};
|
||
}
|
||
|
||
app.get('/api/convs',(req,res)=>{
|
||
res.json(db.prepare('SELECT * FROM convs ORDER BY utime DESC').all());
|
||
});
|
||
app.post('/api/convs',(req,res)=>{
|
||
const id=uuidv4();
|
||
db.prepare("INSERT INTO convs VALUES(?,?,datetime('now'),datetime('now'))").run(id,'新对话');
|
||
res.json({id});
|
||
});
|
||
app.get('/api/convs/:id',(req,res)=>{
|
||
const c=db.prepare('SELECT * FROM convs WHERE id=?').get(req.params.id);
|
||
if(!c)return res.status(404).json({e:'not found'});
|
||
c.messages=db.prepare('SELECT role,content FROM msgs WHERE cid=? ORDER BY id ASC').all(req.params.id);
|
||
res.json(c);
|
||
});
|
||
app.delete('/api/convs/:id',(req,res)=>{
|
||
db.prepare('DELETE FROM msgs WHERE cid=?').run(req.params.id);
|
||
db.prepare('DELETE FROM convs WHERE id=?').run(req.params.id);
|
||
res.json({ok:1});
|
||
});
|
||
app.post('/api/chat',async(req,res)=>{
|
||
const{conversation_id,message}=req.body;
|
||
if(!message)return res.status(400).json({e:'no msg'});
|
||
let cid=conversation_id||uuidv4();
|
||
if(!conversation_id)db.prepare("INSERT INTO convs VALUES(?,?,datetime('now'),datetime('now'))").run(cid,'新对话');
|
||
db.prepare("INSERT INTO msgs(cid,role,content,ctime) VALUES(?,'user',?,datetime('now'))").run(cid,message);
|
||
const hist=db.prepare('SELECT role,content FROM msgs WHERE cid=? ORDER BY id DESC LIMIT 30').all(cid).reverse();
|
||
const t=db.prepare('SELECT title FROM convs WHERE id=?').get(cid);
|
||
if(t&&t.title==='新对话')db.prepare('UPDATE convs SET title=? WHERE id=?').run(message.slice(0,25),cid);
|
||
|
||
const r=reality();
|
||
const ctx='[Server Reality - '+r.host+']'+'\nCPU:\n'+r.cpu+'\nMemory:\n'+r.mem+'\nDisk:\n'+r.disk+'\nPM2:\n'+r.pm2;
|
||
|
||
const sys='你是光湖团队技术助手，由铸渊ICE-GL-ZY001部署。技术主控Awen的AI助手。服务器: 冰朔6台+团队2台。广州是中枢(43.139.217.141)。每次回答时用户消息末尾会附带当前服务器真实采集数据，请基于真实数据回答，不要编造。';
|
||
|
||
try{
|
||
const resp=await axios.post('https://api.deepseek.com/v1/chat/completions',{
|
||
model:'deepseek-chat',
|
||
messages:[{role:'system',content:sys},...hist.slice(0,-1),{role:'user',content:message+'\n\n'+ctx}],
|
||
temperature:0.5,max_tokens:2000
|
||
},{headers:{'Authorization':'Bearer '+DEEPSEEK_KEY,'Content-Type':'application/json'}});
|
||
const reply=resp.data.choices[0].message.content;
|
||
db.prepare("INSERT INTO msgs(cid,role,content,ctime) VALUES(?,'assistant',?,datetime('now'))").run(cid,reply);
|
||
db.prepare("UPDATE convs SET utime=datetime('now') WHERE id=?").run(cid);
|
||
res.json({conversation_id:cid,reply,reality:r});
|
||
}catch(e){res.status(500).json({e:'API:'+e.message})}
|
||
});
|
||
app.get('/api/reality',(req,res)=>res.json(reality()));
|
||
app.get('/api/health',(req,res)=>res.json({ok:1}));
|
||
app.listen(3940,'127.0.0.1',()=>console.log('Agent OK'));
|