ai 2.2.34 → 2.2.36

package/dist/index.d.ts

@@ -826,6 +826,17 @@ declare function GoogleGenerativeAIStream(response: {
 
 declare function HuggingFaceStream(res: AsyncGenerator<any>, callbacks?: AIStreamCallbacksAndOptions): ReadableStream;
 
+type InkeepOnFinalMetadata = {
+    chat_session_id: string;
+    records_cited: any;
+};
+type InkeepChatResultCallbacks = {
+    onFinal?: (completion: string, metadata?: InkeepOnFinalMetadata) => Promise<void> | void;
+    onRecordsCited?: (records_cited: InkeepOnFinalMetadata['records_cited']) => void;
+};
+type InkeepAIStreamCallbacksAndOptions = AIStreamCallbacksAndOptions & InkeepChatResultCallbacks;
+declare function InkeepStream(res: Response, callbacks?: InkeepAIStreamCallbacksAndOptions): ReadableStream;
+
 declare function LangChainStream(callbacks?: AIStreamCallbacksAndOptions): {
     stream: ReadableStream<any>;
     writer: WritableStreamDefaultWriter<any>;
@@ -958,4 +969,4 @@ declare function streamToResponse(res: ReadableStream, response: ServerResponse,
     status?: number;
 }): void;
 
-export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantMessage, COMPLEX_HEADER, ChatRequest, ChatRequestOptions, CohereStream, CompletionUsage, CreateMessage, DataMessage, Function, FunctionCall, FunctionCallHandler, FunctionCallPayload, GoogleGenerativeAIStream, HuggingFaceStream, IdGenerator, JSONValue, LangChainStream, Message$1 as Message, OpenAIStream, OpenAIStreamCallbacks, ReactResponseRow, ReplicateStream, RequestOptions, StreamString, StreamingTextResponse, Tool, ToolCall, ToolCallHandler, ToolCallPayload, ToolChoice, UseChatOptions, UseCompletionOptions, createCallbacksTransformer, createChunkDecoder, createEventStreamTransformer, createStreamDataTransformer, experimental_AssistantResponse, experimental_StreamData, experimental_StreamingReactResponse, isStreamStringEqualToType, nanoid, readableFromAsyncIterable, streamToResponse, trimStartOfStreamHelper };
+export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantMessage, COMPLEX_HEADER, ChatRequest, ChatRequestOptions, CohereStream, CompletionUsage, CreateMessage, DataMessage, Function, FunctionCall, FunctionCallHandler, FunctionCallPayload, GoogleGenerativeAIStream, HuggingFaceStream, IdGenerator, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, JSONValue, LangChainStream, Message$1 as Message, OpenAIStream, OpenAIStreamCallbacks, ReactResponseRow, ReplicateStream, RequestOptions, StreamString, StreamingTextResponse, Tool, ToolCall, ToolCallHandler, ToolCallPayload, ToolChoice, UseChatOptions, UseCompletionOptions, createCallbacksTransformer, createChunkDecoder, createEventStreamTransformer, createStreamDataTransformer, experimental_AssistantResponse, experimental_StreamData, experimental_StreamingReactResponse, isStreamStringEqualToType, nanoid, readableFromAsyncIterable, streamToResponse, trimStartOfStreamHelper };

package/dist/index.js

@@ -30,6 +30,7 @@ __export(streams_exports, {
   CohereStream: () => CohereStream,
   GoogleGenerativeAIStream: () => GoogleGenerativeAIStream,
   HuggingFaceStream: () => HuggingFaceStream,
+  InkeepStream: () => InkeepStream,
   LangChainStream: () => LangChainStream,
   OpenAIStream: () => OpenAIStream,
   ReplicateStream: () => ReplicateStream,
@@ -730,6 +731,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();
@@ -1061,6 +1100,7 @@ function createFunctionCallTransformer(callbacks) {
             controller.enqueue(
               isComplexMode ? textEncoder.encode(formatStreamPart("text", functionResponse)) : textEncoder.encode(functionResponse)
             );
+            aggregatedFinalCompletionResponse = functionResponse;
             return;
           }
           const filteredCallbacks = {
@@ -1389,6 +1429,7 @@ function streamToResponse(res, response, init) {
   CohereStream,
   GoogleGenerativeAIStream,
   HuggingFaceStream,
+  InkeepStream,
   LangChainStream,
   OpenAIStream,
   ReplicateStream,