ai-sdk-ollama
Version:
Vercel AI SDK Provider for Ollama using official ollama-js library
28 lines • 1.03 kB
TypeScript
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