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
@@ -1,27 +1,23 @@
|
||||
import { LanguageModelV1, ProviderV1 } from "@ai-sdk/provider";
|
||||
import { loadApiKey, withoutTrailingSlash } from "@ai-sdk/provider-utils";
|
||||
import { ProviderV2 } from '@ai-sdk/provider';
|
||||
import { LanguageModelV2 } from '@ai-sdk/provider';
|
||||
import { withoutTrailingSlash } from "@ai-sdk/provider-utils";
|
||||
import { Mem0ChatModelId, Mem0ChatSettings, Mem0Config } from "./mem0-types";
|
||||
import { OpenAIProviderSettings } from "@ai-sdk/openai";
|
||||
import { Mem0GenericLanguageModel } from "./mem0-generic-language-model";
|
||||
import { OpenAIChatSettings } from "@ai-sdk/openai/internal";
|
||||
import { AnthropicMessagesSettings } from "@ai-sdk/anthropic/internal";
|
||||
import { AnthropicProviderSettings } from "@ai-sdk/anthropic";
|
||||
import { LLMProviderSettings } from "./mem0-types";
|
||||
|
||||
export interface Mem0Provider extends ProviderV1 {
|
||||
(modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV1;
|
||||
export interface Mem0Provider extends ProviderV2 {
|
||||
(modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV2;
|
||||
|
||||
chat(modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV1;
|
||||
completion(modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV1;
|
||||
chat(modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV2;
|
||||
completion(modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV2;
|
||||
|
||||
languageModel(
|
||||
modelId: Mem0ChatModelId,
|
||||
settings?: Mem0ChatSettings
|
||||
): LanguageModelV1;
|
||||
): LanguageModelV2;
|
||||
}
|
||||
|
||||
export interface Mem0ProviderSettings
|
||||
extends OpenAIChatSettings,
|
||||
AnthropicMessagesSettings {
|
||||
export interface Mem0ProviderSettings {
|
||||
baseURL?: string;
|
||||
/**
|
||||
* Custom fetch implementation. You can use it as a middleware to intercept
|
||||
@@ -46,7 +42,7 @@ export interface Mem0ProviderSettings
|
||||
/**
|
||||
* The configuration for the provider.
|
||||
*/
|
||||
config?: OpenAIProviderSettings | AnthropicProviderSettings;
|
||||
config?: LLMProviderSettings ;
|
||||
}
|
||||
|
||||
export function createMem0(
|
||||
|
||||
Reference in New Issue
Block a user