jorel
Version:
The easiest way to use LLMs, including streams, images, documents, tools and various agent scenarios.
17 lines (16 loc) • 942 B
TypeScript
import { Ollama } from "ollama";
import { LlmCoreProvider, LlmGenerationConfig, LlmMessage, LlmResponse, LlmStreamResponse, LlmStreamResponseChunk, LlmStreamResponseWithToolCalls } from "../../providers";
export interface OllamaConfig {
name?: string;
}
/** Provides access to local Ollama server */
export declare class OllamaProvider implements LlmCoreProvider {
readonly name: string;
static readonly defaultName = "ollama";
get client(): Ollama;
constructor({ name }?: OllamaConfig);
generateResponse(model: string, messages: LlmMessage[], config?: LlmGenerationConfig): Promise<LlmResponse>;
generateResponseStream(model: string, messages: LlmMessage[], config?: LlmGenerationConfig): AsyncGenerator<LlmStreamResponseChunk | LlmStreamResponse | LlmStreamResponseWithToolCalls, void, unknown>;
getAvailableModels(): Promise<string[]>;
createEmbedding(model: string, text: string): Promise<number[]>;
}