@hugex/ai 2.9.1 → 2.9.4

package/dist/index.d.ts

@@ -1,438 +1,27 @@
-import * as ai from 'ai';
-import { Agent, Connection, ConnectionContext, WSMessage } from 'agents';
-import { NodeType, UserMessagePayload, UserInterface, Message, MessageType, ModelEntry, EffectiveParams, ResolvedModel } from './common.js';
-export { ApprovalRequestPayload, Capabilities, ChatHistoryItem, EventItem, FlowConfig, FlowNode, GENERATION_PARAM_KEYS, GENERATION_PRESETS, GenerationParamKey, GenerationPresetId, MAX_USER_IDENTITY_LENGTH, MODEL_REGISTRY_SNAPSHOT, ModelRegistrySnapshot, NODETYPES, OutVar, PRESET_PARAM_KEYS, ParamAdjustment, ParamRange, ParamSupport, REASONING_EFFORT_LEVELS, ReasoningCapability, ReasoningEffortLevel, ResolveInput, ResolveOptions, StepInfo, TOOL_REGISTRY_SNAPSHOT, ToolExecutionResultPayload, TypingMessage, UserAuthPayload, VariableMapping, WsClientMessage, WsClientMessageType, WsComplete, WsError, WsHistory, WsMessage, WsServerMessage, WsServerMessageType, applyPresetToOverrides, buildReasoningProviderOptions, deriveParamSupport, getDefaultOutVars, getEffectiveOutVars, getModelEntry, logc, messageTypes, normalizeEffectiveParams, normalizeReasoning, openAiReasoningParamSupport, openAiStandardParamSupport, parseUserAuth, parseUserMessage, pickNodeOverrides, reasoningEffortLevels, reasoningExposable, resolveOutVarRef, resolvePreset, resolveReasoningEffort, resolveTemplate, sanitizeUserIdentity, stripUnsupportedFromNodeData, toStreamTextSettings, walkPath, workersAiParamSupport, wsDebug, xaiParamSupport } from './common.js';
-import { z } from 'zod';
-
-type MemoryMode = 'raw' | 'summarize';
-type ContextDomain = 'tool' | 'summary' | 'keydata' | 'info' | 'skill';
-interface MemoryConfig {
-    enabled: boolean;
-    mode: MemoryMode;
-    summaryModel: string;
-    summaryThreshold: number;
-    toolCache: boolean;
-}
-interface ContextRow {
-    ref: string;
-    domain: ContextDomain;
-    description: string;
-    data?: string;
-    cycle: number;
-    meta?: string;
-    ts: number;
-}
-type ContextIndexEntry = Pick<ContextRow, 'ref' | 'domain' | 'description' | 'cycle' | 'ts' | 'meta'>;
-type ContextUpsertInput = Omit<ContextRow, 'ts'> & {
-    ts?: number;
-};
-
-interface FlowNode {
-    id: string;
-    type: string;
-    data: any;
-}
-interface FlowEdge {
-    id: string;
-    source: string;
-    target: string;
-    sourceHandle?: string;
-}
-interface FlowState {
-    nodes: FlowNode[];
-    edges: FlowEdge[];
-}
-interface FlowData {
-    tenant: string;
-    flowId: string;
-    chatId: string;
-    state?: FlowState;
-    /** Monotonic version from FlowDO, bumped on every state save. Undefined for flows
-     *  never re-saved since versioning shipped — treated as "unknown", never stale. */
-    stateVersion?: number;
-    detail?: Record<string, any>;
-}
-
-interface TelemetryEvent {
-    userId?: string;
-    tenant?: string;
-    flowId?: string;
-    sessionId?: string;
-    type: string;
-    payload: any;
-    timestamp?: string;
-}
-declare const sendEvent: (event: TelemetryEvent, env: Env) => Promise<void>;
-
-type FlowEngine = (input?: string | UserMessagePayload) => Promise<void>;
-interface ExecutedNode {
-    id: string;
-    type: NodeType;
-    label?: string;
-}
-interface FlowEngineState {
-    flowData?: FlowData;
-    executedNodes: ExecutedNode[];
-    context?: Record<string, any>;
-    results?: Record<string, any>;
-    lastNodeId?: string;
-    lastUserMessage?: string;
-    /** Monotonic flow-run counter for session context memory. */
-    cycle?: number;
-    tokenUsage?: {
-        promptTokens: number;
-        completionTokens: number;
-        totalTokens: number;
-    };
-}
-interface FlowEngineEvent {
-    type: string;
-    content: any;
-    timestamp?: number;
-}
-interface FlowEngineCallbacks {
-    onChangeState: (state: FlowEngineState) => void;
-    onServerEvent: (event: FlowEngineEvent) => void;
-    onTelemetry: (event: TelemetryEvent) => void;
-    onError: (error: string) => void;
-}
-declare function startEngine(sessionEnv: SessionEnv, state: FlowEngineState, callbacks?: FlowEngineCallbacks): FlowEngine;
-
-type ComposioSessionMeta = {
-    sessionId: string;
-    userId: string;
-    /** Fingerprint of execution config; mismatch forces a new Composio session. */
-    configKey?: string;
-};
-
-type SecretsMap = Readonly<Record<string, string>>;
-
-interface SessionEnv {
-    env?: Env;
-    ctx?: DurableObjectState;
-    conn?: Connection;
-    getMessages?: () => Promise<Message[]>;
-    setMessages?: (msg: Message) => Promise<void>;
-    getOraModel?: (modelId: string) => Promise<any>;
-    memoryConfig?: MemoryConfig;
-    cycle?: number;
-    setContext?: (row: ContextUpsertInput) => Promise<void>;
-    getContextIndex?: () => Promise<ContextIndexEntry[]>;
-    getContextData?: (ref: string) => Promise<string | null>;
-    getContext?: (ref: string) => Promise<ContextRow | null>;
-    resolveContext?: (ref: string) => Promise<ContextRow | null>;
-    pruneContext?: (upToCycle: number) => Promise<void>;
-    /** Tenant-scoped secrets resolved at session boot. Server-only — never copy into engineState or any client-bound event. */
-    secrets?: SecretsMap;
-    /** End-user-supplied ephemeral credentials sent live over WebSocket (`AUT`). DO memory only — never persisted, never copied into engineState/state/telemetry, masked in events. Resolved by `{{ auth.NAME }}`. */
-    userAuth?: SecretsMap;
-    /** Custom tool definitions for the tenant, lazy-loaded on first custom-tool reference and cached for the session. Keyed by row.id. */
-    toolDefs?: Record<string, any>;
-    /** In-memory cache for skill references fetched during this session. Keyed by toolId:refName. */
-    skillRefCache?: Map<string, string>;
-    /** Composio session metadata keyed by composio tool row id — reused within the agent chat session. */
-    composioSessions?: Record<string, ComposioSessionMeta>;
-    /** Logged-in user from the WebSocket upgrade (fallback when engineState.context.user is missing). */
-    user?: UserInterface;
-    /** Origin derived from the WebSocket upgrade request (preferred over ORA_PUBLIC_URL for callbacks). */
-    publicOrigin?: string;
-}
-
-interface State {
-    user: UserInterface | undefined;
-    /** Client-supplied identity fallback when no JWT user (via `UID` frame or HTTP body). */
-    injectedUserId?: string | null;
-    sessionId: string | null;
-    lastUpdated: Date | null;
-    engineState?: FlowEngineState;
-    error?: string;
-    modelsCache?: Record<string, any>;
-}
-declare class AgentDO extends Agent<Env, State> {
-    initialState: State;
-    /** JWT email wins; explicit client identity is fallback only. */
-    private resolveSessionUserId;
-    private secrets;
-    private userAuth;
-    /** Composio tool-router session metadata keyed by composio tool row id — DO memory only. */
-    private composioSessions;
-    /** In-memory cache for skill references fetched during this session — DO memory only. */
-    private skillRefCache;
-    /** Public origin from the active socket upgrade (used for OAuth callback URLs). */
-    private publicOrigin?;
-    /** Throttles the per-turn flow-version KV probe (DO memory only; resets on hibernation,
-     *  which just means the next turn re-probes — correct, only loses the throttle benefit). */
-    private lastFlowVersionCheckTs;
-    onRequest(request: Request): Promise<Response>;
-    onConnect(connection: Connection, ctx: ConnectionContext): Promise<void>;
-    onMessage(connection: Connection, message: WSMessage): Promise<void>;
-    onUserMessage(connection: Connection, message: string): Promise<void>;
-    /**
-     * Store a client-supplied session identity in persisted DO state.
-     * Used as `sessions.userId` fallback when no JWT user is present; JWT always wins.
-     */
-    onUserIdentity(connection: Connection, content: string): Promise<void>;
-    /**
-     * Store an end-user-supplied runtime credential in DO memory only.
-     * MUST NOT persist to SQL/chat history, touch this.state, or start the engine.
-     * Empty value clears that name. ACK carries the NAME only — never the value.
-     */
-    onUserAuth(connection: Connection, content: string): Promise<void>;
-    onClearHistory(connection?: Connection): Promise<void>;
-    onStart(): Promise<void>;
-    onClose(connection: Connection, code: number, reason: string, wasClean: boolean): Promise<void>;
-    onError(error: unknown): Promise<void>;
-}
-
-/**
- * Temperature conversion tool with static parameter support
- *
- * Usage examples:
- * - Static params (node.tools): { id: 'convertTempMessure', params: { source: 'C', target: 'F' } }
- * - Runtime override: convertTempMessure.execute({ temperature: 25, source: 'C', target: 'K' })
- * - Mixed: Static params set defaults, runtime args can override
- */
-declare function convertTempMessure(sessionEnv: SessionEnv, engineState: FlowEngineState, opts?: {
-    source?: string;
-    target?: string;
-}, onServerEvent?: (event: any) => void): ai.Tool<{
-    temperature: number;
-    source?: string | undefined;
-    target?: string | undefined;
-}, {
-    input: {
-        temperature: number;
-        source: string;
-        target: string;
-    };
-    output: {
-        value: number;
-        unit: string;
-        important: string;
-    };
-}>;
-
-declare const tempTool: (sessionEnv: SessionEnv, engineState: FlowEngineState, opts?: any, onServerEvent?: (event: any) => void) => ai.Tool<{
-    location: string;
-}, {
-    location: string;
-    temperature: number;
-}>;
-
-declare function twitter(sessionEnv: SessionEnv, engineState: FlowEngineState, opts?: any, onServerEvent?: (event: any) => void): ai.Tool<any, {
-    type: any;
-    userId: any;
-    userName: any;
-    raw: any;
-    natural: string;
-    result: any;
-    tweetId?: undefined;
-    query?: undefined;
-    queryType?: undefined;
-    woeid?: undefined;
-    count?: undefined;
-    username?: undefined;
-    error?: undefined;
-} | {
-    type: any;
-    userName: any;
-    raw: any;
-    natural: string;
-    result: any;
-    userId?: undefined;
-    tweetId?: undefined;
-    query?: undefined;
-    queryType?: undefined;
-    woeid?: undefined;
-    count?: undefined;
-    username?: undefined;
-    error?: undefined;
-} | {
-    type: any;
-    tweetId: any;
-    raw: any;
-    natural: string;
-    result: any;
-    userId?: undefined;
-    userName?: undefined;
-    query?: undefined;
-    queryType?: undefined;
-    woeid?: undefined;
-    count?: undefined;
-    username?: undefined;
-    error?: undefined;
-} | {
-    type: any;
-    query: any;
-    queryType: any;
-    raw: any;
-    natural: string;
-    result: any;
-    userId?: undefined;
-    userName?: undefined;
-    tweetId?: undefined;
-    woeid?: undefined;
-    count?: undefined;
-    username?: undefined;
-    error?: undefined;
-} | {
-    type: any;
-    woeid: any;
-    count: any;
-    raw: any;
-    natural: string;
-    result: any;
-    userId?: undefined;
-    userName?: undefined;
-    tweetId?: undefined;
-    query?: undefined;
-    queryType?: undefined;
-    username?: undefined;
-    error?: undefined;
-} | {
-    type: any;
-    username: any;
-    raw: any;
-    natural: string;
-    result: any;
-    userId?: undefined;
-    userName?: undefined;
-    tweetId?: undefined;
-    query?: undefined;
-    queryType?: undefined;
-    woeid?: undefined;
-    count?: undefined;
-    error?: undefined;
-} | {
-    type: string;
-    error: string;
-    natural: string;
-    userId?: undefined;
-    userName?: undefined;
-    raw?: undefined;
-    result?: undefined;
-    tweetId?: undefined;
-    query?: undefined;
-    queryType?: undefined;
-    woeid?: undefined;
-    count?: undefined;
-    username?: undefined;
-}>;
-
-declare function vectorize(sessionEnv: SessionEnv, engineState: FlowEngineState, opts?: {
-    namespace?: string;
-}): ai.Tool<{
-    query: string;
-    topK?: number;
-}, {
-    results: never[];
-    query?: undefined;
-    topK?: undefined;
-    matches?: undefined;
-} | {
-    query: string;
-    topK: number;
-    matches: any;
-    results?: undefined;
-}>;
-
-/**
- * Sends a message to the client over the WebSocket connection.
- *
- * This function formats the message as "type::content" and sends it via the provided connection.
- * It is the primary mechanism for server-to-client communication in the agent system.
- *
- * @param connection - The WebSocket connection instance to send the message over
- * @param type - The message type from the MessageType enum, used to categorize the message
- * @param content - The message content as a string (typically JSON.stringify'd data)
- */
-declare const serverSend: (connection: Connection, type: MessageType, content: string) => void;
-
-/**
- * Safely parse JSON string, with fallback to extract JSON-like objects
- * @param s - The string to parse
- * @returns Parsed JSON object or null if parsing fails
- */
-declare const safeParseJSON: (s: string) => any;
-/**
- * Build a URL with query parameters
- * @param base - Base URL
- * @param path - Path to append
- * @param params - Query parameters object
- * @returns Full URL string
- */
-declare const buildUrl: (base: string, path: string, params?: Record<string, any>) => string;
-
-/** Normalize registry rows that may carry JSON columns as strings (D1 fallback). */
-declare function normalizeModelEntry(entry: ModelEntry): ModelEntry;
-/**
- * Resolves a model by its ID, returning a provider-agnostic handle suitable for `streamText`.
- */
-declare function resolveModelById(modelId: string, sessionEnv?: SessionEnv, overrides?: EffectiveParams): Promise<ResolvedModel | undefined>;
-/**
- * Registers a custom provider resolver for runtime extension.
- *
- * @param name - The provider name to register.
- * @param resolver - A function that takes a ModelEntry, optional SessionEnv, and optional overrides, returning the model handle.
- *
- * Example usage:
- * ```typescript
- * registerProvider('custom', (entry, sessionEnv, overrides) => customSDK(entry.providerModel, overrides));
- * ```
- */
-declare function registerProvider(name: string, resolver: (entry: ModelEntry, sessionEnv?: SessionEnv, overrides?: Record<string, any>) => Promise<any> | any): void;
-
-/**
- * Converts a JSON Schema to a Zod schema.
- * Supports object schemas with properties, arrays, and primitive types.
- * Treats empty object ({}) as z.object({}).
- * Supports nested object properties recursively; respects schema.required (makes properties optional unless required).
- * A property `default` is an overridable fallback: it becomes z.default(...) so an
- * omitted arg is filled (LLM-supplied value still wins). Precedence: default > required > optional.
- * Supports arrays of primitives: { type: 'array', items: { type: 'string' } } -> z.array(z.string()).
- * Returns z.any() for unsupported shapes as fallback.
- */
-declare function jsonSchemaToZod(schema: any): z.ZodSchema;
-
-declare const hashText: (text: string) => Promise<string>;
-declare const embedText: (sessionEnv: SessionEnv, text: string | string[], fallback?: boolean) => Promise<any>;
-
-declare function validateFlowStart(flow: FlowData): any;
-
-declare function verifyUser(request: Request, env: Env, ctx: ExecutionContext): Promise<boolean | UserInterface | null>;
-declare function verifyToken(env: Env, token?: string | null): Promise<any>;
-
-declare function parseCookies(cookieHeader: string | null | undefined): Record<string, string>;
-
-declare function handleCommonRoutes(request: Request, env: Env, ctx: ExecutionContext): Promise<Response | undefined>;
-
-declare function handleAgentsRoutes(request: Request, env: Env, ctx: ExecutionContext): Promise<Response | undefined>;
-
-declare function handleIntancesRoutes(request: Request, env: Env, ctx: ExecutionContext): Promise<Response | undefined>;
-
-declare function handleAuthRoutes(request: Request, env: Env, ctx: ExecutionContext): Promise<Response | any>;
-
-type SqlContext = {
-    sql: any;
-};
-declare function ensureMessagesTable(ctx: SqlContext): Promise<void>;
-declare function setMessage(ctx: SqlContext, msg: Message): Promise<void>;
-declare function getAllMessages(ctx: SqlContext): Promise<Message[]>;
-declare function clearAllMessages(ctx: SqlContext): Promise<void>;
-declare function ensureContextTable(ctx: SqlContext): Promise<void>;
-declare function setContext(ctx: SqlContext, row: ContextUpsertInput): Promise<void>;
-declare function getContextIndex(ctx: SqlContext): Promise<ContextIndexEntry[]>;
-declare function getContext(ctx: SqlContext, ref: string): Promise<ContextRow | null>;
-declare function getContextData(ctx: SqlContext, ref: string): Promise<string | null>;
-/**
- * Resolve a context ref the model echoed back: exact match first, else a unique prefix match
- * (the working-memory block exposes shortened hash tails). Ambiguous prefixes resolve to null.
- */
-declare function resolveContextRef(ctx: SqlContext, ref: string): Promise<ContextRow | null>;
-/** Evict tool-domain rows up to and including a cycle (used after a summary folds them in). */
-declare function deleteContextToolsUpToCycle(ctx: SqlContext, cycle: number): Promise<void>;
-declare function getMaxContextCycle(ctx: SqlContext): Promise<number>;
-declare function clearAllContext(ctx: SqlContext): Promise<void>;
-declare function ensureAgentTables(ctx: SqlContext): Promise<void>;
-
-export { AgentDO, type ComposioSessionMeta, EffectiveParams, type ExecutedNode, type FlowEngine, type FlowEngineCallbacks, type FlowEngineEvent, type FlowEngineState, Message, MessageType, ModelEntry, NodeType, ResolvedModel, type SessionEnv, type SqlContext, type State, type TelemetryEvent, UserInterface, UserMessagePayload, buildUrl, clearAllContext, clearAllMessages, convertTempMessure, deleteContextToolsUpToCycle, embedText, ensureAgentTables, ensureContextTable, ensureMessagesTable, getAllMessages, getContext, getContextData, getContextIndex, getMaxContextCycle, handleAgentsRoutes, handleAuthRoutes, handleCommonRoutes, handleIntancesRoutes, hashText, jsonSchemaToZod, normalizeModelEntry, parseCookies, registerProvider, resolveContextRef, resolveModelById, safeParseJSON, sendEvent, serverSend, setContext, setMessage, startEngine, tempTool, twitter, validateFlowStart, vectorize, verifyToken, verifyUser };
+export * from './server/tools/convertTempMessure';
+export * from './server/tools/tempTool';
+export * from './server/tools/twitter';
+export * from './server/tools/vectorize';
+export * from './server/core/AgentDO';
+export * from './server/logic/serverMessages';
+export * from './server/logic/FlowEngine';
+export * from './server/utils/apiUtils';
+export * from './server/utils/modelProvider';
+export * from './server/utils/jsonSchemaToZod';
+export * from './server/utils/embeddings';
+export * from './server/utils/flowValidation';
+export * from './server/utils/telemetry';
+export * from './server/utils/auth';
+export * from './server/utils/cookies';
+export * from './server/api/commons';
+export * from './server/api/agents';
+export * from './server/api/instances';
+export * from './server/api/auth';
+export * from './server/runners/memoNodeRunner';
+export * from './server/runners/answerNodeRunner';
+export * from './server/runners/startNodeRunner';
+export * from './server/runners/classifierNodeRunner';
+export * from './server/runners/llmNodeRunner';
+export * from './server/sql/agentSQL';
+export * from './index.common';
+export * from './common/utils/logc';