chat-agent-toolkit 1.2.11 → 1.2.13

package/dist/tools/search/doc-utils.d.ts

@@ -7,5 +7,20 @@ export interface R2CredentialsInput {
 }
 export declare function buildFallbackDocs(query: string): Document[];
 export declare function normalizeSourcesOutput(output: unknown, query: string): Document[];
+/**
+ * Resolves an uploaded fileId to its extracted `{ title, content }` payload.
+ * Hosts register a loader (e.g. reading the native R2 binding) so the search
+ * pipeline does not depend on filesystem or credential-based access.
+ */
+export type UploadFileLoader = (fileId: string) => Promise<{
+    title: string;
+    content: string;
+} | null>;
+/**
+ * Registers the loader used by {@link rerankDocs} to resolve uploaded
+ * fileIds to extracted content. The registered loader takes precedence over
+ * the S3-credentials fallback.
+ */
+export declare function registerUploadFileLoader(loader: UploadFileLoader): void;
 export declare function rerankDocs(query: string, docs: Document[], fileIds: string[], optimizationMode: "speed" | "balanced" | "quality", r2Credentials?: R2CredentialsInput): Promise<Document[]>;
 export declare function processDocs(docs: Document[]): string;

package/dist/tools/search/index.d.ts

@@ -9,4 +9,5 @@ export { default as generateSuggestions } from './suggestionGeneratorAgent';
 export { groupAndSummarizeDocs } from './link-summarizer';
 export type { Document } from './document';
 export { splitTextIntoChunks } from './document';
-export { buildFallbackDocs, rerankDocs, processDocs, normalizeSourcesOutput } from './doc-utils';
+export { buildFallbackDocs, rerankDocs, processDocs, normalizeSourcesOutput, registerUploadFileLoader } from './doc-utils';
+export type { UploadFileLoader } from './doc-utils';

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "chat-agent-toolkit",
-  "version": "1.2.11",
+  "version": "1.2.13",
   "description": "Multi-provider AI agent toolkit: generate language responses, search the web, extract content, and manage memory across 10+ LLM providers.",
   "author": "vtempest <grokthiscontact@gmail.com>",
   "license": "AGPL-3.0",

package/src/tools/search/doc-utils.ts

@@ -36,6 +36,26 @@ export function normalizeSourcesOutput(output: unknown, query: string): Document
   return buildFallbackDocs(query);
 }
 
+/**
+ * Resolves an uploaded fileId to its extracted `{ title, content }` payload.
+ * Hosts register a loader (e.g. reading the native R2 binding) so the search
+ * pipeline does not depend on filesystem or credential-based access.
+ */
+export type UploadFileLoader = (
+  fileId: string,
+) => Promise<{ title: string; content: string } | null>;
+
+let uploadFileLoader: UploadFileLoader | null = null;
+
+/**
+ * Registers the loader used by {@link rerankDocs} to resolve uploaded
+ * fileIds to extracted content. The registered loader takes precedence over
+ * the S3-credentials fallback.
+ */
+export function registerUploadFileLoader(loader: UploadFileLoader): void {
+  uploadFileLoader = loader;
+}
+
 async function downloadExtractedContent(fileId: string, r2Credentials: R2CredentialsInput): Promise<{ title: string; content: string } | null> {
   try {
     const { manageStorage } = await import("manage-storage");
@@ -69,9 +89,25 @@ export async function rerankDocs(
 
   let filesData: { title: string; content: string }[] = [];
 
-  if (fileIds.length > 0 && r2Credentials) {
+  if (fileIds.length > 0) {
     const results = await Promise.all(
-      fileIds.map((fileId) => downloadExtractedContent(fileId, r2Credentials))
+      fileIds.map(async (fileId) => {
+        if (uploadFileLoader) {
+          try {
+            const loaded = await uploadFileLoader(fileId);
+            if (loaded) return loaded;
+          } catch (error) {
+            console.error(
+              `[rerankDocs] Registered upload loader failed for fileId ${fileId}:`,
+              error,
+            );
+          }
+        }
+        if (r2Credentials) {
+          return downloadExtractedContent(fileId, r2Credentials);
+        }
+        return null;
+      }),
     );
     filesData = results.filter((r) => r !== null);
   }

package/src/tools/search/index.ts

@@ -10,4 +10,5 @@ export { default as generateSuggestions } from "./suggestionGeneratorAgent";
 export { groupAndSummarizeDocs } from "./link-summarizer";
 export type { Document } from "./document";
 export { splitTextIntoChunks } from "./document";
-export { buildFallbackDocs, rerankDocs, processDocs, normalizeSourcesOutput } from "./doc-utils";
+export { buildFallbackDocs, rerankDocs, processDocs, normalizeSourcesOutput, registerUploadFileLoader } from "./doc-utils";
+export type { UploadFileLoader } from "./doc-utils";