sascha
/
Notes
mirror of https://github.com/TriliumNext/Notes
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

move more prompts to the constants file

pull/1325/head
perf3ct 2025-03-26 18:01:20 +07:00
parent c49883fdfa
commit a50575c12c
No known key found for this signature in database
GPG Key ID: 569C4EEC436F5232
2 changed files with 43 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/services/llm/chat_service.ts
Unescape Escape View File

@ -2,6 +2,7 @@ import type { Message, ChatCompletionOptions } from './ai_interface.js';
import aiServiceManager from './ai_service_manager.js'; import aiServiceManager from './ai_service_manager.js';
import chatStorageService from './chat_storage_service.js'; import chatStorageService from './chat_storage_service.js';
import log from '../log.js'; import log from '../log.js';
import { CONTEXT_PROMPTS } from './constants/llm_prompt_constants.js';
export interface ChatSession { export interface ChatSession {
id: string; id: string;
@ -132,7 +133,10 @@ export class ChatService {
// Add error message so user knows something went wrong // Add error message so user knows something went wrong
const errorMessage: Message = { const errorMessage: Message = {
role: 'assistant', role: 'assistant',
content: `Error: Failed to generate response. ${error.message || 'Please check AI settings and try again.'}` content: CONTEXT_PROMPTS.ERROR_MESSAGES.GENERAL_ERROR.replace(
'{errorMessage}',
error.message || 'Please check AI settings and try again.'
)
}; };
session.messages.push(errorMessage); session.messages.push(errorMessage);
@ -177,7 +181,7 @@ export class ChatService {
const contextMessage: Message = { const contextMessage: Message = {
role: 'user', role: 'user',
content: `Here is the content of the note I want to discuss:\n\n${context}\n\nPlease help me with this information.` content: CONTEXT_PROMPTS.NOTE_CONTEXT_PROMPT.replace('{context}', context)
}; };
session.messages.push(contextMessage); session.messages.push(contextMessage);
@ -203,7 +207,9 @@ export class ChatService {
const contextMessage: Message = { const contextMessage: Message = {
role: 'user', role: 'user',
content: `Here is the relevant information from my notes based on my query "${query}":\n\n${context}\n\nPlease help me understand this information in relation to my query.` content: CONTEXT_PROMPTS.SEMANTIC_NOTE_CONTEXT_PROMPT
.replace('{query}', query)
.replace('{context}', context)
}; };
session.messages.push(contextMessage); session.messages.push(contextMessage);
@ -261,7 +267,10 @@ export class ChatService {
// Prepend a system message with context // Prepend a system message with context
const systemMessage: Message = { const systemMessage: Message = {
role: 'system', role: 'system',
content: `You are an AI assistant helping with Trilium Notes. Use this context to answer the user's question:\n\n${enhancedContext}` content: CONTEXT_PROMPTS.CONTEXT_AWARE_SYSTEM_PROMPT.replace(
'{enhancedContext}',
enhancedContext
)
}; };
// Create messages array with system message // Create messages array with system message
@ -307,7 +316,10 @@ export class ChatService {
// Add error message // Add error message
const errorMessage: Message = { const errorMessage: Message = {
role: 'assistant', role: 'assistant',
content: `Error: Failed to generate response with note context. ${error.message || 'Please try again.'}` content: CONTEXT_PROMPTS.ERROR_MESSAGES.CONTEXT_ERROR.replace(
'{errorMessage}',
error.message || 'Please try again.'
)
}; };
session.messages.push(errorMessage); session.messages.push(errorMessage);

27
src/services/llm/constants/llm_prompt_constants.ts
Unescape Escape View File

@ -83,7 +83,32 @@ Now, based on the above information, please answer: {query}`,
// Context for index service // Context for index service
INDEX_NO_NOTES_CONTEXT: INDEX_NO_NOTES_CONTEXT:
"I'm an AI assistant helping with your Trilium notes. I couldn't find specific notes related to your query, but I'll try to assist based on general knowledge." "I'm an AI assistant helping with your Trilium notes. I couldn't find specific notes related to your query, but I'll try to assist based on general knowledge.",
// Prompt for adding note context to chat
NOTE_CONTEXT_PROMPT: `Here is the content of the note I want to discuss:
{context}
Please help me with this information.`,
// Prompt for adding semantic note context to chat
SEMANTIC_NOTE_CONTEXT_PROMPT: `Here is the relevant information from my notes based on my query "{query}":
{context}
Please help me understand this information in relation to my query.`,
// System message prompt for context-aware chat
CONTEXT_AWARE_SYSTEM_PROMPT: `You are an AI assistant helping with Trilium Notes. Use this context to answer the user's question:
{enhancedContext}`,
// Error messages
ERROR_MESSAGES: {
GENERAL_ERROR: `Error: Failed to generate response. {errorMessage}`,
CONTEXT_ERROR: `Error: Failed to generate response with note context. {errorMessage}`
}
}; };
// Agent tool prompts // Agent tool prompts