ai 2.2.34 → 2.2.36

package/dist/index.mjs

@@ -681,6 +681,44 @@ function HuggingFaceStream(res, callbacks) {
   );
 }
 
+// streams/inkeep-stream.ts
+function InkeepStream(res, callbacks) {
+  if (!res.body) {
+    throw new Error("Response body is null");
+  }
+  let chat_session_id = "";
+  let records_cited;
+  const inkeepEventParser = (data, options) => {
+    var _a, _b;
+    const { event } = options;
+    if (event === "records_cited") {
+      records_cited = JSON.parse(data);
+      (_a = callbacks == null ? void 0 : callbacks.onRecordsCited) == null ? void 0 : _a.call(callbacks, records_cited);
+    }
+    if (event === "message_chunk") {
+      const inkeepMessageChunk = JSON.parse(data);
+      chat_session_id = (_b = inkeepMessageChunk.chat_session_id) != null ? _b : chat_session_id;
+      return inkeepMessageChunk.content_chunk;
+    }
+    return;
+  };
+  let { onRecordsCited, ...passThroughCallbacks } = callbacks || {};
+  passThroughCallbacks = {
+    ...passThroughCallbacks,
+    onFinal: (completion) => {
+      var _a;
+      const inkeepOnFinalMetadata = {
+        chat_session_id,
+        records_cited
+      };
+      (_a = callbacks == null ? void 0 : callbacks.onFinal) == null ? void 0 : _a.call(callbacks, completion, inkeepOnFinalMetadata);
+    }
+  };
+  return AIStream(res, inkeepEventParser, passThroughCallbacks).pipeThrough(
+    createStreamDataTransformer(passThroughCallbacks == null ? void 0 : passThroughCallbacks.experimental_streamData)
+  );
+}
+
 // streams/langchain-stream.ts
 function LangChainStream(callbacks) {
   const stream = new TransformStream();
@@ -1012,6 +1050,7 @@ function createFunctionCallTransformer(callbacks) {
             controller.enqueue(
               isComplexMode ? textEncoder.encode(formatStreamPart("text", functionResponse)) : textEncoder.encode(functionResponse)
             );
+            aggregatedFinalCompletionResponse = functionResponse;
             return;
           }
           const filteredCallbacks = {
@@ -1339,6 +1378,7 @@ export {
   CohereStream,
   GoogleGenerativeAIStream,
   HuggingFaceStream,
+  InkeepStream,
   LangChainStream,
   OpenAIStream,
   ReplicateStream,