@absolutejs/rag 0.0.11 → 0.0.12

package/dist/ai/rag/index.js

@@ -21714,6 +21714,7 @@ var MAX_INGEST_JOBS = 20;
 var MAX_ADMIN_ACTIONS = 20;
 var MAX_ADMIN_JOBS = 20;
 var DEFAULT_STALE_AFTER_MS = 1000 * 60 * 60 * 24 * 7;
+var WEBSOCKET_OPEN = 1;
 var REQUEST_USER_SUB_HEADER = "x-absolutejs-user-sub";
 var HTML_HEADERS = { "Content-Type": "text/html; charset=utf-8" };
 var defaultParseProvider = (content) => {
@@ -22157,9 +22158,23 @@ var persistSearchTraceIfConfigured = async (input) => {
     await input.onPrune?.(input.retention);
   }
 };
-var buildRAGContextFromQuery = async (config, topK, scoreThreshold, queryText, ragModel, embedding, embeddingModel) => {
+var sendWebSocketError = (ws, error, conversationId, messageId) => {
+  if (typeof ws.readyState === "number" && ws.readyState !== WEBSOCKET_OPEN) {
+    return;
+  }
+  try {
+    ws.send(JSON.stringify({
+      conversationId,
+      message: error instanceof Error ? error.message : String(error),
+      messageId,
+      type: "error"
+    }));
+  } catch {}
+};
+var getProviderDefaultEmbeddingModel = (embedding) => embedding !== undefined && typeof embedding === "object" && typeof embedding.defaultModel === "string" ? embedding.defaultModel : undefined;
+var buildRAGContextFromQuery = async (config, topK, scoreThreshold, queryText, embedding, embeddingModel) => {
   const collection = config.collection ?? (config.ragStore ? createRAGCollection({
-    defaultModel: embeddingModel ?? ragModel,
+    defaultModel: embeddingModel,
     defaultTopK: topK,
     embedding,
     rerank: config.rerank,
@@ -22172,7 +22187,7 @@ var buildRAGContextFromQuery = async (config, topK, scoreThreshold, queryText, r
     };
   }
   const queried = await collection.searchWithTrace({
-    model: embeddingModel ?? ragModel,
+    model: embeddingModel,
     query: queryText,
     scoreThreshold,
     topK
@@ -22195,7 +22210,27 @@ var buildRAGContextFromQuery = async (config, topK, scoreThreshold, queryText, r
     trace: queried.trace
   };
 };
+var assertEmbeddingModelConfigured = (config) => {
+  if (config.collection !== undefined) {
+    return;
+  }
+  if (config.ragStore === undefined) {
+    return;
+  }
+  if (config.embeddingModel !== undefined) {
+    return;
+  }
+  const { embedding } = config;
+  if (embedding === undefined || typeof embedding !== "object") {
+    return;
+  }
+  if (getProviderDefaultEmbeddingModel(embedding) !== undefined) {
+    return;
+  }
+  throw new Error("ragChat: an embedding provider is configured but no embedding model is set. " + "Pass `embeddingModel` to ragChat() or set `defaultModel` on the embedding provider. " + "The chat model cannot be used for embeddings.");
+};
 var ragChat = (config) => {
+  assertEmbeddingModelConfigured(config);
   const path = config.path ?? DEFAULT_PATH;
   const authorizeRAGAction = config.authorizeRAGAction;
   const resolveRAGAccessScope = config.resolveRAGAccessScope;
@@ -22657,53 +22692,57 @@ var ragChat = (config) => {
     }));
   };
   const handleMessage = async (ws, rawContent, rawConversationId, rawAttachments) => {
-    const parsed = parseProvider(rawContent);
-    const { content, providerName } = parsed;
-    const userMessageId = generateId2();
     const assistantMessageId = generateId2();
     const conversationId = rawConversationId ?? generateId2();
-    const conversation = await store.getOrCreate(conversationId);
-    const history = buildHistory(conversation);
-    const model = resolveModel(config, parsed);
-    const ragModel = parsed.model ?? model;
-    appendMessage(conversation, {
-      attachments: rawAttachments,
-      content,
-      conversationId,
-      id: userMessageId,
-      role: "user",
-      timestamp: Date.now()
-    });
-    await store.set(conversationId, conversation);
-    const retrievalStartedAt = Date.now();
-    handleRAGRetrieving(ws, conversationId, assistantMessageId, retrievalStartedAt);
-    const provider = config.provider(providerName);
-    const rag = await buildRAGContextFromQuery(config, topK, scoreThreshold, content, ragModel, config.embedding, config.embeddingModel);
-    const controller = new AbortController;
-    abortControllers.set(conversationId, controller);
-    const { ragContext, sources, trace } = rag;
-    const retrievedAt = Date.now();
-    const retrievalDurationMs = retrievedAt - retrievalStartedAt;
-    handleRAGRetrieved(ws, conversationId, assistantMessageId, sources, retrievalStartedAt, retrievedAt, retrievalDurationMs, trace);
-    await streamAI(ws, conversationId, assistantMessageId, {
-      completeMeta: includeCompleteSources ? { sources } : undefined,
-      maxTurns: config.maxTurns,
-      messages: [
-        ...history,
-        buildUserMessage(content, rawAttachments, ragContext)
-      ],
-      model,
-      provider,
-      signal: controller.signal,
-      systemPrompt: config.systemPrompt,
-      thinking: resolveThinking(config, providerName, model),
-      tools: resolveTools(config, providerName, model),
-      onComplete: async (fullResponse, usage) => {
-        await appendAssistantMessage(conversationId, assistantMessageId, fullResponse, sources, usage, model, retrievalStartedAt, retrievedAt, retrievalDurationMs, trace);
-        abortControllers.delete(conversationId);
-        config.onComplete?.(conversationId, fullResponse, usage, sources);
-      }
-    });
+    try {
+      const parsed = parseProvider(rawContent);
+      const { content, providerName } = parsed;
+      const userMessageId = generateId2();
+      const conversation = await store.getOrCreate(conversationId);
+      const history = buildHistory(conversation);
+      const model = resolveModel(config, parsed);
+      appendMessage(conversation, {
+        attachments: rawAttachments,
+        content,
+        conversationId,
+        id: userMessageId,
+        role: "user",
+        timestamp: Date.now()
+      });
+      await store.set(conversationId, conversation);
+      const retrievalStartedAt = Date.now();
+      handleRAGRetrieving(ws, conversationId, assistantMessageId, retrievalStartedAt);
+      const provider = config.provider(providerName);
+      const rag = await buildRAGContextFromQuery(config, topK, scoreThreshold, content, config.embedding, config.embeddingModel);
+      const controller = new AbortController;
+      abortControllers.set(conversationId, controller);
+      const { ragContext, sources, trace } = rag;
+      const retrievedAt = Date.now();
+      const retrievalDurationMs = retrievedAt - retrievalStartedAt;
+      handleRAGRetrieved(ws, conversationId, assistantMessageId, sources, retrievalStartedAt, retrievedAt, retrievalDurationMs, trace);
+      await streamAI(ws, conversationId, assistantMessageId, {
+        completeMeta: includeCompleteSources ? { sources } : undefined,
+        maxTurns: config.maxTurns,
+        messages: [
+          ...history,
+          buildUserMessage(content, rawAttachments, ragContext)
+        ],
+        model,
+        provider,
+        signal: controller.signal,
+        systemPrompt: config.systemPrompt,
+        thinking: resolveThinking(config, providerName, model),
+        tools: resolveTools(config, providerName, model),
+        onComplete: async (fullResponse, usage) => {
+          await appendAssistantMessage(conversationId, assistantMessageId, fullResponse, sources, usage, model, retrievalStartedAt, retrievedAt, retrievalDurationMs, trace);
+          abortControllers.delete(conversationId);
+          config.onComplete?.(conversationId, fullResponse, usage, sources);
+        }
+      });
+    } catch (error) {
+      abortControllers.delete(conversationId);
+      sendWebSocketError(ws, error, conversationId, assistantMessageId);
+    }
   };
   const resolveCollection = () => config.collection ?? (ragStore ? createRAGCollection({
     defaultModel: config.embeddingModel,
@@ -29422,7 +29461,6 @@ var ragChat = (config) => {
       const parsed = parseProvider(lastMessage.content);
       const { content, providerName } = parsed;
       const model = resolveModel(config, parsed);
-      const ragModel = parsed.model ?? model;
       const assistantMessageId = generateId2();
       const retrievalStartedAt = Date.now();
       yield {
@@ -29434,7 +29472,7 @@ var ragChat = (config) => {
         event: "retrieval"
       };
       const provider = config.provider(providerName);
-      const { ragContext, sources, trace } = await buildRAGContextFromQuery(config, topK, scoreThreshold, content, ragModel, config.embedding, config.embeddingModel);
+      const { ragContext, sources, trace } = await buildRAGContextFromQuery(config, topK, scoreThreshold, content, config.embedding, config.embeddingModel);
       const retrievedAt = Date.now();
       const retrievalDurationMs = retrievedAt - retrievalStartedAt;
       yield {
@@ -29485,16 +29523,20 @@ var ragChat = (config) => {
       if (!message) {
         return;
       }
-      if (message.type === "cancel") {
-        handleCancel(message.conversationId);
-        return;
-      }
-      if (message.type === "branch") {
-        await handleBranch(ws, message.messageId, message.conversationId);
-        return;
-      }
-      if (message.type === "message") {
-        await handleMessage(ws, message.content, message.conversationId, message.attachments);
+      try {
+        if (message.type === "cancel") {
+          handleCancel(message.conversationId);
+          return;
+        }
+        if (message.type === "branch") {
+          await handleBranch(ws, message.messageId, message.conversationId);
+          return;
+        }
+        if (message.type === "message") {
+          await handleMessage(ws, message.content, message.conversationId, message.attachments);
+        }
+      } catch (error) {
+        sendWebSocketError(ws, error, "conversationId" in message ? message.conversationId : undefined);
       }
     }
   }).post(`${path}/search`, async ({ body, request, set }) => {
@@ -31977,7 +32019,9 @@ var toErrorMessage3 = async (response, label) => {
   return new Error(`${label}: ${response.status} ${response.statusText}${detailMessage ? ` (${detailMessage})` : ""}`);
 };
 var normalizeString = (value) => typeof value === "string" && value.trim().length > 0 ? value.trim() : undefined;
-var toUniqueStringArray = (values) => [...new Set(values.filter((value) => typeof value === "string" && value.length > 0))];
+var toUniqueStringArray = (values) => [
+  ...new Set(values.filter((value) => typeof value === "string" && value.length > 0))
+];
 var toContactTitle = (person, fallbackId) => {
   const primaryName = person.names?.find((name) => normalizeString(name.displayName) !== undefined);
   const displayName = normalizeString(primaryName?.displayName);
@@ -37605,5 +37649,5 @@ export {
   addRAGEvaluationSuiteCase
 };
 
-//# debugId=20E465BEF0B96FD164756E2164756E21
+//# debugId=3E3026F16C216BA364756E2164756E21
 //# sourceMappingURL=index.js.map