orkajs 1.0.0

package/dist/chunk-7ADDQ53B.js

@@ -0,0 +1,88 @@
+import { PromptTemplate } from './chunk-TMWVWAPX.js';
+
+// src/templates/few-shot-prompt-template.ts
+var FewShotPromptTemplate = class _FewShotPromptTemplate {
+  examples;
+  examplePrompt;
+  prefix;
+  suffix;
+  inputVariables;
+  exampleSeparator;
+  partialVariables;
+  constructor(options) {
+    this.examples = options.examples;
+    this.prefix = options.prefix ?? "";
+    this.suffix = options.suffix;
+    this.inputVariables = options.inputVariables;
+    this.exampleSeparator = options.exampleSeparator ?? "\n\n";
+    this.partialVariables = options.partialVariables ?? {};
+    const exampleVars = PromptTemplate.extractVariables(options.examplePrompt);
+    this.examplePrompt = new PromptTemplate({
+      template: options.examplePrompt,
+      inputVariables: exampleVars,
+      validateTemplate: false
+    });
+  }
+  format(variables) {
+    const allVars = { ...this.resolvePartials(), ...variables };
+    const missing = this.inputVariables.filter((v) => !(v in allVars));
+    if (missing.length > 0) {
+      throw new Error(`Missing variables: ${missing.join(", ")}`);
+    }
+    const formattedExamples = this.examples.map(
+      (example) => this.examplePrompt.format(example)
+    );
+    const parts = [];
+    if (this.prefix) {
+      let prefix = this.prefix;
+      for (const [key, value] of Object.entries(allVars)) {
+        prefix = prefix.replace(new RegExp(`\\{\\{\\s*${key}\\s*\\}\\}`, "g"), value);
+      }
+      parts.push(prefix);
+    }
+    parts.push(formattedExamples.join(this.exampleSeparator));
+    let suffix = this.suffix;
+    for (const [key, value] of Object.entries(allVars)) {
+      suffix = suffix.replace(new RegExp(`\\{\\{\\s*${key}\\s*\\}\\}`, "g"), value);
+    }
+    parts.push(suffix);
+    return parts.join(this.exampleSeparator);
+  }
+  addExample(example) {
+    this.examples.push(example);
+  }
+  addExamples(examples) {
+    this.examples.push(...examples);
+  }
+  getExamples() {
+    return [...this.examples];
+  }
+  getInputVariables() {
+    return [...this.inputVariables];
+  }
+  partial(variables) {
+    const newPartials = { ...this.partialVariables, ...variables };
+    const partialKeys = Object.keys(newPartials);
+    const remaining = this.inputVariables.filter((v) => !partialKeys.includes(v));
+    return new _FewShotPromptTemplate({
+      examples: this.examples,
+      examplePrompt: this.examplePrompt.getTemplate(),
+      prefix: this.prefix,
+      suffix: this.suffix,
+      inputVariables: remaining,
+      exampleSeparator: this.exampleSeparator,
+      partialVariables: newPartials
+    });
+  }
+  resolvePartials() {
+    const resolved = {};
+    for (const [key, value] of Object.entries(this.partialVariables)) {
+      resolved[key] = typeof value === "function" ? value() : value;
+    }
+    return resolved;
+  }
+};
+
+export { FewShotPromptTemplate };
+//# sourceMappingURL=chunk-7ADDQ53B.js.map
+//# sourceMappingURL=chunk-7ADDQ53B.js.map

package/dist/chunk-7ADDQ53B.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/templates/few-shot-prompt-template.ts"],"names":[],"mappings":";;;AAGO,IAAM,qBAAA,GAAN,MAAM,sBAAA,CAAsB;AAAA,EACzB,QAAA;AAAA,EACA,aAAA;AAAA,EACA,MAAA;AAAA,EACA,MAAA;AAAA,EACA,cAAA;AAAA,EACA,gBAAA;AAAA,EACA,gBAAA;AAAA,EAER,YAAY,OAAA,EAAuC;AACjD,IAAA,IAAA,CAAK,WAAW,OAAA,CAAQ,QAAA;AACxB,IAAA,IAAA,CAAK,MAAA,GAAS,QAAQ,MAAA,IAAU,EAAA;AAChC,IAAA,IAAA,CAAK,SAAS,OAAA,CAAQ,MAAA;AACtB,IAAA,IAAA,CAAK,iBAAiB,OAAA,CAAQ,cAAA;AAC9B,IAAA,IAAA,CAAK,gBAAA,GAAmB,QAAQ,gBAAA,IAAoB,MAAA;AACpD,IAAA,IAAA,CAAK,gBAAA,GAAmB,OAAA,CAAQ,gBAAA,IAAoB,EAAC;AAErD,IAAA,MAAM,WAAA,GAAc,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,aAAa,CAAA;AACzE,IAAA,IAAA,CAAK,aAAA,GAAgB,IAAI,cAAA,CAAe;AAAA,MACtC,UAAU,OAAA,CAAQ,aAAA;AAAA,MAClB,cAAA,EAAgB,WAAA;AAAA,MAChB,gBAAA,EAAkB;AAAA,KACnB,CAAA;AAAA,EACH;AAAA,EAEA,OAAO,SAAA,EAA2C;AAChD,IAAA,MAAM,UAAU,EAAE,GAAG,KAAK,eAAA,EAAgB,EAAG,GAAG,SAAA,EAAU;AAE1D,IAAA,MAAM,UAAU,IAAA,CAAK,cAAA,CAAe,OAAO,CAAA,CAAA,KAAK,EAAE,KAAK,OAAA,CAAQ,CAAA;AAC/D,IAAA,IAAI,OAAA,CAAQ,SAAS,CAAA,EAAG;AACtB,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,mBAAA,EAAsB,QAAQ,IAAA,CAAK,IAAI,CAAC,CAAA,CAAE,CAAA;AAAA,IAC5D;AAGA,IAAA,MAAM,iBAAA,GAAoB,KAAK,QAAA,CAAS,GAAA;AAAA,MAAI,CAAA,OAAA,KAC1C,IAAA,CAAK,aAAA,CAAc,MAAA,CAAO,OAAO;AAAA,KACnC;AAGA,IAAA,MAAM,QAAkB,EAAC;AAEzB,IAAA,IAAI,KAAK,MAAA,EAAQ;AACf,MAAA,IAAI,SAAS,IAAA,CAAK,MAAA;AAClB,MAAA,KAAA,MAAW,CAAC,GAAA,EAAK,KAAK,KAAK,MAAA,CAAO,OAAA,CAAQ,OAAO,CAAA,EAAG;AAClD,QAAA,MAAA,GAAS,MAAA,CAAO,QAAQ,IAAI,MAAA,CAAO,aAAa,GAAG,CAAA,UAAA,CAAA,EAAc,GAAG,CAAA,EAAG,KAAK,CAAA;AAAA,MAC9E;AACA,MAAA,KAAA,CAAM,KAAK,MAAM,CAAA;AAAA,IACnB;AAEA,IAAA,KAAA,CAAM,IAAA,CAAK,iBAAA,CAAkB,IAAA,CAAK,IAAA,CAAK,gBAAgB,CAAC,CAAA;AAExD,IAAA,IAAI,SAAS,IAAA,CAAK,MAAA;AAClB,IAAA,KAAA,MAAW,CAAC,GAAA,EAAK,KAAK,KAAK,MAAA,CAAO,OAAA,CAAQ,OAAO,CAAA,EAAG;AAClD,MAAA,MAAA,GAAS,MAAA,CAAO,QAAQ,IAAI,MAAA,CAAO,aAAa,GAAG,CAAA,UAAA,CAAA,EAAc,GAAG,CAAA,EAAG,KAAK,CAAA;AAAA,IAC9E;AACA,IAAA,KAAA,CAAM,KAAK,MAAM,CAAA;AAEjB,IAAA,OAAO,KAAA,CAAM,IAAA,CAAK,IAAA,CAAK,gBAAgB,CAAA;AAAA,EACzC;AAAA,EAEA,WAAW,OAAA,EAA+B;AACxC,IAAA,IAAA,CAAK,QAAA,CAAS,KAAK,OAAO,CAAA;AAAA,EAC5B;AAAA,EAEA,YAAY,QAAA,EAAkC;AAC5C,IAAA,IAAA,CAAK,QAAA,CAAS,IAAA,CAAK,GAAG,QAAQ,CAAA;AAAA,EAChC;AAAA,EAEA,WAAA,GAAgC;AAC9B,IAAA,OAAO,CAAC,GAAG,IAAA,CAAK,QAAQ,CAAA;AAAA,EAC1B;AAAA,EAEA,iBAAA,GAA8B;AAC5B,IAAA,OAAO,CAAC,GAAG,IAAA,CAAK,cAAc,CAAA;AAAA,EAChC;AAAA,EAEA,QAAQ,SAAA,EAA2E;AACjF,IAAA,MAAM,cAAc,EAAE,GAAG,IAAA,CAAK,gBAAA,EAAkB,GAAG,SAAA,EAAU;AAC7D,IAAA,MAAM,WAAA,GAAc,MAAA,CAAO,IAAA,CAAK,WAAW,CAAA;AAC3C,IAAA,MAAM,SAAA,GAAY,KAAK,cAAA,CAAe,MAAA,CAAO,OAAK,CAAC,WAAA,CAAY,QAAA,CAAS,CAAC,CAAC,CAAA;AAE1E,IAAA,OAAO,IAAI,sBAAA,CAAsB;AAAA,MAC/B,UAAU,IAAA,CAAK,QAAA;AAAA,MACf,aAAA,EAAe,IAAA,CAAK,aAAA,CAAc,WAAA,EAAY;AAAA,MAC9C,QAAQ,IAAA,CAAK,MAAA;AAAA,MACb,QAAQ,IAAA,CAAK,MAAA;AAAA,MACb,cAAA,EAAgB,SAAA;AAAA,MAChB,kBAAkB,IAAA,CAAK,gBAAA;AAAA,MACvB,gBAAA,EAAkB;AAAA,KACnB,CAAA;AAAA,EACH;AAAA,EAEQ,eAAA,GAA0C;AAChD,IAAA,MAAM,WAAmC,EAAC;AAC1C,IAAA,KAAA,MAAW,CAAC,KAAK,KAAK,CAAA,IAAK,OAAO,OAAA,CAAQ,IAAA,CAAK,gBAAgB,CAAA,EAAG;AAChE,MAAA,QAAA,CAAS,GAAG,CAAA,GAAI,OAAO,KAAA,KAAU,UAAA,GAAa,OAAM,GAAI,KAAA;AAAA,IAC1D;AACA,IAAA,OAAO,QAAA;AAAA,EACT;AACF","file":"chunk-7ADDQ53B.js","sourcesContent":["import type { FewShotPromptTemplateOptions, FewShotExample } from './types.js';\nimport { PromptTemplate } from './prompt-template.js';\n\nexport class FewShotPromptTemplate {\n  private examples: FewShotExample[];\n  private examplePrompt: PromptTemplate;\n  private prefix: string;\n  private suffix: string;\n  private inputVariables: string[];\n  private exampleSeparator: string;\n  private partialVariables: Record<string, string | (() => string)>;\n\n  constructor(options: FewShotPromptTemplateOptions) {\n    this.examples = options.examples;\n    this.prefix = options.prefix ?? '';\n    this.suffix = options.suffix;\n    this.inputVariables = options.inputVariables;\n    this.exampleSeparator = options.exampleSeparator ?? '\\n\\n';\n    this.partialVariables = options.partialVariables ?? {};\n\n    const exampleVars = PromptTemplate.extractVariables(options.examplePrompt);\n    this.examplePrompt = new PromptTemplate({\n      template: options.examplePrompt,\n      inputVariables: exampleVars,\n      validateTemplate: false,\n    });\n  }\n\n  format(variables: Record<string, string>): string {\n    const allVars = { ...this.resolvePartials(), ...variables };\n\n    const missing = this.inputVariables.filter(v => !(v in allVars));\n    if (missing.length > 0) {\n      throw new Error(`Missing variables: ${missing.join(', ')}`);\n    }\n\n    // Format examples\n    const formattedExamples = this.examples.map(example =>\n      this.examplePrompt.format(example)\n    );\n\n    // Build the full prompt\n    const parts: string[] = [];\n\n    if (this.prefix) {\n      let prefix = this.prefix;\n      for (const [key, value] of Object.entries(allVars)) {\n        prefix = prefix.replace(new RegExp(`\\\\{\\\\{\\\\s*${key}\\\\s*\\\\}\\\\}`, 'g'), value);\n      }\n      parts.push(prefix);\n    }\n\n    parts.push(formattedExamples.join(this.exampleSeparator));\n\n    let suffix = this.suffix;\n    for (const [key, value] of Object.entries(allVars)) {\n      suffix = suffix.replace(new RegExp(`\\\\{\\\\{\\\\s*${key}\\\\s*\\\\}\\\\}`, 'g'), value);\n    }\n    parts.push(suffix);\n\n    return parts.join(this.exampleSeparator);\n  }\n\n  addExample(example: FewShotExample): void {\n    this.examples.push(example);\n  }\n\n  addExamples(examples: FewShotExample[]): void {\n    this.examples.push(...examples);\n  }\n\n  getExamples(): FewShotExample[] {\n    return [...this.examples];\n  }\n\n  getInputVariables(): string[] {\n    return [...this.inputVariables];\n  }\n\n  partial(variables: Record<string, string | (() => string)>): FewShotPromptTemplate {\n    const newPartials = { ...this.partialVariables, ...variables };\n    const partialKeys = Object.keys(newPartials);\n    const remaining = this.inputVariables.filter(v => !partialKeys.includes(v));\n\n    return new FewShotPromptTemplate({\n      examples: this.examples,\n      examplePrompt: this.examplePrompt.getTemplate(),\n      prefix: this.prefix,\n      suffix: this.suffix,\n      inputVariables: remaining,\n      exampleSeparator: this.exampleSeparator,\n      partialVariables: newPartials,\n    });\n  }\n\n  private resolvePartials(): Record<string, string> {\n    const resolved: Record<string, string> = {};\n    for (const [key, value] of Object.entries(this.partialVariables)) {\n      resolved[key] = typeof value === 'function' ? value() : value;\n    }\n    return resolved;\n  }\n}\n"]}

package/dist/chunk-7CS4NTB4.cjs

@@ -0,0 +1,118 @@
+'use strict';
+
+// src/adapters/mistral/index.ts
+var MistralAdapter = class {
+  name = "mistral";
+  apiKey;
+  model;
+  embeddingModel;
+  baseURL;
+  timeoutMs;
+  constructor(config) {
+    this.apiKey = config.apiKey;
+    this.model = config.model ?? "mistral-small-latest";
+    this.embeddingModel = config.embeddingModel ?? "mistral-embed";
+    this.baseURL = config.baseURL ?? "https://api.mistral.ai/v1";
+    this.timeoutMs = config.timeoutMs ?? 6e4;
+  }
+  async generate(prompt, options = {}) {
+    const messages = [];
+    if (options.systemPrompt) {
+      messages.push({ role: "system", content: options.systemPrompt });
+    }
+    messages.push({ role: "user", content: prompt });
+    const controller = new AbortController();
+    const timeout = setTimeout(() => controller.abort(), this.timeoutMs);
+    let response;
+    try {
+      response = await fetch(`${this.baseURL}/chat/completions`, {
+        method: "POST",
+        signal: controller.signal,
+        headers: {
+          "Content-Type": "application/json",
+          "Authorization": `Bearer ${this.apiKey}`
+        },
+        body: JSON.stringify({
+          model: this.model,
+          messages,
+          temperature: options.temperature ?? 0.7,
+          max_tokens: options.maxTokens ?? 1024,
+          stop: options.stopSequences
+        })
+      });
+    } catch (error) {
+      clearTimeout(timeout);
+      if (error.name === "AbortError") {
+        throw new Error(`Mistral API request timed out after ${this.timeoutMs}ms`);
+      }
+      throw error;
+    } finally {
+      clearTimeout(timeout);
+    }
+    if (!response.ok) {
+      const error = await response.text();
+      throw new Error(`Mistral API error: ${response.status} - ${error}`);
+    }
+    const data = await response.json();
+    return {
+      content: data.choices[0]?.message?.content ?? "",
+      usage: {
+        promptTokens: data.usage.prompt_tokens,
+        completionTokens: data.usage.completion_tokens,
+        totalTokens: data.usage.total_tokens
+      },
+      model: data.model,
+      finishReason: this.mapFinishReason(data.choices[0]?.finish_reason)
+    };
+  }
+  async embed(texts) {
+    const input = Array.isArray(texts) ? texts : [texts];
+    const controller = new AbortController();
+    const timeout = setTimeout(() => controller.abort(), this.timeoutMs);
+    let response;
+    try {
+      response = await fetch(`${this.baseURL}/embeddings`, {
+        method: "POST",
+        signal: controller.signal,
+        headers: {
+          "Content-Type": "application/json",
+          "Authorization": `Bearer ${this.apiKey}`
+        },
+        body: JSON.stringify({
+          model: this.embeddingModel,
+          input
+        })
+      });
+    } catch (error) {
+      clearTimeout(timeout);
+      if (error.name === "AbortError") {
+        throw new Error(`Mistral Embeddings API request timed out after ${this.timeoutMs}ms`);
+      }
+      throw error;
+    } finally {
+      clearTimeout(timeout);
+    }
+    if (!response.ok) {
+      const error = await response.text();
+      throw new Error(`Mistral Embeddings API error: ${response.status} - ${error}`);
+    }
+    const data = await response.json();
+    return data.data.sort((a, b) => a.index - b.index).map((d) => d.embedding);
+  }
+  mapFinishReason(reason) {
+    switch (reason) {
+      case "stop":
+        return "stop";
+      case "length":
+        return "length";
+      case "tool_calls":
+        return "tool_calls";
+      default:
+        return "stop";
+    }
+  }
+};
+
+exports.MistralAdapter = MistralAdapter;
+//# sourceMappingURL=chunk-7CS4NTB4.cjs.map
+//# sourceMappingURL=chunk-7CS4NTB4.cjs.map

package/dist/chunk-7CS4NTB4.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/adapters/mistral/index.ts"],"names":[],"mappings":";;;AAUO,IAAM,iBAAN,MAA2C;AAAA,EACvC,IAAA,GAAO,SAAA;AAAA,EACR,MAAA;AAAA,EACA,KAAA;AAAA,EACA,cAAA;AAAA,EACA,OAAA;AAAA,EACA,SAAA;AAAA,EAER,YAAY,MAAA,EAA8B;AACxC,IAAA,IAAA,CAAK,SAAS,MAAA,CAAO,MAAA;AACrB,IAAA,IAAA,CAAK,KAAA,GAAQ,OAAO,KAAA,IAAS,sBAAA;AAC7B,IAAA,IAAA,CAAK,cAAA,GAAiB,OAAO,cAAA,IAAkB,eAAA;AAC/C,IAAA,IAAA,CAAK,OAAA,GAAU,OAAO,OAAA,IAAW,2BAAA;AACjC,IAAA,IAAA,CAAK,SAAA,GAAY,OAAO,SAAA,IAAa,GAAA;AAAA,EACvC;AAAA,EAEA,MAAM,QAAA,CAAS,MAAA,EAAgB,OAAA,GAA8B,EAAC,EAAuB;AACnF,IAAA,MAAM,WAAqD,EAAC;AAE5D,IAAA,IAAI,QAAQ,YAAA,EAAc;AACxB,MAAA,QAAA,CAAS,KAAK,EAAE,IAAA,EAAM,UAAU,OAAA,EAAS,OAAA,CAAQ,cAAc,CAAA;AAAA,IACjE;AACA,IAAA,QAAA,CAAS,KAAK,EAAE,IAAA,EAAM,MAAA,EAAQ,OAAA,EAAS,QAAQ,CAAA;AAE/C,IAAA,MAAM,UAAA,GAAa,IAAI,eAAA,EAAgB;AACvC,IAAA,MAAM,UAAU,UAAA,CAAW,MAAM,WAAW,KAAA,EAAM,EAAG,KAAK,SAAS,CAAA;AAEnE,IAAA,IAAI,QAAA;AACJ,IAAA,IAAI;AACF,MAAA,QAAA,GAAW,MAAM,KAAA,CAAM,CAAA,EAAG,IAAA,CAAK,OAAO,CAAA,iBAAA,CAAA,EAAqB;AAAA,QACzD,MAAA,EAAQ,MAAA;AAAA,QACR,QAAQ,UAAA,CAAW,MAAA;AAAA,QACnB,OAAA,EAAS;AAAA,UACP,cAAA,EAAgB,kBAAA;AAAA,UAChB,eAAA,EAAiB,CAAA,OAAA,EAAU,IAAA,CAAK,MAAM,CAAA;AAAA,SACxC;AAAA,QACA,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,UACnB,OAAO,IAAA,CAAK,KAAA;AAAA,UACZ,QAAA;AAAA,UACA,WAAA,EAAa,QAAQ,WAAA,IAAe,GAAA;AAAA,UACpC,UAAA,EAAY,QAAQ,SAAA,IAAa,IAAA;AAAA,UACjC,MAAM,OAAA,CAAQ;AAAA,SACf;AAAA,OACF,CAAA;AAAA,IACH,SAAS,KAAA,EAAO;AACd,MAAA,YAAA,CAAa,OAAO,CAAA;AACpB,MAAA,IAAK,KAAA,CAAgB,SAAS,YAAA,EAAc;AAC1C,QAAA,MAAM,IAAI,KAAA,CAAM,CAAA,oCAAA,EAAuC,IAAA,CAAK,SAAS,CAAA,EAAA,CAAI,CAAA;AAAA,MAC3E;AACA,MAAA,MAAM,KAAA;AAAA,IACR,CAAA,SAAE;AACA,MAAA,YAAA,CAAa,OAAO,CAAA;AAAA,IACtB;AAEA,IAAA,IAAI,CAAC,SAAS,EAAA,EAAI;AAChB,MAAA,MAAM,KAAA,GAAQ,MAAM,QAAA,CAAS,IAAA,EAAK;AAClC,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,mBAAA,EAAsB,SAAS,MAAM,CAAA,GAAA,EAAM,KAAK,CAAA,CAAE,CAAA;AAAA,IACpE;AAEA,IAAA,MAAM,IAAA,GAAO,MAAM,QAAA,CAAS,IAAA,EAAK;AAMjC,IAAA,OAAO;AAAA,MACL,SAAS,IAAA,CAAK,OAAA,CAAQ,CAAC,CAAA,EAAG,SAAS,OAAA,IAAW,EAAA;AAAA,MAC9C,KAAA,EAAO;AAAA,QACL,YAAA,EAAc,KAAK,KAAA,CAAM,aAAA;AAAA,QACzB,gBAAA,EAAkB,KAAK,KAAA,CAAM,iBAAA;AAAA,QAC7B,WAAA,EAAa,KAAK,KAAA,CAAM;AAAA,OAC1B;AAAA,MACA,OAAO,IAAA,CAAK,KAAA;AAAA,MACZ,cAAc,IAAA,CAAK,eAAA,CAAgB,KAAK,OAAA,CAAQ,CAAC,GAAG,aAAa;AAAA,KACnE;AAAA,EACF;AAAA,EAEA,MAAM,MAAM,KAAA,EAA+C;AACzD,IAAA,MAAM,QAAQ,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,GAAI,KAAA,GAAQ,CAAC,KAAK,CAAA;AAEnD,IAAA,MAAM,UAAA,GAAa,IAAI,eAAA,EAAgB;AACvC,IAAA,MAAM,UAAU,UAAA,CAAW,MAAM,WAAW,KAAA,EAAM,EAAG,KAAK,SAAS,CAAA;AAEnE,IAAA,IAAI,QAAA;AACJ,IAAA,IAAI;AACF,MAAA,QAAA,GAAW,MAAM,KAAA,CAAM,CAAA,EAAG,IAAA,CAAK,OAAO,CAAA,WAAA,CAAA,EAAe;AAAA,QACnD,MAAA,EAAQ,MAAA;AAAA,QACR,QAAQ,UAAA,CAAW,MAAA;AAAA,QACnB,OAAA,EAAS;AAAA,UACP,cAAA,EAAgB,kBAAA;AAAA,UAChB,eAAA,EAAiB,CAAA,OAAA,EAAU,IAAA,CAAK,MAAM,CAAA;AAAA,SACxC;AAAA,QACA,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,UACnB,OAAO,IAAA,CAAK,cAAA;AAAA,UACZ;AAAA,SACD;AAAA,OACF,CAAA;AAAA,IACH,SAAS,KAAA,EAAO;AACd,MAAA,YAAA,CAAa,OAAO,CAAA;AACpB,MAAA,IAAK,KAAA,CAAgB,SAAS,YAAA,EAAc;AAC1C,QAAA,MAAM,IAAI,KAAA,CAAM,CAAA,+CAAA,EAAkD,IAAA,CAAK,SAAS,CAAA,EAAA,CAAI,CAAA;AAAA,MACtF;AACA,MAAA,MAAM,KAAA;AAAA,IACR,CAAA,SAAE;AACA,MAAA,YAAA,CAAa,OAAO,CAAA;AAAA,IACtB;AAEA,IAAA,IAAI,CAAC,SAAS,EAAA,EAAI;AAChB,MAAA,MAAM,KAAA,GAAQ,MAAM,QAAA,CAAS,IAAA,EAAK;AAClC,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,8BAAA,EAAiC,SAAS,MAAM,CAAA,GAAA,EAAM,KAAK,CAAA,CAAE,CAAA;AAAA,IAC/E;AAEA,IAAA,MAAM,IAAA,GAAO,MAAM,QAAA,CAAS,IAAA,EAAK;AAIjC,IAAA,OAAO,IAAA,CAAK,IAAA,CACT,IAAA,CAAK,CAAC,GAAG,CAAA,KAAM,CAAA,CAAE,KAAA,GAAQ,CAAA,CAAE,KAAK,CAAA,CAChC,GAAA,CAAI,CAAA,CAAA,KAAK,EAAE,SAAS,CAAA;AAAA,EACzB;AAAA,EAEQ,gBAAgB,MAAA,EAA2C;AACjE,IAAA,QAAQ,MAAA;AAAQ,MACd,KAAK,MAAA;AAAQ,QAAA,OAAO,MAAA;AAAA,MACpB,KAAK,QAAA;AAAU,QAAA,OAAO,QAAA;AAAA,MACtB,KAAK,YAAA;AAAc,QAAA,OAAO,YAAA;AAAA,MAC1B;AAAS,QAAA,OAAO,MAAA;AAAA;AAClB,EACF;AACF","file":"chunk-7CS4NTB4.cjs","sourcesContent":["import type { LLMAdapter, LLMGenerateOptions, LLMResult } from '../../types/index.js';\n\nexport interface MistralAdapterConfig {\n  apiKey: string;\n  model?: string;\n  embeddingModel?: string;\n  baseURL?: string;\n  timeoutMs?: number;\n}\n\nexport class MistralAdapter implements LLMAdapter {\n  readonly name = 'mistral';\n  private apiKey: string;\n  private model: string;\n  private embeddingModel: string;\n  private baseURL: string;\n  private timeoutMs: number;\n\n  constructor(config: MistralAdapterConfig) {\n    this.apiKey = config.apiKey;\n    this.model = config.model ?? 'mistral-small-latest';\n    this.embeddingModel = config.embeddingModel ?? 'mistral-embed';\n    this.baseURL = config.baseURL ?? 'https://api.mistral.ai/v1';\n    this.timeoutMs = config.timeoutMs ?? 60_000;\n  }\n\n  async generate(prompt: string, options: LLMGenerateOptions = {}): Promise<LLMResult> {\n    const messages: Array<{ role: string; content: string }> = [];\n    \n    if (options.systemPrompt) {\n      messages.push({ role: 'system', content: options.systemPrompt });\n    }\n    messages.push({ role: 'user', content: prompt });\n\n    const controller = new AbortController();\n    const timeout = setTimeout(() => controller.abort(), this.timeoutMs);\n\n    let response: Response;\n    try {\n      response = await fetch(`${this.baseURL}/chat/completions`, {\n        method: 'POST',\n        signal: controller.signal,\n        headers: {\n          'Content-Type': 'application/json',\n          'Authorization': `Bearer ${this.apiKey}`,\n        },\n        body: JSON.stringify({\n          model: this.model,\n          messages,\n          temperature: options.temperature ?? 0.7,\n          max_tokens: options.maxTokens ?? 1024,\n          stop: options.stopSequences,\n        }),\n      });\n    } catch (error) {\n      clearTimeout(timeout);\n      if ((error as Error).name === 'AbortError') {\n        throw new Error(`Mistral API request timed out after ${this.timeoutMs}ms`);\n      }\n      throw error;\n    } finally {\n      clearTimeout(timeout);\n    }\n\n    if (!response.ok) {\n      const error = await response.text();\n      throw new Error(`Mistral API error: ${response.status} - ${error}`);\n    }\n\n    const data = await response.json() as {\n      choices: Array<{ message: { content: string }; finish_reason: string }>;\n      usage: { prompt_tokens: number; completion_tokens: number; total_tokens: number };\n      model: string;\n    };\n\n    return {\n      content: data.choices[0]?.message?.content ?? '',\n      usage: {\n        promptTokens: data.usage.prompt_tokens,\n        completionTokens: data.usage.completion_tokens,\n        totalTokens: data.usage.total_tokens,\n      },\n      model: data.model,\n      finishReason: this.mapFinishReason(data.choices[0]?.finish_reason),\n    };\n  }\n\n  async embed(texts: string | string[]): Promise<number[][]> {\n    const input = Array.isArray(texts) ? texts : [texts];\n\n    const controller = new AbortController();\n    const timeout = setTimeout(() => controller.abort(), this.timeoutMs);\n\n    let response: Response;\n    try {\n      response = await fetch(`${this.baseURL}/embeddings`, {\n        method: 'POST',\n        signal: controller.signal,\n        headers: {\n          'Content-Type': 'application/json',\n          'Authorization': `Bearer ${this.apiKey}`,\n        },\n        body: JSON.stringify({\n          model: this.embeddingModel,\n          input,\n        }),\n      });\n    } catch (error) {\n      clearTimeout(timeout);\n      if ((error as Error).name === 'AbortError') {\n        throw new Error(`Mistral Embeddings API request timed out after ${this.timeoutMs}ms`);\n      }\n      throw error;\n    } finally {\n      clearTimeout(timeout);\n    }\n\n    if (!response.ok) {\n      const error = await response.text();\n      throw new Error(`Mistral Embeddings API error: ${response.status} - ${error}`);\n    }\n\n    const data = await response.json() as {\n      data: Array<{ embedding: number[]; index: number }>;\n    };\n\n    return data.data\n      .sort((a, b) => a.index - b.index)\n      .map(d => d.embedding);\n  }\n\n  private mapFinishReason(reason: string): LLMResult['finishReason'] {\n    switch (reason) {\n      case 'stop': return 'stop';\n      case 'length': return 'length';\n      case 'tool_calls': return 'tool_calls';\n      default: return 'stop';\n    }\n  }\n}\n"]}

package/dist/chunk-7MCLKMJP.cjs

@@ -0,0 +1,57 @@
+'use strict';
+
+// src/resilience/retry.ts
+var DEFAULT_RETRY = {
+  maxRetries: 3,
+  initialDelayMs: 1e3,
+  maxDelayMs: 3e4,
+  backoffMultiplier: 2
+};
+async function withRetry(fn, options = {}) {
+  const {
+    maxRetries = DEFAULT_RETRY.maxRetries,
+    initialDelayMs = DEFAULT_RETRY.initialDelayMs,
+    maxDelayMs = DEFAULT_RETRY.maxDelayMs,
+    backoffMultiplier = DEFAULT_RETRY.backoffMultiplier,
+    retryableErrors,
+    onRetry
+  } = options;
+  let lastError;
+  for (let attempt = 0; attempt <= maxRetries; attempt++) {
+    try {
+      return await fn();
+    } catch (error) {
+      lastError = error;
+      if (attempt >= maxRetries) break;
+      if (retryableErrors && !isRetryable(lastError, retryableErrors)) {
+        throw lastError;
+      }
+      if (onRetry) {
+        onRetry(lastError, attempt + 1);
+      }
+      const delay = Math.min(
+        initialDelayMs * Math.pow(backoffMultiplier, attempt),
+        maxDelayMs
+      );
+      const jitter = delay * 0.1 * Math.random();
+      await sleep(delay + jitter);
+    }
+  }
+  throw lastError;
+}
+function isRetryable(error, patterns) {
+  const message = error.message;
+  return patterns.some((pattern) => {
+    if (typeof pattern === "string") {
+      return message.includes(pattern);
+    }
+    return pattern.test(message);
+  });
+}
+function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+
+exports.withRetry = withRetry;
+//# sourceMappingURL=chunk-7MCLKMJP.cjs.map
+//# sourceMappingURL=chunk-7MCLKMJP.cjs.map

package/dist/chunk-7MCLKMJP.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/resilience/retry.ts"],"names":[],"mappings":";;;AASA,IAAM,aAAA,GAA6E;AAAA,EACjF,UAAA,EAAY,CAAA;AAAA,EACZ,cAAA,EAAgB,GAAA;AAAA,EAChB,UAAA,EAAY,GAAA;AAAA,EACZ,iBAAA,EAAmB;AACrB,CAAA;AAEA,eAAsB,SAAA,CACpB,EAAA,EACA,OAAA,GAAwB,EAAC,EACb;AACZ,EAAA,MAAM;AAAA,IACJ,aAAa,aAAA,CAAc,UAAA;AAAA,IAC3B,iBAAiB,aAAA,CAAc,cAAA;AAAA,IAC/B,aAAa,aAAA,CAAc,UAAA;AAAA,IAC3B,oBAAoB,aAAA,CAAc,iBAAA;AAAA,IAClC,eAAA;AAAA,IACA;AAAA,GACF,GAAI,OAAA;AAEJ,EAAA,IAAI,SAAA;AAEJ,EAAA,KAAA,IAAS,OAAA,GAAU,CAAA,EAAG,OAAA,IAAW,UAAA,EAAY,OAAA,EAAA,EAAW;AACtD,IAAA,IAAI;AACF,MAAA,OAAO,MAAM,EAAA,EAAG;AAAA,IAClB,SAAS,KAAA,EAAO;AACd,MAAA,SAAA,GAAY,KAAA;AAEZ,MAAA,IAAI,WAAW,UAAA,EAAY;AAE3B,MAAA,IAAI,eAAA,IAAmB,CAAC,WAAA,CAAY,SAAA,EAAW,eAAe,CAAA,EAAG;AAC/D,QAAA,MAAM,SAAA;AAAA,MACR;AAEA,MAAA,IAAI,OAAA,EAAS;AACX,QAAA,OAAA,CAAQ,SAAA,EAAW,UAAU,CAAC,CAAA;AAAA,MAChC;AAEA,MAAA,MAAM,QAAQ,IAAA,CAAK,GAAA;AAAA,QACjB,cAAA,GAAiB,IAAA,CAAK,GAAA,CAAI,iBAAA,EAAmB,OAAO,CAAA;AAAA,QACpD;AAAA,OACF;AACA,MAAA,MAAM,MAAA,GAAS,KAAA,GAAQ,GAAA,GAAM,IAAA,CAAK,MAAA,EAAO;AACzC,MAAA,MAAM,KAAA,CAAM,QAAQ,MAAM,CAAA;AAAA,IAC5B;AAAA,EACF;AAEA,EAAA,MAAM,SAAA;AACR;AAEA,SAAS,WAAA,CAAY,OAAc,QAAA,EAAwC;AACzE,EAAA,MAAM,UAAU,KAAA,CAAM,OAAA;AACtB,EAAA,OAAO,QAAA,CAAS,KAAK,CAAA,OAAA,KAAW;AAC9B,IAAA,IAAI,OAAO,YAAY,QAAA,EAAU;AAC/B,MAAA,OAAO,OAAA,CAAQ,SAAS,OAAO,CAAA;AAAA,IACjC;AACA,IAAA,OAAO,OAAA,CAAQ,KAAK,OAAO,CAAA;AAAA,EAC7B,CAAC,CAAA;AACH;AAEA,SAAS,MAAM,EAAA,EAA2B;AACxC,EAAA,OAAO,IAAI,OAAA,CAAQ,CAAA,OAAA,KAAW,UAAA,CAAW,OAAA,EAAS,EAAE,CAAC,CAAA;AACvD","file":"chunk-7MCLKMJP.cjs","sourcesContent":["export interface RetryOptions {\n  maxRetries?: number;\n  initialDelayMs?: number;\n  maxDelayMs?: number;\n  backoffMultiplier?: number;\n  retryableErrors?: (string | RegExp)[];\n  onRetry?: (error: Error, attempt: number) => void;\n}\n\nconst DEFAULT_RETRY: Required<Omit<RetryOptions, 'retryableErrors' | 'onRetry'>> = {\n  maxRetries: 3,\n  initialDelayMs: 1000,\n  maxDelayMs: 30000,\n  backoffMultiplier: 2,\n};\n\nexport async function withRetry<T>(\n  fn: () => Promise<T>,\n  options: RetryOptions = {},\n): Promise<T> {\n  const {\n    maxRetries = DEFAULT_RETRY.maxRetries,\n    initialDelayMs = DEFAULT_RETRY.initialDelayMs,\n    maxDelayMs = DEFAULT_RETRY.maxDelayMs,\n    backoffMultiplier = DEFAULT_RETRY.backoffMultiplier,\n    retryableErrors,\n    onRetry,\n  } = options;\n\n  let lastError: Error | undefined;\n\n  for (let attempt = 0; attempt <= maxRetries; attempt++) {\n    try {\n      return await fn();\n    } catch (error) {\n      lastError = error as Error;\n\n      if (attempt >= maxRetries) break;\n\n      if (retryableErrors && !isRetryable(lastError, retryableErrors)) {\n        throw lastError;\n      }\n\n      if (onRetry) {\n        onRetry(lastError, attempt + 1);\n      }\n\n      const delay = Math.min(\n        initialDelayMs * Math.pow(backoffMultiplier, attempt),\n        maxDelayMs,\n      );\n      const jitter = delay * 0.1 * Math.random();\n      await sleep(delay + jitter);\n    }\n  }\n\n  throw lastError;\n}\n\nfunction isRetryable(error: Error, patterns: (string | RegExp)[]): boolean {\n  const message = error.message;\n  return patterns.some(pattern => {\n    if (typeof pattern === 'string') {\n      return message.includes(pattern);\n    }\n    return pattern.test(message);\n  });\n}\n\nfunction sleep(ms: number): Promise<void> {\n  return new Promise(resolve => setTimeout(resolve, ms));\n}\n"]}

package/dist/chunk-7NVGLURS.js

@@ -0,0 +1,141 @@
+import { BaseAgent } from './chunk-QQAP3IF5.js';
+
+// src/agent/openai-functions-agent.ts
+var OpenAIFunctionsAgent = class extends BaseAgent {
+  functions;
+  constructor(config, llm, memory) {
+    super(config, llm, memory, "openai-functions");
+    this.temperature = config.temperature ?? 0.1;
+    this.functions = config.functions ?? this.buildFunctionsFromTools();
+  }
+  async run(input) {
+    const startTime = Date.now();
+    const steps = [];
+    const memoryContext = this.getMemoryContext();
+    this.emit({ type: "step:start", agentType: this.agentType, input });
+    for (let step = 0; step < this.maxSteps; step++) {
+      const stepStart = Date.now();
+      const prompt = this.buildPrompt(input, steps, memoryContext);
+      const result = await this.llm.generate(prompt, {
+        temperature: this.temperature,
+        maxTokens: 2048,
+        systemPrompt: this.buildSystemPrompt()
+      });
+      const parsed = this.parseFunctionCall(result.content);
+      if (!parsed.functionCall) {
+        const finalAnswer = parsed.content;
+        this.saveToMemory(input, finalAnswer);
+        this.emit({ type: "complete", agentType: this.agentType, output: finalAnswer });
+        const agentResult = this.buildResult(input, finalAnswer, steps, startTime, { stepsUsed: steps.length });
+        agentResult.totalTokens += result.usage.totalTokens;
+        return agentResult;
+      }
+      const { name, arguments: args } = parsed.functionCall;
+      const observation = await this.executeTool(name, args);
+      steps.push({
+        thought: `Calling function: ${name}`,
+        action: name,
+        actionInput: args,
+        observation,
+        latencyMs: Date.now() - stepStart,
+        usage: {
+          promptTokens: result.usage.promptTokens,
+          completionTokens: result.usage.completionTokens,
+          totalTokens: result.usage.totalTokens
+        }
+      });
+      this.emit({ type: "step:end", agentType: this.agentType, step, output: observation });
+    }
+    const fallback = steps.length > 0 ? `Reached max steps. Last result: ${steps[steps.length - 1].observation}` : "Unable to process request.";
+    this.saveToMemory(input, fallback);
+    return this.buildResult(input, fallback, steps, startTime, { maxStepsReached: true });
+  }
+  buildFunctionsFromTools() {
+    return [...this.tools.values()].map((tool) => ({
+      name: tool.name,
+      description: tool.description,
+      parameters: {
+        type: "object",
+        properties: Object.fromEntries(
+          (tool.parameters ?? []).map((p) => [
+            p.name,
+            {
+              type: p.type,
+              description: p.description
+            }
+          ])
+        ),
+        required: (tool.parameters ?? []).filter((p) => p.required).map((p) => p.name)
+      }
+    }));
+  }
+  buildSystemPrompt() {
+    const functionsJson = JSON.stringify(this.functions, null, 2);
+    const rules = this.getRulesString();
+    return `You are an AI assistant with access to functions. Your goal: ${this.goal}
+
+Available functions:
+${functionsJson}
+
+${rules ? `Rules:
+${rules}
+` : ""}
+To call a function, respond with a JSON block:
+\`\`\`function_call
+{"name": "function_name", "arguments": {"param1": "value1"}}
+\`\`\`
+
+If you do NOT need to call a function, respond normally with your answer.
+When you have enough information from function results, provide your final answer as plain text (no function_call block).
+
+${this.systemPrompt ?? ""}`;
+  }
+  buildPrompt(input, steps, memoryContext) {
+    let prompt = `${memoryContext}
+User: ${input}
+`;
+    for (const step of steps) {
+      prompt += `
+Function call: ${step.action}(${JSON.stringify(step.actionInput)})
+`;
+      prompt += `Result: ${step.observation}
+`;
+    }
+    if (steps.length > 0) {
+      prompt += "\nBased on the function results above, continue or provide your final answer:\n";
+    }
+    return prompt;
+  }
+  parseFunctionCall(content) {
+    const functionCallMatch = content.match(/```function_call\s*\n?([\s\S]*?)\n?\s*```/);
+    if (functionCallMatch) {
+      try {
+        const parsed = JSON.parse(functionCallMatch[1].trim());
+        return {
+          content: content.replace(functionCallMatch[0], "").trim(),
+          functionCall: {
+            name: parsed.name,
+            arguments: parsed.arguments ?? {}
+          }
+        };
+      } catch {
+      }
+    }
+    const jsonMatch = content.match(/\{"name":\s*"([^"]+)",\s*"arguments":\s*(\{[\s\S]*?\})\}/);
+    if (jsonMatch) {
+      try {
+        const args = JSON.parse(jsonMatch[2]);
+        return {
+          content: content.replace(jsonMatch[0], "").trim(),
+          functionCall: { name: jsonMatch[1], arguments: args }
+        };
+      } catch {
+      }
+    }
+    return { content };
+  }
+};
+
+export { OpenAIFunctionsAgent };
+//# sourceMappingURL=chunk-7NVGLURS.js.map
+//# sourceMappingURL=chunk-7NVGLURS.js.map

package/dist/chunk-7NVGLURS.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/agent/openai-functions-agent.ts"],"names":[],"mappings":";;;AAUO,IAAM,oBAAA,GAAN,cAAmC,SAAA,CAAU;AAAA,EAC1C,SAAA;AAAA,EAER,WAAA,CAAY,MAAA,EAAoC,GAAA,EAAiB,MAAA,EAAiB;AAChF,IAAA,KAAA,CAAM,MAAA,EAAQ,GAAA,EAAK,MAAA,EAAQ,kBAAkB,CAAA;AAC7C,IAAA,IAAA,CAAK,WAAA,GAAc,OAAO,WAAA,IAAe,GAAA;AACzC,IAAA,IAAA,CAAK,SAAA,GAAY,MAAA,CAAO,SAAA,IAAa,IAAA,CAAK,uBAAA,EAAwB;AAAA,EACpE;AAAA,EAEA,MAAM,IAAI,KAAA,EAAqC;AAC7C,IAAA,MAAM,SAAA,GAAY,KAAK,GAAA,EAAI;AAC3B,IAAA,MAAM,QAA2B,EAAC;AAClC,IAAA,MAAM,aAAA,GAAgB,KAAK,gBAAA,EAAiB;AAE5C,IAAA,IAAA,CAAK,IAAA,CAAK,EAAE,IAAA,EAAM,YAAA,EAAc,WAAW,IAAA,CAAK,SAAA,EAAW,OAAO,CAAA;AAElE,IAAA,KAAA,IAAS,IAAA,GAAO,CAAA,EAAG,IAAA,GAAO,IAAA,CAAK,UAAU,IAAA,EAAA,EAAQ;AAC/C,MAAA,MAAM,SAAA,GAAY,KAAK,GAAA,EAAI;AAC3B,MAAA,MAAM,MAAA,GAAS,IAAA,CAAK,WAAA,CAAY,KAAA,EAAO,OAAO,aAAa,CAAA;AAE3D,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,GAAA,CAAI,SAAS,MAAA,EAAQ;AAAA,QAC7C,aAAa,IAAA,CAAK,WAAA;AAAA,QAClB,SAAA,EAAW,IAAA;AAAA,QACX,YAAA,EAAc,KAAK,iBAAA;AAAkB,OACtC,CAAA;AAED,MAAA,MAAM,MAAA,GAAS,IAAA,CAAK,iBAAA,CAAkB,MAAA,CAAO,OAAO,CAAA;AAEpD,MAAA,IAAI,CAAC,OAAO,YAAA,EAAc;AACxB,QAAA,MAAM,cAAc,MAAA,CAAO,OAAA;AAC3B,QAAA,IAAA,CAAK,YAAA,CAAa,OAAO,WAAW,CAAA;AACpC,QAAA,IAAA,CAAK,IAAA,CAAK,EAAE,IAAA,EAAM,UAAA,EAAY,WAAW,IAAA,CAAK,SAAA,EAAW,MAAA,EAAQ,WAAA,EAAa,CAAA;AAE9E,QAAA,MAAM,WAAA,GAAc,IAAA,CAAK,WAAA,CAAY,KAAA,EAAO,WAAA,EAAa,KAAA,EAAO,SAAA,EAAW,EAAE,SAAA,EAAW,KAAA,CAAM,MAAA,EAAQ,CAAA;AACtG,QAAA,WAAA,CAAY,WAAA,IAAe,OAAO,KAAA,CAAM,WAAA;AACxC,QAAA,OAAO,WAAA;AAAA,MACT;AAEA,MAAA,MAAM,EAAE,IAAA,EAAM,SAAA,EAAW,IAAA,KAAS,MAAA,CAAO,YAAA;AACzC,MAAA,MAAM,WAAA,GAAc,MAAM,IAAA,CAAK,WAAA,CAAY,MAAM,IAAI,CAAA;AAErD,MAAA,KAAA,CAAM,IAAA,CAAK;AAAA,QACT,OAAA,EAAS,qBAAqB,IAAI,CAAA,CAAA;AAAA,QAClC,MAAA,EAAQ,IAAA;AAAA,QACR,WAAA,EAAa,IAAA;AAAA,QACb,WAAA;AAAA,QACA,SAAA,EAAW,IAAA,CAAK,GAAA,EAAI,GAAI,SAAA;AAAA,QACxB,KAAA,EAAO;AAAA,UACL,YAAA,EAAc,OAAO,KAAA,CAAM,YAAA;AAAA,UAC3B,gBAAA,EAAkB,OAAO,KAAA,CAAM,gBAAA;AAAA,UAC/B,WAAA,EAAa,OAAO,KAAA,CAAM;AAAA;AAC5B,OACD,CAAA;AAED,MAAA,IAAA,CAAK,IAAA,CAAK,EAAE,IAAA,EAAM,UAAA,EAAY,SAAA,EAAW,KAAK,SAAA,EAAW,IAAA,EAAM,MAAA,EAAQ,WAAA,EAAa,CAAA;AAAA,IACtF;AAEA,IAAA,MAAM,QAAA,GAAW,KAAA,CAAM,MAAA,GAAS,CAAA,GAC5B,CAAA,gCAAA,EAAmC,KAAA,CAAM,KAAA,CAAM,MAAA,GAAS,CAAC,CAAA,CAAE,WAAW,CAAA,CAAA,GACtE,4BAAA;AAEJ,IAAA,IAAA,CAAK,YAAA,CAAa,OAAO,QAAQ,CAAA;AACjC,IAAA,OAAO,IAAA,CAAK,YAAY,KAAA,EAAO,QAAA,EAAU,OAAO,SAAA,EAAW,EAAE,eAAA,EAAiB,IAAA,EAAM,CAAA;AAAA,EACtF;AAAA,EAEQ,uBAAA,GAA4C;AAClD,IAAA,OAAO,CAAC,GAAG,IAAA,CAAK,KAAA,CAAM,QAAQ,CAAA,CAAE,IAAI,CAAA,IAAA,MAAS;AAAA,MAC3C,MAAM,IAAA,CAAK,IAAA;AAAA,MACX,aAAa,IAAA,CAAK,WAAA;AAAA,MAClB,UAAA,EAAY;AAAA,QACV,IAAA,EAAM,QAAA;AAAA,QACN,YAAY,MAAA,CAAO,WAAA;AAAA,UAAA,CAChB,IAAA,CAAK,UAAA,IAAc,EAAC,EAAG,IAAI,CAAA,CAAA,KAAK;AAAA,YAC/B,CAAA,CAAE,IAAA;AAAA,YACF;AAAA,cACE,MAAM,CAAA,CAAE,IAAA;AAAA,cACR,aAAa,CAAA,CAAE;AAAA;AACjB,WACD;AAAA,SACH;AAAA,QACA,QAAA,EAAA,CAAW,IAAA,CAAK,UAAA,IAAc,EAAC,EAAG,MAAA,CAAO,CAAA,CAAA,KAAK,CAAA,CAAE,QAAQ,CAAA,CAAE,GAAA,CAAI,CAAA,CAAA,KAAK,EAAE,IAAI;AAAA;AAC3E,KACF,CAAE,CAAA;AAAA,EACJ;AAAA,EAEQ,iBAAA,GAA4B;AAClC,IAAA,MAAM,gBAAgB,IAAA,CAAK,SAAA,CAAU,IAAA,CAAK,SAAA,EAAW,MAAM,CAAC,CAAA;AAC5D,IAAA,MAAM,KAAA,GAAQ,KAAK,cAAA,EAAe;AAElC,IAAA,OAAO,CAAA,6DAAA,EAAgE,KAAK,IAAI;;AAAA;AAAA,EAGlF,aAAa;;AAAA,EAEb,KAAA,GAAQ,CAAA;AAAA,EAAW,KAAK;AAAA,CAAA,GAAO,EAAE;AAAA;AAAA;AAAA;AAAA;;AAAA;AAAA;;AAAA,EASjC,IAAA,CAAK,gBAAgB,EAAE,CAAA,CAAA;AAAA,EACvB;AAAA,EAEQ,WAAA,CAAY,KAAA,EAAe,KAAA,EAA0B,aAAA,EAA+B;AAC1F,IAAA,IAAI,MAAA,GAAS,GAAG,aAAa;AAAA,MAAA,EAAW,KAAK;AAAA,CAAA;AAE7C,IAAA,KAAA,MAAW,QAAQ,KAAA,EAAO;AACxB,MAAA,MAAA,IAAU;AAAA,eAAA,EAAoB,KAAK,MAAM,CAAA,CAAA,EAAI,KAAK,SAAA,CAAU,IAAA,CAAK,WAAW,CAAC,CAAA;AAAA,CAAA;AAC7E,MAAA,MAAA,IAAU,CAAA,QAAA,EAAW,KAAK,WAAW;AAAA,CAAA;AAAA,IACvC;AAEA,IAAA,IAAI,KAAA,CAAM,SAAS,CAAA,EAAG;AACpB,MAAA,MAAA,IAAU,iFAAA;AAAA,IACZ;AAEA,IAAA,OAAO,MAAA;AAAA,EACT;AAAA,EAEQ,kBAAkB,OAAA,EAA2G;AACnI,IAAA,MAAM,iBAAA,GAAoB,OAAA,CAAQ,KAAA,CAAM,2CAA2C,CAAA;AAEnF,IAAA,IAAI,iBAAA,EAAmB;AACrB,MAAA,IAAI;AACF,QAAA,MAAM,SAAS,IAAA,CAAK,KAAA,CAAM,kBAAkB,CAAC,CAAA,CAAE,MAAM,CAAA;AACrD,QAAA,OAAO;AAAA,UACL,OAAA,EAAS,QAAQ,OAAA,CAAQ,iBAAA,CAAkB,CAAC,CAAA,EAAG,EAAE,EAAE,IAAA,EAAK;AAAA,UACxD,YAAA,EAAc;AAAA,YACZ,MAAM,MAAA,CAAO,IAAA;AAAA,YACb,SAAA,EAAW,MAAA,CAAO,SAAA,IAAa;AAAC;AAClC,SACF;AAAA,MACF,CAAA,CAAA,MAAQ;AAAA,MAER;AAAA,IACF;AAGA,IAAA,MAAM,SAAA,GAAY,OAAA,CAAQ,KAAA,CAAM,0DAA0D,CAAA;AAC1F,IAAA,IAAI,SAAA,EAAW;AACb,MAAA,IAAI;AACF,QAAA,MAAM,IAAA,GAAO,IAAA,CAAK,KAAA,CAAM,SAAA,CAAU,CAAC,CAAC,CAAA;AACpC,QAAA,OAAO;AAAA,UACL,OAAA,EAAS,QAAQ,OAAA,CAAQ,SAAA,CAAU,CAAC,CAAA,EAAG,EAAE,EAAE,IAAA,EAAK;AAAA,UAChD,cAAc,EAAE,IAAA,EAAM,UAAU,CAAC,CAAA,EAAG,WAAW,IAAA;AAAK,SACtD;AAAA,MACF,CAAA,CAAA,MAAQ;AAAA,MAER;AAAA,IACF;AAEA,IAAA,OAAO,EAAE,OAAA,EAAQ;AAAA,EACnB;AACF","file":"chunk-7NVGLURS.js","sourcesContent":["import type { LLMAdapter } from '../types/index.js';\nimport type { Memory } from '../memory/memory.js';\nimport type {\n  OpenAIFunctionsAgentConfig,\n  OpenAIFunction,\n  AgentResult,\n  AgentStepResult,\n} from './types.js';\nimport { BaseAgent } from './base-agent.js';\n\nexport class OpenAIFunctionsAgent extends BaseAgent {\n  private functions: OpenAIFunction[];\n\n  constructor(config: OpenAIFunctionsAgentConfig, llm: LLMAdapter, memory?: Memory) {\n    super(config, llm, memory, 'openai-functions');\n    this.temperature = config.temperature ?? 0.1;\n    this.functions = config.functions ?? this.buildFunctionsFromTools();\n  }\n\n  async run(input: string): Promise<AgentResult> {\n    const startTime = Date.now();\n    const steps: AgentStepResult[] = [];\n    const memoryContext = this.getMemoryContext();\n\n    this.emit({ type: 'step:start', agentType: this.agentType, input });\n\n    for (let step = 0; step < this.maxSteps; step++) {\n      const stepStart = Date.now();\n      const prompt = this.buildPrompt(input, steps, memoryContext);\n\n      const result = await this.llm.generate(prompt, {\n        temperature: this.temperature,\n        maxTokens: 2048,\n        systemPrompt: this.buildSystemPrompt(),\n      });\n\n      const parsed = this.parseFunctionCall(result.content);\n\n      if (!parsed.functionCall) {\n        const finalAnswer = parsed.content;\n        this.saveToMemory(input, finalAnswer);\n        this.emit({ type: 'complete', agentType: this.agentType, output: finalAnswer });\n\n        const agentResult = this.buildResult(input, finalAnswer, steps, startTime, { stepsUsed: steps.length });\n        agentResult.totalTokens += result.usage.totalTokens;\n        return agentResult;\n      }\n\n      const { name, arguments: args } = parsed.functionCall;\n      const observation = await this.executeTool(name, args);\n\n      steps.push({\n        thought: `Calling function: ${name}`,\n        action: name,\n        actionInput: args,\n        observation,\n        latencyMs: Date.now() - stepStart,\n        usage: {\n          promptTokens: result.usage.promptTokens,\n          completionTokens: result.usage.completionTokens,\n          totalTokens: result.usage.totalTokens,\n        },\n      });\n\n      this.emit({ type: 'step:end', agentType: this.agentType, step, output: observation });\n    }\n\n    const fallback = steps.length > 0\n      ? `Reached max steps. Last result: ${steps[steps.length - 1].observation}`\n      : 'Unable to process request.';\n\n    this.saveToMemory(input, fallback);\n    return this.buildResult(input, fallback, steps, startTime, { maxStepsReached: true });\n  }\n\n  private buildFunctionsFromTools(): OpenAIFunction[] {\n    return [...this.tools.values()].map(tool => ({\n      name: tool.name,\n      description: tool.description,\n      parameters: {\n        type: 'object' as const,\n        properties: Object.fromEntries(\n          (tool.parameters ?? []).map(p => [\n            p.name,\n            {\n              type: p.type,\n              description: p.description,\n            },\n          ])\n        ),\n        required: (tool.parameters ?? []).filter(p => p.required).map(p => p.name),\n      },\n    }));\n  }\n\n  private buildSystemPrompt(): string {\n    const functionsJson = JSON.stringify(this.functions, null, 2);\n    const rules = this.getRulesString();\n\n    return `You are an AI assistant with access to functions. Your goal: ${this.goal}\n\nAvailable functions:\n${functionsJson}\n\n${rules ? `Rules:\\n${rules}\\n` : ''}\nTo call a function, respond with a JSON block:\n\\`\\`\\`function_call\n{\"name\": \"function_name\", \"arguments\": {\"param1\": \"value1\"}}\n\\`\\`\\`\n\nIf you do NOT need to call a function, respond normally with your answer.\nWhen you have enough information from function results, provide your final answer as plain text (no function_call block).\n\n${this.systemPrompt ?? ''}`;\n  }\n\n  private buildPrompt(input: string, steps: AgentStepResult[], memoryContext: string): string {\n    let prompt = `${memoryContext}\\nUser: ${input}\\n`;\n\n    for (const step of steps) {\n      prompt += `\\nFunction call: ${step.action}(${JSON.stringify(step.actionInput)})\\n`;\n      prompt += `Result: ${step.observation}\\n`;\n    }\n\n    if (steps.length > 0) {\n      prompt += '\\nBased on the function results above, continue or provide your final answer:\\n';\n    }\n\n    return prompt;\n  }\n\n  private parseFunctionCall(content: string): { content: string; functionCall?: { name: string; arguments: Record<string, unknown> } } {\n    const functionCallMatch = content.match(/```function_call\\s*\\n?([\\s\\S]*?)\\n?\\s*```/);\n\n    if (functionCallMatch) {\n      try {\n        const parsed = JSON.parse(functionCallMatch[1].trim());\n        return {\n          content: content.replace(functionCallMatch[0], '').trim(),\n          functionCall: {\n            name: parsed.name,\n            arguments: parsed.arguments ?? {},\n          },\n        };\n      } catch {\n        // Fall through to JSON detection\n      }\n    }\n\n    // Also try to detect inline JSON function calls\n    const jsonMatch = content.match(/\\{\"name\":\\s*\"([^\"]+)\",\\s*\"arguments\":\\s*(\\{[\\s\\S]*?\\})\\}/);\n    if (jsonMatch) {\n      try {\n        const args = JSON.parse(jsonMatch[2]);\n        return {\n          content: content.replace(jsonMatch[0], '').trim(),\n          functionCall: { name: jsonMatch[1], arguments: args },\n        };\n      } catch {\n        // Fall through\n      }\n    }\n\n    return { content };\n  }\n}\n"]}

package/dist/chunk-AD2GLQK7.js

@@ -0,0 +1,34 @@
+// src/parsers/comma-separated-list-parser.ts
+var CommaSeparatedListParser = class {
+  trim;
+  removeDuplicates;
+  constructor(options = {}) {
+    this.trim = options.trim ?? true;
+    this.removeDuplicates = options.removeDuplicates ?? false;
+  }
+  parse(text) {
+    const cleaned = text.replace(/```[\s\S]*?```/g, "").trim();
+    let items = cleaned.split(",");
+    if (this.trim) {
+      items = items.map((item) => item.trim());
+    }
+    items = items.filter((item) => item.length > 0);
+    if (this.removeDuplicates) {
+      items = [...new Set(items)];
+    }
+    if (items.length === 0) {
+      throw new Error(`CommaSeparatedListParser: No items found in input.
+Input: ${text.slice(0, 200)}`);
+    }
+    return items;
+  }
+  getFormatInstructions() {
+    return `Your response must be a comma-separated list of values.
+Example: item1, item2, item3
+Do not include any additional text, numbering, or formatting.`;
+  }
+};
+
+export { CommaSeparatedListParser };
+//# sourceMappingURL=chunk-AD2GLQK7.js.map
+//# sourceMappingURL=chunk-AD2GLQK7.js.map

package/dist/chunk-AD2GLQK7.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/parsers/comma-separated-list-parser.ts"],"names":[],"mappings":";AAOO,IAAM,2BAAN,MAAiE;AAAA,EAC9D,IAAA;AAAA,EACA,gBAAA;AAAA,EAER,WAAA,CAAY,OAAA,GAA2C,EAAC,EAAG;AACzD,IAAA,IAAA,CAAK,IAAA,GAAO,QAAQ,IAAA,IAAQ,IAAA;AAC5B,IAAA,IAAA,CAAK,gBAAA,GAAmB,QAAQ,gBAAA,IAAoB,KAAA;AAAA,EACtD;AAAA,EAEA,MAAM,IAAA,EAAwB;AAC5B,IAAA,MAAM,UAAU,IAAA,CAAK,OAAA,CAAQ,iBAAA,EAAmB,EAAE,EAAE,IAAA,EAAK;AAEzD,IAAA,IAAI,KAAA,GAAQ,OAAA,CAAQ,KAAA,CAAM,GAAG,CAAA;AAE7B,IAAA,IAAI,KAAK,IAAA,EAAM;AACb,MAAA,KAAA,GAAQ,KAAA,CAAM,GAAA,CAAI,CAAA,IAAA,KAAQ,IAAA,CAAK,MAAM,CAAA;AAAA,IACvC;AAEA,IAAA,KAAA,GAAQ,KAAA,CAAM,MAAA,CAAO,CAAA,IAAA,KAAQ,IAAA,CAAK,SAAS,CAAC,CAAA;AAE5C,IAAA,IAAI,KAAK,gBAAA,EAAkB;AACzB,MAAA,KAAA,GAAQ,CAAC,GAAG,IAAI,GAAA,CAAI,KAAK,CAAC,CAAA;AAAA,IAC5B;AAEA,IAAA,IAAI,KAAA,CAAM,WAAW,CAAA,EAAG;AACtB,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA;AAAA,OAAA,EAA8D,IAAA,CAAK,KAAA,CAAM,CAAA,EAAG,GAAG,CAAC,CAAA,CAAE,CAAA;AAAA,IACpG;AAEA,IAAA,OAAO,KAAA;AAAA,EACT;AAAA,EAEA,qBAAA,GAAgC;AAC9B,IAAA,OAAO,CAAA;AAAA;AAAA,6DAAA,CAAA;AAAA,EACT;AACF","file":"chunk-AD2GLQK7.js","sourcesContent":["import type { OutputParser } from './types.js';\n\nexport interface CommaSeparatedListParserOptions {\n  trim?: boolean;\n  removeDuplicates?: boolean;\n}\n\nexport class CommaSeparatedListParser implements OutputParser<string[]> {\n  private trim: boolean;\n  private removeDuplicates: boolean;\n\n  constructor(options: CommaSeparatedListParserOptions = {}) {\n    this.trim = options.trim ?? true;\n    this.removeDuplicates = options.removeDuplicates ?? false;\n  }\n\n  parse(text: string): string[] {\n    const cleaned = text.replace(/```[\\s\\S]*?```/g, '').trim();\n\n    let items = cleaned.split(',');\n\n    if (this.trim) {\n      items = items.map(item => item.trim());\n    }\n\n    items = items.filter(item => item.length > 0);\n\n    if (this.removeDuplicates) {\n      items = [...new Set(items)];\n    }\n\n    if (items.length === 0) {\n      throw new Error(`CommaSeparatedListParser: No items found in input.\\nInput: ${text.slice(0, 200)}`);\n    }\n\n    return items;\n  }\n\n  getFormatInstructions(): string {\n    return `Your response must be a comma-separated list of values.\\nExample: item1, item2, item3\\nDo not include any additional text, numbering, or formatting.`;\n  }\n}\n"]}

package/dist/chunk-BBUZ6P5S.cjs

@@ -0,0 +1,16 @@
+'use strict';
+
+var crypto = require('crypto');
+
+// src/utils/id.ts
+var counter = 0;
+function generateId() {
+  const timestamp = Date.now().toString(36);
+  const random = crypto.randomBytes(6).toString("hex");
+  const seq = (counter++ & 65535).toString(36);
+  return `${timestamp}_${random}_${seq}`;
+}
+
+exports.generateId = generateId;
+//# sourceMappingURL=chunk-BBUZ6P5S.cjs.map
+//# sourceMappingURL=chunk-BBUZ6P5S.cjs.map

package/dist/chunk-BBUZ6P5S.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/utils/id.ts"],"names":["randomBytes"],"mappings":";;;;;AAEA,IAAI,OAAA,GAAU,CAAA;AAEP,SAAS,UAAA,GAAqB;AACnC,EAAA,MAAM,SAAA,GAAY,IAAA,CAAK,GAAA,EAAI,CAAE,SAAS,EAAE,CAAA;AACxC,EAAA,MAAM,MAAA,GAASA,kBAAA,CAAY,CAAC,CAAA,CAAE,SAAS,KAAK,CAAA;AAC5C,EAAA,MAAM,GAAA,GAAA,CAAO,OAAA,EAAA,GAAY,KAAA,EAAQ,QAAA,CAAS,EAAE,CAAA;AAC5C,EAAA,OAAO,CAAA,EAAG,SAAS,CAAA,CAAA,EAAI,MAAM,IAAI,GAAG,CAAA,CAAA;AACtC","file":"chunk-BBUZ6P5S.cjs","sourcesContent":["import { randomBytes } from 'crypto';\n\nlet counter = 0;\n\nexport function generateId(): string {\n  const timestamp = Date.now().toString(36);\n  const random = randomBytes(6).toString('hex');\n  const seq = (counter++ & 0xFFFF).toString(36);\n  return `${timestamp}_${random}_${seq}`;\n}\n"]}

package/dist/chunk-BDIK4SAE.cjs

@@ -0,0 +1,65 @@
+'use strict';
+
+// src/retrievers/parent-document-retriever.ts
+var ParentDocumentRetriever = class {
+  vectorDB;
+  llm;
+  childTopK;
+  parentTopK;
+  minScore;
+  constructor(options) {
+    this.vectorDB = options.vectorDB;
+    this.llm = options.llm;
+    this.childTopK = options.childTopK ?? 10;
+    this.parentTopK = options.parentTopK ?? 3;
+    this.minScore = options.minScore;
+  }
+  async retrieve(query, collection) {
+    const [embedding] = await this.llm.embed([query]);
+    const childResults = await this.vectorDB.search(collection, embedding, {
+      topK: this.childTopK,
+      minScore: this.minScore
+    });
+    if (childResults.length === 0) return [];
+    const parentMap = /* @__PURE__ */ new Map();
+    for (const child of childResults) {
+      const parentId = child.metadata?.parentId ?? child.metadata?.documentId ?? child.id;
+      const existing = parentMap.get(parentId);
+      if (existing) {
+        existing.children.push(child);
+        existing.bestScore = Math.max(existing.bestScore, child.score);
+      } else {
+        parentMap.set(parentId, {
+          children: [child],
+          bestScore: child.score
+        });
+      }
+    }
+    const sortedParents = Array.from(parentMap.entries()).sort(([, a], [, b]) => b.bestScore - a.bestScore).slice(0, this.parentTopK);
+    const results = [];
+    for (const [parentId, { children, bestScore }] of sortedParents) {
+      const sortedChildren = children.sort((a, b) => {
+        const indexA = a.metadata?.chunkIndex ?? 0;
+        const indexB = b.metadata?.chunkIndex ?? 0;
+        return indexA - indexB;
+      });
+      const parentContent = sortedChildren[0]?.metadata?.parentContent;
+      const content = parentContent ? parentContent : sortedChildren.map((c) => c.content ?? "").join("\n\n");
+      results.push({
+        id: parentId,
+        score: bestScore,
+        content,
+        metadata: {
+          ...sortedChildren[0]?.metadata,
+          childCount: children.length,
+          retrieverType: "parent-document"
+        }
+      });
+    }
+    return results;
+  }
+};
+
+exports.ParentDocumentRetriever = ParentDocumentRetriever;
+//# sourceMappingURL=chunk-BDIK4SAE.cjs.map
+//# sourceMappingURL=chunk-BDIK4SAE.cjs.map

package/dist/chunk-BDIK4SAE.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/retrievers/parent-document-retriever.ts"],"names":[],"mappings":";;;AAGO,IAAM,0BAAN,MAAmD;AAAA,EAChD,QAAA;AAAA,EACA,GAAA;AAAA,EACA,SAAA;AAAA,EACA,UAAA;AAAA,EACA,QAAA;AAAA,EAER,YAAY,OAAA,EAAyC;AACnD,IAAA,IAAA,CAAK,WAAW,OAAA,CAAQ,QAAA;AACxB,IAAA,IAAA,CAAK,MAAM,OAAA,CAAQ,GAAA;AACnB,IAAA,IAAA,CAAK,SAAA,GAAY,QAAQ,SAAA,IAAa,EAAA;AACtC,IAAA,IAAA,CAAK,UAAA,GAAa,QAAQ,UAAA,IAAc,CAAA;AACxC,IAAA,IAAA,CAAK,WAAW,OAAA,CAAQ,QAAA;AAAA,EAC1B;AAAA,EAEA,MAAM,QAAA,CAAS,KAAA,EAAe,UAAA,EAAmD;AAC/E,IAAA,MAAM,CAAC,SAAS,CAAA,GAAI,MAAM,KAAK,GAAA,CAAI,KAAA,CAAM,CAAC,KAAK,CAAC,CAAA;AAGhD,IAAA,MAAM,eAAe,MAAM,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,YAAY,SAAA,EAAW;AAAA,MACrE,MAAM,IAAA,CAAK,SAAA;AAAA,MACX,UAAU,IAAA,CAAK;AAAA,KAChB,CAAA;AAED,IAAA,IAAI,YAAA,CAAa,MAAA,KAAW,CAAA,EAAG,OAAO,EAAC;AAGvC,IAAA,MAAM,SAAA,uBAAgB,GAAA,EAGnB;AAEH,IAAA,KAAA,MAAW,SAAS,YAAA,EAAc;AAChC,MAAA,MAAM,WAAY,KAAA,CAAM,QAAA,EAAU,YAAwB,KAAA,CAAM,QAAA,EAAU,cAAyB,KAAA,CAAM,EAAA;AAEzG,MAAA,MAAM,QAAA,GAAW,SAAA,CAAU,GAAA,CAAI,QAAQ,CAAA;AACvC,MAAA,IAAI,QAAA,EAAU;AACZ,QAAA,QAAA,CAAS,QAAA,CAAS,KAAK,KAAK,CAAA;AAC5B,QAAA,QAAA,CAAS,YAAY,IAAA,CAAK,GAAA,CAAI,QAAA,CAAS,SAAA,EAAW,MAAM,KAAK,CAAA;AAAA,MAC/D,CAAA,MAAO;AACL,QAAA,SAAA,CAAU,IAAI,QAAA,EAAU;AAAA,UACtB,QAAA,EAAU,CAAC,KAAK,CAAA;AAAA,UAChB,WAAW,KAAA,CAAM;AAAA,SAClB,CAAA;AAAA,MACH;AAAA,IACF;AAGA,IAAA,MAAM,aAAA,GAAgB,KAAA,CAAM,IAAA,CAAK,SAAA,CAAU,OAAA,EAAS,CAAA,CACjD,IAAA,CAAK,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAA,KAAM,CAAA,CAAE,SAAA,GAAY,CAAA,CAAE,SAAS,CAAA,CAChD,KAAA,CAAM,CAAA,EAAG,IAAA,CAAK,UAAU,CAAA;AAG3B,IAAA,MAAM,UAAgC,EAAC;AAEvC,IAAA,KAAA,MAAW,CAAC,QAAA,EAAU,EAAE,UAAU,SAAA,EAAW,KAAK,aAAA,EAAe;AAE/D,MAAA,MAAM,cAAA,GAAiB,QAAA,CAAS,IAAA,CAAK,CAAC,GAAG,CAAA,KAAM;AAC7C,QAAA,MAAM,MAAA,GAAU,CAAA,CAAE,QAAA,EAAU,UAAA,IAAyB,CAAA;AACrD,QAAA,MAAM,MAAA,GAAU,CAAA,CAAE,QAAA,EAAU,UAAA,IAAyB,CAAA;AACrD,QAAA,OAAO,MAAA,GAAS,MAAA;AAAA,MAClB,CAAC,CAAA;AAGD,MAAA,MAAM,aAAA,GAAgB,cAAA,CAAe,CAAC,CAAA,EAAG,QAAA,EAAU,aAAA;AAEnD,MAAA,MAAM,OAAA,GAAU,aAAA,GACZ,aAAA,GACA,cAAA,CAAe,GAAA,CAAI,CAAA,CAAA,KAAK,CAAA,CAAE,OAAA,IAAW,EAAE,CAAA,CAAE,IAAA,CAAK,MAAM,CAAA;AAExD,MAAA,OAAA,CAAQ,IAAA,CAAK;AAAA,QACX,EAAA,EAAI,QAAA;AAAA,QACJ,KAAA,EAAO,SAAA;AAAA,QACP,OAAA;AAAA,QACA,QAAA,EAAU;AAAA,UACR,GAAG,cAAA,CAAe,CAAC,CAAA,EAAG,QAAA;AAAA,UACtB,YAAY,QAAA,CAAS,MAAA;AAAA,UACrB,aAAA,EAAe;AAAA;AACjB,OACD,CAAA;AAAA,IACH;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AACF","file":"chunk-BDIK4SAE.cjs","sourcesContent":["import type { VectorSearchResult, LLMAdapter, VectorDBAdapter } from '../types/index.js';\nimport type { Retriever, ParentDocumentRetrieverOptions } from './types.js';\n\nexport class ParentDocumentRetriever implements Retriever {\n  private vectorDB: VectorDBAdapter;\n  private llm: LLMAdapter;\n  private childTopK: number;\n  private parentTopK: number;\n  private minScore?: number;\n\n  constructor(options: ParentDocumentRetrieverOptions) {\n    this.vectorDB = options.vectorDB;\n    this.llm = options.llm;\n    this.childTopK = options.childTopK ?? 10;\n    this.parentTopK = options.parentTopK ?? 3;\n    this.minScore = options.minScore;\n  }\n\n  async retrieve(query: string, collection: string): Promise<VectorSearchResult[]> {\n    const [embedding] = await this.llm.embed([query]);\n\n    // Step 1: Search for child chunks (small, precise)\n    const childResults = await this.vectorDB.search(collection, embedding, {\n      topK: this.childTopK,\n      minScore: this.minScore,\n    });\n\n    if (childResults.length === 0) return [];\n\n    // Step 2: Group by parent document ID\n    const parentMap = new Map<string, {\n      children: VectorSearchResult[];\n      bestScore: number;\n    }>();\n\n    for (const child of childResults) {\n      const parentId = (child.metadata?.parentId as string) ?? (child.metadata?.documentId as string) ?? child.id;\n\n      const existing = parentMap.get(parentId);\n      if (existing) {\n        existing.children.push(child);\n        existing.bestScore = Math.max(existing.bestScore, child.score);\n      } else {\n        parentMap.set(parentId, {\n          children: [child],\n          bestScore: child.score,\n        });\n      }\n    }\n\n    // Step 3: Sort parents by best child score and take top N\n    const sortedParents = Array.from(parentMap.entries())\n      .sort(([, a], [, b]) => b.bestScore - a.bestScore)\n      .slice(0, this.parentTopK);\n\n    // Step 4: Reconstruct parent context from children\n    const results: VectorSearchResult[] = [];\n\n    for (const [parentId, { children, bestScore }] of sortedParents) {\n      // Sort children by their index/position within the parent\n      const sortedChildren = children.sort((a, b) => {\n        const indexA = (a.metadata?.chunkIndex as number) ?? 0;\n        const indexB = (b.metadata?.chunkIndex as number) ?? 0;\n        return indexA - indexB;\n      });\n\n      // If parent content is stored in metadata, use it\n      const parentContent = sortedChildren[0]?.metadata?.parentContent as string | undefined;\n\n      const content = parentContent\n        ? parentContent\n        : sortedChildren.map(c => c.content ?? '').join('\\n\\n');\n\n      results.push({\n        id: parentId,\n        score: bestScore,\n        content,\n        metadata: {\n          ...sortedChildren[0]?.metadata,\n          childCount: children.length,\n          retrieverType: 'parent-document',\n        },\n      });\n    }\n\n    return results;\n  }\n}\n"]}