@arbidocs/sdk 0.3.13 → 0.3.14

package/dist/index.cjs

@@ -514,11 +514,13 @@ async function streamSSE(response, callbacks = {}) {
   let text = "";
   let assistantMessageExtId = null;
   const agentSteps = [];
+  let toolCallCount = 0;
   const errors = [];
   const artifacts = [];
   let userMessage = null;
   let metadata = null;
   let usage = null;
+  let context = null;
   const eventHandlers = {
     // OpenAI Responses API events (dot-separated names from server)
     "response.created": (raw) => {
@@ -562,6 +564,9 @@ async function streamSSE(response, callbacks = {}) {
         metadata = meta;
         callbacks.onMetadata?.(meta);
       }
+      if (data.context) {
+        context = data.context;
+      }
       if (data.t != null) callbacks.onElapsedTime?.(data.t);
       callbacks.onComplete?.();
     },
@@ -576,6 +581,10 @@ async function streamSSE(response, callbacks = {}) {
       const data = JSON.parse(raw);
       const label = formatAgentStepLabel(data);
       if (label) agentSteps.push(label);
+      const detail = data.detail;
+      if (detail && Array.isArray(detail)) {
+        toolCallCount += detail.filter((d) => d.tool).length;
+      }
       callbacks.onAgentStep?.(data);
       if (data.t != null) callbacks.onElapsedTime?.(data.t);
     },
@@ -626,11 +635,13 @@ async function streamSSE(response, callbacks = {}) {
     text,
     assistantMessageExtId,
     agentSteps,
+    toolCallCount,
     errors,
     userMessage,
     metadata,
     artifacts,
-    usage
+    usage,
+    context
   };
 }
 var consumeSSEStream = streamSSE;
@@ -829,6 +840,13 @@ function formatWsMessage(msg) {
   if (client.isMessageType(msg, "presence_update")) {
     return { text: `${msg.user_id} is ${msg.status}`, level: "info" };
   }
+  if (client.isMessageType(msg, "response_complete")) {
+    const icon = msg.status === "completed" ? "\u2713" : "\u2717";
+    return {
+      text: `${icon} Task ${msg.response_id} ${msg.status}`,
+      level: msg.status === "completed" ? "success" : "error"
+    };
+  }
   if (isNotification(msg)) {
     const sender = msg.sender?.email || "someone";
     const content = msg.content ?? msg.type;
@@ -1885,6 +1903,69 @@ async function uploadZip(auth, workspaceId, zipPath) {
   return { doc_ext_ids: allDocIds, duplicates: allDuplicates, folders };
 }
 
+// src/operations/responses.ts
+var responses_exports = {};
+__export(responses_exports, {
+  extractResponseText: () => extractResponseText,
+  getResponse: () => getResponse,
+  submitBackgroundQuery: () => submitBackgroundQuery
+});
+async function submitBackgroundQuery(options) {
+  const { workspaceId, question, docIds, previousResponseId, model, ...auth } = options;
+  const tools = {};
+  if (docIds.length > 0) {
+    tools.retrieval_chunk = {
+      name: "retrieval_chunk",
+      description: "retrieval chunk",
+      tool_args: { doc_ext_ids: docIds },
+      tool_responses: {}
+    };
+    tools.retrieval_full_context = {
+      name: "retrieval_full_context",
+      description: "retrieval full context",
+      tool_args: { doc_ext_ids: [] },
+      tool_responses: {}
+    };
+  }
+  const body = {
+    input: question,
+    workspace_ext_id: workspaceId,
+    stream: false,
+    background: true,
+    store: true,
+    tools,
+    ...previousResponseId ? { previous_response_id: previousResponseId } : {},
+    ...model ? { model } : {}
+  };
+  const res = await authenticatedFetch({
+    ...auth,
+    path: "/v1/responses",
+    method: "POST",
+    body: JSON.stringify(body),
+    headers: { "Content-Type": "application/json" }
+  });
+  return await res.json();
+}
+async function getResponse(auth, responseId) {
+  const res = await authenticatedFetch({
+    ...auth,
+    path: `/v1/responses/${responseId}`,
+    method: "GET"
+  });
+  return await res.json();
+}
+function extractResponseText(response) {
+  const parts = [];
+  for (const msg of response.output) {
+    for (const item of msg.content) {
+      if ("type" in item && item.type === "output_text" && "text" in item) {
+        parts.push(item.text);
+      }
+    }
+  }
+  return parts.join("");
+}
+
 exports.Arbi = Arbi;
 exports.ArbiApiError = ArbiApiError;
 exports.ArbiError = ArbiError;
@@ -1924,6 +2005,7 @@ exports.requireData = requireData;
 exports.requireOk = requireOk;
 exports.resolveAuth = resolveAuth;
 exports.resolveWorkspace = resolveWorkspace;
+exports.responses = responses_exports;
 exports.selectWorkspace = selectWorkspace;
 exports.selectWorkspaceById = selectWorkspaceById;
 exports.settings = settings_exports;