Add Infer Property (#2452)

This commit is contained in:
Saket Aryan
2025-03-27 21:18:42 +05:30
committed by GitHub
parent 0eb53b9f27
commit bc4ab3db77
10 changed files with 202 additions and 349 deletions

View File

@@ -112,7 +112,7 @@ export class Memory {
runId,
metadata = {},
filters = {},
prompt,
infer = true,
} = config;
if (userId) filters.userId = metadata.userId = userId;
@@ -136,6 +136,7 @@ export class Memory {
final_parsedMessages,
metadata,
filters,
infer,
);
// Add to graph store if available
@@ -161,7 +162,27 @@ export class Memory {
messages: Message[],
metadata: Record<string, any>,
filters: SearchFilters,
infer: boolean,
): Promise<MemoryItem[]> {
if (!infer) {
const returnedMemories: MemoryItem[] = [];
for (const message of messages) {
if (message.content === "system") {
continue;
}
const memoryId = await this.createMemory(
message.content as string,
{},
metadata,
);
returnedMemories.push({
id: memoryId,
memory: message.content as string,
metadata: { event: "ADD" },
});
}
return returnedMemories;
}
const parsedMessages = messages.map((m) => m.content).join("\n");
// Get prompts

View File

@@ -10,7 +10,7 @@ export interface Entity {
export interface AddMemoryOptions extends Entity {
metadata?: Record<string, any>;
filters?: SearchFilters;
prompt?: string;
infer?: boolean;
}
export interface SearchMemoryOptions extends Entity {