55 lines
1.4 KiB
TypeScript
55 lines
1.4 KiB
TypeScript
|
import { Memory } from "mem0ai/oss";
|
||
|
|
|
||
|
|
const memory = new Memory({
|
||
|
|
version: "v1.1",
|
||
|
|
embedder: {
|
||
|
|
provider: "openai",
|
||
|
|
config: {
|
||
|
|
baseURL: "https://api.siliconflow.cn/v1/",
|
||
|
|
apiKey: "sk-tnynjffzcdudlflrydljyxttorrtstpviyuodnyuyyalzfan",
|
||
|
|
model: "BAAI/bge-m3"
|
||
|
|
}
|
||
|
|
},
|
||
|
|
llm: {
|
||
|
|
provider: "openai",
|
||
|
|
config: {
|
||
|
|
baseURL: "https://api.siliconflow.cn/v1/",
|
||
|
|
apiKey: "sk-tnynjffzcdudlflrydljyxttorrtstpviyuodnyuyyalzfan",
|
||
|
|
model: "Qwen/Qwen2.5-7B-Instruct"
|
||
|
|
}
|
||
|
|
},
|
||
|
|
vectorStore: {
|
||
|
|
provider: "memory",
|
||
|
|
config: {
|
||
|
|
collectionName: "memories",
|
||
|
|
dimension: 1024
|
||
|
|
}
|
||
|
|
},
|
||
|
|
historyDbPath: "./memory.db"
|
||
|
|
})
|
||
|
|
|
||
|
|
const messages = [
|
||
|
|
{ role: "user", content: "I'm planning to watch a movie tonight. Any recommendations?" },
|
||
|
|
{ role: "assistant", content: "How about thriller movies? They can be quite engaging." },
|
||
|
|
{ role: "user", content: "I'm not a big fan of thriller movies but I love sci-fi movies." },
|
||
|
|
{ role: "assistant", content: "Got it! I'll avoid thriller recommendations and suggest sci-fi movies in the future." }
|
||
|
|
];
|
||
|
|
|
||
|
|
try {
|
||
|
|
const insertContent = await memory.add(messages, {"userId": "huyang", metadata: {
|
||
|
|
category: "movie_recommendations"
|
||
|
|
}})
|
||
|
|
if (insertContent) {
|
||
|
|
console.log(
|
||
|
|
"Insert Ok...",
|
||
|
|
"\r\n",
|
||
|
|
insertContent.results
|
||
|
|
)
|
||
|
|
} else {
|
||
|
|
console.log(
|
||
|
|
"Insert Error"
|
||
|
|
)
|
||
|
|
}
|
||
|
|
}catch (e) {
|
||
|
|
console.log(e)
|
||
|
|
}
|