UNPKG

ai-sdk-ollama

Version:

Vercel AI SDK Provider for Ollama using official ollama-js library

35 lines 1.31 kB
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