feat(vercel-ai-sdk): migrate to Vercel AI SDK v6 (#4741)
Co-authored-by: kartik-mem0 <kartik.labhshetwar@mem0.ai>
This commit is contained in:
co-authored by
kartik-mem0
parent
b819d95d18
commit
b4a50e3dc8
@@ -1,5 +1,5 @@
|
||||
import { addMemories, retrieveMemories } from "../src";
|
||||
import { LanguageModelV2Prompt } from '@ai-sdk/provider';
|
||||
import { LanguageModelV3Prompt } from '@ai-sdk/provider';
|
||||
import { testConfig } from "../config/test-config";
|
||||
|
||||
describe("Memory Core Functions", () => {
|
||||
@@ -8,7 +8,7 @@ describe("Memory Core Functions", () => {
|
||||
|
||||
describe("addMemories", () => {
|
||||
it("should successfully add memories and return correct format", async () => {
|
||||
const messages: LanguageModelV2Prompt = [
|
||||
const messages: LanguageModelV3Prompt = [
|
||||
{
|
||||
role: "user",
|
||||
content: [
|
||||
@@ -34,7 +34,7 @@ describe("Memory Core Functions", () => {
|
||||
describe("retrieveMemories", () => {
|
||||
beforeEach(async () => {
|
||||
// Add some test memories before each retrieval test
|
||||
const messages: LanguageModelV2Prompt = [
|
||||
const messages: LanguageModelV3Prompt = [
|
||||
{
|
||||
role: "user",
|
||||
content: [
|
||||
@@ -56,7 +56,7 @@ describe("Memory Core Functions", () => {
|
||||
});
|
||||
|
||||
it("should retrieve memories with array of prompts", async () => {
|
||||
const messages: LanguageModelV2Prompt = [
|
||||
const messages: LanguageModelV3Prompt = [
|
||||
{
|
||||
role: "user",
|
||||
content: [
|
||||
|
||||
Reference in New Issue
Block a user