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

Vercel AI SDK migration to V5 (#3223)

Browse Source 
Co-authored-by: Saket Aryan <94069182+whysosaket@users.noreply.github.com>
This commit is contained in:
Parshva Daftari
2025-08-05 20:27:57 +05:30
committed by GitHub
co-authored by Saket Aryan
parent 42e60d6724
commit 0e18f54d36
23 changed files with 321 additions and 423 deletions
Download Patch File Download Diff File Expand all files Collapse all files
vercel-ai-sdk/tests/mem0-provider-tests/mem0-google.test.ts
+5 -4
View File
@@ -2,7 +2,8 @@ import dotenv from "dotenv";
dotenv.config();
import { createMem0 } from "../../src";
import { generateText, LanguageModelV1Prompt } from "ai";
import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config";
describe("GOOGLE MEM0 Tests", () => {
@@ -22,7 +23,7 @@ describe("GOOGLE MEM0 Tests", () => {
});
it("should retrieve memories and generate text using Google provider", async () => {
const messages: LanguageModelV1Prompt = [
const messages: LanguageModelV2Prompt = [
{
role: "user",
content: [
@@ -34,7 +35,7 @@ describe("GOOGLE MEM0 Tests", () => {
const { text } = await generateText({
// @ts-ignore
model: mem0("gemini-2.5-pro-preview-05-06"),
model: mem0("gemini-1.5-flash"),
messages: messages
});
@@ -48,7 +49,7 @@ describe("GOOGLE MEM0 Tests", () => {
const { text } = await generateText({
// @ts-ignore
model: mem0("gemini-2.5-pro-preview-05-06"),
model: mem0("gemini-1.5-flash"),
prompt: prompt
});