ai-sdk-ollama
Version:
Vercel AI SDK Provider for Ollama using official ollama-js library
35 lines • 1.31 kB
TypeScript
import { EmbeddingModelV4, EmbeddingModelV4Embedding, SharedV4ProviderMetadata, SharedV4ProviderOptions, SharedV4Warning } from '@ai-sdk/provider';
import { Ollama } from 'ollama';
import { OllamaEmbeddingSettings } from '../provider.js';
export interface OllamaEmbeddingConfig {
client: Ollama;
provider: string;
}
export declare class OllamaEmbeddingModel implements EmbeddingModelV4 {
private readonly settings;
private readonly config;
readonly specificationVersion: 'v4';
readonly modelId: string;
readonly maxEmbeddingsPerCall = 2048;
readonly supportsParallelCalls = true;
constructor(modelId: string, settings: OllamaEmbeddingSettings, config: OllamaEmbeddingConfig);
get provider(): string;
doEmbed(parameters: {
values: string[];
abortSignal?: AbortSignal;
providerOptions?: SharedV4ProviderOptions;
headers?: Record<string, string | undefined>;
}): Promise<{
embeddings: EmbeddingModelV4Embedding[];
usage?: {
tokens: number;
};
providerMetadata?: SharedV4ProviderMetadata;
response?: {
headers?: Record<string, string>;
body?: unknown;
};
warnings: SharedV4Warning[];
}>;
}
//# sourceMappingURL=embedding-model.d.ts.map