feat: route admin chat through Hermes Agent API (DeepSeek fallback)
All checks were successful
Deploy API Server / build-and-deploy (push) Successful in 11s

This commit is contained in:
WangDL 2026-05-22 00:20:34 +08:00
parent 2b0bc92ebb
commit a550ce2e67

View File

@ -2,6 +2,10 @@
import { DeepSeekProvider } from '../ai/providers/deepseek.provider'; import { DeepSeekProvider } from '../ai/providers/deepseek.provider';
import type { AiChatDto } from './dto/ai-chat.dto'; import type { AiChatDto } from './dto/ai-chat.dto';
const HERMES_API_URL = 'http://10.2.0.7:8642/v1/chat/completions';
const HERMES_API_KEY = 'zhixi-hermes-key-2026';
const SYSTEM_PROMPT = '你是知习管理后台的 AI 任务助理,帮助管理员处理日常任务、解答问题。请用简洁专业的中文回复。';
@Injectable() @Injectable()
export class AdminAiChatService { export class AdminAiChatService {
private readonly logger = new Logger(AdminAiChatService.name); private readonly logger = new Logger(AdminAiChatService.name);
@ -14,11 +18,48 @@ export class AdminAiChatService {
const messages = hasSystemPrompt const messages = hasSystemPrompt
? dto.messages ? dto.messages
: [ : [{ role: 'system' as const, content: SYSTEM_PROMPT }, ...dto.messages];
{ role: 'system' as const, content: '你是知习管理后台的 AI 任务助理,帮助管理员处理日常任务、解答问题。请用简洁专业的中文回复。' },
...dto.messages,
];
// Try Hermes Agent first, fall back to DeepSeek
try {
return await this.callHermes(messages);
} catch (err: any) {
this.logger.warn('Hermes unavailable, falling back to DeepSeek: ' + err.message);
return await this.callDeepSeek(messages);
}
}
private async callHermes(messages: Array<{ role: string; content: string }>) {
const start = Date.now();
const resp = await fetch(HERMES_API_URL, {
method: 'POST',
headers: {
'Content-Type': 'application/json',
Authorization: 'Bearer ' + HERMES_API_KEY,
},
body: JSON.stringify({
model: 'hermes-agent',
messages,
temperature: 0.7,
max_tokens: 4096,
}),
signal: AbortSignal.timeout(120_000),
});
if (!resp.ok) {
throw new Error('Hermes returned ' + resp.status);
}
const data = await resp.json();
const content = data.choices?.[0]?.message?.content || '';
const usage = data.usage || {};
this.logger.log('Hermes chat: ' + (Date.now() - start) + 'ms, tokens: ' +
(usage.prompt_tokens || 0) + '/' + (usage.completion_tokens || 0));
return { content, usage: { model: 'hermes-agent', inputTokens: usage.prompt_tokens, outputTokens: usage.completion_tokens } };
}
private async callDeepSeek(messages: Array<{ role: string; content: string }>) {
const start = Date.now(); const start = Date.now();
const result = await this.deepseek.generate({ const result = await this.deepseek.generate({
model: 'deepseek-chat', model: 'deepseek-chat',
@ -26,20 +67,15 @@ export class AdminAiChatService {
temperature: 0.7, temperature: 0.7,
maxTokens: 4096, maxTokens: 4096,
}); });
this.logger.log('DeepSeek chat: ' + (Date.now() - start) + 'ms');
this.logger.log(`AI chat completed in ${Date.now() - start}ms, tokens: ${result.usage?.inputTokens ?? 0}/${result.usage?.outputTokens ?? 0}`); return { content: result.rawText, usage: result.usage };
return {
content: result.rawText,
usage: result.usage,
};
} }
getDashboardConfig() { getDashboardConfig() {
return { return {
url: 'http://10.2.0.7:9119', url: 'http://10.2.0.7:9119',
token: 'v8dAkvcsHaHe_yok5kjO0QErKtogQcq3IKaJT8PNczQ', apiUrl: 'http://10.2.0.7:8642/v1',
description: 'Hermes Agent Dashboard — 4核4G 上的 AI Agent,可执行服务器管理、文件操作等高级任务', description: 'Hermes Agent Dashboard — 4核4G 上的 AI Agent',
}; };
} }
} }