Max
121 lines
4.9 KiB
TypeScript
121 lines
4.9 KiB
TypeScript
import { HumanMessage, AIMessage, SystemMessage, BaseMessage } from '@langchain/core/messages';
|
|
import { ChatPromptTemplate, MessagesPlaceholder } from '@langchain/core/prompts';
|
|
import { createReactAgent } from '@langchain/langgraph/prebuilt';
|
|
import { MemorySaver } from '@langchain/langgraph';
|
|
import gigachat from './gigachat';
|
|
import { SupportContextTool } from './support-context-tool';
|
|
|
|
export interface SupportAgentConfig {
|
|
temperature?: number;
|
|
threadId?: string;
|
|
}
|
|
|
|
export interface SupportResponse {
|
|
content: string;
|
|
success: boolean;
|
|
error?: string;
|
|
}
|
|
|
|
export class SupportAgent {
|
|
private llm: any;
|
|
private memorySaver: MemorySaver;
|
|
private agent: any;
|
|
private systemPrompt: string;
|
|
private threadId: string;
|
|
private isFirstMessage: boolean;
|
|
private userId: string;
|
|
|
|
constructor(config: SupportAgentConfig = {}) {
|
|
this.systemPrompt = this.getDefaultSystemPrompt();
|
|
this.threadId = config.threadId || 'default';
|
|
this.userId = this.threadId;
|
|
this.memorySaver = new MemorySaver();
|
|
this.isFirstMessage = true;
|
|
|
|
this.llm = gigachat;
|
|
if (config.temperature !== undefined) {
|
|
this.llm.temperature = config.temperature;
|
|
}
|
|
|
|
const tools = [
|
|
new SupportContextTool(this.userId)
|
|
];
|
|
|
|
this.agent = createReactAgent({
|
|
llm: this.llm,
|
|
tools: tools,
|
|
checkpointSaver: this.memorySaver
|
|
});
|
|
}
|
|
|
|
private getDefaultSystemPrompt(): string {
|
|
return `Ты - профессиональный агент службы поддержки.
|
|
|
|
Твои основные задачи:
|
|
- Помогать пользователям решать их вопросы и проблемы
|
|
- Отвечать вежливо, профессионально и по существу
|
|
- Предоставлять четкие и понятные инструкции
|
|
- Проявлять эмпатию к проблемам пользователей
|
|
- Если не знаешь ответ, честно сообщить об этом и предложить альтернативные способы получения помощи
|
|
|
|
ВАЖНО: У тебя есть доступ к инструменту get_support_context, который позволяет получить историю предыдущих сообщений пользователя.
|
|
ВСЕГДА используй этот инструмент ПЕРВЫМ ДЕЛОМ при получении каждого нового сообщения, чтобы понять контекст и предыдущие обращения пользователя.
|
|
Только после получения контекста отвечай на вопрос пользователя.
|
|
|
|
Если в истории есть предыдущие обращения, обязательно ссылайся на них в своем ответе, показывая что помнишь предыдущее общение.
|
|
|
|
Всегда отвечай на русском языке и старайся быть максимально полезным.`;
|
|
}
|
|
|
|
public async processMessage(userMessage: string): Promise<SupportResponse> {
|
|
try {
|
|
const messages: BaseMessage[] = [];
|
|
|
|
if (this.isFirstMessage) {
|
|
messages.push(new SystemMessage(this.systemPrompt));
|
|
this.isFirstMessage = false;
|
|
}
|
|
|
|
messages.push(new HumanMessage(userMessage));
|
|
|
|
const response = await this.agent.invoke({
|
|
messages: messages
|
|
}, {
|
|
configurable: {
|
|
thread_id: this.threadId
|
|
}
|
|
});
|
|
|
|
const lastMessage = response.messages[response.messages.length - 1];
|
|
|
|
return {
|
|
content: lastMessage.content || 'Извините, не удалось сформировать ответ.',
|
|
success: true
|
|
};
|
|
|
|
} catch (error) {
|
|
console.error('Ошибка при обработке сообщения:', error);
|
|
return {
|
|
content: 'Извините, произошла ошибка при обработке вашего запроса. Попробуйте позже.',
|
|
success: false,
|
|
error: error instanceof Error ? error.message : 'Неизвестная ошибка'
|
|
};
|
|
}
|
|
}
|
|
|
|
public async clearHistory(): Promise<void> {
|
|
this.memorySaver = new MemorySaver();
|
|
|
|
const tools = [
|
|
new SupportContextTool(this.userId)
|
|
];
|
|
|
|
this.agent = createReactAgent({
|
|
llm: this.llm,
|
|
tools: tools,
|
|
checkpointSaver: this.memorySaver
|
|
});
|
|
|
|
this.isFirstMessage = true;
|
|
}
|
|
}
|