illuma-agents 1.0.2

package/dist/esm/utils/misc.mjs

@@ -0,0 +1,53 @@
+function isPresent(value) {
+    return value != null && value !== '';
+}
+/**
+ * Unescapes a c-escaped string
+ * @param str The string to unescape
+ * @returns The unescaped string
+ */
+const unescapeString = (string) => string.replace(/\\(.)/g, (_, char) => {
+    switch (char) {
+        case 'n':
+            return '\n';
+        case 't':
+            return '\t';
+        case 'r':
+            return '\r';
+        case '"':
+            return '"';
+        case '\'':
+            return '\'';
+        case '\\':
+            return '\\';
+        default:
+            return char;
+    }
+});
+/**
+ * Recursively unescapes all string values in an object
+ * @param obj The object to unescape
+ * @returns The unescaped object
+ */
+function unescapeObject(obj, key) {
+    if (typeof obj === 'string') {
+        let unescaped = unescapeString(obj);
+        if (key === 'filePath' && unescaped.match(/^"(.+)"$/)) {
+            unescaped = unescaped.substring(1, unescaped.length - 1);
+        }
+        return unescaped;
+    }
+    if (Array.isArray(obj)) {
+        return obj.map((value) => unescapeObject(value, key === 'contextPaths' ? 'filePath' : ''));
+    }
+    if (typeof obj === 'object' && obj !== null) {
+        return Object.fromEntries(Object.entries(obj).map(([key, value]) => [
+            key,
+            unescapeObject(value, key),
+        ]));
+    }
+    return obj;
+}
+
+export { isPresent, unescapeObject };
+//# sourceMappingURL=misc.mjs.map

package/dist/esm/utils/misc.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"misc.mjs","sources":["../../../src/utils/misc.ts"],"sourcesContent":["export function isPresent(value: string | null | undefined): value is string {\r\n  return value != null && value !== '';\r\n}\r\n\r\n/**\r\n * Unescapes a c-escaped string\r\n * @param str The string to unescape\r\n * @returns The unescaped string\r\n */\r\nconst unescapeString = (string: string): string =>\r\n  string.replace(/\\\\(.)/g, (_, char) => {\r\n    switch (char) {\r\n    case 'n':\r\n      return '\\n';\r\n    case 't':\r\n      return '\\t';\r\n    case 'r':\r\n      return '\\r';\r\n    case '\"':\r\n      return '\"';\r\n    case '\\'':\r\n      return '\\'';\r\n    case '\\\\':\r\n      return '\\\\';\r\n    default:\r\n      return char;\r\n    }\r\n  });\r\n\r\n/**\r\n * Recursively unescapes all string values in an object\r\n * @param obj The object to unescape\r\n * @returns The unescaped object\r\n */\r\nexport function unescapeObject(obj: unknown, key?: string): unknown {\r\n  if (typeof obj === 'string') {\r\n    let unescaped = unescapeString(obj);\r\n    if (key === 'filePath' && unescaped.match(/^\"(.+)\"$/)) {\r\n      unescaped = unescaped.substring(1, unescaped.length - 1);\r\n    }\r\n    return unescaped;\r\n  }\r\n  if (Array.isArray(obj)) {\r\n    return obj.map((value) =>\r\n      unescapeObject(value, key === 'contextPaths' ? 'filePath' : '')\r\n    );\r\n  }\r\n  if (typeof obj === 'object' && obj !== null) {\r\n    return Object.fromEntries(\r\n      Object.entries(obj).map(([key, value]) => [\r\n        key,\r\n        unescapeObject(value, key),\r\n      ])\r\n    );\r\n  }\r\n  return obj;\r\n}\r\n"],"names":[],"mappings":"AAAM,SAAU,SAAS,CAAC,KAAgC,EAAA;AACxD,IAAA,OAAO,KAAK,IAAI,IAAI,IAAI,KAAK,KAAK,EAAE;AACtC;AAEA;;;;AAIG;AACH,MAAM,cAAc,GAAG,CAAC,MAAc,KACpC,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,IAAI,KAAI;IACnC,QAAQ,IAAI;AACZ,QAAA,KAAK,GAAG;AACN,YAAA,OAAO,IAAI;AACb,QAAA,KAAK,GAAG;AACN,YAAA,OAAO,IAAI;AACb,QAAA,KAAK,GAAG;AACN,YAAA,OAAO,IAAI;AACb,QAAA,KAAK,GAAG;AACN,YAAA,OAAO,GAAG;AACZ,QAAA,KAAK,IAAI;AACP,YAAA,OAAO,IAAI;AACb,QAAA,KAAK,IAAI;AACP,YAAA,OAAO,IAAI;AACb,QAAA;AACE,YAAA,OAAO,IAAI;;AAEf,CAAC,CAAC;AAEJ;;;;AAIG;AACa,SAAA,cAAc,CAAC,GAAY,EAAE,GAAY,EAAA;AACvD,IAAA,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE;AAC3B,QAAA,IAAI,SAAS,GAAG,cAAc,CAAC,GAAG,CAAC;QACnC,IAAI,GAAG,KAAK,UAAU,IAAI,SAAS,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;AACrD,YAAA,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC,CAAC,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC;;AAE1D,QAAA,OAAO,SAAS;;AAElB,IAAA,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;QACtB,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC,KAAK,KACnB,cAAc,CAAC,KAAK,EAAE,GAAG,KAAK,cAAc,GAAG,UAAU,GAAG,EAAE,CAAC,CAChE;;IAEH,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,KAAK,IAAI,EAAE;QAC3C,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,KAAK;YACxC,GAAG;AACH,YAAA,cAAc,CAAC,KAAK,EAAE,GAAG,CAAC;AAC3B,SAAA,CAAC,CACH;;AAEH,IAAA,OAAO,GAAG;AACZ;;;;"}

package/dist/esm/utils/run.mjs

@@ -0,0 +1,66 @@
+import { Runnable, patchConfig, mergeConfigs } from '@langchain/core/runnables';
+import { AsyncLocalStorageProviderSingleton } from '@langchain/core/singletons';
+
+/**
+ * Delays the execution for a specified number of milliseconds.
+ *
+ * @param {number} ms - The number of milliseconds to delay.
+ * @return {Promise<void>} A promise that resolves after the specified delay.
+ */
+function sleep(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}
+class RunnableCallable extends Runnable {
+    lc_namespace = ['langgraph'];
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    func;
+    tags;
+    config;
+    trace = true;
+    recurse = true;
+    constructor(fields) {
+        super();
+        this.name = fields.name ?? fields.func.name;
+        this.func = fields.func;
+        this.config = fields.tags ? { tags: fields.tags } : undefined;
+        this.trace = fields.trace ?? this.trace;
+        this.recurse = fields.recurse ?? this.recurse;
+    }
+    async _tracedInvoke(input, config, runManager) {
+        return new Promise((resolve, reject) => {
+            const childConfig = patchConfig(config, {
+                callbacks: runManager?.getChild(),
+            });
+            void AsyncLocalStorageProviderSingleton.runWithConfig(childConfig, async () => {
+                try {
+                    const output = await this.func(input, childConfig);
+                    resolve(output);
+                }
+                catch (e) {
+                    reject(e);
+                }
+            });
+        });
+    }
+    async invoke(
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    input, options
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    ) {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        let returnValue;
+        if (this.trace) {
+            returnValue = await this._callWithConfig(this._tracedInvoke, input, mergeConfigs(this.config, options));
+        }
+        else {
+            returnValue = await this.func(input, mergeConfigs(this.config, options));
+        }
+        if (Runnable.isRunnable(returnValue) && this.recurse) {
+            return await returnValue.invoke(input, options);
+        }
+        return returnValue;
+    }
+}
+
+export { RunnableCallable, sleep };
+//# sourceMappingURL=run.mjs.map

package/dist/esm/utils/run.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"run.mjs","sources":["../../../src/utils/run.ts"],"sourcesContent":["import { CallbackManagerForChainRun } from '@langchain/core/callbacks/manager';\r\nimport {\r\n  mergeConfigs,\r\n  patchConfig,\r\n  Runnable,\r\n  RunnableConfig,\r\n} from '@langchain/core/runnables';\r\nimport { AsyncLocalStorageProviderSingleton } from '@langchain/core/singletons';\r\n\r\n/**\r\n * Delays the execution for a specified number of milliseconds.\r\n *\r\n * @param {number} ms - The number of milliseconds to delay.\r\n * @return {Promise<void>} A promise that resolves after the specified delay.\r\n */\r\nexport function sleep(ms: number): Promise<void> {\r\n  return new Promise((resolve) => setTimeout(resolve, ms));\r\n}\r\n\r\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\r\nexport interface RunnableCallableArgs extends Partial<any> {\r\n  name?: string;\r\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\r\n  func: (...args: any[]) => any;\r\n  tags?: string[];\r\n  trace?: boolean;\r\n  recurse?: boolean;\r\n}\r\n\r\nexport class RunnableCallable<I = unknown, O = unknown> extends Runnable<I, O> {\r\n  lc_namespace: string[] = ['langgraph'];\r\n\r\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\r\n  func: (...args: any[]) => any;\r\n\r\n  tags?: string[];\r\n\r\n  config?: RunnableConfig;\r\n\r\n  trace: boolean = true;\r\n\r\n  recurse: boolean = true;\r\n\r\n  constructor(fields: RunnableCallableArgs) {\r\n    super();\r\n    this.name = fields.name ?? fields.func.name;\r\n    this.func = fields.func;\r\n    this.config = fields.tags ? { tags: fields.tags } : undefined;\r\n    this.trace = fields.trace ?? this.trace;\r\n    this.recurse = fields.recurse ?? this.recurse;\r\n  }\r\n\r\n  protected async _tracedInvoke(\r\n    input: I,\r\n    config?: Partial<RunnableConfig>,\r\n    runManager?: CallbackManagerForChainRun\r\n  ): Promise<O> {\r\n    return new Promise<O>((resolve, reject) => {\r\n      const childConfig = patchConfig(config, {\r\n        callbacks: runManager?.getChild(),\r\n      });\r\n      void AsyncLocalStorageProviderSingleton.runWithConfig(\r\n        childConfig,\r\n        async () => {\r\n          try {\r\n            const output = await this.func(input, childConfig);\r\n            resolve(output);\r\n          } catch (e) {\r\n            reject(e);\r\n          }\r\n        }\r\n      );\r\n    });\r\n  }\r\n\r\n  async invoke(\r\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\r\n    input: any,\r\n    options?: Partial<RunnableConfig> | undefined\r\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\r\n  ): Promise<any> {\r\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\r\n    let returnValue: any;\r\n\r\n    if (this.trace) {\r\n      returnValue = await this._callWithConfig(\r\n        this._tracedInvoke,\r\n        input,\r\n        mergeConfigs(this.config, options)\r\n      );\r\n    } else {\r\n      returnValue = await this.func(input, mergeConfigs(this.config, options));\r\n    }\r\n\r\n    if (Runnable.isRunnable(returnValue) && this.recurse) {\r\n      return await returnValue.invoke(input, options);\r\n    }\r\n\r\n    return returnValue;\r\n  }\r\n}"],"names":[],"mappings":";;;AASA;;;;;AAKG;AACG,SAAU,KAAK,CAAC,EAAU,EAAA;AAC9B,IAAA,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,KAAK,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;AAC1D;AAYM,MAAO,gBAA2C,SAAQ,QAAc,CAAA;AAC5E,IAAA,YAAY,GAAa,CAAC,WAAW,CAAC;;AAGtC,IAAA,IAAI;AAEJ,IAAA,IAAI;AAEJ,IAAA,MAAM;IAEN,KAAK,GAAY,IAAI;IAErB,OAAO,GAAY,IAAI;AAEvB,IAAA,WAAA,CAAY,MAA4B,EAAA;AACtC,QAAA,KAAK,EAAE;AACP,QAAA,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI;AAC3C,QAAA,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI;QACvB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,IAAI,GAAG,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,GAAG,SAAS;QAC7D,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK;QACvC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO;;AAGrC,IAAA,MAAM,aAAa,CAC3B,KAAQ,EACR,MAAgC,EAChC,UAAuC,EAAA;QAEvC,OAAO,IAAI,OAAO,CAAI,CAAC,OAAO,EAAE,MAAM,KAAI;AACxC,YAAA,MAAM,WAAW,GAAG,WAAW,CAAC,MAAM,EAAE;AACtC,gBAAA,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE;AAClC,aAAA,CAAC;YACF,KAAK,kCAAkC,CAAC,aAAa,CACnD,WAAW,EACX,YAAW;AACT,gBAAA,IAAI;oBACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,WAAW,CAAC;oBAClD,OAAO,CAAC,MAAM,CAAC;;gBACf,OAAO,CAAC,EAAE;oBACV,MAAM,CAAC,CAAC,CAAC;;AAEb,aAAC,CACF;AACH,SAAC,CAAC;;AAGJ,IAAA,MAAM,MAAM;;AAEV,IAAA,KAAU,EACV;;;;AAIA,QAAA,IAAI,WAAgB;AAEpB,QAAA,IAAI,IAAI,CAAC,KAAK,EAAE;YACd,WAAW,GAAG,MAAM,IAAI,CAAC,eAAe,CACtC,IAAI,CAAC,aAAa,EAClB,KAAK,EACL,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CACnC;;aACI;AACL,YAAA,WAAW,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;;QAG1E,IAAI,QAAQ,CAAC,UAAU,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE;YACpD,OAAO,MAAM,WAAW,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC;;AAGjD,QAAA,OAAO,WAAW;;AAErB;;;;"}

package/dist/esm/utils/title.mjs

@@ -0,0 +1,108 @@
+import { z } from 'zod';
+import { ChatPromptTemplate } from '@langchain/core/prompts';
+import { RunnableLambda, RunnableSequence } from '@langchain/core/runnables';
+import { ContentTypes } from '../common/enum.mjs';
+
+const defaultTitlePrompt = `Analyze this conversation and provide:
+1. The detected language of the conversation
+2. A concise title in the detected language (5 words or less, no punctuation or quotation)
+
+{convo}`;
+const titleSchema = z.object({
+    title: z
+        .string()
+        .describe('A concise title for the conversation in 5 words or less, without punctuation or quotation'),
+});
+const combinedSchema = z.object({
+    language: z.string().describe('The detected language of the conversation'),
+    title: z
+        .string()
+        .describe('A concise title for the conversation in 5 words or less, without punctuation or quotation'),
+});
+const createTitleRunnable = async (model, _titlePrompt) => {
+    // Disabled since this works fine
+    // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+    /* @ts-ignore */
+    const titleLLM = model.withStructuredOutput(titleSchema);
+    // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+    /* @ts-ignore */
+    const combinedLLM = model.withStructuredOutput(combinedSchema);
+    const titlePrompt = ChatPromptTemplate.fromTemplate(_titlePrompt ?? defaultTitlePrompt).withConfig({ runName: 'TitlePrompt' });
+    const titleOnlyInnerChain = RunnableSequence.from([titlePrompt, titleLLM]);
+    const combinedInnerChain = RunnableSequence.from([titlePrompt, combinedLLM]);
+    /** Wrap titleOnlyChain in RunnableLambda to create parent span */
+    const titleOnlyChain = new RunnableLambda({
+        func: async (input, config) => {
+            return await titleOnlyInnerChain.invoke(input, config);
+        },
+    }).withConfig({ runName: 'TitleOnlyChain' });
+    /** Wrap combinedChain in RunnableLambda to create parent span */
+    const combinedChain = new RunnableLambda({
+        func: async (input, config) => {
+            return await combinedInnerChain.invoke(input, config);
+        },
+    }).withConfig({ runName: 'TitleLanguageChain' });
+    /** Runnable to add default values if needed */
+    const addDefaults = new RunnableLambda({
+        func: (result) => ({
+            language: result?.language ?? 'English',
+            title: result?.title ?? '',
+        }),
+    }).withConfig({ runName: 'AddDefaults' });
+    const combinedChainInner = RunnableSequence.from([
+        combinedChain,
+        addDefaults,
+    ]);
+    /** Wrap combinedChainWithDefaults in RunnableLambda to create parent span */
+    const combinedChainWithDefaults = new RunnableLambda({
+        func: async (input, config) => {
+            return await combinedChainInner.invoke(input, config);
+        },
+    }).withConfig({ runName: 'CombinedChainWithDefaults' });
+    return new RunnableLambda({
+        func: async (input, config) => {
+            const invokeInput = { convo: input.convo };
+            if (input.skipLanguage) {
+                return (await titleOnlyChain.invoke(invokeInput, config));
+            }
+            return await combinedChainWithDefaults.invoke(invokeInput, config);
+        },
+    }).withConfig({ runName: 'TitleGenerator' });
+};
+const defaultCompletionPrompt = `Provide a concise, 5-word-or-less title for the conversation, using title case conventions. Only return the title itself.
+
+Conversation:
+{convo}`;
+const createCompletionTitleRunnable = async (model, titlePrompt) => {
+    const completionPrompt = ChatPromptTemplate.fromTemplate(titlePrompt ?? defaultCompletionPrompt).withConfig({ runName: 'CompletionTitlePrompt' });
+    /** Runnable to extract content from model response */
+    const extractContent = new RunnableLambda({
+        func: (response) => {
+            let content = '';
+            if (typeof response.content === 'string') {
+                content = response.content;
+            }
+            else if (Array.isArray(response.content)) {
+                content = response.content
+                    .filter((part) => part.type === ContentTypes.TEXT)
+                    .map((part) => part.text)
+                    .join('');
+            }
+            return { title: content.trim() };
+        },
+    }).withConfig({ runName: 'ExtractTitle' });
+    const innerChain = RunnableSequence.from([
+        completionPrompt,
+        model,
+        extractContent,
+    ]);
+    /** Wrap in RunnableLambda to create a parent span for LangFuse */
+    return new RunnableLambda({
+        func: async (input, config) => {
+            return await innerChain.invoke(input, config);
+        },
+    }).withConfig({ runName: 'CompletionTitleChain' });
+};
+
+export { createCompletionTitleRunnable, createTitleRunnable };
+//# sourceMappingURL=title.mjs.map

package/dist/esm/utils/title.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"title.mjs","sources":["../../../src/utils/title.ts"],"sourcesContent":["import { z } from 'zod';\r\nimport { ChatPromptTemplate } from '@langchain/core/prompts';\r\nimport { RunnableLambda, RunnableSequence } from '@langchain/core/runnables';\r\nimport type { Runnable, RunnableConfig } from '@langchain/core/runnables';\r\nimport type { AIMessage } from '@langchain/core/messages';\r\nimport type * as t from '@/types';\r\nimport { ContentTypes } from '@/common';\r\n\r\nconst defaultTitlePrompt = `Analyze this conversation and provide:\r\n1. The detected language of the conversation\r\n2. A concise title in the detected language (5 words or less, no punctuation or quotation)\r\n\r\n{convo}`;\r\n\r\nconst titleSchema = z.object({\r\n  title: z\r\n    .string()\r\n    .describe(\r\n      'A concise title for the conversation in 5 words or less, without punctuation or quotation'\r\n    ),\r\n});\r\n\r\nconst combinedSchema = z.object({\r\n  language: z.string().describe('The detected language of the conversation'),\r\n  title: z\r\n    .string()\r\n    .describe(\r\n      'A concise title for the conversation in 5 words or less, without punctuation or quotation'\r\n    ),\r\n});\r\n\r\nexport const createTitleRunnable = async (\r\n  model: t.ChatModelInstance,\r\n  _titlePrompt?: string\r\n): Promise<Runnable> => {\r\n  // Disabled since this works fine\r\n  // eslint-disable-next-line @typescript-eslint/ban-ts-comment\r\n  /* @ts-ignore */\r\n  const titleLLM = model.withStructuredOutput(titleSchema);\r\n  // eslint-disable-next-line @typescript-eslint/ban-ts-comment\r\n  /* @ts-ignore */\r\n  const combinedLLM = model.withStructuredOutput(combinedSchema);\r\n\r\n  const titlePrompt = ChatPromptTemplate.fromTemplate(\r\n    _titlePrompt ?? defaultTitlePrompt\r\n  ).withConfig({ runName: 'TitlePrompt' });\r\n\r\n  const titleOnlyInnerChain = RunnableSequence.from([titlePrompt, titleLLM]);\r\n  const combinedInnerChain = RunnableSequence.from([titlePrompt, combinedLLM]);\r\n\r\n  /** Wrap titleOnlyChain in RunnableLambda to create parent span */\r\n  const titleOnlyChain = new RunnableLambda({\r\n    func: async (\r\n      input: { convo: string },\r\n      config?: Partial<RunnableConfig>\r\n    ): Promise<{ title: string }> => {\r\n      return await titleOnlyInnerChain.invoke(input, config);\r\n    },\r\n  }).withConfig({ runName: 'TitleOnlyChain' });\r\n\r\n  /** Wrap combinedChain in RunnableLambda to create parent span */\r\n  const combinedChain = new RunnableLambda({\r\n    func: async (\r\n      input: { convo: string },\r\n      config?: Partial<RunnableConfig>\r\n    ): Promise<{ language: string; title: string }> => {\r\n      return await combinedInnerChain.invoke(input, config);\r\n    },\r\n  }).withConfig({ runName: 'TitleLanguageChain' });\r\n\r\n  /** Runnable to add default values if needed */\r\n  const addDefaults = new RunnableLambda({\r\n    func: (\r\n      result: { language: string; title: string } | undefined\r\n    ): { language: string; title: string } => ({\r\n      language: result?.language ?? 'English',\r\n      title: result?.title ?? '',\r\n    }),\r\n  }).withConfig({ runName: 'AddDefaults' });\r\n\r\n  const combinedChainInner = RunnableSequence.from([\r\n    combinedChain,\r\n    addDefaults,\r\n  ]);\r\n\r\n  /** Wrap combinedChainWithDefaults in RunnableLambda to create parent span */\r\n  const combinedChainWithDefaults = new RunnableLambda({\r\n    func: async (\r\n      input: { convo: string },\r\n      config?: Partial<RunnableConfig>\r\n    ): Promise<{ language: string; title: string }> => {\r\n      return await combinedChainInner.invoke(input, config);\r\n    },\r\n  }).withConfig({ runName: 'CombinedChainWithDefaults' });\r\n\r\n  return new RunnableLambda({\r\n    func: async (\r\n      input: {\r\n        convo: string;\r\n        inputText: string;\r\n        skipLanguage: boolean;\r\n      },\r\n      config?: Partial<RunnableConfig>\r\n    ): Promise<{ language: string; title: string } | { title: string }> => {\r\n      const invokeInput = { convo: input.convo };\r\n\r\n      if (input.skipLanguage) {\r\n        return (await titleOnlyChain.invoke(invokeInput, config)) as {\r\n          title: string;\r\n        };\r\n      }\r\n\r\n      return await combinedChainWithDefaults.invoke(invokeInput, config);\r\n    },\r\n  }).withConfig({ runName: 'TitleGenerator' });\r\n};\r\n\r\nconst defaultCompletionPrompt = `Provide a concise, 5-word-or-less title for the conversation, using title case conventions. Only return the title itself.\r\n\r\nConversation:\r\n{convo}`;\r\n\r\nexport const createCompletionTitleRunnable = async (\r\n  model: t.ChatModelInstance,\r\n  titlePrompt?: string\r\n): Promise<Runnable> => {\r\n  const completionPrompt = ChatPromptTemplate.fromTemplate(\r\n    titlePrompt ?? defaultCompletionPrompt\r\n  ).withConfig({ runName: 'CompletionTitlePrompt' });\r\n\r\n  /** Runnable to extract content from model response */\r\n  const extractContent = new RunnableLambda({\r\n    func: (response: AIMessage): { title: string } => {\r\n      let content = '';\r\n      if (typeof response.content === 'string') {\r\n        content = response.content;\r\n      } else if (Array.isArray(response.content)) {\r\n        content = response.content\r\n          .filter(\r\n            (part): part is { type: ContentTypes.TEXT; text: string } =>\r\n              part.type === ContentTypes.TEXT\r\n          )\r\n          .map((part) => part.text)\r\n          .join('');\r\n      }\r\n      return { title: content.trim() };\r\n    },\r\n  }).withConfig({ runName: 'ExtractTitle' });\r\n\r\n  const innerChain = RunnableSequence.from([\r\n    completionPrompt,\r\n    model,\r\n    extractContent,\r\n  ]);\r\n\r\n  /** Wrap in RunnableLambda to create a parent span for LangFuse */\r\n  return new RunnableLambda({\r\n    func: async (\r\n      input: { convo: string },\r\n      config?: Partial<RunnableConfig>\r\n    ): Promise<{ title: string }> => {\r\n      return await innerChain.invoke(input, config);\r\n    },\r\n  }).withConfig({ runName: 'CompletionTitleChain' });\r\n};\r\n"],"names":[],"mappings":";;;;;AAQA,MAAM,kBAAkB,GAAG,CAAA;;;;QAInB;AAER,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC;AAC3B,IAAA,KAAK,EAAE;AACJ,SAAA,MAAM;SACN,QAAQ,CACP,2FAA2F,CAC5F;AACJ,CAAA,CAAC;AAEF,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9B,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,2CAA2C,CAAC;AAC1E,IAAA,KAAK,EAAE;AACJ,SAAA,MAAM;SACN,QAAQ,CACP,2FAA2F,CAC5F;AACJ,CAAA,CAAC;AAEW,MAAA,mBAAmB,GAAG,OACjC,KAA0B,EAC1B,YAAqB,KACA;;;;IAIrB,MAAM,QAAQ,GAAG,KAAK,CAAC,oBAAoB,CAAC,WAAW,CAAC;;;IAGxD,MAAM,WAAW,GAAG,KAAK,CAAC,oBAAoB,CAAC,cAAc,CAAC;AAE9D,IAAA,MAAM,WAAW,GAAG,kBAAkB,CAAC,YAAY,CACjD,YAAY,IAAI,kBAAkB,CACnC,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;AAExC,IAAA,MAAM,mBAAmB,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;AAC1E,IAAA,MAAM,kBAAkB,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;;AAG5E,IAAA,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC;AACxC,QAAA,IAAI,EAAE,OACJ,KAAwB,EACxB,MAAgC,KACF;YAC9B,OAAO,MAAM,mBAAmB,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC;SACvD;KACF,CAAC,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;;AAG5C,IAAA,MAAM,aAAa,GAAG,IAAI,cAAc,CAAC;AACvC,QAAA,IAAI,EAAE,OACJ,KAAwB,EACxB,MAAgC,KACgB;YAChD,OAAO,MAAM,kBAAkB,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC;SACtD;KACF,CAAC,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,oBAAoB,EAAE,CAAC;;AAGhD,IAAA,MAAM,WAAW,GAAG,IAAI,cAAc,CAAC;AACrC,QAAA,IAAI,EAAE,CACJ,MAAuD,MACd;AACzC,YAAA,QAAQ,EAAE,MAAM,EAAE,QAAQ,IAAI,SAAS;AACvC,YAAA,KAAK,EAAE,MAAM,EAAE,KAAK,IAAI,EAAE;SAC3B,CAAC;KACH,CAAC,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;AAEzC,IAAA,MAAM,kBAAkB,GAAG,gBAAgB,CAAC,IAAI,CAAC;QAC/C,aAAa;QACb,WAAW;AACZ,KAAA,CAAC;;AAGF,IAAA,MAAM,yBAAyB,GAAG,IAAI,cAAc,CAAC;AACnD,QAAA,IAAI,EAAE,OACJ,KAAwB,EACxB,MAAgC,KACgB;YAChD,OAAO,MAAM,kBAAkB,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC;SACtD;KACF,CAAC,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,2BAA2B,EAAE,CAAC;IAEvD,OAAO,IAAI,cAAc,CAAC;AACxB,QAAA,IAAI,EAAE,OACJ,KAIC,EACD,MAAgC,KACoC;YACpE,MAAM,WAAW,GAAG,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE;AAE1C,YAAA,IAAI,KAAK,CAAC,YAAY,EAAE;gBACtB,QAAQ,MAAM,cAAc,CAAC,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC;;YAK1D,OAAO,MAAM,yBAAyB,CAAC,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC;SACnE;KACF,CAAC,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;AAC9C;AAEA,MAAM,uBAAuB,GAAG,CAAA;;;QAGxB;AAEK,MAAA,6BAA6B,GAAG,OAC3C,KAA0B,EAC1B,WAAoB,KACC;AACrB,IAAA,MAAM,gBAAgB,GAAG,kBAAkB,CAAC,YAAY,CACtD,WAAW,IAAI,uBAAuB,CACvC,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,uBAAuB,EAAE,CAAC;;AAGlD,IAAA,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC;AACxC,QAAA,IAAI,EAAE,CAAC,QAAmB,KAAuB;YAC/C,IAAI,OAAO,GAAG,EAAE;AAChB,YAAA,IAAI,OAAO,QAAQ,CAAC,OAAO,KAAK,QAAQ,EAAE;AACxC,gBAAA,OAAO,GAAG,QAAQ,CAAC,OAAO;;iBACrB,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE;gBAC1C,OAAO,GAAG,QAAQ,CAAC;AAChB,qBAAA,MAAM,CACL,CAAC,IAAI,KACH,IAAI,CAAC,IAAI,KAAK,YAAY,CAAC,IAAI;qBAElC,GAAG,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI;qBACvB,IAAI,CAAC,EAAE,CAAC;;YAEb,OAAO,EAAE,KAAK,EAAE,OAAO,CAAC,IAAI,EAAE,EAAE;SACjC;KACF,CAAC,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC;AAE1C,IAAA,MAAM,UAAU,GAAG,gBAAgB,CAAC,IAAI,CAAC;QACvC,gBAAgB;QAChB,KAAK;QACL,cAAc;AACf,KAAA,CAAC;;IAGF,OAAO,IAAI,cAAc,CAAC;AACxB,QAAA,IAAI,EAAE,OACJ,KAAwB,EACxB,MAAgC,KACF;YAC9B,OAAO,MAAM,UAAU,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC;SAC9C;KACF,CAAC,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,sBAAsB,EAAE,CAAC;AACpD;;;;"}

package/dist/esm/utils/tokens.mjs

@@ -0,0 +1,62 @@
+import { Tiktoken } from 'js-tiktoken/lite';
+import { ContentTypes } from '../common/enum.mjs';
+
+function getTokenCountForMessage(message, getTokenCount) {
+    const tokensPerMessage = 3;
+    const processValue = (value) => {
+        if (Array.isArray(value)) {
+            for (const item of value) {
+                if (!item ||
+                    !item.type ||
+                    item.type === ContentTypes.ERROR ||
+                    item.type === ContentTypes.IMAGE_URL) {
+                    continue;
+                }
+                if (item.type === ContentTypes.TOOL_CALL && item.tool_call != null) {
+                    const toolName = item.tool_call?.name || '';
+                    if (toolName != null && toolName && typeof toolName === 'string') {
+                        numTokens += getTokenCount(toolName);
+                    }
+                    const args = item.tool_call?.args || '';
+                    if (args != null && args && typeof args === 'string') {
+                        numTokens += getTokenCount(args);
+                    }
+                    const output = item.tool_call?.output || '';
+                    if (output != null && output && typeof output === 'string') {
+                        numTokens += getTokenCount(output);
+                    }
+                    continue;
+                }
+                const nestedValue = item[item.type];
+                if (!nestedValue) {
+                    continue;
+                }
+                processValue(nestedValue);
+            }
+        }
+        else if (typeof value === 'string') {
+            numTokens += getTokenCount(value);
+        }
+        else if (typeof value === 'number') {
+            numTokens += getTokenCount(value.toString());
+        }
+        else if (typeof value === 'boolean') {
+            numTokens += getTokenCount(value.toString());
+        }
+    };
+    let numTokens = tokensPerMessage;
+    processValue(message.content);
+    return numTokens;
+}
+const createTokenCounter = async () => {
+    const res = await fetch('https://tiktoken.pages.dev/js/o200k_base.json');
+    const o200k_base = await res.json();
+    const countTokens = (text) => {
+        const enc = new Tiktoken(o200k_base);
+        return enc.encode(text).length;
+    };
+    return (message) => getTokenCountForMessage(message, countTokens);
+};
+
+export { createTokenCounter, getTokenCountForMessage };
+//# sourceMappingURL=tokens.mjs.map

package/dist/esm/utils/tokens.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"tokens.mjs","sources":["../../../src/utils/tokens.ts"],"sourcesContent":["import { Tiktoken } from 'js-tiktoken/lite';\r\nimport type { BaseMessage } from '@langchain/core/messages';\r\nimport { ContentTypes } from '@/common/enum';\r\n\r\nexport function getTokenCountForMessage(message: BaseMessage, getTokenCount: (text: string) => number): number {\r\n  const tokensPerMessage = 3;\r\n\r\n  const processValue = (value: unknown): void => {\r\n    if (Array.isArray(value)) {\r\n      for (const item of value) {\r\n        if (\r\n          !item ||\r\n          !item.type ||\r\n          item.type === ContentTypes.ERROR ||\r\n          item.type === ContentTypes.IMAGE_URL\r\n        ) {\r\n          continue;\r\n        }\r\n\r\n        if (item.type === ContentTypes.TOOL_CALL && item.tool_call != null) {\r\n          const toolName = item.tool_call?.name || '';\r\n          if (toolName != null && toolName && typeof toolName === 'string') {\r\n            numTokens += getTokenCount(toolName);\r\n          }\r\n\r\n          const args = item.tool_call?.args || '';\r\n          if (args != null && args && typeof args === 'string') {\r\n            numTokens += getTokenCount(args);\r\n          }\r\n\r\n          const output = item.tool_call?.output || '';\r\n          if (output != null && output && typeof output === 'string') {\r\n            numTokens += getTokenCount(output);\r\n          }\r\n          continue;\r\n        }\r\n\r\n        const nestedValue = item[item.type];\r\n\r\n        if (!nestedValue) {\r\n          continue;\r\n        }\r\n\r\n        processValue(nestedValue);\r\n      }\r\n    } else if (typeof value === 'string') {\r\n      numTokens += getTokenCount(value);\r\n    } else if (typeof value === 'number') {\r\n      numTokens += getTokenCount(value.toString());\r\n    } else if (typeof value === 'boolean') {\r\n      numTokens += getTokenCount(value.toString());\r\n    }\r\n  };\r\n\r\n  let numTokens = tokensPerMessage;\r\n  processValue(message.content);\r\n  return numTokens;\r\n}\r\n\r\nexport const createTokenCounter = async () => {\r\n  const res = await fetch('https://tiktoken.pages.dev/js/o200k_base.json');\r\n  const o200k_base = await res.json();\r\n\r\n  const countTokens = (text: string): number => {\r\n    const enc = new Tiktoken(o200k_base);\r\n    return enc.encode(text).length;\r\n  };\r\n\r\n  return (message: BaseMessage): number => getTokenCountForMessage(message, countTokens);\r\n};"],"names":[],"mappings":";;;AAIgB,SAAA,uBAAuB,CAAC,OAAoB,EAAE,aAAuC,EAAA;IACnG,MAAM,gBAAgB,GAAG,CAAC;AAE1B,IAAA,MAAM,YAAY,GAAG,CAAC,KAAc,KAAU;AAC5C,QAAA,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE;AACxB,YAAA,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;AACxB,gBAAA,IACE,CAAC,IAAI;oBACL,CAAC,IAAI,CAAC,IAAI;AACV,oBAAA,IAAI,CAAC,IAAI,KAAK,YAAY,CAAC,KAAK;AAChC,oBAAA,IAAI,CAAC,IAAI,KAAK,YAAY,CAAC,SAAS,EACpC;oBACA;;AAGF,gBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,EAAE;oBAClE,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,EAAE,IAAI,IAAI,EAAE;oBAC3C,IAAI,QAAQ,IAAI,IAAI,IAAI,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE;AAChE,wBAAA,SAAS,IAAI,aAAa,CAAC,QAAQ,CAAC;;oBAGtC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,EAAE,IAAI,IAAI,EAAE;oBACvC,IAAI,IAAI,IAAI,IAAI,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE;AACpD,wBAAA,SAAS,IAAI,aAAa,CAAC,IAAI,CAAC;;oBAGlC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,MAAM,IAAI,EAAE;oBAC3C,IAAI,MAAM,IAAI,IAAI,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE;AAC1D,wBAAA,SAAS,IAAI,aAAa,CAAC,MAAM,CAAC;;oBAEpC;;gBAGF,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;gBAEnC,IAAI,CAAC,WAAW,EAAE;oBAChB;;gBAGF,YAAY,CAAC,WAAW,CAAC;;;AAEtB,aAAA,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE;AACpC,YAAA,SAAS,IAAI,aAAa,CAAC,KAAK,CAAC;;AAC5B,aAAA,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE;YACpC,SAAS,IAAI,aAAa,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;;AACvC,aAAA,IAAI,OAAO,KAAK,KAAK,SAAS,EAAE;YACrC,SAAS,IAAI,aAAa,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;;AAEhD,KAAC;IAED,IAAI,SAAS,GAAG,gBAAgB;AAChC,IAAA,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC;AAC7B,IAAA,OAAO,SAAS;AAClB;AAEa,MAAA,kBAAkB,GAAG,YAAW;AAC3C,IAAA,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,+CAA+C,CAAC;AACxE,IAAA,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE;AAEnC,IAAA,MAAM,WAAW,GAAG,CAAC,IAAY,KAAY;AAC3C,QAAA,MAAM,GAAG,GAAG,IAAI,QAAQ,CAAC,UAAU,CAAC;QACpC,OAAO,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM;AAChC,KAAC;IAED,OAAO,CAAC,OAAoB,KAAa,uBAAuB,CAAC,OAAO,EAAE,WAAW,CAAC;AACxF;;;;"}

package/dist/types/common/enum.d.ts

@@ -0,0 +1,128 @@
+/**
+ * Enum representing the various event types emitted during the execution of runnables.
+ * These events provide real-time information about the progress and state of different components.
+ *
+ * @enum {string}
+ */
+export declare enum GraphEvents {
+    /** [Custom] Agent update event in multi-agent graph/workflow */
+    ON_AGENT_UPDATE = "on_agent_update",
+    /** [Custom] Delta event for run steps (message creation and tool calls) */
+    ON_RUN_STEP = "on_run_step",
+    /** [Custom] Delta event for run steps (tool calls) */
+    ON_RUN_STEP_DELTA = "on_run_step_delta",
+    /** [Custom] Completed event for run steps (tool calls) */
+    ON_RUN_STEP_COMPLETED = "on_run_step_completed",
+    /** [Custom] Delta events for messages */
+    ON_MESSAGE_DELTA = "on_message_delta",
+    /** [Custom] Reasoning Delta events for messages */
+    ON_REASONING_DELTA = "on_reasoning_delta",
+    /** Custom event, emitted by system */
+    ON_CUSTOM_EVENT = "on_custom_event",
+    /** Emitted when a chat model starts processing. */
+    CHAT_MODEL_START = "on_chat_model_start",
+    /** Emitted when a chat model streams a chunk of its response. */
+    CHAT_MODEL_STREAM = "on_chat_model_stream",
+    /** Emitted when a chat model completes its processing. */
+    CHAT_MODEL_END = "on_chat_model_end",
+    /** Emitted when a language model starts processing. */
+    LLM_START = "on_llm_start",
+    /** Emitted when a language model streams a chunk of its response. */
+    LLM_STREAM = "on_llm_stream",
+    /** Emitted when a language model completes its processing. */
+    LLM_END = "on_llm_end",
+    /** Emitted when a chain starts processing. */
+    CHAIN_START = "on_chain_start",
+    /** Emitted when a chain streams a chunk of its output. */
+    CHAIN_STREAM = "on_chain_stream",
+    /** Emitted when a chain completes its processing. */
+    CHAIN_END = "on_chain_end",
+    /** Emitted when a tool starts its operation. */
+    TOOL_START = "on_tool_start",
+    /** Emitted when a tool completes its operation. */
+    TOOL_END = "on_tool_end",
+    /** Emitted when a retriever starts its operation. */
+    RETRIEVER_START = "on_retriever_start",
+    /** Emitted when a retriever completes its operation. */
+    RETRIEVER_END = "on_retriever_end",
+    /** Emitted when a prompt starts processing. */
+    PROMPT_START = "on_prompt_start",
+    /** Emitted when a prompt completes its processing. */
+    PROMPT_END = "on_prompt_end"
+}
+export declare enum Providers {
+    OPENAI = "openAI",
+    BEDROCK_LEGACY = "bedrock_legacy",
+    VERTEXAI = "vertexai",
+    BEDROCK = "bedrock",
+    ANTHROPIC = "anthropic",
+    MISTRALAI = "mistralai",
+    MISTRAL = "mistral",
+    OLLAMA = "ollama",
+    GOOGLE = "google",
+    AZURE = "azureOpenAI",
+    DEEPSEEK = "deepseek",
+    OPENROUTER = "openrouter",
+    XAI = "xai"
+}
+export declare enum GraphNodeKeys {
+    TOOLS = "tools",
+    AGENT = "agent",
+    PRE_TOOLS = "pre_tools",
+    POST_TOOLS = "post_tools"
+}
+export declare enum GraphNodeActions {
+    TOOL_NODE = "tool_node",
+    CALL_MODEL = "call_model",
+    ROUTE_MESSAGE = "route_message"
+}
+export declare enum CommonEvents {
+    LANGGRAPH = "LangGraph"
+}
+export declare enum StepTypes {
+    TOOL_CALLS = "tool_calls",
+    MESSAGE_CREATION = "message_creation"
+}
+export declare enum ContentTypes {
+    TEXT = "text",
+    ERROR = "error",
+    THINK = "think",
+    TOOL_CALL = "tool_call",
+    IMAGE_URL = "image_url",
+    IMAGE_FILE = "image_file",
+    /** Anthropic */
+    THINKING = "thinking",
+    /** Vertex AI / Google Common */
+    REASONING = "reasoning",
+    /** Multi-Agent Switch */
+    AGENT_UPDATE = "agent_update",
+    /** Bedrock */
+    REASONING_CONTENT = "reasoning_content"
+}
+export declare enum ToolCallTypes {
+    FUNCTION = "function",
+    RETRIEVAL = "retrieval",
+    FILE_SEARCH = "file_search",
+    CODE_INTERPRETER = "code_interpreter",
+    TOOL_CALL = "tool_call"
+}
+export declare enum Callback {
+    TOOL_ERROR = "handleToolError",
+    TOOL_START = "handleToolStart",
+    TOOL_END = "handleToolEnd"
+}
+export declare enum Constants {
+    OFFICIAL_CODE_BASEURL = "https://api.illuma.ai/v1",
+    EXECUTE_CODE = "execute_code",
+    WEB_SEARCH = "web_search",
+    CONTENT_AND_ARTIFACT = "content_and_artifact"
+}
+export declare enum TitleMethod {
+    STRUCTURED = "structured",
+    FUNCTIONS = "functions",
+    COMPLETION = "completion"
+}
+export declare enum EnvVar {
+    CODE_API_KEY = "CODE_EXECUTOR_API_KEY",
+    CODE_BASEURL = "CODE_EXECUTOR_API_BASEURL"
+}

package/dist/types/common/index.d.ts

@@ -0,0 +1 @@
+export * from './enum';

package/dist/types/events.d.ts

@@ -0,0 +1,29 @@
+import type { UsageMetadata, BaseMessageFields } from '@langchain/core/messages';
+import type { Graph } from '@/graphs';
+import type * as t from '@/types';
+export declare class HandlerRegistry {
+    private handlers;
+    register(eventType: string, handler: t.EventHandler): void;
+    getHandler(eventType: string): t.EventHandler | undefined;
+}
+export declare class ModelEndHandler implements t.EventHandler {
+    collectedUsage?: UsageMetadata[];
+    constructor(collectedUsage?: UsageMetadata[]);
+    handle(event: string, data: t.ModelEndData, metadata?: Record<string, unknown>, graph?: Graph): void;
+}
+export declare class ToolEndHandler implements t.EventHandler {
+    private callback?;
+    private omitOutput?;
+    constructor(callback?: t.ToolEndCallback, omitOutput?: (name?: string) => boolean);
+    handle(event: string, data: t.StreamEventData | undefined, metadata?: Record<string, unknown>, graph?: Graph): void;
+}
+export declare class TestLLMStreamHandler implements t.EventHandler {
+    handle(event: string, data: t.StreamEventData | undefined): void;
+}
+export declare class TestChatStreamHandler implements t.EventHandler {
+    handle(event: string, data: t.StreamEventData | undefined): void;
+}
+export declare class LLMStreamHandler implements t.EventHandler {
+    handle(event: string, data: t.StreamEventData | undefined, metadata?: Record<string, unknown>): void;
+}
+export declare const createMetadataAggregator: (_collected?: Record<string, NonNullable<BaseMessageFields["response_metadata"]>>[]) => t.MetadataAggregatorResult;

package/dist/types/graphs/Graph.d.ts

@@ -0,0 +1,122 @@
+import { ToolNode } from '@langchain/langgraph/prebuilt';
+import { START } from '@langchain/langgraph';
+import { Runnable, RunnableConfig } from '@langchain/core/runnables';
+import { SystemMessage } from '@langchain/core/messages';
+import type { BaseMessage, UsageMetadata } from '@langchain/core/messages';
+import type * as t from '@/types';
+import { Providers, GraphNodeKeys, Callback, ContentTypes } from '@/common';
+import type { ToolCall } from '@langchain/core/messages/tool';
+import { ToolNode as CustomToolNode } from '@/tools/ToolNode';
+import { createPruneMessages } from '@/messages';
+import { HandlerRegistry } from '@/events';
+export type GraphNode = GraphNodeKeys | typeof START;
+export type ClientCallback<T extends unknown[]> = (graph: StandardGraph, ...args: T) => void;
+export type ClientCallbacks = {
+    [Callback.TOOL_ERROR]?: ClientCallback<[Error, string]>;
+    [Callback.TOOL_START]?: ClientCallback<unknown[]>;
+    [Callback.TOOL_END]?: ClientCallback<unknown[]>;
+};
+export type SystemCallbacks = {
+    [K in keyof ClientCallbacks]: ClientCallbacks[K] extends ClientCallback<infer Args> ? (...args: Args) => void : never;
+};
+export declare abstract class Graph<T extends t.BaseGraphState = t.BaseGraphState, TNodeName extends string = string> {
+    abstract resetValues(): void;
+    abstract createGraphState(): t.GraphStateChannels<T>;
+    abstract initializeTools(): CustomToolNode<T> | ToolNode<T>;
+    abstract initializeModel(): Runnable;
+    abstract getRunMessages(): BaseMessage[] | undefined;
+    abstract getContentParts(): t.MessageContentComplex[] | undefined;
+    abstract generateStepId(stepKey: string): [string, number];
+    abstract getKeyList(metadata: Record<string, unknown> | undefined): (string | number | undefined)[];
+    abstract getStepKey(metadata: Record<string, unknown> | undefined): string;
+    abstract checkKeyList(keyList: (string | number | undefined)[]): boolean;
+    abstract getStepIdByKey(stepKey: string, index?: number): string;
+    abstract getRunStep(stepId: string): t.RunStep | undefined;
+    abstract dispatchRunStep(stepKey: string, stepDetails: t.StepDetails): string;
+    abstract dispatchRunStepDelta(id: string, delta: t.ToolCallDelta): void;
+    abstract dispatchMessageDelta(id: string, delta: t.MessageDelta): void;
+    abstract dispatchReasoningDelta(stepId: string, delta: t.ReasoningDelta): void;
+    abstract handleToolCallCompleted(data: t.ToolEndData, metadata?: Record<string, unknown>, omitOutput?: boolean): void;
+    abstract createCallModel(): (state: T, config?: RunnableConfig) => Promise<Partial<T>>;
+    abstract createWorkflow(): t.CompiledWorkflow<T>;
+    lastToken?: string;
+    tokenTypeSwitch?: 'reasoning' | 'content';
+    reasoningKey: 'reasoning_content' | 'reasoning';
+    currentTokenType: ContentTypes.TEXT | ContentTypes.THINK | 'think_and_text';
+    messageStepHasToolCalls: Map<string, boolean>;
+    messageIdsByStepKey: Map<string, string>;
+    prelimMessageIdsByStepKey: Map<string, string>;
+    config: RunnableConfig | undefined;
+    contentData: t.RunStep[];
+    stepKeyIds: Map<string, string[]>;
+    contentIndexMap: Map<string, number>;
+    toolCallStepIds: Map<string, string>;
+    currentUsage: Partial<UsageMetadata> | undefined;
+    indexTokenCountMap: Record<string, number | undefined>;
+    maxContextTokens: number | undefined;
+    pruneMessages?: ReturnType<typeof createPruneMessages>;
+    /** The amount of time that should pass before another consecutive API call */
+    streamBuffer: number | undefined;
+    tokenCounter?: t.TokenCounter;
+    signal?: AbortSignal;
+    /** Set of invoked tool call IDs from non-message run steps completed mid-run, if any */
+    invokedToolIds?: Set<string>;
+    handlerRegistry: HandlerRegistry | undefined;
+}
+export declare class StandardGraph extends Graph<t.BaseGraphState, GraphNode> {
+    private graphState;
+    clientOptions: t.ClientOptions;
+    boundModel?: Runnable;
+    /** The last recorded timestamp that a stream API call was invoked */
+    lastStreamCall: number | undefined;
+    systemMessage: SystemMessage | undefined;
+    messages: BaseMessage[];
+    runId: string | undefined;
+    tools?: t.GraphTools;
+    toolMap?: t.ToolMap;
+    startIndex: number;
+    provider: Providers;
+    toolEnd: boolean;
+    signal?: AbortSignal;
+    constructor({ runId, tools, signal, toolMap, provider, streamBuffer, instructions, reasoningKey, clientOptions, toolEnd, additional_instructions, }: t.StandardGraphInput);
+    resetValues(keepContent?: boolean): void;
+    getRunStep(stepId: string): t.RunStep | undefined;
+    getStepKey(metadata: Record<string, unknown> | undefined): string;
+    getStepIdByKey(stepKey: string, index?: number): string;
+    generateStepId(stepKey: string): [string, number];
+    getKeyList(metadata: Record<string, unknown> | undefined): (string | number | undefined)[];
+    checkKeyList(keyList: (string | number | undefined)[]): boolean;
+    getRunMessages(): BaseMessage[] | undefined;
+    getContentParts(): t.MessageContentComplex[] | undefined;
+    createGraphState(): t.GraphStateChannels<t.BaseGraphState>;
+    initializeTools(): CustomToolNode<t.BaseGraphState> | ToolNode<t.BaseGraphState>;
+    initializeModel(): Runnable;
+    overrideTestModel(responses: string[], sleep?: number, toolCalls?: ToolCall[]): void;
+    getNewModel({ provider, clientOptions, omitOptions, }: {
+        provider: Providers;
+        clientOptions?: t.ClientOptions;
+        omitOptions?: Set<string>;
+    }): t.ChatModelInstance;
+    storeUsageMetadata(finalMessage?: BaseMessage): void;
+    cleanupSignalListener(): void;
+    createCallModel(): (state: t.BaseGraphState, config?: RunnableConfig) => Promise<Partial<t.BaseGraphState>>;
+    createWorkflow(): t.CompiledWorkflow<t.BaseGraphState>;
+    /**
+     * Dispatches a run step to the client, returns the step ID
+     */
+    dispatchRunStep(stepKey: string, stepDetails: t.StepDetails): string;
+    handleToolCallCompleted(data: t.ToolEndData, metadata?: Record<string, unknown>, omitOutput?: boolean): void;
+    /**
+     * Static version of handleToolCallError to avoid creating strong references
+     * that prevent garbage collection
+     */
+    static handleToolCallErrorStatic(graph: StandardGraph, data: t.ToolErrorData, metadata?: Record<string, unknown>): void;
+    /**
+     * Instance method that delegates to the static method
+     * Kept for backward compatibility
+     */
+    handleToolCallError(data: t.ToolErrorData, metadata?: Record<string, unknown>): void;
+    dispatchRunStepDelta(id: string, delta: t.ToolCallDelta): void;
+    dispatchMessageDelta(id: string, delta: t.MessageDelta): void;
+    dispatchReasoningDelta: (stepId: string, delta: t.ReasoningDelta) => void;
+}

package/dist/types/graphs/index.d.ts

@@ -0,0 +1 @@
+export * from './Graph';

package/dist/types/index.d.ts

@@ -0,0 +1,13 @@
+export * from './run';
+export * from './stream';
+export * from './splitStream';
+export * from './events';
+export * from './messages';
+export * from './graphs';
+export * from './tools/CodeExecutor';
+export * from './tools/handlers';
+export * from './tools/search';
+export * from './common';
+export * from './utils';
+export type * from './types';
+export { CustomOpenAIClient } from './llm/openai';

package/dist/types/instrumentation.d.ts

@@ -0,0 +1 @@
+export {};