extraction error logging
This commit is contained in:
@@ -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`, {
|
||||||
|
|||||||
Reference in New Issue
Block a user