wangdl/api-server
2
0
Fork 0
Code Issues 45 Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: remove DeepSeek fallback — Hermes only; fail fast on error
All checks were successful
Deploy API Server / build-and-deploy (push) Successful in 39s
Details

Browse Source
This commit is contained in:
WangDL 2026-05-22 10:10:11 +08:00
parent 6413936472
commit c2e8f92abe
2 changed files with 9 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/modules/admin-ai-chat/admin-ai-chat.module.ts
View File

@ -1,12 +1,11 @@
import { Module } from '@nestjs/common'; import { Module } from '@nestjs/common';
import { AdminAiChatController } from './admin-ai-chat.controller'; import { AdminAiChatController } from './admin-ai-chat.controller';
import { AdminAiChatService } from './admin-ai-chat.service'; import { AdminAiChatService } from './admin-ai-chat.service';
import { DeepSeekProvider } from '../ai/providers/deepseek.provider';
import { AdminAuthGuard } from '../../common/guards/admin-auth.guard'; import { AdminAuthGuard } from '../../common/guards/admin-auth.guard';
import { AdminRolesGuard } from '../../common/guards/admin-roles.guard'; import { AdminRolesGuard } from '../../common/guards/admin-roles.guard';
@Module({ @Module({
controllers: [AdminAiChatController], controllers: [AdminAiChatController],
providers: [AdminAiChatService, DeepSeekProvider, AdminAuthGuard, AdminRolesGuard], providers: [AdminAiChatService, AdminAuthGuard, AdminRolesGuard],
}) })
export class AdminAiChatModule {} export class AdminAiChatModule {}

26
src/modules/admin-ai-chat/admin-ai-chat.service.ts
View File

@ -1,5 +1,4 @@
import { Injectable, Logger } from '@nestjs/common'; import { Injectable, Logger } from '@nestjs/common';
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_URL = 'http://10.2.0.7:8642/v1/chat/completions';
@ -10,7 +9,7 @@ const SYSTEM_PROMPT = '你是知习管理后台的 AI 任务助理，帮助管
export class AdminAiChatService { export class AdminAiChatService {
private readonly logger = new Logger(AdminAiChatService.name); private readonly logger = new Logger(AdminAiChatService.name);
constructor(private readonly deepseek: DeepSeekProvider) {} constructor() {}
async chat(dto: AiChatDto) { async chat(dto: AiChatDto) {
const systemMessages = dto.messages.filter(m => m.role === 'system'); const systemMessages = dto.messages.filter(m => m.role === 'system');
@ -20,13 +19,7 @@ export class AdminAiChatService {
? dto.messages ? dto.messages
: [{ role: 'system' as const, content: SYSTEM_PROMPT }, ...dto.messages]; : [{ role: 'system' as const, content: SYSTEM_PROMPT }, ...dto.messages];
// Try Hermes Agent first, fall back to DeepSeek
try {
return await this.callHermes(messages); 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 }>) { private async callHermes(messages: Array<{ role: string; content: string }>) {
@ -47,7 +40,8 @@ export class AdminAiChatService {
}); });
if (!resp.ok) { if (!resp.ok) {
throw new Error('Hermes returned ' + resp.status); const text = await resp.text().catch(() => '');
throw new Error(`Hermes API error ${resp.status}: ${text}`);
} }
const data = await resp.json(); const data = await resp.json();
@ -59,18 +53,6 @@ export class AdminAiChatService {
return { content, usage: { model: 'hermes-agent', inputTokens: usage.prompt_tokens, outputTokens: usage.completion_tokens } }; 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 result = await this.deepseek.generate({
model: 'deepseek-chat',
messages: messages as Array<{ role: 'system' | 'user' | 'assistant'; content: string }>,
temperature: 0.7,
maxTokens: 4096,
});
this.logger.log('DeepSeek chat: ' + (Date.now() - start) + 'ms');
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',