rawi/dist/libs/providers/ollama/ollama-provider.d.ts

Rawi (راوي) is the developer-friendly AI CLI that brings the power of 11 major AI providers directly to your terminal. With seamless shell integration, persistent conversations, and 200+ specialized prompt templates, Rawi transforms your command line into

import { OllamaChatModelId, OllamaChatSettings } from './ollama-chat-settings.js';
import { ProviderV2, LanguageModelV2, EmbeddingModelV2 } from '@ai-sdk/provider';
import { OllamaEmbeddingModelId, OllamaEmbeddingSettings } from './ollama-embedding-settings.js';
import './ollama-models-list.js';

interface OllamaProvider extends ProviderV2 {
    (modelId: OllamaChatModelId, settings?: OllamaChatSettings): LanguageModelV2;
    chat(modelId: OllamaChatModelId, settings?: OllamaChatSettings): LanguageModelV2;
    embedding(modelId: OllamaEmbeddingModelId, settings?: OllamaEmbeddingSettings): EmbeddingModelV2<string>;
    languageModel(modelId: OllamaChatModelId, settings?: OllamaChatSettings): LanguageModelV2;
    textEmbeddingModel(modelId: OllamaEmbeddingModelId, settings?: OllamaEmbeddingSettings): EmbeddingModelV2<string>;
}
interface OllamaProviderSettings {
    /**
     * Base URL for Ollama API calls.
     */
    baseURL?: string;
    /**
     * Custom fetch implementation. You can use it as a middleware to intercept
     * requests or to provide a custom fetch implementation for e.g. testing
     */
    fetch?: typeof fetch;
    /**
     * @internal
     */
    generateId?: () => string;
    /**
     * Custom headers to include in the requests.
     */
    headers?: Record<string, string>;
}
declare function createOllama(options?: OllamaProviderSettings): OllamaProvider;
declare const ollama: OllamaProvider;

export { type OllamaProvider, type OllamaProviderSettings, createOllama, ollama };