@distri/core 0.2.4 → 0.2.7

package/dist/index.d.ts

@@ -1,14 +1,20 @@
-import { AgentSkill, Message, TaskStatusUpdateEvent, TaskArtifactUpdateEvent, Task, MessageSendParams, Part, Artifact } from '@a2a-js/sdk/client';
+import { MessageSendParams, Message, Task, Part, AgentSkill, TaskStatusUpdateEvent, TaskArtifactUpdateEvent } from '@a2a-js/sdk/client';
 export { AgentCard, Message, MessageSendParams, Task, TaskArtifactUpdateEvent, TaskStatus, TaskStatusUpdateEvent } from '@a2a-js/sdk/client';
 
 type Role = 'user' | 'system' | 'assistant';
 interface RunStartedEvent {
     type: 'run_started';
-    data: {};
+    data: {
+        runId?: string;
+        taskId?: string;
+    };
 }
 interface RunFinishedEvent {
     type: 'run_finished';
-    data: {};
+    data: {
+        runId?: string;
+        taskId?: string;
+    };
 }
 interface RunErrorEvent {
     type: 'run_error';
@@ -39,13 +45,16 @@ interface TextMessageStartEvent {
     type: 'text_message_start';
     data: {
         message_id: string;
+        step_id: string;
         role: Role;
+        is_final?: boolean;
     };
 }
 interface TextMessageContentEvent {
     type: 'text_message_content';
     data: {
         message_id: string;
+        step_id: string;
         delta: string;
     };
 }
@@ -53,34 +62,22 @@ interface TextMessageEndEvent {
     type: 'text_message_end';
     data: {
         message_id: string;
+        step_id: string;
     };
 }
-interface ToolCallStartEvent {
-    type: 'tool_call_start';
+interface ToolExecutionStartEvent {
+    type: 'tool_execution_start';
     data: {
         tool_call_id: string;
         tool_call_name: string;
         parent_message_id?: string;
+        input?: any;
     };
 }
-interface ToolCallArgsEvent {
-    type: 'tool_call_args';
-    data: {
-        tool_call_id: string;
-        delta: string;
-    };
-}
-interface ToolCallEndEvent {
-    type: 'tool_call_end';
-    data: {
-        tool_call_id: string;
-    };
-}
-interface ToolCallResultEvent {
-    type: 'tool_call_result';
+interface ToolExecutionEndEvent {
+    type: 'tool_execution_end';
     data: {
         tool_call_id: string;
-        result: string;
     };
 }
 interface ToolRejectedEvent {
@@ -90,15 +87,6 @@ interface ToolRejectedEvent {
         tool_call_id?: string;
     };
 }
-interface TaskArtifactEvent {
-    type: 'task_artifact';
-    data: {
-        artifact_id: string;
-        artifact_type: string;
-        resolution?: any;
-        content?: any;
-    };
-}
 interface AgentHandoverEvent {
     type: 'agent_handover';
     data: {
@@ -129,7 +117,405 @@ interface FeedbackReceivedEvent {
         feedback: string;
     };
 }
-type DistriEvent = RunStartedEvent | RunFinishedEvent | RunErrorEvent | PlanStartedEvent | PlanFinishedEvent | PlanPrunedEvent | TextMessageStartEvent | TextMessageContentEvent | TextMessageEndEvent | ToolCallStartEvent | ToolCallArgsEvent | ToolCallEndEvent | ToolCallResultEvent | ToolRejectedEvent | StepStartedEvent | StepCompletedEvent | TaskArtifactEvent | AgentHandoverEvent | FeedbackReceivedEvent;
+interface ToolCallsEvent {
+    type: 'tool_calls';
+    data: {
+        tool_calls: Array<{
+            tool_call_id: string;
+            tool_name: string;
+            input: any;
+        }>;
+    };
+}
+interface ToolResultsEvent {
+    type: 'tool_results';
+    data: {
+        results: Array<ToolResult>;
+    };
+}
+interface BrowserScreenshotEvent {
+    type: 'browser_screenshot';
+    data: {
+        image: string;
+        format?: string;
+        filename?: string;
+        size?: number;
+        timestamp_ms?: number;
+    };
+}
+interface InlineHookRequestedEvent {
+    type: 'inline_hook_requested';
+    data: {
+        hook_id: string;
+        hook: string;
+        context: {
+            agent_id: string;
+            thread_id: string;
+            task_id: string;
+            run_id: string;
+        };
+        timeout_ms?: number;
+        fire_and_forget?: boolean;
+        message?: any;
+        plan?: any;
+        result?: any;
+    };
+}
+type DistriEvent = RunStartedEvent | RunFinishedEvent | RunErrorEvent | PlanStartedEvent | PlanFinishedEvent | PlanPrunedEvent | TextMessageStartEvent | TextMessageContentEvent | TextMessageEndEvent | ToolExecutionStartEvent | ToolExecutionEndEvent | ToolRejectedEvent | StepStartedEvent | StepCompletedEvent | AgentHandoverEvent | FeedbackReceivedEvent | ToolCallsEvent | ToolResultsEvent | BrowserScreenshotEvent | InlineHookRequestedEvent;
+
+type ChatCompletionRole = 'system' | 'user' | 'assistant' | 'tool';
+interface ChatCompletionMessage {
+    role: ChatCompletionRole;
+    content: string;
+}
+type ChatCompletionResponseFormat = {
+    type: 'text';
+} | {
+    type: 'json_schema';
+    json_schema: {
+        name: string;
+        schema: Record<string, unknown>;
+        strict?: boolean;
+    };
+};
+interface ChatCompletionRequest {
+    model?: string;
+    messages: ChatCompletionMessage[];
+    temperature?: number;
+    max_tokens?: number;
+    response_format?: ChatCompletionResponseFormat;
+    tools?: unknown[];
+    tool_choice?: 'none' | 'auto' | Record<string, unknown>;
+}
+interface ChatCompletionChoice {
+    index: number;
+    finish_reason?: string | null;
+    message: ChatCompletionMessage;
+}
+interface ChatCompletionResponse {
+    id: string;
+    created: number;
+    model: string;
+    object: string;
+    choices: ChatCompletionChoice[];
+    usage?: {
+        prompt_tokens?: number;
+        completion_tokens?: number;
+        total_tokens?: number;
+    };
+}
+/**
+ * Enhanced Distri Client that wraps A2AClient and adds Distri-specific features
+ *
+ * @example
+ * // Local development
+ * const client = new DistriClient({ baseUrl: 'http://localhost:3033' });
+ *
+ * // Cloud with default URL (https://api.distri.dev)
+ * const client = DistriClient.create();
+ */
+declare class DistriClient {
+    private config;
+    private accessToken?;
+    private refreshToken?;
+    private tokenRefreshSkewMs;
+    private onTokenRefresh?;
+    private refreshPromise?;
+    private agentClients;
+    constructor(config: DistriClientConfig);
+    /**
+     * Create a client with default cloud configuration.
+     *
+     * @param overrides - Optional overrides for the default config
+     */
+    static create(overrides?: Partial<DistriClientConfig>): DistriClient;
+    /**
+     * Check if this client has authentication configured.
+     */
+    hasAuth(): boolean;
+    /**
+     * Check if this client is configured for local development.
+     */
+    isLocal(): boolean;
+    /**
+     * Session store: set a value (optionally with expiry)
+     */
+    setSessionValue(sessionId: string, key: string, value: unknown, expiry?: Date | string): Promise<void>;
+    /**
+     * Session store: get a single value
+     */
+    getSessionValue<T = unknown>(sessionId: string, key: string): Promise<T | null>;
+    /**
+     * Session store: get all values in a session
+     */
+    getSessionValues(sessionId: string): Promise<Record<string, unknown>>;
+    /**
+     * Session store: delete a single key
+     */
+    deleteSessionValue(sessionId: string, key: string): Promise<void>;
+    /**
+     * Session store: clear all keys in a session
+     */
+    clearSession(sessionId: string): Promise<void>;
+    /**
+     * Response from the token endpoint
+     */
+    static readonly TokenType: {
+        readonly Main: "main";
+        readonly Short: "short";
+    };
+    /**
+     * Issue an access token + refresh token for temporary authentication.
+     * Requires an existing authenticated session (bearer token).
+     *
+     * @returns Token response with access/refresh token strings
+     * @throws ApiError if not authenticated or token issuance fails
+     *
+     * @example
+     * ```typescript
+     * const { access_token, refresh_token } = await client.issueToken();
+     * // Persist the refresh token and use access_token for requests
+     * ```
+     */
+    issueToken(): Promise<{
+        access_token: string;
+        refresh_token: string;
+        expires_at: number;
+    }>;
+    /**
+     * Get the current access/refresh tokens.
+     */
+    getTokens(): {
+        accessToken?: string;
+        refreshToken?: string;
+    };
+    /**
+     * Update the access/refresh tokens in memory.
+     */
+    setTokens(tokens: {
+        accessToken?: string;
+        refreshToken?: string;
+    }): void;
+    /**
+     * Start streaming speech-to-text transcription via WebSocket
+     */
+    streamingTranscription(options?: StreamingTranscriptionOptions): Promise<{
+        sendAudio: (audioData: ArrayBuffer) => void;
+        sendText: (text: string) => void;
+        stop: () => void;
+        close: () => void;
+    }>;
+    /**
+     * Transcribe audio blob to text using speech-to-text API
+     */
+    transcribe(audioBlob: Blob, config?: SpeechToTextConfig): Promise<string>;
+    getConfiguration(): Promise<ConfigurationResponse>;
+    updateConfiguration(configuration: DistriConfiguration): Promise<ConfigurationResponse>;
+    /**
+     * Minimal LLM helper that proxies to the Distri server using Distri messages.
+     */
+    llm(messages: DistriMessage[], tools?: unknown[], options?: LlmExecuteOptions): Promise<LLMResponse>;
+    /**
+     * Get all available agents from the Distri server
+     */
+    getAgents(): Promise<AgentDefinition[]>;
+    /**
+     * Get specific agent by ID
+     */
+    getAgent(agentId: string): Promise<AgentConfigWithTools>;
+    /**
+     * Get or create A2AClient for an agent
+     */
+    private getA2AClient;
+    /**
+     * Send a message to an agent
+     */
+    sendMessage(agentId: string, params: MessageSendParams): Promise<Message | Task>;
+    /**
+     * Send a streaming message to an agent
+     */
+    sendMessageStream(agentId: string, params: MessageSendParams): AsyncGenerator<A2AStreamEventData>;
+    /**
+     * Get task details
+     */
+    getTask(agentId: string, taskId: string): Promise<Task>;
+    /**
+     * Cancel a task
+     */
+    cancelTask(agentId: string, taskId: string): Promise<void>;
+    /**
+     * Get threads from Distri server
+     */
+    getThreads(): Promise<DistriThread[]>;
+    getThread(threadId: string): Promise<DistriThread>;
+    /**
+     * Get thread messages
+     */
+    getThreadMessages(threadId: string): Promise<Message[]>;
+    /**
+     * Get messages from a thread as DistriMessage format
+     */
+    getThreadMessagesAsDistri(threadId: string): Promise<DistriMessage[]>;
+    /**
+     * Send a DistriMessage to a thread
+     */
+    sendDistriMessage(threadId: string, message: DistriMessage, context: InvokeContext): Promise<void>;
+    /**
+     * Complete an external tool call
+     */
+    completeTool(agentId: string, result: ToolResult): Promise<void>;
+    /**
+     * Complete an inline hook with a mutation payload.
+     */
+    completeInlineHook(hookId: string, mutation: any): Promise<void>;
+    /**
+     * Get the base URL for making direct requests
+     */
+    get baseUrl(): string;
+    private applyTokens;
+    private ensureAccessToken;
+    private refreshTokens;
+    private performTokenRefresh;
+    private isTokenExpiring;
+    private getTokenExpiry;
+    private decodeJwtPayload;
+    private decodeBase64Url;
+    private applyAuthHeader;
+    /**
+     * Enhanced fetch with retry logic
+     */
+    private fetchAbsolute;
+    /**
+     * Enhanced fetch with retry logic and auth headers.
+     * Exposed publicly for extensions like DistriHomeClient.
+     */
+    fetch(input: RequestInfo | URL, initialInit?: RequestInit): Promise<Response>;
+    /**
+     * Delay utility
+     */
+    private delay;
+    /**
+     * Debug logging
+     */
+    private debug;
+    /**
+     * Helper method to create A2A messages
+     */
+    static initMessage(parts: Part[] | string, role: "agent" | "user" | undefined, message: Omit<Partial<Message>, 'parts' | 'role' | 'kind'>): Message;
+    /**
+     * Create a DistriMessage instance
+     */
+    static initDistriMessage(role: DistriMessage['role'], parts: DistriPart[], id?: string, created_at?: number): DistriMessage;
+    /**
+     * Helper method to create message send parameters
+     */
+    static initMessageParams(message: Message, configuration?: MessageSendParams['configuration'], metadata?: any): MessageSendParams;
+    /**
+     * Create MessageSendParams from a DistriMessage using InvokeContext
+     */
+    static initDistriMessageParams(message: DistriMessage, context: InvokeContext): MessageSendParams;
+}
+declare function uuidv4(): string;
+
+/**
+ * Configuration for Agent invoke method
+ */
+interface InvokeConfig {
+    /** Configuration for the message */
+    configuration?: MessageSendParams['configuration'];
+    /** Context/thread ID */
+    contextId?: string;
+    /** Metadata for the requests */
+    metadata?: any;
+}
+/**
+ * Result from agent invoke
+ */
+interface InvokeResult {
+    /** Final response message */
+    message?: Message;
+    /** Task if created */
+    task?: any;
+    /** Whether the response was streamed */
+    streamed: boolean;
+}
+interface ExternalToolValidationResult {
+    isValid: boolean;
+    requiredTools: string[];
+    providedTools: string[];
+    missingTools: string[];
+    message?: string;
+}
+declare class ExternalToolValidationError extends DistriError {
+    missingTools: string[];
+    requiredTools: string[];
+    providedTools: string[];
+    agentName: string;
+    constructor(agentName: string, result: ExternalToolValidationResult);
+}
+/**
+ * Enhanced Agent class with simple tool system following AG-UI pattern
+ */
+declare class Agent {
+    private client;
+    private agentDefinition;
+    private hookHandlers;
+    private defaultHookHandler;
+    constructor(agentDefinition: AgentConfigWithTools, client: DistriClient);
+    /**
+     * Get agent information
+     */
+    get id(): string;
+    get name(): string;
+    get description(): string | undefined;
+    get agentType(): string | undefined;
+    get iconUrl(): string | undefined;
+    /**
+     * Get the full agent definition (including backend tools)
+     */
+    getDefinition(): AgentConfigWithTools;
+    /**
+     * Fetch messages for a thread (public method for useChat)
+     */
+    getThreadMessages(threadId: string): Promise<Message[]>;
+    /**
+     * Direct (non-streaming) invoke
+     */
+    invoke(params: MessageSendParams, tools?: DistriBaseTool[], hooks?: Record<string, HookHandler>): Promise<Message>;
+    /**
+     * Streaming invoke
+     */
+    invokeStream(params: MessageSendParams, tools?: DistriBaseTool[], hooks?: Record<string, HookHandler>): Promise<AsyncGenerator<DistriChatMessage>>;
+    /**
+     * Validate that required external tools are registered before invoking.
+     */
+    validateExternalTools(tools?: DistriBaseTool[]): ExternalToolValidationResult;
+    /**
+     * Enhance message params with tool definitions
+     */
+    private enhanceParamsWithTools;
+    private assertExternalTools;
+    private getRequiredExternalTools;
+    private resolveToolConfig;
+    private extractToolConfig;
+    private formatExternalToolValidationMessage;
+    /**
+     * Register multiple hooks at once.
+     */
+    registerHooks(hooks: Record<string, HookHandler>, defaultHandler?: HookHandler): void;
+    /**
+     * Create an agent instance from an agent ID
+     */
+    static create(agentIdOrDef: string | AgentConfigWithTools, client: DistriClient): Promise<Agent>;
+    /**
+     * Complete an external tool call by sending the result back to the server
+     */
+    completeTool(result: ToolResult): Promise<void>;
+    /**
+     * List all available agents
+     */
+    static list(client: DistriClient): Promise<Agent[]>;
+}
 
 /**
  * Message roles supported by Distri
@@ -142,7 +528,16 @@ interface DistriMessage {
     id: string;
     role: MessageRole;
     parts: DistriPart[];
-    created_at?: string;
+    created_at: number;
+    step_id?: string;
+    is_final?: boolean;
+}
+interface LlmExecuteOptions {
+    thread_id?: string;
+    parent_task_id?: string;
+    run_id?: string;
+    model_settings?: any;
+    is_sub_task?: boolean;
 }
 interface AssistantWithToolCalls {
     id: string;
@@ -156,6 +551,36 @@ interface AssistantWithToolCalls {
     is_external: boolean;
     reason: string | null;
 }
+interface UseToolsOptions {
+    agent?: Agent;
+    externalTools?: DistriBaseTool[];
+    executionOptions?: ToolExecutionOptions;
+}
+interface ToolExecutionOptions {
+    autoExecute?: boolean;
+}
+interface HookMutation {
+    dynamic_values: Record<string, any>;
+}
+interface HookContext {
+    agent_id: string;
+    thread_id: string;
+    task_id: string;
+    run_id: string;
+}
+interface InlineHookRequest {
+    hook_id: string;
+    hook: string;
+    context: HookContext;
+    timeout_ms?: number;
+    fire_and_forget?: boolean;
+    message?: any;
+    plan?: any;
+    result?: any;
+}
+interface InlineHookEventData extends InlineHookRequest {
+}
+type HookHandler = (req: InlineHookRequest) => Promise<HookMutation> | HookMutation;
 interface ToolResults {
     id: string;
     type: 'tool_results';
@@ -166,15 +591,6 @@ interface ToolResults {
     rejected: boolean;
     reason: string | null;
 }
-interface GenericArtifact {
-    id: string;
-    type: 'artifact';
-    timestamp: number;
-    data: any;
-    artifactId: string;
-    name: string;
-    description: string | null;
-}
 interface DistriPlan {
     id: string;
     type: 'plan';
@@ -184,8 +600,33 @@ interface DistriPlan {
 }
 interface BasePlanStep {
     id: string;
-    title: string;
 }
+interface ThoughtPlanStep extends BasePlanStep {
+    type: 'thought';
+    message: string;
+}
+interface ActionPlanStep extends BasePlanStep {
+    type: 'action';
+    action: PlanAction;
+}
+interface CodePlanStep extends BasePlanStep {
+    type: 'code';
+    code: string;
+    language: string;
+}
+interface FinalResultPlanStep extends BasePlanStep {
+    type: 'final_result';
+    content: string;
+    tool_calls: any[];
+}
+interface PlanAction {
+    tool_name?: string;
+    input?: string;
+    prompt?: string;
+    context?: any[];
+    tool_calling_config?: any;
+}
+type PlanStep = ThoughtPlanStep | ActionPlanStep | CodePlanStep | FinalResultPlanStep;
 interface LlmPlanStep extends BasePlanStep {
     type: 'llm_call';
     prompt: string;
@@ -204,82 +645,71 @@ interface ReactStep extends BasePlanStep {
     thought: string;
     action: string;
 }
-interface FinalResultStep extends BasePlanStep {
-    type: 'final_result';
-    content: string;
-    tool_calls: any[];
-}
-type PlanStep = LlmPlanStep | BatchToolCallsStep | ThoughtStep | ReactStep | FinalResultStep;
-type DistriArtifact = AssistantWithToolCalls | ToolResults | GenericArtifact | DistriPlan;
-type DistriStreamEvent = DistriMessage | DistriEvent | DistriArtifact;
+type DistriStreamEvent = DistriMessage | DistriEvent;
 /**
  * Context required for constructing A2A messages from DistriMessage
  */
 interface InvokeContext {
     thread_id: string;
     run_id?: string;
+    task_id?: string;
     getMetadata?: () => any;
 }
 /**
  * Distri message parts - equivalent to Rust enum Part
  */
 type TextPart = {
-    type: 'text';
-    text: string;
+    part_type: 'text';
+    data: string;
 };
-type CodeObservationPart = {
-    type: 'code_observation';
-    thought: string;
-    code: string;
+type ToolCallPart = {
+    part_type: 'tool_call';
+    data: ToolCall;
 };
-type ImageUrlPart = {
-    type: 'image_url';
-    image: FileUrl;
+type ToolResultRefPart = {
+    part_type: 'tool_result';
+    data: ToolResult;
 };
-type ImageBytesPart = {
-    type: 'image_bytes';
-    image: FileBytes;
+type ImagePart = {
+    part_type: 'image';
+    data: FileType;
 };
-type ImagePart = ImageUrlPart | ImageBytesPart;
 type DataPart = {
-    type: 'data';
-    data: any;
+    part_type: 'data';
+    data: object;
 };
-type ToolCallPart = {
-    type: 'tool_call';
-    tool_call: ToolCall;
-};
-type ToolResultPart = {
-    type: 'tool_result';
-    tool_result: ToolResult;
-};
-type PlanPart = {
-    type: 'plan';
-    plan: string;
-};
-type DistriPart = TextPart | CodeObservationPart | ImagePart | DataPart | ToolCallPart | ToolResultPart | PlanPart;
+type DistriPart = TextPart | ToolCallPart | ToolResultRefPart | ImagePart | DataPart;
 /**
  * File type for images
  */
 interface FileBytes {
+    type: 'bytes';
     mime_type: string;
     data: string;
     name?: string;
 }
 interface FileUrl {
+    type: 'url';
     mime_type: string;
     url: string;
     name?: string;
 }
 type FileType = FileBytes | FileUrl;
+interface ToolDefinition {
+    name: string;
+    description: string;
+    parameters: object;
+    examples?: string;
+    output_schema?: object;
+}
 /**
  * Tool definition interface following AG-UI pattern
  */
-interface DistriBaseTool {
-    name: string;
+interface DistriBaseTool extends ToolDefinition {
     type: 'function' | 'ui';
-    description: string;
-    input_schema: object;
+    is_final?: boolean;
+    autoExecute?: boolean;
+    isExternal?: boolean;
 }
 interface DistriFnTool extends DistriBaseTool {
     type: 'function';
@@ -290,7 +720,7 @@ interface DistriFnTool extends DistriBaseTool {
  * Tool handler function
  */
 interface ToolHandler {
-    (input: any): Promise<string | number | boolean | null | object>;
+    (input: any): Promise<string | number | boolean | null | DistriPart[] | object>;
 }
 /**
  * Tool call from agent
@@ -301,15 +731,41 @@ interface ToolCall {
     input: any;
 }
 /**
- * Tool result for responding to tool calls
+ * Tool result structure that can come from backend or frontend
  */
 interface ToolResult {
-    tool_call_id: string;
-    tool_name: string;
-    result: string | number | boolean | null;
+    readonly tool_call_id: string;
+    readonly tool_name: string;
+    readonly parts: readonly DistriPart[];
+}
+/**
+ * Tool result data that goes inside the parts array
+ */
+interface ToolResultData {
+    result: string | number | boolean | null | object;
     success: boolean;
     error?: string;
 }
+declare function isArrayParts(result: any): boolean;
+/**
+ * Type-safe helper to create a successful ToolResult
+ * Uses proper DistriPart structure - conversion to backend format happens in encoder
+ */
+declare function createSuccessfulToolResult(toolCallId: string, toolName: string, result: string | number | boolean | null | object | DistriPart[]): ToolResult;
+/**
+ * Type-safe helper to create a failed ToolResult
+ * Uses proper DistriPart structure - conversion to backend format happens in encoder
+ */
+declare function createFailedToolResult(toolCallId: string, toolName: string, error: string, result?: string | number | boolean | null | object): ToolResult;
+/**
+ * Type-safe helper to extract ToolResultData from a ToolResult
+ * Handles both frontend DistriPart format and backend BackendPart format
+ */
+declare function extractToolResultData(toolResult: ToolResult): ToolResultData | null;
+interface AgentConfigWithTools extends AgentDefinition {
+    markdown?: string;
+    resolved_tools?: ToolDefinition[];
+}
 /**
  * Distri-specific Agent type that wraps A2A AgentCard
  */
@@ -317,6 +773,8 @@ interface AgentDefinition {
     /** The name of the agent. */
     name: string;
     id: string;
+    /** Optional package identifier (workspace/plugin) that registered the agent */
+    package_name?: string | null;
     /** A brief description of the agent's purpose. */
     description?: string;
     /** The version of the agent. */
@@ -327,6 +785,8 @@ interface AgentDefinition {
     mcp_servers?: McpDefinition[];
     /** Settings related to the model used by the agent. */
     model_settings?: ModelSettings;
+    /** Secondary Model Settings used for analysis */
+    analysis_model_settings?: ModelSettings;
     /** The size of the history to maintain for the agent. */
     history_size?: number;
     /** The planning configuration for the agent, if any. */
@@ -337,6 +797,21 @@ interface AgentDefinition {
     skills?: AgentSkill[];
     /** List of sub-agents that this agent can transfer control to */
     sub_agents?: string[];
+    agent_type?: string;
+    context_size?: number;
+    tools?: DistriBaseTool[];
+    browser_config?: BrowserAgentConfig;
+}
+interface BrowserAgentConfig {
+    enabled?: boolean;
+    persist_session?: boolean;
+    runtime?: DistriBrowserRuntimeConfig | null;
+}
+interface DistriBrowserRuntimeConfig {
+    window_size?: [number, number];
+    headless?: boolean;
+    enable_stealth_mode?: boolean;
+    enable_real_emulation?: boolean;
 }
 interface McpDefinition {
     /** The filter applied to the tools in this MCP definition. */
@@ -350,18 +825,31 @@ interface ModelSettings {
     model: string;
     temperature: number;
     max_tokens: number;
+    context_size: number;
     top_p: number;
     frequency_penalty: number;
     presence_penalty: number;
-    max_iterations: number;
-    provider: ModelProvider;
+    provider: ModelProviderConfig;
     /** Additional parameters for the agent, if any. */
     parameters?: any;
     /** The format of the response, if specified. */
     response_format?: any;
 }
 type McpServerType = 'tool' | 'agent';
-type ModelProvider = 'openai' | 'aigateway';
+type ModelProviderName = 'openai' | 'openai_compat' | 'vllora' | string;
+type ModelProviderConfig = {
+    name: 'openai';
+} | {
+    name: 'openai_compat';
+    base_url: string;
+    project_id?: string;
+} | {
+    name: 'vllora';
+    base_url?: string;
+} | {
+    name: string;
+    [key: string]: any;
+};
 /**
  * Distri Thread type for conversation management
  */
@@ -388,22 +876,121 @@ interface ChatProps {
     agent: AgentDefinition;
     onThreadUpdate?: () => void;
 }
+interface SpeechToTextConfig {
+    model?: 'whisper-1';
+    language?: string;
+    temperature?: number;
+}
+interface StreamingTranscriptionOptions {
+    onTranscript?: (text: string, isFinal: boolean) => void;
+    onError?: (error: Error) => void;
+    onStart?: () => void;
+    onEnd?: () => void;
+}
 /**
  * Connection Status
  */
 type ConnectionStatus = 'connecting' | 'connected' | 'disconnected' | 'error';
+/**
+ * Default base URL for the Distri cloud service
+ */
+declare const DEFAULT_BASE_URL = "https://api.distri.dev";
 /**
  * Distri Client Configuration
  */
 interface DistriClientConfig {
+    /**
+     * Base URL of the Distri server
+     * Default: https://api.distri.dev
+     */
     baseUrl: string;
+    /**
+     * API version (currently unused)
+     */
     apiVersion?: string;
+    /**
+     * Request timeout in milliseconds (default: 30000)
+     */
     timeout?: number;
+    /**
+     * Number of retry attempts for failed requests (default: 3)
+     */
     retryAttempts?: number;
+    /**
+     * Delay between retry attempts in milliseconds (default: 1000)
+     */
     retryDelay?: number;
+    /**
+     * Enable debug logging
+     */
     debug?: boolean;
+    /**
+     * Custom headers to include in all requests
+     */
     headers?: Record<string, string>;
+    /**
+     * Request interceptor for modifying requests before sending
+     */
     interceptor?: (init?: RequestInit) => Promise<RequestInit | undefined>;
+    /**
+     * Access token for bearer auth (optional)
+     */
+    accessToken?: string;
+    /**
+     * Refresh token for bearer auth (optional)
+     */
+    refreshToken?: string;
+    /**
+     * Token refresh skew in milliseconds (default: 60000)
+     */
+    tokenRefreshSkewMs?: number;
+    /**
+     * Callback invoked when tokens are refreshed
+     */
+    onTokenRefresh?: (tokens: {
+        accessToken: string;
+        refreshToken: string;
+    }) => void;
+}
+interface LLMResponse {
+    finish_reason: string;
+    content: string;
+    tool_calls: ToolCall[];
+    token_usage: number;
+    tools?: any[];
+}
+interface ExternalMcpServer {
+    name: string;
+    [key: string]: any;
+}
+interface ServerConfig {
+    base_url?: string;
+    host?: string;
+    port?: number;
+    [key: string]: any;
+}
+interface DistriConfiguration {
+    name: string;
+    version: string;
+    description?: string;
+    license?: string;
+    working_directory?: string;
+    agents?: string[];
+    mcp_servers?: ExternalMcpServer[];
+    server?: ServerConfig;
+    model_settings?: ModelSettings;
+    analysis_model_settings?: ModelSettings;
+    keywords?: string[];
+    [key: string]: any;
+}
+interface ConfigurationMeta {
+    base_path: string;
+    overrides_path: string;
+    overrides_active: boolean;
+}
+interface ConfigurationResponse {
+    configuration: DistriConfiguration;
+    meta: ConfigurationMeta;
 }
 /**
  * Error Types
@@ -427,183 +1014,7 @@ declare class ConnectionError extends DistriError {
 type A2AStreamEventData = Message | TaskStatusUpdateEvent | TaskArtifactUpdateEvent | Task;
 declare function isDistriMessage(event: DistriStreamEvent): event is DistriMessage;
 declare function isDistriEvent(event: DistriStreamEvent): event is DistriEvent;
-declare function isDistriPlan(event: DistriStreamEvent): event is DistriPlan;
-declare function isDistriArtifact(event: DistriStreamEvent): event is DistriArtifact;
-type DistriChatMessage = DistriEvent | DistriMessage | DistriArtifact;
-
-/**
- * Enhanced Distri Client that wraps A2AClient and adds Distri-specific features
- */
-declare class DistriClient {
-    private config;
-    private agentClients;
-    constructor(config: DistriClientConfig);
-    /**
-     * Get all available agents from the Distri server
-     */
-    getAgents(): Promise<AgentDefinition[]>;
-    /**
-     * Get specific agent by ID
-     */
-    getAgent(agentId: string): Promise<AgentDefinition>;
-    /**
-     * Get or create A2AClient for an agent
-     */
-    private getA2AClient;
-    /**
-     * Send a message to an agent
-     */
-    sendMessage(agentId: string, params: MessageSendParams): Promise<Message | Task>;
-    /**
-     * Send a streaming message to an agent
-     */
-    sendMessageStream(agentId: string, params: MessageSendParams): AsyncGenerator<A2AStreamEventData>;
-    /**
-     * Get task details
-     */
-    getTask(agentId: string, taskId: string): Promise<Task>;
-    /**
-     * Cancel a task
-     */
-    cancelTask(agentId: string, taskId: string): Promise<void>;
-    /**
-     * Get threads from Distri server
-     */
-    getThreads(): Promise<DistriThread[]>;
-    getThread(threadId: string): Promise<DistriThread>;
-    /**
-     * Get thread messages
-     */
-    getThreadMessages(threadId: string): Promise<Message[]>;
-    /**
-     * Get messages from a thread as DistriMessage format
-     */
-    getThreadMessagesAsDistri(threadId: string): Promise<DistriMessage[]>;
-    /**
-     * Send a DistriMessage to a thread
-     */
-    sendDistriMessage(threadId: string, message: DistriMessage, context: InvokeContext): Promise<void>;
-    /**
-     * Get the base URL for making direct requests
-     */
-    get baseUrl(): string;
-    /**
-     * Enhanced fetch with retry logic
-     */
-    private fetchAbsolute;
-    /**
-     * Enhanced fetch with retry logic
-     */
-    private fetch;
-    /**
-     * Delay utility
-     */
-    private delay;
-    /**
-     * Debug logging
-     */
-    private debug;
-    /**
-     * Helper method to create A2A messages
-     */
-    static initMessage(parts: Part[] | string, role: "agent" | "user" | undefined, message: Omit<Partial<Message>, 'parts' | 'role' | 'kind'>): Message;
-    /**
-     * Create a DistriMessage instance
-     */
-    static initDistriMessage(role: DistriMessage['role'], parts: DistriPart[], id?: string, created_at?: string): DistriMessage;
-    /**
-     * Helper method to create message send parameters
-     */
-    static initMessageParams(message: Message, configuration?: MessageSendParams['configuration'], metadata?: any): MessageSendParams;
-    /**
-     * Create MessageSendParams from a DistriMessage using InvokeContext
-     */
-    static initDistriMessageParams(message: DistriMessage, context: InvokeContext): MessageSendParams;
-}
-declare function uuidv4(): string;
-
-/**
- * Configuration for Agent invoke method
- */
-interface InvokeConfig {
-    /** Configuration for the message */
-    configuration?: MessageSendParams['configuration'];
-    /** Context/thread ID */
-    contextId?: string;
-    /** Metadata for the requests */
-    metadata?: any;
-}
-/**
- * Result from agent invoke
- */
-interface InvokeResult {
-    /** Final response message */
-    message?: Message;
-    /** Task if created */
-    task?: any;
-    /** Whether the response was streamed */
-    streamed: boolean;
-}
-/**
- * Enhanced Agent class with simple tool system following AG-UI pattern
- */
-declare class Agent {
-    private client;
-    private agentDefinition;
-    private tools;
-    constructor(agentDefinition: AgentDefinition, client: DistriClient);
-    /**
-     * Add a tool to the agent (AG-UI style)
-     */
-    registerTool(tool: DistriBaseTool): void;
-    /**
-     * Add multiple tools at once
-     */
-    registerTools(tools: DistriBaseTool[]): void;
-    /**
-     * Remove a tool
-     */
-    unregisterTool(toolName: string): void;
-    /**
-     * Get all registered tools
-     */
-    getTools(): DistriBaseTool[];
-    /**
-     * Check if a tool is registered
-     */
-    hasTool(toolName: string): boolean;
-    /**
-     * Get agent information
-     */
-    get id(): string;
-    get name(): string;
-    get description(): string | undefined;
-    get iconUrl(): string | undefined;
-    /**
-     * Fetch messages for a thread (public method for useChat)
-     */
-    getThreadMessages(threadId: string): Promise<Message[]>;
-    /**
-     * Direct (non-streaming) invoke
-     */
-    invoke(params: MessageSendParams): Promise<Message>;
-    /**
-     * Streaming invoke
-     */
-    invokeStream(params: MessageSendParams): Promise<AsyncGenerator<DistriChatMessage>>;
-    /**
-     * Enhance message params with tool definitions
-     */
-    private enhanceParamsWithTools;
-    /**
-     * Create an agent instance from an agent ID
-     */
-    static create(agentIdOrDef: string | AgentDefinition, client: DistriClient): Promise<Agent>;
-    /**
-     * List all available agents
-     */
-    static list(client: DistriClient): Promise<Agent[]>;
-}
+type DistriChatMessage = DistriEvent | DistriMessage;
 
 /**
  * Converts an A2A Message to a DistriMessage
@@ -613,10 +1024,6 @@ declare function convertA2AMessageToDistri(a2aMessage: Message): DistriMessage;
  * Converts A2A status-update events to DistriEvent based on metadata type
  */
 declare function convertA2AStatusUpdateToDistri(statusUpdate: any): DistriEvent | null;
-/**
- * Converts A2A artifacts to DistriArtifact, DistriPlan, or DistriMessage based on content
- */
-declare function convertA2AArtifactToDistri(artifact: Artifact): DistriArtifact | DistriMessage | null;
 /**
  * Enhanced decoder for A2A stream events that properly handles all event types
  */
@@ -624,7 +1031,7 @@ declare function decodeA2AStreamEvent(event: any): DistriChatMessage | null;
 /**
  * Process A2A stream data (like from stream.json) and convert to DistriMessage/DistriEvent/DistriArtifact array
  */
-declare function processA2AStreamData(streamData: any[]): (DistriMessage | DistriEvent | DistriArtifact)[];
+declare function processA2AStreamData(streamData: any[]): (DistriMessage | DistriEvent)[];
 /**
  * Process A2A messages.json data and convert to DistriMessage array
  */
@@ -654,4 +1061,4 @@ declare function extractToolCallsFromDistriMessage(message: DistriMessage): any[
  */
 declare function extractToolResultsFromDistriMessage(message: DistriMessage): any[];
 
-export { A2AProtocolError, type A2AStreamEventData, Agent, type AgentDefinition, type AgentHandoverEvent, ApiError, type AssistantWithToolCalls, type BasePlanStep, type BatchToolCallsStep, type ChatProps, type CodeObservationPart, ConnectionError, type ConnectionStatus, type DataPart, type DistriArtifact, type DistriBaseTool, type DistriChatMessage, DistriClient, type DistriClientConfig, DistriError, type DistriEvent, type DistriFnTool, type DistriMessage, type DistriPart, type DistriPlan, type DistriStreamEvent, type DistriThread, type FeedbackReceivedEvent, type FileBytes, type FileType, type FileUrl, type FinalResultStep, type GenericArtifact, type ImageBytesPart, type ImagePart, type ImageUrlPart, type InvokeConfig, type InvokeContext, type InvokeResult, type LlmPlanStep, type McpDefinition, type McpServerType, type MessageRole, type ModelProvider, type ModelSettings, type PlanFinishedEvent, type PlanPart, type PlanPrunedEvent, type PlanStartedEvent, type PlanStep, type ReactStep, type Role, type RunErrorEvent, type RunFinishedEvent, type RunStartedEvent, type StepCompletedEvent, type StepStartedEvent, type TaskArtifactEvent, type TextMessageContentEvent, type TextMessageEndEvent, type TextMessageStartEvent, type TextPart, type ThoughtStep, type Thread, type ToolCall, type ToolCallArgsEvent, type ToolCallEndEvent, type ToolCallPart, type ToolCallResultEvent, type ToolCallStartEvent, type ToolHandler, type ToolRejectedEvent, type ToolResult, type ToolResultPart, type ToolResults, convertA2AArtifactToDistri, convertA2AMessageToDistri, convertA2APartToDistri, convertA2AStatusUpdateToDistri, convertDistriMessageToA2A, convertDistriPartToA2A, decodeA2AStreamEvent, extractTextFromDistriMessage, extractToolCallsFromDistriMessage, extractToolResultsFromDistriMessage, isDistriArtifact, isDistriEvent, isDistriMessage, isDistriPlan, processA2AMessagesData, processA2AStreamData, uuidv4 };
+export { A2AProtocolError, type A2AStreamEventData, type ActionPlanStep, Agent, type AgentConfigWithTools, type AgentDefinition, type AgentHandoverEvent, ApiError, type AssistantWithToolCalls, type BasePlanStep, type BatchToolCallsStep, type BrowserAgentConfig, type BrowserScreenshotEvent, type ChatCompletionChoice, type ChatCompletionMessage, type ChatCompletionRequest, type ChatCompletionResponse, type ChatCompletionResponseFormat, type ChatCompletionRole, type ChatProps, type CodePlanStep, type ConfigurationMeta, type ConfigurationResponse, ConnectionError, type ConnectionStatus, DEFAULT_BASE_URL, type DataPart, type DistriBaseTool, type DistriBrowserRuntimeConfig, type DistriChatMessage, DistriClient, type DistriClientConfig, type DistriConfiguration, DistriError, type DistriEvent, type DistriFnTool, type DistriMessage, type DistriPart, type DistriPlan, type DistriStreamEvent, type DistriThread, type ExternalMcpServer, ExternalToolValidationError, type ExternalToolValidationResult, type FeedbackReceivedEvent, type FileBytes, type FileType, type FileUrl, type FinalResultPlanStep, type HookContext, type HookHandler, type HookMutation, type ImagePart, type InlineHookEventData, type InlineHookRequest, type InlineHookRequestedEvent, type InvokeConfig, type InvokeContext, type InvokeResult, type LLMResponse, type LlmExecuteOptions, type LlmPlanStep, type McpDefinition, type McpServerType, type MessageRole, type ModelProviderConfig, type ModelProviderName, type ModelSettings, type PlanAction, type PlanFinishedEvent, type PlanPrunedEvent, type PlanStartedEvent, type PlanStep, type ReactStep, type Role, type RunErrorEvent, type RunFinishedEvent, type RunStartedEvent, type ServerConfig, type SpeechToTextConfig, type StepCompletedEvent, type StepStartedEvent, type StreamingTranscriptionOptions, type TextMessageContentEvent, type TextMessageEndEvent, type TextMessageStartEvent, type TextPart, type ThoughtPlanStep, type ThoughtStep, type Thread, type ToolCall, type ToolCallPart, type ToolCallsEvent, type ToolDefinition, type ToolExecutionEndEvent, type ToolExecutionOptions, type ToolExecutionStartEvent, type ToolHandler, type ToolRejectedEvent, type ToolResult, type ToolResultData, type ToolResultRefPart, type ToolResults, type ToolResultsEvent, type UseToolsOptions, convertA2AMessageToDistri, convertA2APartToDistri, convertA2AStatusUpdateToDistri, convertDistriMessageToA2A, convertDistriPartToA2A, createFailedToolResult, createSuccessfulToolResult, decodeA2AStreamEvent, extractTextFromDistriMessage, extractToolCallsFromDistriMessage, extractToolResultData, extractToolResultsFromDistriMessage, isArrayParts, isDistriEvent, isDistriMessage, processA2AMessagesData, processA2AStreamData, uuidv4 };