UNPKG

@nomyx/assistant

Version:

A powerful assistant library and cli for your AI projects. works with Vertex AI (Claude and Gemini)

22 lines (21 loc) 1.16 kB
import { ChatMessage, ChatOptions, ProviderResponse } from '../../../types/chat'; import { ProviderCapabilities, EmbeddingProvider } from '../../../types/provider'; import { Tool, GenericToolSchema, StandardizedToolCall } from '../../../types/tool'; import { ILogger } from '../../../types/common'; import { BaseProvider } from '../BaseProvider'; import { VertexProviderConfig } from './types'; export declare class VertexProvider extends BaseProvider implements EmbeddingProvider { private config; private cache; constructor(config: VertexProviderConfig, logger: ILogger); chat(messages: ChatMessage[], options: ChatOptions, tools?: Tool[]): Promise<ProviderResponse>; streamChat(messages: ChatMessage[], options: ChatOptions, tools?: Tool[]): AsyncIterableIterator<ProviderResponse>; getCapabilities(): ProviderCapabilities; embed(text: string): Promise<number[]>; embedBatch(texts: string[]): Promise<number[][]>; convertToolSchema(schema: GenericToolSchema): any; convertToolCall(call: any): StandardizedToolCall; clearCache(): void; } export * from './types'; export { VertexProviderError } from './errors';