adding in entity extraction layer

2026-04-17 05:37:24 -07:00
parent bb05d1508d
commit ec44b935d1
2 changed files with 72 additions and 0 deletions
--- a/packages/memory-service/src/episodic/index.js
+++ b/packages/memory-service/src/episodic/index.js
@@ -1,6 +1,7 @@
 const {getDB} = require('../db');
 const { EPISODIC, getEnv, SERVICES, parseRow, formatEpisodeText } = require('@nexusai/shared');
 const semantic = require('../semantic');
+const { extractAndStoreEntities } = require('../entities/extraction')

 // --Sessions --------------------------------------------------

@@ -127,6 +128,10 @@ async function createEpisode(sessionId, userMessage, aiResponse, tokenCount = nu
    }))
    .catch(err => console.error(`Failed to embed episode ${episode.id}:`, err.message));

+  extractAndStoreEntities(userMessage, aiResponse)
+    .catch(err => console.error(`Failed to extract entities for episode ${episode.id}:`, err.message));
+
+
  return episode;
 }