From 980053a0ee0a25db5c76c14d23dcc675775844d7 Mon Sep 17 00:00:00 2001 From: Storme-bit Date: Mon, 20 Apr 2026 23:25:31 -0700 Subject: [PATCH] extraction error logging --- packages/memory-service/src/entities/extraction.js | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/packages/memory-service/src/entities/extraction.js b/packages/memory-service/src/entities/extraction.js index d6ce6dd..65b7ff7 100644 --- a/packages/memory-service/src/entities/extraction.js +++ b/packages/memory-service/src/entities/extraction.js @@ -3,7 +3,7 @@ const { getEnv, SERVICES, formatEpisodeText } = require('@nexusai/shared'); const { upsertEntity } = require('./index'); const EXTRACTION_URL = getEnv('EXTRACTION_URL', 'http://localhost:11434'); -const EXTRACTION_MODEL = getEnv('EXTRACTION_MODEL', 'qwen2.5:3b'); +const EXTRACTION_MODEL = getEnv('EXTRACTION_MODEL', 'phi3:3.8ab'); const EMBEDDING_SERVICE_URL = getEnv('EMBEDDING_SERVICE_URL', SERVICES.EMBEDDING_URL); const ENTITY_TYPES = ['person', 'place', 'project', 'technology', 'concept', 'organization']; @@ -68,7 +68,7 @@ async function extractAndStoreEntities(userMessage, aiResponse, projectId=null) stream: false, options: { temperature: 0.1, - num_predict: 512, + num_predict: 768, }, }), }); @@ -77,10 +77,12 @@ async function extractAndStoreEntities(userMessage, aiResponse, projectId=null) const data = await res.json(); const raw = data.response?.trim() ?? ''; - console.log('[entities] raw response:', raw.slice(0, 300)); + console.log('[entities] raw response:', JSON.stringify(raw.slice(0, 300))); // Extract just the JSON array — everything from [ to the last ] - const match = raw.match(/\[[\s\S]*\]/); + const stripped = raw.replace(/```(?:json)?/g, '').trim(); + const match = stripped.match(/\[[\s\S]*\]/); + if (!match) throw new Error('No JSON array found in response'); const clean = match[0]; const entities = JSON.parse(clean);