Source code
Revision control
Copy as Markdown
Other Tools
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
import { CATEGORIES_LIST, INTENTS_LIST } from "./MemoriesConstants.sys.mjs";
/**
* JSON Schema for initial memories generation
*/
export const INITIAL_MEMORIES_SCHEMA = {
type: "array",
minItems: 1,
items: {
type: "object",
additionalProperties: false,
required: [
"category",
"intent",
"memory_summary",
"score",
"why",
"evidence",
],
properties: {
category: {
type: ["string", "null"],
enum: [...CATEGORIES_LIST, null],
},
intent: {
type: ["string", "null"],
enum: [...INTENTS_LIST, null],
},
memory_summary: { type: ["string", "null"] },
score: { type: "integer" },
why: { type: "string", minLength: 12, maxLength: 200 },
evidence: {
type: "array",
minItems: 1,
maxItems: 4,
items: {
type: "object",
required: ["type", "value"],
additionalProperties: false,
properties: {
type: {
type: "string",
enum: ["domain", "title", "search", "chat", "user"],
},
value: { type: "string" },
weight: { type: "number", minimum: 0, maximum: 1 },
session_ids: {
type: "array",
items: { type: ["integer", "string"] },
},
},
},
},
},
},
};
/**
* JSON Schema for memories deduplication
*/
export const MEMORIES_DEDUPLICATION_SCHEMA = {
type: "array",
minItems: 1,
items: {
type: "object",
additionalProperties: false,
required: ["unique_memories"],
properties: {
unique_memories: {
type: "array",
minItems: 1,
items: {
type: "object",
additionalProperties: false,
required: ["main_memory", "duplicates"],
properties: {
main_memory: { type: "string" },
duplicates: {
type: "array",
minItems: 1,
items: { type: "string" },
},
},
},
},
},
},
};
/**
* JSON schema for filtering sensitive memories
*/
export const MEMORIES_NON_SENSITIVE_SCHEMA = {
type: "array",
minItems: 1,
items: {
type: "object",
additionalProperties: false,
required: ["non_sensitive_memories"],
properties: {
non_sensitive_memories: {
type: "array",
minItems: 1,
items: { type: "string" },
},
},
},
};
/**
* JSON schema for classifying message category and intent
*/
export const MEMORIES_MESSAGE_CLASSIFY_SCHEMA = {
name: "ClassifyMessage",
schema: {
type: "object",
additionalProperties: false,
required: ["categories", "intents"],
properties: {
category: {
type: "array",
minItems: 1,
items: {
type: ["string", "null"],
enum: [...CATEGORIES_LIST, null],
},
},
intent: {
type: "array",
minItems: 1,
items: {
type: ["string", "null"],
enum: [...INTENTS_LIST, null],
},
},
},
},
};