@absolutejs/rag 0.0.26 → 0.0.28

package/dist/adapter-kit/index.js

@@ -20813,7 +20813,10 @@ var createRAGCollection = (options) => {
   const embeddingProvider = resolveRAGEmbeddingProvider(options.embedding, options.store.embed, options.defaultModel);
   const getExpectedDimensions = () => embeddingProvider.dimensions ?? getStatus?.()?.dimensions;
   const embed = async (input, context) => {
-    const vector = await embeddingProvider.embed(input);
+    const vector = await embeddingProvider.embed({
+      ...input,
+      kind: input.kind ?? (context === "query" ? "query" : "passage")
+    });
     validateRAGEmbeddingDimensions(vector, getExpectedDimensions(), context);
     return vector;
   };
@@ -21273,6 +21276,115 @@ var createRAGCollection = (options) => {
     }))).flat();
     await options.store.upsert({ chunks });
   };
+  const buildSourceUpsertInput = async (sourceId, input) => {
+    const sharedMetadata = input.metadata;
+    if (input.document) {
+      return buildRAGUpsertInputFromDocuments({
+        chunkingRegistry: input.chunkingRegistry,
+        defaultChunking: input.chunking,
+        documents: [
+          {
+            ...input.document,
+            chunking: input.document.chunking ?? input.chunking,
+            metadata: { ...input.document.metadata ?? {}, ...sharedMetadata },
+            source: input.document.source ?? sourceId
+          }
+        ]
+      });
+    }
+    if (input.upload) {
+      return buildRAGUpsertInputFromUploads({
+        chunkingRegistry: input.chunkingRegistry,
+        extractorRegistry: input.extractorRegistry,
+        extractors: input.extractors,
+        uploads: [
+          {
+            ...input.upload,
+            chunking: input.upload.chunking ?? input.chunking,
+            metadata: { ...input.upload.metadata ?? {}, ...sharedMetadata },
+            source: input.upload.source ?? sourceId
+          }
+        ]
+      });
+    }
+    if (input.url) {
+      return buildRAGUpsertInputFromURLs({
+        chunkingRegistry: input.chunkingRegistry,
+        extractorRegistry: input.extractorRegistry,
+        extractors: input.extractors,
+        urls: [
+          {
+            ...input.url,
+            chunking: input.url.chunking ?? input.chunking,
+            extractorRegistry: input.url.extractorRegistry ?? input.extractorRegistry,
+            extractors: input.url.extractors ?? input.extractors,
+            metadata: { ...input.url.metadata ?? {}, ...sharedMetadata },
+            source: input.url.source ?? sourceId
+          }
+        ]
+      });
+    }
+    throw new Error("ingestSource requires one of upload, url, or document to be provided.");
+  };
+  const removeSource = async (input) => {
+    const sourceId = input.sourceId?.trim();
+    if (!sourceId) {
+      throw new Error("removeSource requires a non-empty sourceId.");
+    }
+    const deleteFromStore = options.store.delete;
+    if (typeof deleteFromStore !== "function") {
+      throw new Error("removeSource requires a store that implements delete().");
+    }
+    let deleted = 0;
+    const chunkIds = input.chunkIds ?? (typeof input.chunkCount === "number" && input.chunkCount > 0 ? Array.from({ length: input.chunkCount }, (_unused, index) => `${sourceId}#${index}`) : undefined);
+    if (chunkIds && chunkIds.length > 0) {
+      deleted += await deleteFromStore({ chunkIds });
+    }
+    if (input.filterDelete !== false) {
+      try {
+        deleted += await deleteFromStore({ filter: { sourceId } });
+      } catch {}
+    }
+    return { deleted, sourceId };
+  };
+  const ingestSource = async (input) => {
+    const sourceId = input.sourceId?.trim();
+    if (!sourceId) {
+      throw new Error("ingestSource requires a non-empty sourceId.");
+    }
+    if (input.replace !== false) {
+      await removeSource({
+        chunkCount: input.previousChunkCount,
+        sourceId
+      });
+    }
+    const built = await buildSourceUpsertInput(sourceId, input);
+    const embedKind = input.embedKind ?? "passage";
+    const chunks = await Promise.all(built.chunks.map(async (chunk, index) => {
+      const chunkId = `${sourceId}#${index}`;
+      const embedding = chunk.embedding ?? await embed({
+        kind: embedKind,
+        model: options.defaultModel,
+        text: chunk.text
+      }, "chunk");
+      return {
+        ...chunk,
+        chunkId,
+        embedding,
+        metadata: {
+          ...chunk.metadata ?? {},
+          sourceId
+        },
+        source: chunk.source ?? sourceId
+      };
+    }));
+    await ingest({ chunks });
+    return {
+      chunkCount: chunks.length,
+      chunkIds: chunks.map((chunk) => chunk.chunkId),
+      sourceId
+    };
+  };
   return {
     clear: typeof options.store.clear === "function" ? () => options.store.clear?.() : undefined,
     getCapabilities: typeof getCapabilities === "function" ? () => getCapabilities() : undefined,
@@ -21280,12 +21392,16 @@ var createRAGCollection = (options) => {
     searchWithTrace,
     search,
     store: options.store,
-    ingest
+    ingest,
+    ingestSource,
+    removeSource
   };
 };
 var ingestDocuments = async (collection, input) => collection.ingest(input);
 var ingestRAGDocuments = async (collection, input) => collection.ingest(buildRAGUpsertInputFromDocuments(input));
 var searchDocuments = async (collection, input) => collection.search(input);
+var ingestRAGSource = async (collection, input) => collection.ingestSource(input);
+var removeRAGSource = async (collection, input) => collection.removeSource(input);
 // src/chat/chat.ts
 import { Elysia, sse } from "elysia";
 import { createMemoryStore } from "@absolutejs/ai";
@@ -22231,7 +22347,7 @@ var toAssistantTextBlock = (content) => [
   { content, type: "text" }
 ];
 var resolveTools = (config, providerName, model) => typeof config.tools === "function" ? config.tools(providerName, model) : config.tools;
-var resolveThinking = (config, providerName, model) => typeof config.thinking === "function" ? config.thinking(providerName, model) : config.thinking;
+var resolveReasoning = (config, providerName, model) => typeof config.reasoning === "function" ? config.reasoning(providerName, model) : config.reasoning;
 var resolveModel = (config, parsed) => {
   if (parsed.model) {
     return parsed.model;
@@ -22895,7 +23011,7 @@ var ragChat = (config) => {
         provider,
         signal: controller.signal,
         systemPrompt: config.systemPrompt,
-        thinking: resolveThinking(config, providerName, model),
+        reasoning: resolveReasoning(config, providerName, model),
         tools: resolveTools(config, providerName, model),
         onComplete: async (fullResponse, usage) => {
           await appendAssistantMessage(conversationId, assistantMessageId, fullResponse, sources, usage, model, retrievalStartedAt, retrievedAt, retrievalDurationMs, trace);
@@ -29668,7 +29784,7 @@ var ragChat = (config) => {
         provider,
         signal: controller.signal,
         systemPrompt: config.systemPrompt,
-        thinking: resolveThinking(config, providerName, model),
+        reasoning: resolveReasoning(config, providerName, model),
         tools: resolveTools(config, providerName, model),
         onComplete: async (fullResponse, usage) => {
           await appendAssistantMessage(conversationId, assistantMessageId, fullResponse, sources, usage, model, retrievalStartedAt, retrievedAt, retrievalDurationMs);
@@ -31176,5 +31292,5 @@ export {
   RAG_NATIVE_QUERY_CANDIDATE_LIMIT
 };
 
-//# debugId=EBDDF31AF25C9C1C64756E2164756E21
+//# debugId=545F8D8FE97848CE64756E2164756E21
 //# sourceMappingURL=index.js.map