UNPKG

@cellular-ai/engine

Version:

API for building custom AI coding editors/agents/platforms

60 lines (59 loc) 1.89 kB
import { Content, FunctionDeclaration } from '@google/genai'; export interface ToolRequestData { callId: string; name: string; args: Record<string, unknown>; } export interface ToolStartData { callId: string; name: string; args: Record<string, unknown>; } export interface ToolResultData { callId: string; name: string; args: Record<string, unknown>; result: string; duration: number; success: boolean; } export interface ToolErrorData { callId: string; name: string; args: Record<string, unknown>; error: string; duration: number; } export type EngineConfig = { dir: string; fullContext?: boolean; model?: 'pro' | 'flash' | 'mini'; apikey?: string; sessionId?: string; debug: boolean; }; declare class EngineService { private client; private config; private apikey; private sessionId; private debug; private toolRegistry?; private initialized; private memoryContent; constructor(config: EngineConfig); getHistory(): Promise<Content[]>; setHistory(history: Content[]): Promise<void>; clearHistory(): Promise<void>; stream(message: string, context?: string): AsyncGenerator<string, void, unknown>; streamWithToolEvents(message: string, context?: string): AsyncGenerator<{ type: 'text' | 'tool_request' | 'tool_start' | 'tool_result' | 'tool_error' | 'chat_compressed'; data: string | ToolRequestData | ToolStartData | ToolResultData | ToolErrorData | unknown; }, void, unknown>; getTools(): Promise<FunctionDeclaration[]>; executeTool(toolName: string, params: Record<string, unknown>): Promise<import("@google/gemini-cli-core").ToolResult>; getMemoryContent(): string; private ensureInitialized; } declare const createEngine: (config: EngineConfig) => EngineService; export { createEngine, EngineService };