UNPKG

ai-sdk-ollama

Version:

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

28 lines 1.03 kB
import { ChatResponse, Ollama } from 'ollama'; import { OllamaChatSettings } from '../provider.js'; import { OllamaChatLanguageModel } from './chat-language-model.js'; export declare const mockOllamaClient: Ollama; export declare function createModel(settings?: OllamaChatSettings): OllamaChatLanguageModel; /** * Mock a streaming chat response (consistent with AI SDK provider patterns). */ export declare function mockChatStream(data: ChatResponse[]): void; /** * The V4 usage shape for a given token count. `raw` is the provider-shaped * passthrough; its contents are asserted in the dedicated `usage.raw` tests. */ export declare function createExpectedUsage(inputTokens: number, outputTokens: number): { inputTokens: { total: number; noCache: number; cacheRead: undefined; cacheWrite: undefined; }; outputTokens: { total: number; text: number; reasoning: undefined; }; raw: any; }; //# sourceMappingURL=chat-language-model.test-helpers.d.ts.map