orkajs 1.0.0

package/dist/chunk-KXVNH7ZX.cjs

@@ -0,0 +1,149 @@
+'use strict';
+
+// src/adapters/openai/index.ts
+var OpenAIAdapter = class {
+  name = "openai";
+  apiKey;
+  model;
+  embeddingModel;
+  baseURL;
+  timeoutMs;
+  constructor(config) {
+    this.apiKey = config.apiKey;
+    this.model = config.model ?? "gpt-4o-mini";
+    this.embeddingModel = config.embeddingModel ?? "text-embedding-3-small";
+    this.baseURL = config.baseURL ?? "https://api.openai.com/v1";
+    this.timeoutMs = config.timeoutMs ?? 6e4;
+  }
+  async generate(prompt, options = {}) {
+    const messages = [];
+    if (options.messages) {
+      for (const msg of options.messages) {
+        messages.push({
+          role: msg.role,
+          content: typeof msg.content === "string" ? msg.content : this.mapContentParts(msg.content)
+        });
+      }
+    } else {
+      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(`OpenAI API request timed out after ${this.timeoutMs}ms`);
+      }
+      throw error;
+    } finally {
+      clearTimeout(timeout);
+    }
+    if (!response.ok) {
+      const error = await response.text();
+      throw new Error(`OpenAI 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(`OpenAI 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(`OpenAI 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);
+  }
+  mapContentParts(parts) {
+    return parts.map((part) => {
+      switch (part.type) {
+        case "text":
+          return { type: "text", text: part.text };
+        case "image_url":
+          return { type: "image_url", image_url: part.image_url };
+        case "image_base64":
+          return {
+            type: "image_url",
+            image_url: { url: `data:${part.mimeType};base64,${part.data}` }
+          };
+        case "audio":
+          return {
+            type: "input_audio",
+            input_audio: { data: part.data, format: part.format }
+          };
+        default:
+          return part;
+      }
+    });
+  }
+  mapFinishReason(reason) {
+    switch (reason) {
+      case "stop":
+        return "stop";
+      case "length":
+        return "length";
+      case "tool_calls":
+        return "tool_calls";
+      default:
+        return "stop";
+    }
+  }
+};
+
+exports.OpenAIAdapter = OpenAIAdapter;
+//# sourceMappingURL=chunk-KXVNH7ZX.cjs.map
+//# sourceMappingURL=chunk-KXVNH7ZX.cjs.map

package/dist/chunk-KXVNH7ZX.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/adapters/openai/index.ts"],"names":[],"mappings":";;;AAUO,IAAM,gBAAN,MAA0C;AAAA,EACtC,IAAA,GAAO,QAAA;AAAA,EACR,MAAA;AAAA,EACA,KAAA;AAAA,EACA,cAAA;AAAA,EACA,OAAA;AAAA,EACA,SAAA;AAAA,EAER,YAAY,MAAA,EAA6B;AACvC,IAAA,IAAA,CAAK,SAAS,MAAA,CAAO,MAAA;AACrB,IAAA,IAAA,CAAK,KAAA,GAAQ,OAAO,KAAA,IAAS,aAAA;AAC7B,IAAA,IAAA,CAAK,cAAA,GAAiB,OAAO,cAAA,IAAkB,wBAAA;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,WAAiE,EAAC;AAExE,IAAA,IAAI,QAAQ,QAAA,EAAU;AACpB,MAAA,KAAA,MAAW,GAAA,IAAO,QAAQ,QAAA,EAAU;AAClC,QAAA,QAAA,CAAS,IAAA,CAAK;AAAA,UACZ,MAAM,GAAA,CAAI,IAAA;AAAA,UACV,OAAA,EAAS,OAAO,GAAA,CAAI,OAAA,KAAY,QAAA,GAC5B,IAAI,OAAA,GACJ,IAAA,CAAK,eAAA,CAAgB,GAAA,CAAI,OAAO;AAAA,SACrC,CAAA;AAAA,MACH;AAAA,IACF,CAAA,MAAO;AACL,MAAA,IAAI,QAAQ,YAAA,EAAc;AACxB,QAAA,QAAA,CAAS,KAAK,EAAE,IAAA,EAAM,UAAU,OAAA,EAAS,OAAA,CAAQ,cAAc,CAAA;AAAA,MACjE;AACA,MAAA,QAAA,CAAS,KAAK,EAAE,IAAA,EAAM,MAAA,EAAQ,OAAA,EAAS,QAAQ,CAAA;AAAA,IACjD;AAEA,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,mCAAA,EAAsC,IAAA,CAAK,SAAS,CAAA,EAAA,CAAI,CAAA;AAAA,MAC1E;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,kBAAA,EAAqB,SAAS,MAAM,CAAA,GAAA,EAAM,KAAK,CAAA,CAAE,CAAA;AAAA,IACnE;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,8CAAA,EAAiD,IAAA,CAAK,SAAS,CAAA,EAAA,CAAI,CAAA;AAAA,MACrF;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,6BAAA,EAAgC,SAAS,MAAM,CAAA,GAAA,EAAM,KAAK,CAAA,CAAE,CAAA;AAAA,IAC9E;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,KAAA,EAAiC;AACvD,IAAA,OAAO,KAAA,CAAM,IAAI,CAAA,IAAA,KAAQ;AACvB,MAAA,QAAQ,KAAK,IAAA;AAAM,QACjB,KAAK,MAAA;AACH,UAAA,OAAO,EAAE,IAAA,EAAM,MAAA,EAAQ,IAAA,EAAM,KAAK,IAAA,EAAK;AAAA,QACzC,KAAK,WAAA;AACH,UAAA,OAAO,EAAE,IAAA,EAAM,WAAA,EAAa,SAAA,EAAW,KAAK,SAAA,EAAU;AAAA,QACxD,KAAK,cAAA;AACH,UAAA,OAAO;AAAA,YACL,IAAA,EAAM,WAAA;AAAA,YACN,SAAA,EAAW,EAAE,GAAA,EAAK,CAAA,KAAA,EAAQ,KAAK,QAAQ,CAAA,QAAA,EAAW,IAAA,CAAK,IAAI,CAAA,CAAA;AAAG,WAChE;AAAA,QACF,KAAK,OAAA;AACH,UAAA,OAAO;AAAA,YACL,IAAA,EAAM,aAAA;AAAA,YACN,aAAa,EAAE,IAAA,EAAM,KAAK,IAAA,EAAM,MAAA,EAAQ,KAAK,MAAA;AAAO,WACtD;AAAA,QACF;AACE,UAAA,OAAO,IAAA;AAAA;AACX,IACF,CAAC,CAAA;AAAA,EACH;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-KXVNH7ZX.cjs","sourcesContent":["import type { LLMAdapter, LLMGenerateOptions, LLMResult, ContentPart } from '../../types/index.js';\n\nexport interface OpenAIAdapterConfig {\n  apiKey: string;\n  model?: string;\n  embeddingModel?: string;\n  baseURL?: string;\n  timeoutMs?: number;\n}\n\nexport class OpenAIAdapter implements LLMAdapter {\n  readonly name = 'openai';\n  private apiKey: string;\n  private model: string;\n  private embeddingModel: string;\n  private baseURL: string;\n  private timeoutMs: number;\n\n  constructor(config: OpenAIAdapterConfig) {\n    this.apiKey = config.apiKey;\n    this.model = config.model ?? 'gpt-4o-mini';\n    this.embeddingModel = config.embeddingModel ?? 'text-embedding-3-small';\n    this.baseURL = config.baseURL ?? 'https://api.openai.com/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 | unknown[] }> = [];\n\n    if (options.messages) {\n      for (const msg of options.messages) {\n        messages.push({\n          role: msg.role,\n          content: typeof msg.content === 'string'\n            ? msg.content\n            : this.mapContentParts(msg.content),\n        });\n      }\n    } else {\n      if (options.systemPrompt) {\n        messages.push({ role: 'system', content: options.systemPrompt });\n      }\n      messages.push({ role: 'user', content: prompt });\n    }\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(`OpenAI 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(`OpenAI 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(`OpenAI 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(`OpenAI 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 mapContentParts(parts: ContentPart[]): unknown[] {\n    return parts.map(part => {\n      switch (part.type) {\n        case 'text':\n          return { type: 'text', text: part.text };\n        case 'image_url':\n          return { type: 'image_url', image_url: part.image_url };\n        case 'image_base64':\n          return {\n            type: 'image_url',\n            image_url: { url: `data:${part.mimeType};base64,${part.data}` },\n          };\n        case 'audio':\n          return {\n            type: 'input_audio',\n            input_audio: { data: part.data, format: part.format },\n          };\n        default:\n          return part;\n      }\n    });\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-L2VBPZTO.js

@@ -0,0 +1,58 @@
+// src/retrievers/multi-query-retriever.ts
+var MultiQueryRetriever = class {
+  options;
+  constructor(options) {
+    this.options = {
+      queryCount: 3,
+      topK: 5,
+      deduplicateByContent: true,
+      ...options
+    };
+  }
+  async retrieve(query, collection) {
+    const queries = await this.generateQueries(query);
+    const allResults = [];
+    for (const q of queries) {
+      const [embedding] = await this.options.llm.embed([q]);
+      const results = await this.options.vectorDB.search(collection, embedding, {
+        topK: this.options.topK,
+        minScore: this.options.minScore
+      });
+      allResults.push(...results);
+    }
+    return this.deduplicateAndRank(allResults);
+  }
+  async generateQueries(originalQuery) {
+    const prompt = `You are an AI assistant helping to generate alternative search queries.
+Given the original query, generate ${this.options.queryCount} alternative versions that capture different aspects or phrasings of the same information need.
+Return ONLY the queries, one per line, without numbering or bullets.
+
+Original query: ${originalQuery}
+
+Alternative queries:`;
+    const result = await this.options.llm.generate(prompt, {
+      temperature: 0.7,
+      maxTokens: 256
+    });
+    const queries = result.content.split("\n").map((q) => q.trim()).filter((q) => q.length > 0).slice(0, this.options.queryCount);
+    return [originalQuery, ...queries];
+  }
+  deduplicateAndRank(results) {
+    if (!this.options.deduplicateByContent) {
+      return results.sort((a, b) => b.score - a.score).slice(0, this.options.topK);
+    }
+    const seen = /* @__PURE__ */ new Map();
+    for (const result of results) {
+      const key = result.content ?? result.id;
+      const existing = seen.get(key);
+      if (!existing || result.score > existing.score) {
+        seen.set(key, result);
+      }
+    }
+    return Array.from(seen.values()).sort((a, b) => b.score - a.score).slice(0, this.options.topK);
+  }
+};
+
+export { MultiQueryRetriever };
+//# sourceMappingURL=chunk-L2VBPZTO.js.map
+//# sourceMappingURL=chunk-L2VBPZTO.js.map

package/dist/chunk-L2VBPZTO.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/retrievers/multi-query-retriever.ts"],"names":[],"mappings":";AAGO,IAAM,sBAAN,MAA+C;AAAA,EAC5C,OAAA;AAAA,EAER,YAAY,OAAA,EAAqC;AAC/C,IAAA,IAAA,CAAK,OAAA,GAAU;AAAA,MACb,UAAA,EAAY,CAAA;AAAA,MACZ,IAAA,EAAM,CAAA;AAAA,MACN,oBAAA,EAAsB,IAAA;AAAA,MACtB,GAAG;AAAA,KACL;AAAA,EACF;AAAA,EAEA,MAAM,QAAA,CAAS,KAAA,EAAe,UAAA,EAAmD;AAC/E,IAAA,MAAM,OAAA,GAAU,MAAM,IAAA,CAAK,eAAA,CAAgB,KAAK,CAAA;AAChD,IAAA,MAAM,aAAmC,EAAC;AAE1C,IAAA,KAAA,MAAW,KAAK,OAAA,EAAS;AACvB,MAAA,MAAM,CAAC,SAAS,CAAA,GAAI,MAAM,IAAA,CAAK,QAAQ,GAAA,CAAI,KAAA,CAAM,CAAC,CAAC,CAAC,CAAA;AACpD,MAAA,MAAM,UAAU,MAAM,IAAA,CAAK,QAAQ,QAAA,CAAS,MAAA,CAAO,YAAY,SAAA,EAAW;AAAA,QACxE,IAAA,EAAM,KAAK,OAAA,CAAQ,IAAA;AAAA,QACnB,QAAA,EAAU,KAAK,OAAA,CAAQ;AAAA,OACxB,CAAA;AACD,MAAA,UAAA,CAAW,IAAA,CAAK,GAAG,OAAO,CAAA;AAAA,IAC5B;AAEA,IAAA,OAAO,IAAA,CAAK,mBAAmB,UAAU,CAAA;AAAA,EAC3C;AAAA,EAEA,MAAc,gBAAgB,aAAA,EAA0C;AACtE,IAAA,MAAM,MAAA,GAAS,CAAA;AAAA,mCAAA,EACkB,IAAA,CAAK,QAAQ,UAAU,CAAA;AAAA;;AAAA,gBAAA,EAG1C,aAAa;;AAAA,oBAAA,CAAA;AAI3B,IAAA,MAAM,SAAS,MAAM,IAAA,CAAK,OAAA,CAAQ,GAAA,CAAI,SAAS,MAAA,EAAQ;AAAA,MACrD,WAAA,EAAa,GAAA;AAAA,MACb,SAAA,EAAW;AAAA,KACZ,CAAA;AAED,IAAA,MAAM,OAAA,GAAU,OAAO,OAAA,CACpB,KAAA,CAAM,IAAI,CAAA,CACV,GAAA,CAAI,CAAA,CAAA,KAAK,CAAA,CAAE,IAAA,EAAM,EACjB,MAAA,CAAO,CAAA,CAAA,KAAK,EAAE,MAAA,GAAS,CAAC,EACxB,KAAA,CAAM,CAAA,EAAG,IAAA,CAAK,OAAA,CAAQ,UAAU,CAAA;AAEnC,IAAA,OAAO,CAAC,aAAA,EAAe,GAAG,OAAO,CAAA;AAAA,EACnC;AAAA,EAEQ,mBAAmB,OAAA,EAAqD;AAC9E,IAAA,IAAI,CAAC,IAAA,CAAK,OAAA,CAAQ,oBAAA,EAAsB;AACtC,MAAA,OAAO,OAAA,CAAQ,IAAA,CAAK,CAAC,CAAA,EAAG,MAAM,CAAA,CAAE,KAAA,GAAQ,CAAA,CAAE,KAAK,CAAA,CAAE,KAAA,CAAM,CAAA,EAAG,IAAA,CAAK,QAAQ,IAAI,CAAA;AAAA,IAC7E;AAEA,IAAA,MAAM,IAAA,uBAAW,GAAA,EAAgC;AAEjD,IAAA,KAAA,MAAW,UAAU,OAAA,EAAS;AAC5B,MAAA,MAAM,GAAA,GAAM,MAAA,CAAO,OAAA,IAAW,MAAA,CAAO,EAAA;AACrC,MAAA,MAAM,QAAA,GAAW,IAAA,CAAK,GAAA,CAAI,GAAG,CAAA;AAE7B,MAAA,IAAI,CAAC,QAAA,IAAY,MAAA,CAAO,KAAA,GAAQ,SAAS,KAAA,EAAO;AAC9C,QAAA,IAAA,CAAK,GAAA,CAAI,KAAK,MAAM,CAAA;AAAA,MACtB;AAAA,IACF;AAEA,IAAA,OAAO,MAAM,IAAA,CAAK,IAAA,CAAK,QAAQ,CAAA,CAC5B,KAAK,CAAC,CAAA,EAAG,MAAM,CAAA,CAAE,KAAA,GAAQ,EAAE,KAAK,CAAA,CAChC,MAAM,CAAA,EAAG,IAAA,CAAK,QAAQ,IAAI,CAAA;AAAA,EAC/B;AACF","file":"chunk-L2VBPZTO.js","sourcesContent":["import type { VectorSearchResult } from '../types/index.js';\nimport type { Retriever, MultiQueryRetrieverOptions } from './types.js';\n\nexport class MultiQueryRetriever implements Retriever {\n  private options: Required<Pick<MultiQueryRetrieverOptions, 'llm' | 'vectorDB'>> & MultiQueryRetrieverOptions;\n\n  constructor(options: MultiQueryRetrieverOptions) {\n    this.options = {\n      queryCount: 3,\n      topK: 5,\n      deduplicateByContent: true,\n      ...options,\n    };\n  }\n\n  async retrieve(query: string, collection: string): Promise<VectorSearchResult[]> {\n    const queries = await this.generateQueries(query);\n    const allResults: VectorSearchResult[] = [];\n\n    for (const q of queries) {\n      const [embedding] = await this.options.llm.embed([q]);\n      const results = await this.options.vectorDB.search(collection, embedding, {\n        topK: this.options.topK,\n        minScore: this.options.minScore,\n      });\n      allResults.push(...results);\n    }\n\n    return this.deduplicateAndRank(allResults);\n  }\n\n  private async generateQueries(originalQuery: string): Promise<string[]> {\n    const prompt = `You are an AI assistant helping to generate alternative search queries.\nGiven the original query, generate ${this.options.queryCount} alternative versions that capture different aspects or phrasings of the same information need.\nReturn ONLY the queries, one per line, without numbering or bullets.\n\nOriginal query: ${originalQuery}\n\nAlternative queries:`;\n\n    const result = await this.options.llm.generate(prompt, {\n      temperature: 0.7,\n      maxTokens: 256,\n    });\n\n    const queries = result.content\n      .split('\\n')\n      .map(q => q.trim())\n      .filter(q => q.length > 0)\n      .slice(0, this.options.queryCount);\n\n    return [originalQuery, ...queries];\n  }\n\n  private deduplicateAndRank(results: VectorSearchResult[]): VectorSearchResult[] {\n    if (!this.options.deduplicateByContent) {\n      return results.sort((a, b) => b.score - a.score).slice(0, this.options.topK);\n    }\n\n    const seen = new Map<string, VectorSearchResult>();\n\n    for (const result of results) {\n      const key = result.content ?? result.id;\n      const existing = seen.get(key);\n\n      if (!existing || result.score > existing.score) {\n        seen.set(key, result);\n      }\n    }\n\n    return Array.from(seen.values())\n      .sort((a, b) => b.score - a.score)\n      .slice(0, this.options.topK);\n  }\n}\n"]}

package/dist/chunk-LBU7ZGOB.cjs

@@ -0,0 +1,160 @@
+'use strict';
+
+var chunkHW4O3SWN_cjs = require('./chunk-HW4O3SWN.cjs');
+
+// src/agent/structured-chat-agent.ts
+var StructuredChatAgent = class extends chunkHW4O3SWN_cjs.BaseAgent {
+  outputSchema;
+  constructor(config, llm, memory) {
+    super(config, llm, memory, "structured-chat");
+    this.outputSchema = config.outputSchema;
+  }
+  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.parseStructuredResponse(result.content);
+      if (parsed.type === "final") {
+        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 observation = await this.executeTool(parsed.action, parsed.actionInput);
+      steps.push({
+        thought: parsed.thought,
+        action: parsed.action,
+        actionInput: parsed.actionInput,
+        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 observation: ${steps[steps.length - 1].observation}` : "Unable to process request.";
+    this.saveToMemory(input, fallback);
+    return this.buildResult(input, fallback, steps, startTime, { maxStepsReached: true });
+  }
+  buildSystemPrompt() {
+    const toolSchemas = [...this.tools.values()].map((t) => {
+      const schema = {
+        name: t.name,
+        description: t.description
+      };
+      if (t.parameters && t.parameters.length > 0) {
+        schema.parameters = {
+          type: "object",
+          properties: Object.fromEntries(
+            t.parameters.map((p) => [p.name, { type: p.type, description: p.description }])
+          ),
+          required: t.parameters.filter((p) => p.required).map((p) => p.name)
+        };
+      }
+      return schema;
+    });
+    const outputSchemaStr = this.outputSchema ? `
+Your final answer MUST conform to this JSON schema:
+${JSON.stringify(this.outputSchema, null, 2)}
+` : "";
+    const rules = this.getRulesString();
+    return `You are a structured chat agent. Your goal: ${this.goal}
+
+You have access to the following tools (as JSON schemas):
+\`\`\`json
+${JSON.stringify(toolSchemas, null, 2)}
+\`\`\`
+
+${rules ? `Rules:
+${rules}
+` : ""}${outputSchemaStr}
+Respond with a JSON blob in one of two formats:
+
+To use a tool:
+\`\`\`json
+{
+  "thought": "your reasoning",
+  "action": "tool_name",
+  "action_input": { "param": "value" }
+}
+\`\`\`
+
+To give your final answer:
+\`\`\`json
+{
+  "thought": "your final reasoning",
+  "action": "final_answer",
+  "action_input": "your comprehensive answer"
+}
+\`\`\`
+
+ALWAYS respond with a JSON blob. Never respond with plain text.
+
+${this.systemPrompt ?? ""}`;
+  }
+  buildPrompt(input, steps, memoryContext) {
+    let prompt = `${memoryContext}
+Human: ${input}
+`;
+    for (const step of steps) {
+      prompt += `
+Assistant action: ${step.action}(${JSON.stringify(step.actionInput)})
+`;
+      prompt += `Observation: ${step.observation}
+`;
+    }
+    if (steps.length > 0) {
+      prompt += "\nContinue with the next action or provide your final answer:\n";
+    }
+    return prompt;
+  }
+  parseStructuredResponse(content) {
+    const jsonBlockMatch = content.match(/```(?:json)?\s*\n?([\s\S]*?)\n?\s*```/);
+    const jsonStr = jsonBlockMatch ? jsonBlockMatch[1].trim() : content.trim();
+    try {
+      const jsonMatch = jsonStr.match(/\{[\s\S]*\}/);
+      if (jsonMatch) {
+        const parsed = JSON.parse(jsonMatch[0]);
+        const thought = parsed.thought ?? "";
+        const action = parsed.action ?? "final_answer";
+        const actionInput = typeof parsed.action_input === "string" ? { answer: parsed.action_input } : parsed.action_input ?? {};
+        if (action === "final_answer") {
+          return {
+            type: "final",
+            thought,
+            action,
+            actionInput,
+            content: actionInput.answer ?? thought
+          };
+        }
+        return { type: "action", thought, action, actionInput, content: "" };
+      }
+    } catch {
+    }
+    return {
+      type: "final",
+      thought: content,
+      action: "final_answer",
+      actionInput: { answer: content },
+      content
+    };
+  }
+};
+
+exports.StructuredChatAgent = StructuredChatAgent;
+//# sourceMappingURL=chunk-LBU7ZGOB.cjs.map
+//# sourceMappingURL=chunk-LBU7ZGOB.cjs.map

package/dist/chunk-LBU7ZGOB.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/agent/structured-chat-agent.ts"],"names":["BaseAgent"],"mappings":";;;;;AASO,IAAM,mBAAA,GAAN,cAAkCA,2BAAA,CAAU;AAAA,EACzC,YAAA;AAAA,EAER,WAAA,CAAY,MAAA,EAAmC,GAAA,EAAiB,MAAA,EAAiB;AAC/E,IAAA,KAAA,CAAM,MAAA,EAAQ,GAAA,EAAK,MAAA,EAAQ,iBAAiB,CAAA;AAC5C,IAAA,IAAA,CAAK,eAAe,MAAA,CAAO,YAAA;AAAA,EAC7B;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,uBAAA,CAAwB,MAAA,CAAO,OAAO,CAAA;AAE1D,MAAA,IAAI,MAAA,CAAO,SAAS,OAAA,EAAS;AAC3B,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,cAAc,MAAM,IAAA,CAAK,YAAY,MAAA,CAAO,MAAA,EAAQ,OAAO,WAAW,CAAA;AAE5E,MAAA,KAAA,CAAM,IAAA,CAAK;AAAA,QACT,SAAS,MAAA,CAAO,OAAA;AAAA,QAChB,QAAQ,MAAA,CAAO,MAAA;AAAA,QACf,aAAa,MAAA,CAAO,WAAA;AAAA,QACpB,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,qCAAA,EAAwC,KAAA,CAAM,KAAA,CAAM,MAAA,GAAS,CAAC,CAAA,CAAE,WAAW,CAAA,CAAA,GAC3E,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,iBAAA,GAA4B;AAClC,IAAA,MAAM,WAAA,GAAc,CAAC,GAAG,IAAA,CAAK,MAAM,MAAA,EAAQ,CAAA,CAAE,GAAA,CAAI,CAAA,CAAA,KAAK;AACpD,MAAA,MAAM,MAAA,GAAkC;AAAA,QACtC,MAAM,CAAA,CAAE,IAAA;AAAA,QACR,aAAa,CAAA,CAAE;AAAA,OACjB;AACA,MAAA,IAAI,CAAA,CAAE,UAAA,IAAc,CAAA,CAAE,UAAA,CAAW,SAAS,CAAA,EAAG;AAC3C,QAAA,MAAA,CAAO,UAAA,GAAa;AAAA,UAClB,IAAA,EAAM,QAAA;AAAA,UACN,YAAY,MAAA,CAAO,WAAA;AAAA,YACjB,CAAA,CAAE,UAAA,CAAW,GAAA,CAAI,CAAA,CAAA,KAAK,CAAC,CAAA,CAAE,IAAA,EAAM,EAAE,IAAA,EAAM,EAAE,IAAA,EAAM,WAAA,EAAa,CAAA,CAAE,WAAA,EAAa,CAAC;AAAA,WAC9E;AAAA,UACA,QAAA,EAAU,CAAA,CAAE,UAAA,CAAW,MAAA,CAAO,CAAA,CAAA,KAAK,CAAA,CAAE,QAAQ,CAAA,CAAE,GAAA,CAAI,CAAA,CAAA,KAAK,CAAA,CAAE,IAAI;AAAA,SAChE;AAAA,MACF;AACA,MAAA,OAAO,MAAA;AAAA,IACT,CAAC,CAAA;AAED,IAAA,MAAM,eAAA,GAAkB,KAAK,YAAA,GACzB;AAAA;AAAA,EAA0D,KAAK,SAAA,CAAU,IAAA,CAAK,YAAA,EAAc,IAAA,EAAM,CAAC,CAAC;AAAA,CAAA,GACpG,EAAA;AAEJ,IAAA,MAAM,KAAA,GAAQ,KAAK,cAAA,EAAe;AAElC,IAAA,OAAO,CAAA,4CAAA,EAA+C,KAAK,IAAI;;AAAA;AAAA;AAAA,EAIjE,IAAA,CAAK,SAAA,CAAU,WAAA,EAAa,IAAA,EAAM,CAAC,CAAC;AAAA;;AAAA,EAGpC,KAAA,GAAQ,CAAA;AAAA,EAAW,KAAK;AAAA,CAAA,GAAO,EAAE,GAAG,eAAe;AAAA;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;AAAA;;AAAA,EAuBnD,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,OAAA,EAAY,KAAK;AAAA,CAAA;AAE9C,IAAA,KAAA,MAAW,QAAQ,KAAA,EAAO;AACxB,MAAA,MAAA,IAAU;AAAA,kBAAA,EAAuB,KAAK,MAAM,CAAA,CAAA,EAAI,KAAK,SAAA,CAAU,IAAA,CAAK,WAAW,CAAC,CAAA;AAAA,CAAA;AAChF,MAAA,MAAA,IAAU,CAAA,aAAA,EAAgB,KAAK,WAAW;AAAA,CAAA;AAAA,IAC5C;AAEA,IAAA,IAAI,KAAA,CAAM,SAAS,CAAA,EAAG;AACpB,MAAA,MAAA,IAAU,iEAAA;AAAA,IACZ;AAEA,IAAA,OAAO,MAAA;AAAA,EACT;AAAA,EAEQ,wBAAwB,OAAA,EAAuI;AAErK,IAAA,MAAM,cAAA,GAAiB,OAAA,CAAQ,KAAA,CAAM,uCAAuC,CAAA;AAC5E,IAAA,MAAM,OAAA,GAAU,iBAAiB,cAAA,CAAe,CAAC,EAAE,IAAA,EAAK,GAAI,QAAQ,IAAA,EAAK;AAEzE,IAAA,IAAI;AACF,MAAA,MAAM,SAAA,GAAY,OAAA,CAAQ,KAAA,CAAM,aAAa,CAAA;AAC7C,MAAA,IAAI,SAAA,EAAW;AACb,QAAA,MAAM,MAAA,GAAS,IAAA,CAAK,KAAA,CAAM,SAAA,CAAU,CAAC,CAAC,CAAA;AACtC,QAAA,MAAM,OAAA,GAAU,OAAO,OAAA,IAAW,EAAA;AAClC,QAAA,MAAM,MAAA,GAAS,OAAO,MAAA,IAAU,cAAA;AAChC,QAAA,MAAM,WAAA,GAAc,OAAO,MAAA,CAAO,YAAA,KAAiB,QAAA,GAC/C,EAAE,MAAA,EAAQ,MAAA,CAAO,YAAA,EAAa,GAC7B,MAAA,CAAO,YAAA,IAAgB,EAAC;AAE7B,QAAA,IAAI,WAAW,cAAA,EAAgB;AAC7B,UAAA,OAAO;AAAA,YACL,IAAA,EAAM,OAAA;AAAA,YACN,OAAA;AAAA,YACA,MAAA;AAAA,YACA,WAAA;AAAA,YACA,OAAA,EAAS,YAAY,MAAA,IAAU;AAAA,WACjC;AAAA,QACF;AAEA,QAAA,OAAO,EAAE,IAAA,EAAM,QAAA,EAAU,SAAS,MAAA,EAAQ,WAAA,EAAa,SAAS,EAAA,EAAG;AAAA,MACrE;AAAA,IACF,CAAA,CAAA,MAAQ;AAAA,IAER;AAEA,IAAA,OAAO;AAAA,MACL,IAAA,EAAM,OAAA;AAAA,MACN,OAAA,EAAS,OAAA;AAAA,MACT,MAAA,EAAQ,cAAA;AAAA,MACR,WAAA,EAAa,EAAE,MAAA,EAAQ,OAAA,EAAQ;AAAA,MAC/B;AAAA,KACF;AAAA,EACF;AACF","file":"chunk-LBU7ZGOB.cjs","sourcesContent":["import type { LLMAdapter } from '../types/index.js';\nimport type { Memory } from '../memory/memory.js';\nimport type {\n  StructuredChatAgentConfig,\n  AgentResult,\n  AgentStepResult,\n} from './types.js';\nimport { BaseAgent } from './base-agent.js';\n\nexport class StructuredChatAgent extends BaseAgent {\n  private outputSchema?: Record<string, unknown>;\n\n  constructor(config: StructuredChatAgentConfig, llm: LLMAdapter, memory?: Memory) {\n    super(config, llm, memory, 'structured-chat');\n    this.outputSchema = config.outputSchema;\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.parseStructuredResponse(result.content);\n\n      if (parsed.type === 'final') {\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 observation = await this.executeTool(parsed.action, parsed.actionInput);\n\n      steps.push({\n        thought: parsed.thought,\n        action: parsed.action,\n        actionInput: parsed.actionInput,\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 observation: ${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 buildSystemPrompt(): string {\n    const toolSchemas = [...this.tools.values()].map(t => {\n      const schema: Record<string, unknown> = {\n        name: t.name,\n        description: t.description,\n      };\n      if (t.parameters && t.parameters.length > 0) {\n        schema.parameters = {\n          type: 'object',\n          properties: Object.fromEntries(\n            t.parameters.map(p => [p.name, { type: p.type, description: p.description }])\n          ),\n          required: t.parameters.filter(p => p.required).map(p => p.name),\n        };\n      }\n      return schema;\n    });\n\n    const outputSchemaStr = this.outputSchema\n      ? `\\nYour final answer MUST conform to this JSON schema:\\n${JSON.stringify(this.outputSchema, null, 2)}\\n`\n      : '';\n\n    const rules = this.getRulesString();\n\n    return `You are a structured chat agent. Your goal: ${this.goal}\n\nYou have access to the following tools (as JSON schemas):\n\\`\\`\\`json\n${JSON.stringify(toolSchemas, null, 2)}\n\\`\\`\\`\n\n${rules ? `Rules:\\n${rules}\\n` : ''}${outputSchemaStr}\nRespond with a JSON blob in one of two formats:\n\nTo use a tool:\n\\`\\`\\`json\n{\n  \"thought\": \"your reasoning\",\n  \"action\": \"tool_name\",\n  \"action_input\": { \"param\": \"value\" }\n}\n\\`\\`\\`\n\nTo give your final answer:\n\\`\\`\\`json\n{\n  \"thought\": \"your final reasoning\",\n  \"action\": \"final_answer\",\n  \"action_input\": \"your comprehensive answer\"\n}\n\\`\\`\\`\n\nALWAYS respond with a JSON blob. Never respond with plain text.\n\n${this.systemPrompt ?? ''}`;\n  }\n\n  private buildPrompt(input: string, steps: AgentStepResult[], memoryContext: string): string {\n    let prompt = `${memoryContext}\\nHuman: ${input}\\n`;\n\n    for (const step of steps) {\n      prompt += `\\nAssistant action: ${step.action}(${JSON.stringify(step.actionInput)})\\n`;\n      prompt += `Observation: ${step.observation}\\n`;\n    }\n\n    if (steps.length > 0) {\n      prompt += '\\nContinue with the next action or provide your final answer:\\n';\n    }\n\n    return prompt;\n  }\n\n  private parseStructuredResponse(content: string): { type: 'action' | 'final'; thought: string; action: string; actionInput: Record<string, unknown>; content: string } {\n    // Try to extract JSON from code blocks\n    const jsonBlockMatch = content.match(/```(?:json)?\\s*\\n?([\\s\\S]*?)\\n?\\s*```/);\n    const jsonStr = jsonBlockMatch ? jsonBlockMatch[1].trim() : content.trim();\n\n    try {\n      const jsonMatch = jsonStr.match(/\\{[\\s\\S]*\\}/);\n      if (jsonMatch) {\n        const parsed = JSON.parse(jsonMatch[0]);\n        const thought = parsed.thought ?? '';\n        const action = parsed.action ?? 'final_answer';\n        const actionInput = typeof parsed.action_input === 'string'\n          ? { answer: parsed.action_input }\n          : (parsed.action_input ?? {});\n\n        if (action === 'final_answer') {\n          return {\n            type: 'final',\n            thought,\n            action,\n            actionInput,\n            content: actionInput.answer ?? thought,\n          };\n        }\n\n        return { type: 'action', thought, action, actionInput, content: '' };\n      }\n    } catch {\n      // Fall through to plain text\n    }\n\n    return {\n      type: 'final',\n      thought: content,\n      action: 'final_answer',\n      actionInput: { answer: content },\n      content,\n    };\n  }\n}\n"]}

package/dist/chunk-LDF6ZMRA.js

@@ -0,0 +1,72 @@
+// src/chains/retrieval-qa-chain.ts
+var RetrievalQAChain = class {
+  llm;
+  retriever;
+  collection;
+  systemPrompt;
+  returnSources;
+  maxSourceTokens;
+  constructor(options) {
+    this.llm = options.llm;
+    this.retriever = options.retriever;
+    this.collection = options.collection;
+    this.returnSources = options.returnSources ?? true;
+    this.maxSourceTokens = options.maxSourceTokens ?? 3e3;
+    this.systemPrompt = options.systemPrompt ?? "You are a helpful assistant. Answer the question based ONLY on the provided context. If the context does not contain enough information, say so clearly.";
+  }
+  async call(question) {
+    const steps = [];
+    const retrieveStart = Date.now();
+    const sources = await this.retriever.retrieve(question, this.collection);
+    steps.push({
+      name: "retrieve",
+      input: question,
+      output: `Found ${sources.length} relevant documents`,
+      latencyMs: Date.now() - retrieveStart
+    });
+    let context = "";
+    let tokenEstimate = 0;
+    const usedSources = [];
+    for (const source of sources) {
+      const text = source.content ?? "";
+      const estimatedTokens = Math.ceil(text.length / 4);
+      if (tokenEstimate + estimatedTokens > this.maxSourceTokens) break;
+      context += `---
+${text}
+
+`;
+      tokenEstimate += estimatedTokens;
+      usedSources.push(source);
+    }
+    const prompt = `Context:
+${context}
+
+Question: ${question}
+
+Answer:`;
+    const generateStart = Date.now();
+    const result = await this.llm.generate(prompt, {
+      systemPrompt: this.systemPrompt,
+      temperature: 0.3
+    });
+    steps.push({
+      name: "generate",
+      input: prompt.slice(0, 200) + "...",
+      output: result.content.slice(0, 200),
+      latencyMs: Date.now() - generateStart
+    });
+    return {
+      answer: result.content,
+      sources: this.returnSources ? usedSources : void 0,
+      intermediateSteps: steps,
+      usage: {
+        ...result.usage,
+        totalTokens: result.usage.totalTokens
+      }
+    };
+  }
+};
+
+export { RetrievalQAChain };
+//# sourceMappingURL=chunk-LDF6ZMRA.js.map
+//# sourceMappingURL=chunk-LDF6ZMRA.js.map

package/dist/chunk-LDF6ZMRA.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/chains/retrieval-qa-chain.ts"],"names":[],"mappings":";AAIO,IAAM,mBAAN,MAAuB;AAAA,EACpB,GAAA;AAAA,EACA,SAAA;AAAA,EACA,UAAA;AAAA,EACA,YAAA;AAAA,EACA,aAAA;AAAA,EACA,eAAA;AAAA,EAER,YAAY,OAAA,EAAkC;AAC5C,IAAA,IAAA,CAAK,MAAM,OAAA,CAAQ,GAAA;AACnB,IAAA,IAAA,CAAK,YAAY,OAAA,CAAQ,SAAA;AACzB,IAAA,IAAA,CAAK,aAAa,OAAA,CAAQ,UAAA;AAC1B,IAAA,IAAA,CAAK,aAAA,GAAgB,QAAQ,aAAA,IAAiB,IAAA;AAC9C,IAAA,IAAA,CAAK,eAAA,GAAkB,QAAQ,eAAA,IAAmB,GAAA;AAClD,IAAA,IAAA,CAAK,YAAA,GAAe,QAAQ,YAAA,IAC1B,0JAAA;AAAA,EACJ;AAAA,EAEA,MAAM,KAAK,QAAA,EAAwC;AACjD,IAAA,MAAM,QAA0C,EAAC;AAGjD,IAAA,MAAM,aAAA,GAAgB,KAAK,GAAA,EAAI;AAC/B,IAAA,MAAM,UAAU,MAAM,IAAA,CAAK,UAAU,QAAA,CAAS,QAAA,EAAU,KAAK,UAAU,CAAA;AACvE,IAAA,KAAA,CAAM,IAAA,CAAK;AAAA,MACT,IAAA,EAAM,UAAA;AAAA,MACN,KAAA,EAAO,QAAA;AAAA,MACP,MAAA,EAAQ,CAAA,MAAA,EAAS,OAAA,CAAQ,MAAM,CAAA,mBAAA,CAAA;AAAA,MAC/B,SAAA,EAAW,IAAA,CAAK,GAAA,EAAI,GAAI;AAAA,KACzB,CAAA;AAGD,IAAA,IAAI,OAAA,GAAU,EAAA;AACd,IAAA,IAAI,aAAA,GAAgB,CAAA;AACpB,IAAA,MAAM,cAAc,EAAC;AAErB,IAAA,KAAA,MAAW,UAAU,OAAA,EAAS;AAC5B,MAAA,MAAM,IAAA,GAAO,OAAO,OAAA,IAAW,EAAA;AAC/B,MAAA,MAAM,eAAA,GAAkB,IAAA,CAAK,IAAA,CAAK,IAAA,CAAK,SAAS,CAAC,CAAA;AACjD,MAAA,IAAI,aAAA,GAAgB,eAAA,GAAkB,IAAA,CAAK,eAAA,EAAiB;AAC5D,MAAA,OAAA,IAAW,CAAA;AAAA,EAAQ,IAAI;;AAAA,CAAA;AACvB,MAAA,aAAA,IAAiB,eAAA;AACjB,MAAA,WAAA,CAAY,KAAK,MAAM,CAAA;AAAA,IACzB;AAGA,IAAA,MAAM,MAAA,GAAS,CAAA;AAAA,EAAa,OAAO;;AAAA,UAAA,EAAiB,QAAQ;;AAAA,OAAA,CAAA;AAE5D,IAAA,MAAM,aAAA,GAAgB,KAAK,GAAA,EAAI;AAC/B,IAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,GAAA,CAAI,SAAS,MAAA,EAAQ;AAAA,MAC7C,cAAc,IAAA,CAAK,YAAA;AAAA,MACnB,WAAA,EAAa;AAAA,KACd,CAAA;AACD,IAAA,KAAA,CAAM,IAAA,CAAK;AAAA,MACT,IAAA,EAAM,UAAA;AAAA,MACN,KAAA,EAAO,MAAA,CAAO,KAAA,CAAM,CAAA,EAAG,GAAG,CAAA,GAAI,KAAA;AAAA,MAC9B,MAAA,EAAQ,MAAA,CAAO,OAAA,CAAQ,KAAA,CAAM,GAAG,GAAG,CAAA;AAAA,MACnC,SAAA,EAAW,IAAA,CAAK,GAAA,EAAI,GAAI;AAAA,KACzB,CAAA;AAED,IAAA,OAAO;AAAA,MACL,QAAQ,MAAA,CAAO,OAAA;AAAA,MACf,OAAA,EAAS,IAAA,CAAK,aAAA,GAAgB,WAAA,GAAc,MAAA;AAAA,MAC5C,iBAAA,EAAmB,KAAA;AAAA,MACnB,KAAA,EAAO;AAAA,QACL,GAAG,MAAA,CAAO,KAAA;AAAA,QACV,WAAA,EAAa,OAAO,KAAA,CAAM;AAAA;AAC5B,KACF;AAAA,EACF;AACF","file":"chunk-LDF6ZMRA.js","sourcesContent":["import type { Retriever } from '../retrievers/types.js';\nimport type { LLMAdapter } from '../types/index.js';\nimport type { ChainResult, RetrievalQAChainOptions } from './types.js';\n\nexport class RetrievalQAChain {\n  private llm: LLMAdapter;\n  private retriever: Retriever;\n  private collection: string;\n  private systemPrompt: string;\n  private returnSources: boolean;\n  private maxSourceTokens: number;\n\n  constructor(options: RetrievalQAChainOptions) {\n    this.llm = options.llm;\n    this.retriever = options.retriever;\n    this.collection = options.collection;\n    this.returnSources = options.returnSources ?? true;\n    this.maxSourceTokens = options.maxSourceTokens ?? 3000;\n    this.systemPrompt = options.systemPrompt ??\n      'You are a helpful assistant. Answer the question based ONLY on the provided context. If the context does not contain enough information, say so clearly.';\n  }\n\n  async call(question: string): Promise<ChainResult> {\n    const steps: ChainResult['intermediateSteps'] = [];\n\n    // Step 1: Retrieve relevant documents\n    const retrieveStart = Date.now();\n    const sources = await this.retriever.retrieve(question, this.collection);\n    steps.push({\n      name: 'retrieve',\n      input: question,\n      output: `Found ${sources.length} relevant documents`,\n      latencyMs: Date.now() - retrieveStart,\n    });\n\n    // Step 2: Build context from sources\n    let context = '';\n    let tokenEstimate = 0;\n    const usedSources = [];\n\n    for (const source of sources) {\n      const text = source.content ?? '';\n      const estimatedTokens = Math.ceil(text.length / 4);\n      if (tokenEstimate + estimatedTokens > this.maxSourceTokens) break;\n      context += `---\\n${text}\\n\\n`;\n      tokenEstimate += estimatedTokens;\n      usedSources.push(source);\n    }\n\n    // Step 3: Generate answer\n    const prompt = `Context:\\n${context}\\n\\nQuestion: ${question}\\n\\nAnswer:`;\n\n    const generateStart = Date.now();\n    const result = await this.llm.generate(prompt, {\n      systemPrompt: this.systemPrompt,\n      temperature: 0.3,\n    });\n    steps.push({\n      name: 'generate',\n      input: prompt.slice(0, 200) + '...',\n      output: result.content.slice(0, 200),\n      latencyMs: Date.now() - generateStart,\n    });\n\n    return {\n      answer: result.content,\n      sources: this.returnSources ? usedSources : undefined,\n      intermediateSteps: steps,\n      usage: {\n        ...result.usage,\n        totalTokens: result.usage.totalTokens,\n      },\n    };\n  }\n}\n"]}

package/dist/chunk-LGJVBZ7V.js

@@ -0,0 +1,59 @@
+// src/retrievers/contextual-compression-retriever.ts
+var ContextualCompressionRetriever = class {
+  options;
+  constructor(options) {
+    this.options = {
+      topK: 10,
+      maxCompressedLength: 500,
+      ...options
+    };
+  }
+  async retrieve(query, collection) {
+    const [embedding] = await this.options.llm.embed([query]);
+    const results = await this.options.vectorDB.search(collection, embedding, {
+      topK: this.options.topK,
+      minScore: this.options.minScore
+    });
+    const compressed = await this.compressResults(query, results);
+    return compressed.filter((r) => r.content && r.content.trim().length > 0);
+  }
+  async compressResults(query, results) {
+    const compressed = [];
+    for (const result of results) {
+      if (!result.content) {
+        compressed.push(result);
+        continue;
+      }
+      const prompt = `Given the following question and document, extract ONLY the parts of the document that are relevant to answering the question. If no part is relevant, respond with "NOT_RELEVANT".
+
+Question: ${query}
+
+Document:
+${result.content}
+
+Relevant extract:`;
+      const llmResult = await this.options.llm.generate(prompt, {
+        temperature: 0,
+        maxTokens: this.options.maxCompressedLength
+      });
+      const extract = llmResult.content.trim();
+      if (extract === "NOT_RELEVANT" || extract.length === 0) {
+        continue;
+      }
+      compressed.push({
+        ...result,
+        content: extract,
+        metadata: {
+          ...result.metadata,
+          originalLength: result.content.length,
+          compressedLength: extract.length
+        }
+      });
+    }
+    return compressed;
+  }
+};
+
+export { ContextualCompressionRetriever };
+//# sourceMappingURL=chunk-LGJVBZ7V.js.map
+//# sourceMappingURL=chunk-LGJVBZ7V.js.map

package/dist/chunk-LGJVBZ7V.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/retrievers/contextual-compression-retriever.ts"],"names":[],"mappings":";AAGO,IAAM,iCAAN,MAA0D;AAAA,EACvD,OAAA;AAAA,EAER,YAAY,OAAA,EAAgD;AAC1D,IAAA,IAAA,CAAK,OAAA,GAAU;AAAA,MACb,IAAA,EAAM,EAAA;AAAA,MACN,mBAAA,EAAqB,GAAA;AAAA,MACrB,GAAG;AAAA,KACL;AAAA,EACF;AAAA,EAEA,MAAM,QAAA,CAAS,KAAA,EAAe,UAAA,EAAmD;AAC/E,IAAA,MAAM,CAAC,SAAS,CAAA,GAAI,MAAM,IAAA,CAAK,QAAQ,GAAA,CAAI,KAAA,CAAM,CAAC,KAAK,CAAC,CAAA;AACxD,IAAA,MAAM,UAAU,MAAM,IAAA,CAAK,QAAQ,QAAA,CAAS,MAAA,CAAO,YAAY,SAAA,EAAW;AAAA,MACxE,IAAA,EAAM,KAAK,OAAA,CAAQ,IAAA;AAAA,MACnB,QAAA,EAAU,KAAK,OAAA,CAAQ;AAAA,KACxB,CAAA;AAED,IAAA,MAAM,UAAA,GAAa,MAAM,IAAA,CAAK,eAAA,CAAgB,OAAO,OAAO,CAAA;AAC5D,IAAA,OAAO,UAAA,CAAW,MAAA,CAAO,CAAA,CAAA,KAAK,CAAA,CAAE,OAAA,IAAW,EAAE,OAAA,CAAQ,IAAA,EAAK,CAAE,MAAA,GAAS,CAAC,CAAA;AAAA,EACxE;AAAA,EAEA,MAAc,eAAA,CAAgB,KAAA,EAAe,OAAA,EAA8D;AACzG,IAAA,MAAM,aAAmC,EAAC;AAE1C,IAAA,KAAA,MAAW,UAAU,OAAA,EAAS;AAC5B,MAAA,IAAI,CAAC,OAAO,OAAA,EAAS;AACnB,QAAA,UAAA,CAAW,KAAK,MAAM,CAAA;AACtB,QAAA;AAAA,MACF;AAEA,MAAA,MAAM,MAAA,GAAS,CAAA;;AAAA,UAAA,EAET,KAAK;;AAAA;AAAA,EAGf,OAAO,OAAO;;AAAA,iBAAA,CAAA;AAIV,MAAA,MAAM,YAAY,MAAM,IAAA,CAAK,OAAA,CAAQ,GAAA,CAAI,SAAS,MAAA,EAAQ;AAAA,QACxD,WAAA,EAAa,CAAA;AAAA,QACb,SAAA,EAAW,KAAK,OAAA,CAAQ;AAAA,OACzB,CAAA;AAED,MAAA,MAAM,OAAA,GAAU,SAAA,CAAU,OAAA,CAAQ,IAAA,EAAK;AAEvC,MAAA,IAAI,OAAA,KAAY,cAAA,IAAkB,OAAA,CAAQ,MAAA,KAAW,CAAA,EAAG;AACtD,QAAA;AAAA,MACF;AAEA,MAAA,UAAA,CAAW,IAAA,CAAK;AAAA,QACd,GAAG,MAAA;AAAA,QACH,OAAA,EAAS,OAAA;AAAA,QACT,QAAA,EAAU;AAAA,UACR,GAAG,MAAA,CAAO,QAAA;AAAA,UACV,cAAA,EAAgB,OAAO,OAAA,CAAQ,MAAA;AAAA,UAC/B,kBAAkB,OAAA,CAAQ;AAAA;AAC5B,OACD,CAAA;AAAA,IACH;AAEA,IAAA,OAAO,UAAA;AAAA,EACT;AACF","file":"chunk-LGJVBZ7V.js","sourcesContent":["import type { VectorSearchResult } from '../types/index.js';\nimport type { Retriever, ContextualCompressionRetrieverOptions } from './types.js';\n\nexport class ContextualCompressionRetriever implements Retriever {\n  private options: Required<Pick<ContextualCompressionRetrieverOptions, 'llm' | 'vectorDB'>> & ContextualCompressionRetrieverOptions;\n\n  constructor(options: ContextualCompressionRetrieverOptions) {\n    this.options = {\n      topK: 10,\n      maxCompressedLength: 500,\n      ...options,\n    };\n  }\n\n  async retrieve(query: string, collection: string): Promise<VectorSearchResult[]> {\n    const [embedding] = await this.options.llm.embed([query]);\n    const results = await this.options.vectorDB.search(collection, embedding, {\n      topK: this.options.topK,\n      minScore: this.options.minScore,\n    });\n\n    const compressed = await this.compressResults(query, results);\n    return compressed.filter(r => r.content && r.content.trim().length > 0);\n  }\n\n  private async compressResults(query: string, results: VectorSearchResult[]): Promise<VectorSearchResult[]> {\n    const compressed: VectorSearchResult[] = [];\n\n    for (const result of results) {\n      if (!result.content) {\n        compressed.push(result);\n        continue;\n      }\n\n      const prompt = `Given the following question and document, extract ONLY the parts of the document that are relevant to answering the question. If no part is relevant, respond with \"NOT_RELEVANT\".\n\nQuestion: ${query}\n\nDocument:\n${result.content}\n\nRelevant extract:`;\n\n      const llmResult = await this.options.llm.generate(prompt, {\n        temperature: 0,\n        maxTokens: this.options.maxCompressedLength,\n      });\n\n      const extract = llmResult.content.trim();\n\n      if (extract === 'NOT_RELEVANT' || extract.length === 0) {\n        continue;\n      }\n\n      compressed.push({\n        ...result,\n        content: extract,\n        metadata: {\n          ...result.metadata,\n          originalLength: result.content.length,\n          compressedLength: extract.length,\n        },\n      });\n    }\n\n    return compressed;\n  }\n}\n"]}