aiwrapper
Version:
A Universal AI Wrapper for JavaScript & TypeScript
25 lines (24 loc) • 969 B
TypeScript
import { LangChatMessages, LangResultWithMessages, LangResultWithString, LanguageProvider } from "../language-provider";
import { Model } from 'aimodels';
export type OllamaLangOptions = {
model?: string;
systemPrompt?: string;
maxTokens?: number;
url?: string;
};
export type OllamaLangConfig = {
model: string;
systemPrompt: string;
maxTokens?: number;
baseURL: string;
};
export declare class OllamaLang extends LanguageProvider {
protected _config: OllamaLangConfig;
protected modelInfo?: Model;
constructor(options: OllamaLangOptions);
protected transformBody(body: Record<string, unknown>): Record<string, unknown>;
ask(prompt: string, onResult?: (result: LangResultWithString) => void): Promise<LangResultWithString>;
chat(messages: LangChatMessages, onResult?: (result: LangResultWithMessages) => void): Promise<LangResultWithMessages>;
private extractThinking;
private processChunkForThinking;
}