extraction error logging

This commit is contained in:
Storme-bit
2026-04-20 23:54:26 -07:00
parent 43dc800a0a
commit 528318b374

View File

@@ -34,7 +34,6 @@ function buildExtractionPrompt(userMessage, aiResponse, knownEntities = []) {
'--- END CONVERSATION ---', '--- END CONVERSATION ---',
'<|im_end|>', '<|im_end|>',
'<|im_start|>assistant', '<|im_start|>assistant',
'{"entities":', // primer nudges it to start filling the array
].join('\n'); ].join('\n');
} }
@@ -59,11 +58,11 @@ async function extractAndStoreEntities(userMessage, aiResponse, projectId=null)
// Fetch existing entities to guide the model toward consistent name/type pairs // Fetch existing entities to guide the model toward consistent name/type pairs
const db = require('../db').getDB(); const db = require('../db').getDB();
console.log('[entities] fetching known entities...'); // add this console.log('[entities] fetching known entities...'); // add this
const knownEntities = db.prepare(`SELECT name, type FROM entities ORDER BY name`).all(); const knownEntities = db.prepare(`SELECT name, type FROM entities ORDER BY rowid DESC LIMIT 20`).all();
console.log('[entities] known entities count:', knownEntities.length); console.log('[entities] known entities count:', knownEntities.length);
const prompt = buildExtractionPrompt(userMessage, aiResponse, knownEntities); const prompt = buildExtractionPrompt(userMessage, aiResponse, knownEntities);
console.log('[entities] prompt preview:', JSON.stringify(prompt.slice(0, 500))); console.log('[entities] prompt preview:', JSON.stringify(prompt.slice(-300)));
const res = await fetch(`${EXTRACTION_URL}/api/generate`, { const res = await fetch(`${EXTRACTION_URL}/api/generate`, {