@measureoneinc/savant 1.0.0

package/dist/ai/response_handlers/base-handler.js

@@ -0,0 +1,153 @@
+"use strict";
+/**
+ * @fileoverview Base Handler for Response Output Types
+ *
+ * Base interface and abstract class for handling different response output types
+ * from the OpenAI Responses API.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BaseHandler = void 0;
+const logger_1 = require("../../lib/utils/logger");
+class BaseHandler {
+    constructor(streaming_utils) {
+        // Initialize logger with the class name
+        this.logger = (0, logger_1.create_compact_logger)(this.constructor.name);
+        this.streaming_utils = streaming_utils;
+    }
+    /**
+     * Extract and parse AgentResponse from content
+     * Handles both string and object content, with JSON extraction from text
+     * @param content - Content from ChatMessage (can be string or object)
+     * @returns Parsed AgentResponse or null if parsing fails
+     */
+    extract_agent_response(content) {
+        if (!content) {
+            this.logger.warn('⚠️ No content provided');
+            return null;
+        }
+        // If content is already an object, check if it's an AgentResponse
+        if (typeof content === 'object' && content !== null) {
+            // Check if it has message_type (indicating it's an AgentResponse)
+            if (content.message_type) {
+                this.logger.debug('✅ Content is already an AgentResponse object');
+                return content;
+            }
+            // If it's an object but not AgentResponse, try to stringify and parse
+            try {
+                const content_str = JSON.stringify(content);
+                return this.parse_content_string(content_str);
+            }
+            catch (e) {
+                this.logger.warn('⚠️ Failed to stringify content object', {
+                    error: e instanceof Error ? e.message : String(e)
+                });
+                return null;
+            }
+        }
+        // If content is a string, parse it
+        if (typeof content === 'string') {
+            return this.parse_content_string(content);
+        }
+        this.logger.warn('⚠️ Content is neither string nor object');
+        return null;
+    }
+    /**
+     * Parse content string to extract AgentResponse
+     * Handles markdown wrappers, invalid escape sequences, and JSON extraction
+     */
+    parse_content_string(content_str) {
+        if (!content_str || content_str.trim().length === 0) {
+            this.logger.warn('⚠️ Empty content string');
+            return null;
+        }
+        this.logger.debug(`🔍 Attempting to extract AgentResponse from content (length: ${content_str.length})`);
+        // First, try direct JSON parse
+        try {
+            const agent_response = JSON.parse(content_str);
+            this.logger.debug('✅ Parsed as pure JSON');
+            return agent_response;
+        }
+        catch (e) {
+            this.logger.debug(`⚠️ Direct JSON parse failed, applying extraction logic`);
+        }
+        // Clean the text
+        let trimmed = content_str.trim();
+        // Strip markdown code blocks
+        const markdown_match = trimmed.match(/```(?:json)?\s*\n?([\s\S]*?)\n?```/);
+        if (markdown_match) {
+            trimmed = markdown_match[1].trim();
+            this.logger.debug('✂️ Stripped markdown wrapper');
+        }
+        // Fix invalid \' escape sequences (Gemini sometimes outputs these)
+        if (trimmed.includes("\\'")) {
+            trimmed = trimmed.replace(/\\'/g, "'");
+            this.logger.debug('🔧 Fixed invalid \' escape sequences');
+        }
+        // Extract JSON from text
+        const first_brace = trimmed.indexOf('{');
+        const last_brace = trimmed.lastIndexOf('}');
+        if (first_brace !== -1 && last_brace !== -1 && last_brace > first_brace) {
+            const json_str = trimmed.substring(first_brace, last_brace + 1);
+            try {
+                const parsed = JSON.parse(json_str);
+                this.logger.debug('✅ Successfully extracted and parsed JSON');
+                return parsed;
+            }
+            catch (e) {
+                this.logger.error(`❌ Failed to parse extracted JSON: ${e instanceof Error ? e.message : String(e)}`);
+                return null;
+            }
+        }
+        this.logger.error('❌ Could not extract JSON from content');
+        return null;
+    }
+    /**
+     * Append Gemini response to conversation history
+     * Formats the response as a ChatMessage with role 'model' and content as JSON string
+     * @param history - Current conversation history
+     * @param result - GenerateContentResponse or ChatMessage from Gemini
+     * @param agent_response - Optional parsed AgentResponse to include
+     * @returns Updated conversation history
+     */
+    append_gemini_response_to_history(history, result, agent_response) {
+        const updated_history = [...history];
+        // If agent_response is provided, use it; otherwise extract from result
+        if (agent_response) {
+            updated_history.push({
+                role: 'model',
+                content: JSON.stringify(agent_response)
+            });
+            return updated_history;
+        }
+        if ('text' in result) {
+            // GenerateContentResponse
+            updated_history.push({
+                role: 'model',
+                content: result.text || ''
+            });
+            return updated_history;
+        }
+        // ChatMessage - already formatted as a ChatMessage object
+        updated_history.push(result);
+        return updated_history;
+    }
+    /**
+     * Append context to conversation history as a special context message
+     * @param history - Current conversation history
+     * @param context - Context object to append
+     * @returns Updated conversation history
+     */
+    append_context_to_history(history, context) {
+        const updated_history = [...history];
+        updated_history.push({
+            role: 'user',
+            content: {
+                type: 'context',
+                context: context
+            }
+        });
+        return updated_history;
+    }
+}
+exports.BaseHandler = BaseHandler;
+//# sourceMappingURL=base-handler.js.map

package/dist/ai/response_handlers/base-handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base-handler.js","sourceRoot":"","sources":["../../../src/ai/response_handlers/base-handler.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAEH,mDAAuE;AAsBvE,MAAsB,WAAW;IAK7B,YAAY,eAA+B;QACvC,wCAAwC;QACxC,IAAI,CAAC,MAAM,GAAG,IAAA,8BAAqB,EAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAC3D,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;IAC3C,CAAC;IAkBD;;;;;OAKG;IACO,sBAAsB,CAAC,OAAqB;QAClD,IAAI,CAAC,OAAO,EAAE,CAAC;YACX,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YAC3C,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,kEAAkE;QAClE,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;YAClD,kEAAkE;YAClE,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;gBACvB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,8CAA8C,CAAC,CAAC;gBAClE,OAAO,OAAwB,CAAC;YACpC,CAAC;YACD,sEAAsE;YACtE,IAAI,CAAC;gBACD,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;gBAC5C,OAAO,IAAI,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;YAClD,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACT,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,uCAAuC,EAAE;oBACtD,KAAK,EAAE,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;iBACpD,CAAC,CAAC;gBACH,OAAO,IAAI,CAAC;YAChB,CAAC;QACL,CAAC;QAED,mCAAmC;QACnC,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAC9B,OAAO,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,yCAAyC,CAAC,CAAC;QAC5D,OAAO,IAAI,CAAC;IAChB,CAAC;IAED;;;OAGG;IACO,oBAAoB,CAAC,WAAmB;QAC9C,IAAI,CAAC,WAAW,IAAI,WAAW,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;YAC5C,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gEAAgE,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC;QAEzG,+BAA+B;QAC/B,IAAI,CAAC;YACD,MAAM,cAAc,GAAkB,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAC9D,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAC3C,OAAO,cAAc,CAAC;QAC1B,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,wDAAwD,CAAC,CAAC;QAChF,CAAC;QAED,iBAAiB;QACjB,IAAI,OAAO,GAAG,WAAW,CAAC,IAAI,EAAE,CAAC;QAEjC,6BAA6B;QAC7B,MAAM,cAAc,GAAG,OAAO,CAAC,KAAK,CAAC,oCAAoC,CAAC,CAAC;QAC3E,IAAI,cAAc,EAAE,CAAC;YACjB,OAAO,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YACnC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;QACtD,CAAC;QAED,mEAAmE;QACnE,IAAI,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YACvC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC9D,CAAC;QAED,yBAAyB;QACzB,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACzC,MAAM,UAAU,GAAG,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAE5C,IAAI,WAAW,KAAK,CAAC,CAAC,IAAI,UAAU,KAAK,CAAC,CAAC,IAAI,UAAU,GAAG,WAAW,EAAE,CAAC;YACtE,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,CAAC,WAAW,EAAE,UAAU,GAAG,CAAC,CAAC,CAAC;YAChE,IAAI,CAAC;gBACD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;gBACpC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;gBAC9D,OAAO,MAAM,CAAC;YAClB,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACT,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qCAAqC,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACrG,OAAO,IAAI,CAAC;YAChB,CAAC;QACL,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uCAAuC,CAAC,CAAC;QAC3D,OAAO,IAAI,CAAC;IAChB,CAAC;IAED;;;;;;;OAOG;IACO,iCAAiC,CACvC,OAAsB,EACtB,MAA6C,EAC7C,cAA8B;QAE9B,MAAM,eAAe,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC;QAErC,uEAAuE;QACvE,IAAI,cAAc,EAAE,CAAC;YACjB,eAAe,CAAC,IAAI,CAAC;gBACjB,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC;aAC1C,CAAC,CAAC;YACH,OAAO,eAAe,CAAC;QAC3B,CAAC;QAED,IAAI,MAAM,IAAI,MAAM,EAAE,CAAC;YACnB,0BAA0B;YAC1B,eAAe,CAAC,IAAI,CAAC;gBACjB,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,MAAM,CAAC,IAAI,IAAI,EAAE;aAC7B,CAAC,CAAC;YACH,OAAO,eAAe,CAAC;QAC3B,CAAC;QACD,0DAA0D;QAC1D,eAAe,CAAC,IAAI,CAAC,MAAqB,CAAC,CAAC;QAC5C,OAAO,eAAe,CAAC;IAC3B,CAAC;IAED;;;;;OAKG;IACO,yBAAyB,CAAC,OAAsB,EAAE,OAAY;QACpE,MAAM,eAAe,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC;QAErC,eAAe,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE;gBACL,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,OAAO;aACnB;SACJ,CAAC,CAAC;QAEH,OAAO,eAAe,CAAC;IAC3B,CAAC;CACJ;AApLD,kCAoLC"}

package/dist/ai/response_handlers/direct-response-handler.d.ts

@@ -0,0 +1,18 @@
+/**
+ * @fileoverview Direct Response Handler
+ *
+ * Handles DIRECT_RESPONSE message_type from AgentResponse.
+ * This is a terminal response type that breaks the execution loop.
+ */
+import { BaseHandler, HandlerContext, HandlerResult } from "./base-handler";
+import { ChatMessage } from "../../lib/schemas";
+export declare class DirectResponseHandler extends BaseHandler {
+    can_handle(message_type: string | null, output_type: string, input: ChatMessage): boolean;
+    handle(input: ChatMessage, context: HandlerContext): Promise<HandlerResult>;
+    /**
+     * Prepare input for OrchestratorAgent
+     * Formats the message for orchestrator routing
+     */
+    prepare_input(last_message: ChatMessage): ChatMessage[];
+}
+//# sourceMappingURL=direct-response-handler.d.ts.map

package/dist/ai/response_handlers/direct-response-handler.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"direct-response-handler.d.ts","sourceRoot":"","sources":["../../../src/ai/response_handlers/direct-response-handler.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAC5E,OAAO,EAAiB,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAE/D,qBAAa,qBAAsB,SAAQ,WAAW;IAClD,UAAU,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI,EAAE,WAAW,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,GAAG,OAAO;IAQnF,MAAM,CAAC,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,cAAc,GAAG,OAAO,CAAC,aAAa,CAAC;IAkDjF;;;OAGG;IACH,aAAa,CAAC,YAAY,EAAE,WAAW,GAAG,WAAW,EAAE;CAuB1D"}

package/dist/ai/response_handlers/direct-response-handler.js

@@ -0,0 +1,86 @@
+"use strict";
+/**
+ * @fileoverview Direct Response Handler
+ *
+ * Handles DIRECT_RESPONSE message_type from AgentResponse.
+ * This is a terminal response type that breaks the execution loop.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DirectResponseHandler = void 0;
+const base_handler_1 = require("./base-handler");
+class DirectResponseHandler extends base_handler_1.BaseHandler {
+    can_handle(message_type, output_type, input) {
+        // first extract the message into an object
+        return input.role === 'model' && message_type === 'DIRECT_RESPONSE';
+    }
+    async handle(input, context) {
+        this.logger.info('Direct response handler received result');
+        const result = this.extract_agent_response(input.content);
+        if (!result) {
+            this.logger.warn('⚠️ No AgentResponse in response');
+            return {
+                success: false,
+                result: [],
+                should_continue: false
+            };
+        }
+        const message_type = result.message_type;
+        if (!message_type) {
+            this.logger.warn('⚠️ No message_type in response');
+            return {
+                success: false,
+                result: [],
+                should_continue: false
+            };
+        }
+        // Extract domain from orchestrator response (for logging)
+        const extracted_domain = result?.domain || result?.content?.domain || null;
+        const available_domains = context?.available_domains || [];
+        // Log domain identification result from orchestrator
+        this.logger.info('🎯 DirectResponseHandler - Orchestrator domain identification', {
+            identified_domain: extracted_domain || 'null',
+            message_type: message_type,
+            available_domains: available_domains,
+            agent_kind: result?.agent_kind
+        });
+        this.logger.info(`✅ Direct response received - breaking execution loop`);
+        // send the message to the user as a direct response through the streaming utils
+        this.streaming_utils.write('direct_response', {
+            message: input.content
+        });
+        return {
+            success: true,
+            result: [],
+            should_continue: false,
+            context: {},
+            message_type: message_type
+        };
+    }
+    /**
+     * Prepare input for OrchestratorAgent
+     * Formats the message for orchestrator routing
+     */
+    prepare_input(last_message) {
+        if (!last_message) {
+            return [{
+                    role: 'user',
+                    content: 'Please help me with this task'
+                }];
+        }
+        // Extract the user's message
+        let user_message = '';
+        if (typeof last_message.content === 'string') {
+            user_message = last_message.content;
+        }
+        else if (last_message.content && typeof last_message.content === 'object') {
+            user_message = last_message.content.message || last_message.content.text || JSON.stringify(last_message.content);
+        }
+        const formatted_message = user_message || 'Please help me with this task';
+        return [{
+                role: 'user',
+                content: formatted_message
+            }];
+    }
+}
+exports.DirectResponseHandler = DirectResponseHandler;
+//# sourceMappingURL=direct-response-handler.js.map

package/dist/ai/response_handlers/direct-response-handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"direct-response-handler.js","sourceRoot":"","sources":["../../../src/ai/response_handlers/direct-response-handler.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAGH,iDAA4E;AAG5E,MAAa,qBAAsB,SAAQ,0BAAW;IAClD,UAAU,CAAC,YAA2B,EAAE,WAAmB,EAAE,KAAkB;QAE3E,2CAA2C;QAG3C,OAAO,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,YAAY,KAAK,iBAAiB,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAkB,EAAE,OAAuB;QACpD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,yCAAyC,CAAC,CAAC;QAE5D,MAAM,MAAM,GAAG,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC1D,IAAI,CAAC,MAAM,EAAE,CAAC;YACV,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;YACpD,OAAO;gBACH,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,EAAE;gBACV,eAAe,EAAE,KAAK;aACzB,CAAC;QACN,CAAC;QAGD,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,YAAY,EAAE,CAAC;YAChB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;YACnD,OAAO;gBACH,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,EAAE;gBACV,eAAe,EAAE,KAAK;aACzB,CAAC;QACN,CAAC;QAED,0DAA0D;QAC1D,MAAM,gBAAgB,GAAI,MAAc,EAAE,MAAM,IAAK,MAAc,EAAE,OAAO,EAAE,MAAM,IAAI,IAAI,CAAC;QAC7F,MAAM,iBAAiB,GAAI,OAAe,EAAE,iBAAiB,IAAI,EAAE,CAAC;QAEpE,qDAAqD;QACrD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,+DAA+D,EAAE;YAC9E,iBAAiB,EAAE,gBAAgB,IAAI,MAAM;YAC7C,YAAY,EAAE,YAAY;YAC1B,iBAAiB,EAAE,iBAAiB;YACpC,UAAU,EAAG,MAAc,EAAE,UAAU;SAC1C,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sDAAsD,CAAC,CAAC;QACzE,gFAAgF;QAChF,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,iBAAiB,EAAE;YAC1C,OAAO,EAAE,KAAK,CAAC,OAAiB;SACnC,CAAC,CAAC;QACH,OAAO;YACH,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,EAAE;YACV,eAAe,EAAE,KAAK;YACtB,OAAO,EAAE,EAAE;YACX,YAAY,EAAE,YAAY;SAC7B,CAAC;IACN,CAAC;IAED;;;OAGG;IACH,aAAa,CAAC,YAAyB;QACnC,IAAI,CAAC,YAAY,EAAE,CAAC;YAChB,OAAO,CAAC;oBACJ,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,+BAA+B;iBAC3C,CAAC,CAAC;QACP,CAAC;QAED,6BAA6B;QAC7B,IAAI,YAAY,GAAG,EAAE,CAAC;QACtB,IAAI,OAAO,YAAY,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YAC3C,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC;QACxC,CAAC;aAAM,IAAI,YAAY,CAAC,OAAO,IAAI,OAAO,YAAY,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YAC1E,YAAY,GAAI,YAAY,CAAC,OAAe,CAAC,OAAO,IAAK,YAAY,CAAC,OAAe,CAAC,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QACvI,CAAC;QAED,MAAM,iBAAiB,GAAG,YAAY,IAAI,+BAA+B,CAAC;QAE1E,OAAO,CAAC;gBACJ,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,iBAAiB;aAC7B,CAAC,CAAC;IACP,CAAC;CACJ;AAtFD,sDAsFC"}

package/dist/ai/response_handlers/error-handler.d.ts

@@ -0,0 +1,13 @@
+/**
+ * @fileoverview Error Handler
+ *
+ * Handles error output types by returning error responses.
+ */
+import type { GenerateContentResponse } from "@google/genai";
+import { BaseHandler, HandlerContext, HandlerResult } from "./base-handler";
+import { ChatMessage } from "../../lib/schemas";
+export declare class ErrorHandler extends BaseHandler {
+    can_handle(message_type: string | null, output_type: string, input: ChatMessage): boolean;
+    handle(result: GenerateContentResponse | ChatMessage, context: HandlerContext): Promise<HandlerResult>;
+}
+//# sourceMappingURL=error-handler.d.ts.map

package/dist/ai/response_handlers/error-handler.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"error-handler.d.ts","sourceRoot":"","sources":["../../../src/ai/response_handlers/error-handler.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AAC7D,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAC5E,OAAO,EAAiB,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAE/D,qBAAa,YAAa,SAAQ,WAAW;IAC5C,UAAU,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI,EAAE,WAAW,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,GAAG,OAAO;IAKnF,MAAM,CAAC,MAAM,EAAE,uBAAuB,GAAG,WAAW,EAAE,OAAO,EAAE,cAAc,GAAG,OAAO,CAAC,aAAa,CAAC;CA6D5G"}

package/dist/ai/response_handlers/error-handler.js

@@ -0,0 +1,72 @@
+"use strict";
+/**
+ * @fileoverview Error Handler
+ *
+ * Handles error output types by returning error responses.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ErrorHandler = void 0;
+const base_handler_1 = require("./base-handler");
+class ErrorHandler extends base_handler_1.BaseHandler {
+    can_handle(message_type, output_type, input) {
+        // Check message_type first, then output_type
+        return message_type === 'ERROR' || output_type === 'error';
+    }
+    async handle(result, context) {
+        // Extract error message from GenerateContentResponse or ChatMessage
+        let error_message;
+        let content_str;
+        if ('text' in result) {
+            // It's a GenerateContentResponse
+            error_message = result.error?.message ||
+                (result.candidates && result.candidates.length === 0 ? 'No candidates in response' : 'Unknown error');
+            content_str = result.text;
+        }
+        else {
+            // It's a ChatMessage
+            error_message = 'Error in response';
+            const content = result.content;
+            content_str = typeof content === 'string' ? content : content.text || JSON.stringify(content);
+        }
+        this.logger.warn(`⚠️ Error in response: ${error_message}`);
+        // Try to parse AgentResponse from text if available
+        let updated_history = [];
+        if (content_str) {
+            // Use base handler extraction method to handle markdown, escape sequences, etc.
+            const agent_response = this.extract_agent_response(content_str);
+            if (agent_response) {
+                updated_history = this.append_gemini_response_to_history([], result, agent_response);
+            }
+            // If extraction fails, just use current history
+        }
+        // Check if we should retry (if we haven't exceeded max attempts)
+        const should_retry = context.attempt_number < context.max_attempts;
+        const handler_context = {
+            error: error_message,
+            error_details: result.error
+        };
+        if (should_retry) {
+            this.logger.info(`🔄 Retrying after error (attempt ${context.attempt_number + 1}/{(context as { attempt_number: number; max_attempts: number }).max_attempts})`);
+            const retry_context = { ...handler_context, retrying: true };
+            const history_with_context = this.append_context_to_history(updated_history, retry_context);
+            return {
+                success: false,
+                result: history_with_context,
+                should_continue: true,
+                context: retry_context,
+                message_type: null // Retry with same agent
+            };
+        }
+        // Max attempts reached - return error
+        const history_with_context = this.append_context_to_history(updated_history, handler_context);
+        return {
+            success: false,
+            result: history_with_context,
+            should_continue: false,
+            context: handler_context,
+            message_type: 'ERROR' // Terminal error state
+        };
+    }
+}
+exports.ErrorHandler = ErrorHandler;
+//# sourceMappingURL=error-handler.js.map

package/dist/ai/response_handlers/error-handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"error-handler.js","sourceRoot":"","sources":["../../../src/ai/response_handlers/error-handler.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAGH,iDAA4E;AAG5E,MAAa,YAAa,SAAQ,0BAAW;IAC5C,UAAU,CAAC,YAA2B,EAAE,WAAmB,EAAE,KAAkB;QAC9E,6CAA6C;QAC7C,OAAO,YAAY,KAAK,OAAO,IAAI,WAAW,KAAK,OAAO,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAA6C,EAAE,OAAuB;QAClF,oEAAoE;QACpE,IAAI,aAAqB,CAAC;QAC1B,IAAI,WAA+B,CAAC;QAEpC,IAAI,MAAM,IAAI,MAAM,EAAE,CAAC;YACtB,iCAAiC;YACjC,aAAa,GAAI,MAAc,CAAC,KAAK,EAAE,OAAO;gBAC7C,CAAC,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC;YACvG,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC;QAC3B,CAAC;aAAM,CAAC;YACP,qBAAqB;YACrB,aAAa,GAAG,mBAAmB,CAAC;YACpC,MAAM,OAAO,GAAI,MAAsB,CAAC,OAAO,CAAC;YAChD,WAAW,GAAG,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAE,OAAe,CAAC,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACxG,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,yBAAyB,aAAa,EAAE,CAAC,CAAC;QAE3D,oDAAoD;QACpD,IAAI,eAAe,GAAkB,EAAmB,CAAC;QACzD,IAAI,WAAW,EAAE,CAAC;YACjB,gFAAgF;YAChF,MAAM,cAAc,GAAG,IAAI,CAAC,sBAAsB,CAAC,WAAW,CAAC,CAAC;YAChE,IAAI,cAAc,EAAE,CAAC;gBACpB,eAAe,GAAG,IAAI,CAAC,iCAAiC,CAAC,EAAE,EAAE,MAAM,EAAE,cAAc,CAAC,CAAC;YACtF,CAAC;YACD,gDAAgD;QACjD,CAAC;QAED,iEAAiE;QACjE,MAAM,YAAY,GAAI,OAA4D,CAAC,cAAc,GAAI,OAA4D,CAAC,YAAY,CAAC;QAE/K,MAAM,eAAe,GAAG;YACvB,KAAK,EAAE,aAAa;YACpB,aAAa,EAAG,MAAc,CAAC,KAAK;SACpC,CAAC;QAEF,IAAI,YAAY,EAAE,CAAC;YAClB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,oCAAqC,OAA4D,CAAC,cAAc,GAAG,CAAC,gFAAgF,CAAC,CAAC;YACvN,MAAM,aAAa,GAAG,EAAE,GAAG,eAAe,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;YAC7D,MAAM,oBAAoB,GAAG,IAAI,CAAC,yBAAyB,CAAC,eAAe,EAAE,aAAa,CAAC,CAAC;YAC5F,OAAO;gBACN,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,oBAAoB;gBAC5B,eAAe,EAAE,IAAI;gBACrB,OAAO,EAAE,aAAa;gBACtB,YAAY,EAAE,IAAI,CAAC,wBAAwB;aAC3C,CAAC;QACH,CAAC;QAED,sCAAsC;QACtC,MAAM,oBAAoB,GAAG,IAAI,CAAC,yBAAyB,CAAC,eAAe,EAAE,eAAe,CAAC,CAAC;QAC9F,OAAO;YACN,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,oBAAoB;YAC5B,eAAe,EAAE,KAAK;YACtB,OAAO,EAAE,eAAe;YACxB,YAAY,EAAE,OAAO,CAAC,uBAAuB;SAC7C,CAAC;IACH,CAAC;CACD;AAnED,oCAmEC"}

package/dist/ai/response_handlers/handler-registry.d.ts

@@ -0,0 +1,51 @@
+/**
+ * @fileoverview Response Handler Registry
+ *
+ * Registry for managing and selecting appropriate handlers.
+ * Routes by message_type (from AgentResponse) first, then falls back to output_type (from OpenAI Response).
+ */
+import { BaseHandler, HandlerContext, HandlerResult } from "./base-handler";
+import type { ChatMessage } from "../../lib/schemas";
+import { Logger } from "../../lib/utils/logger";
+import { StreamingUtils } from "../../lib/streaming-utils";
+export declare class HandlerRegistry {
+    private logger;
+    private streaming_utils;
+    private message_type_handlers;
+    private output_type_handlers;
+    private user_input_handler;
+    constructor(logger: Logger, streaming_utils: StreamingUtils);
+    /**
+     * Get handler for preparing input based on message_type
+     * This is used to format input before calling an agent
+     */
+    get_handler_for_prepare_input(message_type: string | null): BaseHandler | null;
+    /**
+     * Extract message_type from ChatMessage content
+     * Handles both string content and object content
+     */
+    private extract_message_type_from_content;
+    /**
+     * Extract output_type from ChatMessage content
+     * For model messages, we infer output_type from the content structure
+     */
+    private extract_output_type_from_content;
+    /**
+     * Extract JSON object from text that may contain natural language or markdown code blocks
+     * Looks for JSON objects in the text (starting with { and ending with })
+     * PUBLIC so handlers can use this for parsing raw output_text
+     */
+    extract_json_from_text(text: string): string | null;
+    /**
+     * Find and return the appropriate handler for the given ChatMessage
+     * Uses heuristics-based classification:
+     * - If message.role == 'user', returns UserInputHandler
+     * - Otherwise, parses content to extract message_type/output_type and finds appropriate handler
+     */
+    find_handler(message: ChatMessage): BaseHandler | null;
+    /**
+     * Handle the response using the appropriate handler
+     */
+    handle_response(message: ChatMessage, context: HandlerContext): Promise<HandlerResult>;
+}
+//# sourceMappingURL=handler-registry.d.ts.map

package/dist/ai/response_handlers/handler-registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"handler-registry.d.ts","sourceRoot":"","sources":["../../../src/ai/response_handlers/handler-registry.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAC5E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAUrD,OAAO,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AAEhD,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAE3D,qBAAa,eAAe;IAKZ,OAAO,CAAC,MAAM;IAAU,OAAO,CAAC,eAAe;IAJ3D,OAAO,CAAC,qBAAqB,CAAqB;IAClD,OAAO,CAAC,oBAAoB,CAAqB;IACjD,OAAO,CAAC,kBAAkB,CAAmB;gBAEzB,MAAM,EAAE,MAAM,EAAU,eAAe,EAAE,cAAc;IAiB3E;;;OAGG;IACH,6BAA6B,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI,GAAG,WAAW,GAAG,IAAI;IAyB9E;;;OAGG;IACH,OAAO,CAAC,iCAAiC;IAqDzC;;;OAGG;IACH,OAAO,CAAC,gCAAgC;IAoBxC;;;;OAIG;IACI,sBAAsB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAuE1D;;;;;OAKG;IACH,YAAY,CAAC,OAAO,EAAE,WAAW,GAAG,WAAW,GAAG,IAAI;IA6CtD;;OAEG;IACG,eAAe,CAAC,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,cAAc,GAAG,OAAO,CAAC,aAAa,CAAC;CAc/F"}