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/utils-test/google-integration.test.ts
+5 -4
View File
@@ -2,7 +2,8 @@ import dotenv from "dotenv";
dotenv.config();
import { retrieveMemories } from "../../src";
import { generateText, LanguageModelV1Prompt } from "ai";
import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config";
import { createGoogleGenerativeAI } from "@ai-sdk/google";
@@ -18,7 +19,7 @@ describe("GOOGLE Integration Tests", () => {
});
it("should retrieve memories and generate text using Google provider", async () => {
const messages: LanguageModelV1Prompt = [
const messages: LanguageModelV2Prompt = [
{
role: "user",
content: [
@@ -32,7 +33,7 @@ describe("GOOGLE Integration Tests", () => {
const memories = await retrieveMemories(messages, { user_id: userId });
const { text } = await generateText({
model: google("gemini-2.5-pro-preview-05-06"),
model: google("gemini-1.5-flash"),
messages: messages,
system: memories,
});
@@ -47,7 +48,7 @@ describe("GOOGLE Integration Tests", () => {
const memories = await retrieveMemories(prompt, { user_id: userId });
const { text } = await generateText({
model: google("gemini-2.5-pro-preview-05-06"),
model: google("gemini-1.5-flash"),
prompt: prompt,
system: memories
});