illuma-agents 1.0.2

package/dist/esm/events.mjs

@@ -0,0 +1,135 @@
+import { handleToolCalls } from './tools/handlers.mjs';
+import { Providers } from './common/enum.mjs';
+
+class HandlerRegistry {
+    handlers = new Map();
+    register(eventType, handler) {
+        this.handlers.set(eventType, handler);
+    }
+    getHandler(eventType) {
+        return this.handlers.get(eventType);
+    }
+}
+class ModelEndHandler {
+    collectedUsage;
+    constructor(collectedUsage) {
+        if (collectedUsage && !Array.isArray(collectedUsage)) {
+            throw new Error('collectedUsage must be an array');
+        }
+        this.collectedUsage = collectedUsage;
+    }
+    handle(event, data, metadata, graph) {
+        if (!graph || !metadata) {
+            console.warn(`Graph or metadata not found in ${event} event`);
+            return;
+        }
+        const usage = data?.output?.usage_metadata;
+        if (usage != null && this.collectedUsage != null) {
+            this.collectedUsage.push(usage);
+        }
+        if (metadata.ls_provider === 'FakeListChatModel') {
+            return handleToolCalls(data?.output?.tool_calls, metadata, graph);
+        }
+        console.log(`====== ${event.toUpperCase()} ======`);
+        console.dir({
+            usage,
+        }, { depth: null });
+        if (metadata.provider !== Providers.GOOGLE) {
+            return;
+        }
+        handleToolCalls(data?.output?.tool_calls, metadata, graph);
+    }
+}
+class ToolEndHandler {
+    callback;
+    omitOutput;
+    constructor(callback, omitOutput) {
+        this.callback = callback;
+        this.omitOutput = omitOutput;
+    }
+    handle(event, data, metadata, graph) {
+        if (!graph || !metadata) {
+            console.warn(`Graph or metadata not found in ${event} event`);
+            return;
+        }
+        const toolEndData = data;
+        if (!toolEndData?.output) {
+            console.warn('No output found in tool_end event');
+            return;
+        }
+        this.callback?.(toolEndData, metadata);
+        graph.handleToolCallCompleted({ input: toolEndData.input, output: toolEndData.output }, metadata, this.omitOutput?.(toolEndData.output.name));
+    }
+}
+class TestLLMStreamHandler {
+    handle(event, data) {
+        const chunk = data?.chunk;
+        const isMessageChunk = !!(chunk && 'message' in chunk);
+        const msg = isMessageChunk ? chunk.message : undefined;
+        if (msg && msg.tool_call_chunks && msg.tool_call_chunks.length > 0) {
+            console.log(msg.tool_call_chunks);
+        }
+        else if (msg && msg.content) {
+            if (typeof msg.content === 'string') {
+                process.stdout.write(msg.content);
+            }
+        }
+    }
+}
+class TestChatStreamHandler {
+    handle(event, data) {
+        const chunk = data?.chunk;
+        const isContentChunk = !!(chunk && 'content' in chunk);
+        const content = isContentChunk && chunk.content;
+        if (!content || !isContentChunk) {
+            return;
+        }
+        if (chunk.tool_call_chunks && chunk.tool_call_chunks.length > 0) {
+            console.dir(chunk.tool_call_chunks, { depth: null });
+        }
+        if (typeof content === 'string') {
+            process.stdout.write(content);
+        }
+        else {
+            console.dir(content, { depth: null });
+        }
+    }
+}
+class LLMStreamHandler {
+    handle(event, data, metadata) {
+        const chunk = data?.chunk;
+        const isMessageChunk = !!(chunk && 'message' in chunk);
+        const msg = isMessageChunk && chunk.message;
+        if (metadata) {
+            console.log(metadata);
+        }
+        if (msg && msg.tool_call_chunks && msg.tool_call_chunks.length > 0) {
+            console.log(msg.tool_call_chunks);
+        }
+        else if (msg && msg.content) {
+            if (typeof msg.content === 'string') {
+                // const text_delta = msg.content;
+                // dispatchCustomEvent(GraphEvents.CHAT_MODEL_STREAM, { chunk }, config);
+                process.stdout.write(msg.content);
+            }
+        }
+    }
+}
+const createMetadataAggregator = (_collected) => {
+    const collected = _collected || [];
+    const handleLLMEnd = (output) => {
+        const { generations } = output;
+        const lastMessageOutput = generations[generations.length - 1]?.[0];
+        if (!lastMessageOutput) {
+            return;
+        }
+        const { message } = lastMessageOutput;
+        if (message?.response_metadata) {
+            collected.push(message.response_metadata);
+        }
+    };
+    return { handleLLMEnd, collected };
+};
+
+export { HandlerRegistry, LLMStreamHandler, ModelEndHandler, TestChatStreamHandler, TestLLMStreamHandler, ToolEndHandler, createMetadataAggregator };
+//# sourceMappingURL=events.mjs.map

package/dist/esm/events.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"events.mjs","sources":["../../src/events.ts"],"sourcesContent":["/* eslint-disable no-console */\r\n// src/events.ts\r\nimport type {\r\n  UsageMetadata,\r\n  BaseMessageFields,\r\n} from '@langchain/core/messages';\r\nimport type { Graph } from '@/graphs';\r\nimport type * as t from '@/types';\r\nimport { handleToolCalls } from '@/tools/handlers';\r\nimport { Providers } from '@/common';\r\n\r\nexport class HandlerRegistry {\r\n  private handlers: Map<string, t.EventHandler> = new Map();\r\n\r\n  register(eventType: string, handler: t.EventHandler): void {\r\n    this.handlers.set(eventType, handler);\r\n  }\r\n\r\n  getHandler(eventType: string): t.EventHandler | undefined {\r\n    return this.handlers.get(eventType);\r\n  }\r\n}\r\n\r\nexport class ModelEndHandler implements t.EventHandler {\r\n  collectedUsage?: UsageMetadata[];\r\n  constructor(collectedUsage?: UsageMetadata[]) {\r\n    if (collectedUsage && !Array.isArray(collectedUsage)) {\r\n      throw new Error('collectedUsage must be an array');\r\n    }\r\n    this.collectedUsage = collectedUsage;\r\n  }\r\n\r\n  handle(\r\n    event: string,\r\n    data: t.ModelEndData,\r\n    metadata?: Record<string, unknown>,\r\n    graph?: Graph\r\n  ): void {\r\n    if (!graph || !metadata) {\r\n      console.warn(`Graph or metadata not found in ${event} event`);\r\n      return;\r\n    }\r\n\r\n    const usage = data?.output?.usage_metadata;\r\n    if (usage != null && this.collectedUsage != null) {\r\n      this.collectedUsage.push(usage);\r\n    }\r\n\r\n    if (metadata.ls_provider === 'FakeListChatModel') {\r\n      return handleToolCalls(data?.output?.tool_calls, metadata, graph);\r\n    }\r\n\r\n    console.log(`====== ${event.toUpperCase()} ======`);\r\n    console.dir(\r\n      {\r\n        usage,\r\n      },\r\n      { depth: null }\r\n    );\r\n\r\n    if (metadata.provider !== Providers.GOOGLE) {\r\n      return;\r\n    }\r\n\r\n    handleToolCalls(data?.output?.tool_calls, metadata, graph);\r\n  }\r\n}\r\n\r\nexport class ToolEndHandler implements t.EventHandler {\r\n  private callback?: t.ToolEndCallback;\r\n  private omitOutput?: (name?: string) => boolean;\r\n  constructor(\r\n    callback?: t.ToolEndCallback,\r\n    omitOutput?: (name?: string) => boolean\r\n  ) {\r\n    this.callback = callback;\r\n    this.omitOutput = omitOutput;\r\n  }\r\n  handle(\r\n    event: string,\r\n    data: t.StreamEventData | undefined,\r\n    metadata?: Record<string, unknown>,\r\n    graph?: Graph\r\n  ): void {\r\n    if (!graph || !metadata) {\r\n      console.warn(`Graph or metadata not found in ${event} event`);\r\n      return;\r\n    }\r\n\r\n    const toolEndData = data as t.ToolEndData | undefined;\r\n    if (!toolEndData?.output) {\r\n      console.warn('No output found in tool_end event');\r\n      return;\r\n    }\r\n\r\n    this.callback?.(toolEndData, metadata);\r\n    graph.handleToolCallCompleted(\r\n      { input: toolEndData.input, output: toolEndData.output },\r\n      metadata,\r\n      this.omitOutput?.(toolEndData.output.name)\r\n    );\r\n  }\r\n}\r\n\r\nexport class TestLLMStreamHandler implements t.EventHandler {\r\n  handle(event: string, data: t.StreamEventData | undefined): void {\r\n    const chunk = data?.chunk;\r\n    const isMessageChunk = !!(chunk && 'message' in chunk);\r\n    const msg = isMessageChunk ? chunk.message : undefined;\r\n    if (msg && msg.tool_call_chunks && msg.tool_call_chunks.length > 0) {\r\n      console.log(msg.tool_call_chunks);\r\n    } else if (msg && msg.content) {\r\n      if (typeof msg.content === 'string') {\r\n        process.stdout.write(msg.content);\r\n      }\r\n    }\r\n  }\r\n}\r\n\r\nexport class TestChatStreamHandler implements t.EventHandler {\r\n  handle(event: string, data: t.StreamEventData | undefined): void {\r\n    const chunk = data?.chunk;\r\n    const isContentChunk = !!(chunk && 'content' in chunk);\r\n    const content = isContentChunk && chunk.content;\r\n\r\n    if (!content || !isContentChunk) {\r\n      return;\r\n    }\r\n\r\n    if (chunk.tool_call_chunks && chunk.tool_call_chunks.length > 0) {\r\n      console.dir(chunk.tool_call_chunks, { depth: null });\r\n    }\r\n\r\n    if (typeof content === 'string') {\r\n      process.stdout.write(content);\r\n    } else {\r\n      console.dir(content, { depth: null });\r\n    }\r\n  }\r\n}\r\n\r\nexport class LLMStreamHandler implements t.EventHandler {\r\n  handle(\r\n    event: string,\r\n    data: t.StreamEventData | undefined,\r\n    metadata?: Record<string, unknown>\r\n  ): void {\r\n    const chunk = data?.chunk;\r\n    const isMessageChunk = !!(chunk && 'message' in chunk);\r\n    const msg = isMessageChunk && chunk.message;\r\n    if (metadata) {\r\n      console.log(metadata);\r\n    }\r\n    if (msg && msg.tool_call_chunks && msg.tool_call_chunks.length > 0) {\r\n      console.log(msg.tool_call_chunks);\r\n    } else if (msg && msg.content) {\r\n      if (typeof msg.content === 'string') {\r\n        // const text_delta = msg.content;\r\n        // dispatchCustomEvent(GraphEvents.CHAT_MODEL_STREAM, { chunk }, config);\r\n        process.stdout.write(msg.content);\r\n      }\r\n    }\r\n  }\r\n}\r\n\r\nexport const createMetadataAggregator = (\r\n  _collected?: Record<\r\n    string,\r\n    NonNullable<BaseMessageFields['response_metadata']>\r\n  >[]\r\n): t.MetadataAggregatorResult => {\r\n  const collected = _collected || [];\r\n\r\n  const handleLLMEnd: t.HandleLLMEnd = (output) => {\r\n    const { generations } = output;\r\n    const lastMessageOutput = (\r\n      generations[generations.length - 1] as\r\n        | (t.StreamGeneration | undefined)[]\r\n        | undefined\r\n    )?.[0];\r\n    if (!lastMessageOutput) {\r\n      return;\r\n    }\r\n    const { message } = lastMessageOutput;\r\n    if (message?.response_metadata) {\r\n      collected.push(message.response_metadata);\r\n    }\r\n  };\r\n\r\n  return { handleLLMEnd, collected };\r\n};\r\n"],"names":[],"mappings":";;;MAWa,eAAe,CAAA;AAClB,IAAA,QAAQ,GAAgC,IAAI,GAAG,EAAE;IAEzD,QAAQ,CAAC,SAAiB,EAAE,OAAuB,EAAA;QACjD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC;;AAGvC,IAAA,UAAU,CAAC,SAAiB,EAAA;QAC1B,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC;;AAEtC;MAEY,eAAe,CAAA;AAC1B,IAAA,cAAc;AACd,IAAA,WAAA,CAAY,cAAgC,EAAA;QAC1C,IAAI,cAAc,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE;AACpD,YAAA,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC;;AAEpD,QAAA,IAAI,CAAC,cAAc,GAAG,cAAc;;AAGtC,IAAA,MAAM,CACJ,KAAa,EACb,IAAoB,EACpB,QAAkC,EAClC,KAAa,EAAA;AAEb,QAAA,IAAI,CAAC,KAAK,IAAI,CAAC,QAAQ,EAAE;AACvB,YAAA,OAAO,CAAC,IAAI,CAAC,kCAAkC,KAAK,CAAA,MAAA,CAAQ,CAAC;YAC7D;;AAGF,QAAA,MAAM,KAAK,GAAG,IAAI,EAAE,MAAM,EAAE,cAAc;QAC1C,IAAI,KAAK,IAAI,IAAI,IAAI,IAAI,CAAC,cAAc,IAAI,IAAI,EAAE;AAChD,YAAA,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC;;AAGjC,QAAA,IAAI,QAAQ,CAAC,WAAW,KAAK,mBAAmB,EAAE;AAChD,YAAA,OAAO,eAAe,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,CAAC;;QAGnE,OAAO,CAAC,GAAG,CAAC,CAAU,OAAA,EAAA,KAAK,CAAC,WAAW,EAAE,CAAS,OAAA,CAAA,CAAC;QACnD,OAAO,CAAC,GAAG,CACT;YACE,KAAK;AACN,SAAA,EACD,EAAE,KAAK,EAAE,IAAI,EAAE,CAChB;QAED,IAAI,QAAQ,CAAC,QAAQ,KAAK,SAAS,CAAC,MAAM,EAAE;YAC1C;;QAGF,eAAe,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,CAAC;;AAE7D;MAEY,cAAc,CAAA;AACjB,IAAA,QAAQ;AACR,IAAA,UAAU;IAClB,WACE,CAAA,QAA4B,EAC5B,UAAuC,EAAA;AAEvC,QAAA,IAAI,CAAC,QAAQ,GAAG,QAAQ;AACxB,QAAA,IAAI,CAAC,UAAU,GAAG,UAAU;;AAE9B,IAAA,MAAM,CACJ,KAAa,EACb,IAAmC,EACnC,QAAkC,EAClC,KAAa,EAAA;AAEb,QAAA,IAAI,CAAC,KAAK,IAAI,CAAC,QAAQ,EAAE;AACvB,YAAA,OAAO,CAAC,IAAI,CAAC,kCAAkC,KAAK,CAAA,MAAA,CAAQ,CAAC;YAC7D;;QAGF,MAAM,WAAW,GAAG,IAAiC;AACrD,QAAA,IAAI,CAAC,WAAW,EAAE,MAAM,EAAE;AACxB,YAAA,OAAO,CAAC,IAAI,CAAC,mCAAmC,CAAC;YACjD;;QAGF,IAAI,CAAC,QAAQ,GAAG,WAAW,EAAE,QAAQ,CAAC;AACtC,QAAA,KAAK,CAAC,uBAAuB,CAC3B,EAAE,KAAK,EAAE,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,WAAW,CAAC,MAAM,EAAE,EACxD,QAAQ,EACR,IAAI,CAAC,UAAU,GAAG,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAC3C;;AAEJ;MAEY,oBAAoB,CAAA;IAC/B,MAAM,CAAC,KAAa,EAAE,IAAmC,EAAA;AACvD,QAAA,MAAM,KAAK,GAAG,IAAI,EAAE,KAAK;QACzB,MAAM,cAAc,GAAG,CAAC,EAAE,KAAK,IAAI,SAAS,IAAI,KAAK,CAAC;AACtD,QAAA,MAAM,GAAG,GAAG,cAAc,GAAG,KAAK,CAAC,OAAO,GAAG,SAAS;AACtD,QAAA,IAAI,GAAG,IAAI,GAAG,CAAC,gBAAgB,IAAI,GAAG,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE;AAClE,YAAA,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC;;AAC5B,aAAA,IAAI,GAAG,IAAI,GAAG,CAAC,OAAO,EAAE;AAC7B,YAAA,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE;gBACnC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC;;;;AAIxC;MAEY,qBAAqB,CAAA;IAChC,MAAM,CAAC,KAAa,EAAE,IAAmC,EAAA;AACvD,QAAA,MAAM,KAAK,GAAG,IAAI,EAAE,KAAK;QACzB,MAAM,cAAc,GAAG,CAAC,EAAE,KAAK,IAAI,SAAS,IAAI,KAAK,CAAC;AACtD,QAAA,MAAM,OAAO,GAAG,cAAc,IAAI,KAAK,CAAC,OAAO;AAE/C,QAAA,IAAI,CAAC,OAAO,IAAI,CAAC,cAAc,EAAE;YAC/B;;AAGF,QAAA,IAAI,KAAK,CAAC,gBAAgB,IAAI,KAAK,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE;AAC/D,YAAA,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;;AAGtD,QAAA,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE;AAC/B,YAAA,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC;;aACxB;YACL,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;;;AAG1C;MAEY,gBAAgB,CAAA;AAC3B,IAAA,MAAM,CACJ,KAAa,EACb,IAAmC,EACnC,QAAkC,EAAA;AAElC,QAAA,MAAM,KAAK,GAAG,IAAI,EAAE,KAAK;QACzB,MAAM,cAAc,GAAG,CAAC,EAAE,KAAK,IAAI,SAAS,IAAI,KAAK,CAAC;AACtD,QAAA,MAAM,GAAG,GAAG,cAAc,IAAI,KAAK,CAAC,OAAO;QAC3C,IAAI,QAAQ,EAAE;AACZ,YAAA,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC;;AAEvB,QAAA,IAAI,GAAG,IAAI,GAAG,CAAC,gBAAgB,IAAI,GAAG,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE;AAClE,YAAA,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC;;AAC5B,aAAA,IAAI,GAAG,IAAI,GAAG,CAAC,OAAO,EAAE;AAC7B,YAAA,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE;;;gBAGnC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC;;;;AAIxC;AAEY,MAAA,wBAAwB,GAAG,CACtC,UAGG,KAC2B;AAC9B,IAAA,MAAM,SAAS,GAAG,UAAU,IAAI,EAAE;AAElC,IAAA,MAAM,YAAY,GAAmB,CAAC,MAAM,KAAI;AAC9C,QAAA,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM;AAC9B,QAAA,MAAM,iBAAiB,GACrB,WAAW,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAGnC,GAAG,CAAC,CAAC;QACN,IAAI,CAAC,iBAAiB,EAAE;YACtB;;AAEF,QAAA,MAAM,EAAE,OAAO,EAAE,GAAG,iBAAiB;AACrC,QAAA,IAAI,OAAO,EAAE,iBAAiB,EAAE;AAC9B,YAAA,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC;;AAE7C,KAAC;AAED,IAAA,OAAO,EAAE,YAAY,EAAE,SAAS,EAAE;AACpC;;;;"}