Vercel AI SDK migration to V5 (#3223)
Co-authored-by: Saket Aryan <94069182+whysosaket@users.noreply.github.com>
This commit is contained in:
co-authored by
Saket Aryan
parent
42e60d6724
commit
0e18f54d36
@@ -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 { createAnthropic } from "@ai-sdk/anthropic";
|
||||
|
||||
@@ -19,7 +20,7 @@ describe("ANTHROPIC Integration Tests", () => {
|
||||
});
|
||||
|
||||
it("should retrieve memories and generate text using ANTHROPIC provider", async () => {
|
||||
const messages: LanguageModelV1Prompt = [
|
||||
const messages: LanguageModelV2Prompt = [
|
||||
{
|
||||
role: "user",
|
||||
content: [
|
||||
|
||||
@@ -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 { createCohere } from "@ai-sdk/cohere";
|
||||
|
||||
@@ -18,7 +19,7 @@ describe("COHERE Integration Tests", () => {
|
||||
});
|
||||
|
||||
it("should retrieve memories and generate text using COHERE provider", async () => {
|
||||
const messages: LanguageModelV1Prompt = [
|
||||
const messages: LanguageModelV2Prompt = [
|
||||
{
|
||||
role: "user",
|
||||
content: [
|
||||
|
||||
@@ -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
|
||||
});
|
||||
|
||||
@@ -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 { createGroq } from "@ai-sdk/groq";
|
||||
|
||||
@@ -19,7 +20,7 @@ describe("GROQ Integration Tests", () => {
|
||||
});
|
||||
|
||||
it("should retrieve memories and generate text using GROQ provider", async () => {
|
||||
const messages: LanguageModelV1Prompt = [
|
||||
const messages: LanguageModelV2Prompt = [
|
||||
{
|
||||
role: "user",
|
||||
content: [
|
||||
|
||||
@@ -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 { createOpenAI } from "@ai-sdk/openai";
|
||||
|
||||
@@ -18,7 +19,7 @@ describe("OPENAI Integration Tests", () => {
|
||||
});
|
||||
|
||||
it("should retrieve memories and generate text using OpenAI provider", async () => {
|
||||
const messages: LanguageModelV1Prompt = [
|
||||
const messages: LanguageModelV2Prompt = [
|
||||
{
|
||||
role: "user",
|
||||
content: [
|
||||
|
||||
Reference in New Issue
Block a user