diff --git a/packages/memory-service/src/entities/extraction.js b/packages/memory-service/src/entities/extraction.js index b1b4123..6a5418a 100644 --- a/packages/memory-service/src/entities/extraction.js +++ b/packages/memory-service/src/entities/extraction.js @@ -34,7 +34,6 @@ function buildExtractionPrompt(userMessage, aiResponse, knownEntities = []) { '--- END CONVERSATION ---', '<|im_end|>', '<|im_start|>assistant', - '{"entities":', // primer nudges it to start filling the array ].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 const db = require('../db').getDB(); 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); 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`, {