JDSS/mem0
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

style: run prettier on TS SDK files

Browse Source 
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Soumil Rathi
2026-04-13 11:29:06 -07:00
parent 3ae22be8a2
commit e1e033daef
7 changed files with 40 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mem0-ts/package.json
-1
View File
@@ -120,7 +120,6 @@
"better-sqlite3": "^12.6.2",
"cloudflare": "^4.2.0",
"groq-sdk": "0.3.0",
"ollama": "^0.5.14",
"pg": "8.11.3",
"redis": "^4.6.13",
mem0-ts/src/oss/src/memory/index.ts
+22 -54
View File
@@ -168,7 +168,10 @@ export class Memory {
};
// For file-based stores (memory/SQLite), use a separate DB path for entities
if (entityConfig.dbPath) {
entityConfig.dbPath = entityConfig.dbPath.replace(/\.db$/, "_entities.db");
entityConfig.dbPath = entityConfig.dbPath.replace(
/\.db$/,
"_entities.db",
);
}
this._entityStore = VectorStoreFactory.create(
this.config.vectorStore.provider,
@@ -359,8 +362,7 @@ export class Memory {
}
// Phase 2: LLM extraction (single call)
const isAgentScoped =
!!filters.agentId && !filters.userId;
const isAgentScoped = !!filters.agentId && !filters.userId;
let systemPrompt = ADDITIVE_EXTRACTION_PROMPT;
if (isAgentScoped) {
systemPrompt += AGENT_CONTEXT_SUFFIX;
@@ -404,8 +406,7 @@ export class Memory {
extractedMemories = parsed.memory;
} catch {
const fallbackJson = extractJson(cleanResponse);
extractedMemories =
JSON.parse(fallbackJson)?.memory ?? [];
extractedMemories = JSON.parse(fallbackJson)?.memory ?? [];
}
}
} catch (e) {
@@ -565,9 +566,7 @@ export class Memory {
hr.createdAt,
);
} catch (e) {
console.error(
`Failed to add history for ${hr.memoryId}: ${e}`,
);
console.error(`Failed to add history for ${hr.memoryId}: ${e}`);
}
}
}
@@ -582,9 +581,7 @@ export class Memory {
hr.createdAt,
);
} catch (e) {
console.error(
`Failed to add history for ${hr.memoryId}: ${e}`,
);
console.error(`Failed to add history for ${hr.memoryId}: ${e}`);
}
}
}
@@ -601,8 +598,7 @@ export class Memory {
> = {};
for (let idx = 0; idx < records.length; idx++) {
const memoryId = records[idx].memoryId;
const entities =
idx < allEntities.length ? allEntities[idx] : [];
const entities = idx < allEntities.length ? allEntities[idx] : [];
for (const entity of entities) {
const key = entity.text.trim().toLowerCase();
if (key in globalEntities) {
@@ -656,8 +652,7 @@ export class Memory {
const toInsertPayloads: Record<string, any>[] = [];
for (const { index: j, key } of valid) {
const { entityType, entityText, memoryIds } =
globalEntities[key];
const { entityType, entityText, memoryIds } = globalEntities[key];
const entityVec = entityEmbeddings[j]!;
let matches: Array<{
@@ -666,31 +661,20 @@ export class Memory {
payload: Record<string, any>;
}> = [];
try {
matches = await entityStore.search(
entityVec,
1,
searchFilters,
);
matches = await entityStore.search(entityVec, 1, searchFilters);
} catch {}
if (
matches.length > 0 &&
(matches[0].score ?? 0) >= 0.95
) {
if (matches.length > 0 && (matches[0].score ?? 0) >= 0.95) {
// Update existing entity
const match = matches[0];
const payload = match.payload || {};
const linked = new Set<string>(
payload.linkedMemoryIds ?? [],
);
const linked = new Set<string>(payload.linkedMemoryIds ?? []);
for (const mid of memoryIds) linked.add(mid);
payload.linkedMemoryIds = Array.from(linked).sort();
try {
await entityStore.update(match.id, entityVec, payload);
} catch (e) {
console.debug(
`Entity update failed for '${entityText}': ${e}`,
);
console.debug(`Entity update failed for '${entityText}': ${e}`);
}
} else {
// New entity — collect for batch insert
@@ -857,10 +841,7 @@ export class Memory {
// Step 5: Compute BM25 scores from keyword results
const bm25Scores: Record<string, number> = {};
if (keywordResults) {
const [midpoint, steepness] = getBm25Params(
query,
queryLemmatized,
);
const [midpoint, steepness] = getBm25Params(query, queryLemmatized);
for (const mem of keywordResults) {
const memId = String(mem.id);
const rawScore = mem.score ?? 0;
@@ -888,17 +869,14 @@ export class Memory {
if (deduped.length > 0) {
const searchFilters: SearchFilters = {};
if (filters.userId) searchFilters.userId = filters.userId;
if (filters.agentId)
searchFilters.agentId = filters.agentId;
if (filters.agentId) searchFilters.agentId = filters.agentId;
if (filters.runId) searchFilters.runId = filters.runId;
const entityStore = await this.getEntityStore();
for (const entity of deduped) {
try {
const entityEmbedding = await this.embedder.embed(
entity.text,
);
const entityEmbedding = await this.embedder.embed(entity.text);
const matches = await entityStore.search(
entityEmbedding,
500,
@@ -910,22 +888,15 @@ export class Memory {
if (similarity < 0.5) continue;
const payload = match.payload || {};
const linkedMemoryIds =
payload.linkedMemoryIds ?? [];
const linkedMemoryIds = payload.linkedMemoryIds ?? [];
if (!Array.isArray(linkedMemoryIds)) continue;
// Spread-attenuated boost
const numLinked = Math.max(
linkedMemoryIds.length,
1,
);
const numLinked = Math.max(linkedMemoryIds.length, 1);
const memoryCountWeight =
1.0 /
(1.0 + 0.001 * (numLinked - 1) ** 2);
1.0 / (1.0 + 0.001 * (numLinked - 1) ** 2);
const boost =
similarity *
ENTITY_BOOST_WEIGHT *
memoryCountWeight;
similarity * ENTITY_BOOST_WEIGHT * memoryCountWeight;
for (const memoryId of linkedMemoryIds) {
if (memoryId) {
@@ -990,10 +961,7 @@ export class Memory {
metadata: {
...Object.entries(payload)
.filter(([key]) => !excludedKeys.has(key))
.reduce(
(acc, [key, value]) => ({ ...acc, [key]: value }),
{},
),
.reduce((acc, [key, value]) => ({ ...acc, [key]: value }), {}),
scoreBreakdown: scored.scoreBreakdown,
},
...(payload.userId && { userId: payload.userId }),
mem0-ts/src/oss/src/utils/entity_extraction.ts
+12 -10
View File
@@ -425,7 +425,10 @@ function extractProper(text: string): ExtractedEntity[] {
}
const isLabel = i + 1 < tokens.length && tokens[i + 1] === ":";
const isCap = tok.length > 0 && tok.charAt(0) === tok.charAt(0).toUpperCase() && /[A-Z]/.test(tok.charAt(0));
const isCap =
tok.length > 0 &&
tok.charAt(0) === tok.charAt(0).toUpperCase() &&
/[A-Z]/.test(tok.charAt(0));
if (isCap && !isLabel) {
const seq: Array<{ token: string; idx: number }> = [
@@ -508,7 +511,9 @@ function extractCompoundsWithNlp(text: string): ExtractedEntity[] {
if (GENERIC_HEADS.has(head)) {
// Check if there's a specific modifier
const hasSpecificMod = words.some(
(w) => !NON_SPECIFIC_ADJ.has(w.toLowerCase()) && w !== words[words.length - 1],
(w) =>
!NON_SPECIFIC_ADJ.has(w.toLowerCase()) &&
w !== words[words.length - 1],
);
if (!hasSpecificMod) {
continue;
@@ -516,7 +521,9 @@ function extractCompoundsWithNlp(text: string): ExtractedEntity[] {
}
// Filter non-specific adjectives from the beginning
const filtered = words.filter((w) => !NON_SPECIFIC_ADJ.has(w.toLowerCase()));
const filtered = words.filter(
(w) => !NON_SPECIFIC_ADJ.has(w.toLowerCase()),
);
const cleaned = stripGenericEnding(filtered);
if (cleaned.length >= 2) {
@@ -560,16 +567,11 @@ function extractCompoundsRegex(text: string): ExtractedEntity[] {
}
// Also try lowercase compound patterns (e.g., "machine learning", "deep learning")
const lowerCompoundRe =
/\b([a-z]+(?:\s+[a-z]+){1,3})\b/g;
const lowerCompoundRe = /\b([a-z]+(?:\s+[a-z]+){1,3})\b/g;
while ((match = lowerCompoundRe.exec(text)) !== null) {
const phrase = match[1].trim();
const words = phrase.split(/\s+/);
if (
words.length >= 2 &&
words.length <= 4 &&
phrase.length > 5
) {
if (words.length >= 2 && words.length <= 4 && phrase.length > 5) {
const head = words[words.length - 1].toLowerCase();
const allGeneric = words.every(
(w) =>
mem0-ts/src/oss/src/utils/lemmatization.ts
+1 -5
View File
@@ -269,11 +269,7 @@ export function lemmatizeForBm25(text: string): string {
// Also add original if it ends in -ing and differs from stem.
// This handles noun/verb ambiguity (meeting/meet, attending/attend).
if (
word.endsWith("ing") &&
word !== stemmed &&
/^[a-z0-9]+$/.test(word)
) {
if (word.endsWith("ing") && word !== stemmed && /^[a-z0-9]+$/.test(word)) {
tokens.push(word);
}
}
mem0-ts/src/oss/tests/memory.add.test.ts
+3 -4
View File
@@ -111,10 +111,9 @@ describe("Memory - add()", () => {
});
test("result item has a memory string field", async () => {
const result: SearchResult = await memory.add(
"My favorite color is blue",
{ userId },
);
const result: SearchResult = await memory.add("My favorite color is blue", {
userId,
});
expect(typeof result.results[0].memory).toBe("string");
});
mem0-ts/src/oss/tests/memory.crud.test.ts
+1 -3
View File
@@ -30,9 +30,7 @@ jest.mock("../src/llms/openai", () => ({
const newMsgMatch = content.match(
/## New Messages\n([\s\S]*?)(?=\n##|$)/,
);
const extracted = newMsgMatch
? newMsgMatch[1].trim()
: "stored fact";
const extracted = newMsgMatch ? newMsgMatch[1].trim() : "stored fact";
return JSON.stringify({
memory: [
{
mem0-ts/src/oss/tests/memory.init.test.ts
+1 -3
View File
@@ -28,9 +28,7 @@ jest.mock("../src/llms/openai", () => ({
const newMsgMatch = content.match(
/## New Messages\n([\s\S]*?)(?=\n##|$)/,
);
const extracted = newMsgMatch
? newMsgMatch[1].trim()
: "test fact";
const extracted = newMsgMatch ? newMsgMatch[1].trim() : "test fact";
return JSON.stringify({
memory: [
{