@snrraptopack/auwgent-sdk 0.1.0-alpha.21 → 0.1.0-alpha.22

package/auwgent.ts

@@ -14,6 +14,8 @@ export { AuwgentNative };
 
 import {
     AuwgentToolError,
+    AuwgentWarning,
+    AuwgentWarningSource,
 } from './types.js';
 
 import type {
@@ -71,6 +73,8 @@ export class TypedAuwgent<
     private lastTurnRawBlock: string | undefined = undefined;
     private agentStack: string[] = [];
     private helperSessions = new Map<string, SessionState>();
+    private warnings: AuwgentWarning[] = [];
+    private warningHandler: ((warning: AuwgentWarning) => void) | null = null;
 
     constructor(ir: IR, config: AuwgentConfig<IR>) {
         this.ir = ir;
@@ -236,10 +240,11 @@ export class TypedAuwgent<
     }
 
     private async handleMiddlewareEvent(eventJson: string): Promise<string | undefined> {
-        const event = JSON.parse(eventJson);
-        const ctx = this.buildContextFromRuntimeEvent(event);
+        try {
+            const event = JSON.parse(eventJson);
+            const ctx = this.buildContextFromRuntimeEvent(event);
 
-        switch (event?.type) {
+            switch (event?.type) {
             case 'intent': {
                 const value = event?.value && typeof event.value === 'object'
                     ? { ...event.value }
@@ -251,9 +256,13 @@ export class TypedAuwgent<
 
                 for (const m of this.getMiddleware(ctx)) {
                     if (m.onIntent) {
-                        const control = await (m.onIntent as any)(event.name, value, ctx);
-                        if (control !== undefined && control !== null) {
-                            return JSON.stringify(control);
+                        try {
+                            const control = await (m.onIntent as any)(event.name, value, ctx);
+                            if (control !== undefined && control !== null) {
+                                return JSON.stringify(control);
+                            }
+                        } catch (error) {
+                            this.reportWarning('middleware', 'middleware onIntent threw', error, ctx.activeAgent as string);
                         }
                     }
                 }
@@ -265,9 +274,13 @@ export class TypedAuwgent<
 
                 for (const m of this.getMiddleware(ctx)) {
                     if (m.onLLMStart) {
-                        const modified = await (m.onLLMStart as any)(currentPrompt, ctx);
-                        if (typeof modified === 'string') {
-                            currentPrompt = modified;
+                        try {
+                            const modified = await (m.onLLMStart as any)(currentPrompt, ctx);
+                            if (typeof modified === 'string') {
+                                currentPrompt = modified;
+                            }
+                        } catch (error) {
+                            this.reportWarning('middleware', 'middleware onLLMStart threw', error, ctx.activeAgent as string);
                         }
                     }
                 }
@@ -280,7 +293,11 @@ export class TypedAuwgent<
             case 'llm_end': {
                 for (const m of this.getMiddleware(ctx)) {
                     if (m.onLLMEnd) {
-                        await (m.onLLMEnd as any)(event.response ?? {}, ctx);
+                        try {
+                            await (m.onLLMEnd as any)(event.response ?? {}, ctx);
+                        } catch (error) {
+                            this.reportWarning('middleware', 'middleware onLLMEnd threw', error, ctx.activeAgent as string);
+                        }
                     }
                 }
 
@@ -291,7 +308,11 @@ export class TypedAuwgent<
 
                 for (const m of this.getMiddleware(ctx)) {
                     if (m.onRunStart) {
-                        session = await (m.onRunStart as any)(session, ctx);
+                        try {
+                            session = await (m.onRunStart as any)(session, ctx);
+                        } catch (error) {
+                            this.reportWarning('middleware', 'middleware onRunStart threw', error, ctx.activeAgent as string);
+                        }
                     }
                 }
 
@@ -307,7 +328,11 @@ export class TypedAuwgent<
 
                 for (const m of this.getMiddleware(ctx)) {
                     if (m.onRunComplete) {
-                        await (m.onRunComplete as any)(session, ctx);
+                        try {
+                            await (m.onRunComplete as any)(session, ctx);
+                        } catch (error) {
+                            this.reportWarning('middleware', 'middleware onRunComplete threw', error, ctx.activeAgent as string);
+                        }
                     }
                 }
 
@@ -322,9 +347,13 @@ export class TypedAuwgent<
 
                 for (const m of this.getMiddleware(ctx)) {
                     if (m.onError) {
-                        const shouldSwallow = await (m.onError as any)(error, session, ctx);
-                        if (shouldSwallow) {
-                            return JSON.stringify({ swallow: true });
+                        try {
+                            const shouldSwallow = await (m.onError as any)(error, session, ctx);
+                            if (shouldSwallow) {
+                                return JSON.stringify({ swallow: true });
+                            }
+                        } catch (middlewareError) {
+                            this.reportWarning('middleware', 'middleware onError threw', middlewareError, ctx.activeAgent as string);
                         }
                     }
                 }
@@ -333,6 +362,10 @@ export class TypedAuwgent<
             }
             default:
                 return undefined;
+            }
+        } catch (error) {
+            this.reportWarning('onMiddlewareEvent', 'failed to handle middleware event', error);
+            return undefined;
         }
     }
 
@@ -340,11 +373,49 @@ export class TypedAuwgent<
         const ctx = this.getBuildContext();
         for (const m of this.getMiddleware(ctx)) {
             if (m.onError) {
-                await (m as any).onError(error, undefined, ctx);
+                try {
+                    await (m as any).onError(error, undefined, ctx);
+                } catch (middlewareError) {
+                    this.reportWarning('middleware', 'middleware onError handler threw', middlewareError);
+                }
             }
         }
     }
 
+    private reportWarning(source: AuwgentWarningSource, message: string, error?: unknown, agentName?: string): void {
+        const warning: AuwgentWarning = {
+            timestamp: new Date().toISOString(),
+            source,
+            message,
+            detail: error instanceof Error ? error.message : (error != null ? String(error) : undefined),
+            agentName,
+        };
+
+        this.warnings.push(warning);
+
+        try {
+            this.warningHandler?.(warning);
+        } catch {
+            // Keep warning flow non-fatal.
+        }
+
+        const detailSuffix = warning.detail ? `: ${warning.detail}` : '';
+        const agentSuffix = warning.agentName ? ` [agent=${warning.agentName}]` : '';
+        console.warn(`[auwgent][${warning.source}]${agentSuffix} ${warning.message}${detailSuffix}`);
+    }
+
+    onWarning(handler: (warning: AuwgentWarning) => void): void {
+        this.warningHandler = handler;
+    }
+
+    getWarnings(): AuwgentWarning[] {
+        return [...this.warnings];
+    }
+
+    clearWarnings(): void {
+        this.warnings = [];
+    }
+
     /** Generate an embedding for the given text using the configured model */
     async embed(text: string): Promise<number[]> {
         try {
@@ -375,42 +446,89 @@ export class TypedAuwgent<
         const userHandler = this.storedIntentHandler;
 
         this.native.onIntent(async (name: string, value: any, agentName: string) => {
-            const intentCtx = this.getBuildContext();
-            intentCtx.activeAgent = agentName as any;
+            try {
+                const intentCtx = this.getBuildContext();
+                intentCtx.activeAgent = agentName as any;
 
-            if (value && typeof value === 'object' && '_raw' in value) {
-                intentCtx.rawBlock = value._raw;
-                this.lastTurnRawBlock = value._raw;
-                delete value._raw;
-            }
+                if (value && typeof value === 'object' && '_raw' in value) {
+                    intentCtx.rawBlock = value._raw;
+                    this.lastTurnRawBlock = value._raw;
+                    delete value._raw;
+                }
 
-            if (userHandler) {
-                const result = await (userHandler as any)(name, value, agentName);
-                return result ?? null;
+                if (userHandler) {
+                    const result = await (userHandler as any)(name, value, agentName);
+                    return result ?? null;
+                }
+            } catch (error) {
+                this.reportWarning('onIntent', 'intent callback failed', error, agentName);
             }
         });
 
-        if (this.storedPartialHandler) {
-            const partialHandler = this.storedPartialHandler;
-            this.native.onIntentPartial((name: string, value: any, agentName: string) => {
-                (partialHandler as any)(name, value, agentName);
+        const partialHandler = this.storedPartialHandler;
+        this.native.onIntentPartial((name: string, value: any, agentName: string) => {
+                const partialCtx = this.getBuildContext();
+                partialCtx.activeAgent = agentName as any;
+
+                if (partialHandler) {
+                    try {
+                        const maybePromise = (partialHandler as any)(name, value, agentName);
+                        if (maybePromise && typeof maybePromise.then === 'function') {
+                            (maybePromise as Promise<unknown>).catch((error: unknown) => {
+                                this.reportWarning('onIntentPartial', 'partial intent async callback rejected', error, agentName);
+                            });
+                        }
+                    } catch (error) {
+                        this.reportWarning('onIntentPartial', 'partial intent callback failed', error, agentName);
+                    }
+                }
+
+                for (const m of this.getMiddleware(partialCtx)) {
+                    if (m.onIntentPartial) {
+                        try {
+                            const maybePromise = (m.onIntentPartial as any)(name, value, partialCtx);
+                            if (maybePromise && typeof maybePromise.then === 'function') {
+                                (maybePromise as Promise<unknown>).catch((error: unknown) => {
+                                    this.reportWarning('middleware', 'middleware onIntentPartial threw', error, agentName);
+                                });
+                            }
+                        } catch (error) {
+                            this.reportWarning('middleware', 'middleware onIntentPartial threw', error, agentName);
+                        }
+                    }
+                }
             });
-        }
-        this.native.onMiddlewareEvent((eventJson: string) => this.handleMiddlewareEvent(eventJson));
+        this.native.onMiddlewareEvent(async (eventJson: string) => {
+            try {
+                return await this.handleMiddlewareEvent(eventJson);
+            } catch (error) {
+                this.reportWarning('onMiddlewareEvent', 'middleware event callback failed', error);
+                return undefined;
+            }
+        });
 
         this.native.onSubEngineStart(async (helperName: string, emptySessionJson: string) => {
-            const session = this.helperSessions.get(helperName) || (JSON.parse(emptySessionJson) as SessionState);
-            if (session.stack) {
-                this.agentStack = [...session.stack];
+            try {
+                const session = this.helperSessions.get(helperName) || (JSON.parse(emptySessionJson) as SessionState);
+                if (session.stack) {
+                    this.agentStack = [...session.stack];
+                }
+                return JSON.stringify(session);
+            } catch (error) {
+                this.reportWarning('onSubEngineStart', 'sub-engine start callback failed', error, helperName);
+                return emptySessionJson;
             }
-            return JSON.stringify(session);
         });
 
         this.native.onSubEngineComplete(async (helperName: string, completedSessionJson: string) => {
-            const session = JSON.parse(completedSessionJson) as SessionState;
-            this.helperSessions.set(helperName, session);
-            if (session.stack) {
-                this.agentStack = [...session.stack];
+            try {
+                const session = JSON.parse(completedSessionJson) as SessionState;
+                this.helperSessions.set(helperName, session);
+                if (session.stack) {
+                    this.agentStack = [...session.stack];
+                }
+            } catch (error) {
+                this.reportWarning('onSubEngineComplete', 'sub-engine complete callback failed', error, helperName);
             }
         });
     }
@@ -445,6 +563,9 @@ export class TypedAuwgent<
             }
 
             return currentSession;
+        } catch (error) {
+            this.reportWarning('run', 'native run failed', error);
+            throw error;
         } finally {
             // Deactivate listeners after run to release the TSFN ref
             // and allow the Node.js process to exit gracefully.

package/bun.lock

@@ -9,10 +9,10 @@
         "@types/bun": "latest",
       },
       "optionalDependencies": {
-        "@snrraptopack/auwgent-sdk-darwin-arm64": "0.1.0-alpha.21",
-        "@snrraptopack/auwgent-sdk-darwin-x64": "0.1.0-alpha.21",
-        "@snrraptopack/auwgent-sdk-linux-x64-gnu": "0.1.0-alpha.21",
-        "@snrraptopack/auwgent-sdk-win32-x64-msvc": "0.1.0-alpha.21",
+        "@snrraptopack/auwgent-sdk-darwin-arm64": "0.1.0-alpha.22",
+        "@snrraptopack/auwgent-sdk-darwin-x64": "0.1.0-alpha.22",
+        "@snrraptopack/auwgent-sdk-linux-x64-gnu": "0.1.0-alpha.22",
+        "@snrraptopack/auwgent-sdk-win32-x64-msvc": "0.1.0-alpha.22",
       },
       "peerDependencies": {
         "typescript": "^5",

package/middleware.ts

@@ -1,4 +1,4 @@
-import type { AgentIRShape, IntentControl, SessionState, AuwgentIntent, AuwgentModelIntent, AuwgentModelValue, AuwgentResponseValue, AuwgentTargetedResponseValue } from './types.js';
+import type { AgentIRShape, IntentControl, SessionState, AuwgentIntent, AuwgentModelIntent, AuwgentModelValue, AuwgentResponseValue, AuwgentTargetedResponseValue, PartialTextIntentValue, PartialStructuredIntentValue } from './types.js';
 
 // ── Middleware Types ───────────────────────────────────────────────────────
 
@@ -31,7 +31,7 @@ export type MiddlewareContext<IR extends AgentIRShape = any> = (
  */
 interface _MiddlewareHooks<
     IR extends AgentIRShape = any,
-    CustomIntents = never,
+    CustomIntents extends { name: string; value: any } = never,
     Output = any,
     Tools = any,
     T extends MiddlewareContext<IR>['activeAgent'] = any
@@ -61,14 +61,31 @@ interface _MiddlewareHooks<
      */
     onIntent?: (
         ...args: {
-            [K in AuwgentIntent<IR, CustomIntents, Output, Tools>['name']]: [
+            [K in AuwgentIntent<IR, CustomIntents, Output, Tools>['name'] & string]: [
                 name: K,
                 value: Extract<AuwgentIntent<IR, CustomIntents, Output, Tools>, { name: K }>['value'],
                 ctx: Extract<MiddlewareContext<IR>, { activeAgent: T }>
             ];
-        }[AuwgentIntent<IR, CustomIntents, Output, Tools>['name']]
+        }[AuwgentIntent<IR, CustomIntents, Output, Tools>['name'] & string]
     ) => IntentControl | Promise<IntentControl>;
 
+    /**
+     * Fired for streaming partial intent updates.
+     * Observational only; cannot return control signals.
+     */
+    onIntentPartial?: (
+        ...args: {
+            [K in AuwgentIntent<IR, CustomIntents, Output, Tools>['name'] & string]: [
+                name: K,
+                value:
+                    | Extract<AuwgentIntent<IR, CustomIntents, Output, Tools>, { name: K }>['value']
+                    | PartialTextIntentValue
+                    | PartialStructuredIntentValue<any>,
+                ctx: Extract<MiddlewareContext<IR>, { activeAgent: T }>
+            ];
+        }[AuwgentIntent<IR, CustomIntents, Output, Tools>['name'] & string]
+    ) => void | Promise<void>;
+
     /**
      * Fired when the model emits a terminal response (text or schema).
      */
@@ -105,7 +122,7 @@ interface _MiddlewareHooks<
  */
 export type Middleware<
     IR extends AgentIRShape = any,
-    CustomIntents = never,
+    CustomIntents extends { name: string; value: any } = never,
     Output = any,
     Tools = any,
     Target extends MiddlewareContext<IR>['activeAgent'] = MiddlewareContext<IR>['activeAgent']

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@snrraptopack/auwgent-sdk",
-  "version": "0.1.0-alpha.21",
+  "version": "0.1.0-alpha.22",
   "description": "Auwgent IR Runtime — Rust-powered agent engine with type-safe TypeScript bindings",
   "repository": {
     "type": "git",
@@ -45,9 +45,9 @@
     "typescript": "^5"
   },
   "optionalDependencies": {
-    "@snrraptopack/auwgent-sdk-win32-x64-msvc": "0.1.0-alpha.21",
-    "@snrraptopack/auwgent-sdk-darwin-x64": "0.1.0-alpha.21",
-    "@snrraptopack/auwgent-sdk-darwin-arm64": "0.1.0-alpha.21",
-    "@snrraptopack/auwgent-sdk-linux-x64-gnu": "0.1.0-alpha.21"
+    "@snrraptopack/auwgent-sdk-win32-x64-msvc": "0.1.0-alpha.22",
+    "@snrraptopack/auwgent-sdk-darwin-x64": "0.1.0-alpha.22",
+    "@snrraptopack/auwgent-sdk-darwin-arm64": "0.1.0-alpha.22",
+    "@snrraptopack/auwgent-sdk-linux-x64-gnu": "0.1.0-alpha.22"
   }
 }

package/types.ts

@@ -53,6 +53,25 @@ export class AuwgentToolError extends Error {
     }
 }
 
+export type AuwgentWarningSource =
+    | 'onIntent'
+    | 'onIntentPartial'
+    | 'onMiddlewareEvent'
+    | 'onSubEngineStart'
+    | 'onSubEngineComplete'
+    | 'middleware'
+    | 'run'
+    | 'embed'
+    | 'embedBatch';
+
+export interface AuwgentWarning {
+    timestamp: string;
+    source: AuwgentWarningSource;
+    message: string;
+    detail?: string;
+    agentName?: string;
+}
+
 export interface AgentIRShape {
     name: string;
     input: any; // Now mandatory in IR for type extraction
@@ -60,6 +79,7 @@ export interface AgentIRShape {
     tools: readonly IRToolDef[];
     workflows?: readonly IRWorkflowDef[];
     helpers?: readonly IRHelperDef[];
+    types?: Record<string, any>;
     output?: any;
     modelConfig?: IRModelConfigEntry[];
     customIntents?: readonly IRCustomIntentDef[];
@@ -259,10 +279,10 @@ export type GetToolArgs<T> = T extends (args: infer A) => any ? A : Record<strin
 export type IsAny<T> = 0 extends (1 & T) ? true : false;
 export type GetToolResult<T> = T extends (args: any) => Promise<infer R> ? R : any;
 
-export interface ToolCallIntent<K = string, A = any> { name: 'tool_call'; value: { type: K; args: A } }
-export interface ToolResultIntent<K = string, R = any> { name: 'tool_result'; value: { name: K; result: R; overridden?: boolean } }
-export interface ToolErrorIntent<K = string> { name: 'tool_error'; value: { tool: K; message: string } }
-export interface ToolSkippedIntent<K = string, A = any> { name: 'tool_skipped'; value: { type: K; args: A } }
+export interface ToolCallIntent<K extends string = string, A = any> { name: 'tool_call'; value: { type: K; args: A } }
+export interface ToolResultIntent<K extends string = string, R = any> { name: 'tool_result'; value: { name: K; result: R; overridden?: boolean } }
+export interface ToolErrorIntent<K extends string = string> { name: 'tool_error'; value: { tool: K; message: string } }
+export interface ToolSkippedIntent<K extends string = string, A = any> { name: 'tool_skipped'; value: { type: K; args: A } }
 
 export interface WorkflowCallIntent<K = string, A = any> { name: 'workflow_call'; value: { type: K; args: A } }
 export interface WorkflowResultIntent<K = string, R = any> { name: 'workflow_result'; value: { name: K; result: R } }
@@ -271,7 +291,34 @@ export interface HelperCallIntent<K = string, A = any> { name: 'helper_call'; va
 export interface HelperResultIntent<K = string, R = any> { name: 'helper_result'; value: { name: K; result: R } }
 
 export interface ResponseTextIntent { name: 'response_text'; value: { text: string } }
-export interface ResponseSchemaIntent<Output = any> { name: 'response_schema'; value: Output }
+
+type RootOutputSchemaName<IR extends AgentIRShape> =
+    string extends IR['name'] ? never : `${IR['name']}Output`;
+
+type HelperOutputSchemaNames<IR extends AgentIRShape> =
+    IR['helpers'] extends readonly any[]
+    ? IR['helpers'][number] extends { name: infer N extends string }
+        ? `${N}Output`
+        : never
+    : never;
+
+type DeclaredSchemaNames<IR extends AgentIRShape> =
+    IR['types'] extends Record<string, any>
+    ? keyof IR['types'] & string
+    : never;
+
+type ResponseSchemaTypeName<IR extends AgentIRShape> =
+    string extends IR['name']
+    ? string
+    : RootOutputSchemaName<IR> | HelperOutputSchemaNames<IR> | DeclaredSchemaNames<IR>;
+
+export interface ResponseSchemaIntent<Output = any, SchemaType extends string = string> {
+    name: 'response_schema';
+    value: {
+        type: SchemaType;
+        response: Output;
+    };
+}
 export interface PendingStreamValue { $state: 'pending' }
 export interface PartialIntentEnvelope<Snapshot = unknown> {
     partial: true
@@ -298,12 +345,12 @@ export type ToolIntents<Tools> =
     string extends keyof Tools
     ? (ToolCallIntent | ToolResultIntent | ToolErrorIntent | ToolSkippedIntent)
     : {
-        [K in keyof Tools]:
+        [K in keyof Tools & string]:
         | ToolCallIntent<K, GetToolArgs<Tools[K]>>
         | ToolResultIntent<K, GetToolResult<Tools[K]>>
         | ToolErrorIntent<K>
         | ToolSkippedIntent<K, GetToolArgs<Tools[K]>>
-    }[keyof Tools];
+    }[keyof Tools & string];
 
 export type WorkflowIntents<IR extends AgentIRShape> =
     IR['workflows'] extends readonly any[]
@@ -331,8 +378,8 @@ export type CoreIntents<IR extends AgentIRShape, Output = any, Tools = any> = (
     | ToolIntents<Tools>
     | WorkflowIntents<IR>
     | HelperIntents<IR>
-    | (IsAny<Output> extends true ? (ResponseTextIntent | ResponseSchemaIntent) :
-        [Output] extends [never] ? ResponseTextIntent : (ResponseTextIntent | ResponseSchemaIntent<Output>))
+    | (IsAny<Output> extends true ? (ResponseTextIntent | ResponseSchemaIntent<any, ResponseSchemaTypeName<IR>>) :
+        [Output] extends [never] ? ResponseTextIntent : (ResponseTextIntent | ResponseSchemaIntent<Output, ResponseSchemaTypeName<IR>>))
     | ErrorIntent
 );
 
@@ -447,7 +494,7 @@ export type PartialIntentHandler<
                 : PartialStructuredIntentValue<Extract<AuwgentIntent<IR, Custom, Output, Tools>, { name: K }>['value']>,
             agentName: string,
         ];
-    }[AuwgentIntent<IR, Custom, Output, Tools>['name']]
+    }[AuwgentIntent<IR, Custom, Output, Tools>['name'] & string]
 ) => void;
 
 // ── Session Types ────────────────────────────────────────────────────────