@absolutejs/absolute 0.19.0-beta.432 → 0.19.0-beta.434

package/dist/ai/index.js

@@ -2912,6 +2912,95 @@ var ragChat = (config) => {
     return { ok: true };
   }).use(htmxRoutes());
 };
+// src/ai/rag/presentation.ts
+var buildSourceGroupKey = (source) => source.source ?? source.title ?? source.chunkId;
+var buildSourceLabel = (source) => source.source ?? source.title ?? source.chunkId;
+var getLatestAssistantMessage = (messages) => {
+  for (let index = messages.length - 1;index >= 0; index -= 1) {
+    const message = messages[index];
+    if (message?.role === "assistant") {
+      return message;
+    }
+  }
+  return;
+};
+var getLatestRAGSources = (messages) => getLatestAssistantMessage(messages)?.sources ?? [];
+var buildRAGSourceGroups = (sources) => {
+  const groups = new Map;
+  for (const source of sources) {
+    const key = buildSourceGroupKey(source);
+    const existing = groups.get(key);
+    if (existing) {
+      existing.bestScore = Math.max(existing.bestScore, source.score);
+      existing.count += 1;
+      existing.chunks.push(source);
+      continue;
+    }
+    groups.set(key, {
+      bestScore: source.score,
+      chunks: [source],
+      count: 1,
+      key,
+      label: buildSourceLabel(source),
+      source: source.source,
+      title: source.title
+    });
+  }
+  return [...groups.values()].sort((left, right) => {
+    if (right.bestScore !== left.bestScore) {
+      return right.bestScore - left.bestScore;
+    }
+    return left.label.localeCompare(right.label);
+  });
+};
+var buildRAGCitations = (sources) => {
+  const unique = new Map;
+  for (const source of sources) {
+    const key = source.chunkId;
+    const existing = unique.get(key);
+    if (existing && existing.score >= source.score) {
+      continue;
+    }
+    unique.set(key, {
+      chunkId: source.chunkId,
+      key,
+      label: buildSourceLabel(source),
+      metadata: source.metadata,
+      score: source.score,
+      source: source.source,
+      text: source.text,
+      title: source.title
+    });
+  }
+  return [...unique.values()].sort((left, right) => {
+    if (right.score !== left.score) {
+      return right.score - left.score;
+    }
+    return left.label.localeCompare(right.label);
+  });
+};
+var resolveRAGStreamStage = ({
+  error,
+  isStreaming,
+  messages
+}) => {
+  if (error) {
+    return "error";
+  }
+  const assistantMessage = getLatestAssistantMessage(messages);
+  if (!assistantMessage) {
+    return isStreaming ? "submitting" : "idle";
+  }
+  if (!isStreaming) {
+    return "complete";
+  }
+  const hasSources = (assistantMessage.sources?.length ?? 0) > 0;
+  const hasContent = assistantMessage.content.trim().length > 0 || assistantMessage.thinking?.trim().length || (assistantMessage.toolCalls?.length ?? 0) > 0 || (assistantMessage.images?.length ?? 0) > 0;
+  if (hasSources && !hasContent) {
+    return "retrieved";
+  }
+  return "streaming";
+};
 // src/ai/rag/adapters/utils.ts
 var vectorDimensionDefault = 24;
 var normalizeVector = (vector) => {
@@ -3779,6 +3868,7 @@ export {
   streamAI,
   serializeAIMessage,
   searchDocuments,
+  resolveRAGStreamStage,
   resolveAbsoluteSQLiteVecExtensionPath,
   resolveAbsoluteSQLiteVec,
   ragChat as ragPlugin,
@@ -3800,6 +3890,8 @@ export {
   ingestRAGDocuments,
   ingestDocuments,
   google,
+  getLatestRAGSources,
+  getLatestAssistantMessage,
   generateId,
   gemini,
   deepseek,
@@ -3812,9 +3904,11 @@ export {
   createConversationManager,
   buildRAGUpsertInputFromDocuments,
   buildRAGUpsertInputFromDirectory,
+  buildRAGSourceGroups,
+  buildRAGCitations,
   alibaba,
   aiChat
 };
 
-//# debugId=9FA514D5699A09E064756E2164756E21
+//# debugId=65B35A7CB744467564756E2164756E21
 //# sourceMappingURL=index.js.map