@mobilerun/sdk 5.2.1 → 5.3.0

package/src/internal/decoders/line.ts

@@ -0,0 +1,135 @@
+import { concatBytes, decodeUTF8, encodeUTF8 } from '../utils/bytes';
+
+export type Bytes = string | ArrayBuffer | Uint8Array | null | undefined;
+
+/**
+ * A re-implementation of httpx's `LineDecoder` in Python that handles incrementally
+ * reading lines from text.
+ *
+ * https://github.com/encode/httpx/blob/920333ea98118e9cf617f246905d7b202510941c/httpx/_decoders.py#L258
+ */
+export class LineDecoder {
+  // prettier-ignore
+  static NEWLINE_CHARS = new Set(['\n', '\r']);
+  static NEWLINE_REGEXP = /\r\n|[\n\r]/g;
+
+  #buffer: Uint8Array;
+  #carriageReturnIndex: number | null;
+
+  constructor() {
+    this.#buffer = new Uint8Array();
+    this.#carriageReturnIndex = null;
+  }
+
+  decode(chunk: Bytes): string[] {
+    if (chunk == null) {
+      return [];
+    }
+
+    const binaryChunk =
+      chunk instanceof ArrayBuffer ? new Uint8Array(chunk)
+      : typeof chunk === 'string' ? encodeUTF8(chunk)
+      : chunk;
+
+    this.#buffer = concatBytes([this.#buffer, binaryChunk]);
+
+    const lines: string[] = [];
+    let patternIndex;
+    while ((patternIndex = findNewlineIndex(this.#buffer, this.#carriageReturnIndex)) != null) {
+      if (patternIndex.carriage && this.#carriageReturnIndex == null) {
+        // skip until we either get a corresponding `\n`, a new `\r` or nothing
+        this.#carriageReturnIndex = patternIndex.index;
+        continue;
+      }
+
+      // we got double \r or \rtext\n
+      if (
+        this.#carriageReturnIndex != null &&
+        (patternIndex.index !== this.#carriageReturnIndex + 1 || patternIndex.carriage)
+      ) {
+        lines.push(decodeUTF8(this.#buffer.subarray(0, this.#carriageReturnIndex - 1)));
+        this.#buffer = this.#buffer.subarray(this.#carriageReturnIndex);
+        this.#carriageReturnIndex = null;
+        continue;
+      }
+
+      const endIndex =
+        this.#carriageReturnIndex !== null ? patternIndex.preceding - 1 : patternIndex.preceding;
+
+      const line = decodeUTF8(this.#buffer.subarray(0, endIndex));
+      lines.push(line);
+
+      this.#buffer = this.#buffer.subarray(patternIndex.index);
+      this.#carriageReturnIndex = null;
+    }
+
+    return lines;
+  }
+
+  flush(): string[] {
+    if (!this.#buffer.length) {
+      return [];
+    }
+    return this.decode('\n');
+  }
+}
+
+/**
+ * This function searches the buffer for the end patterns, (\r or \n)
+ * and returns an object with the index preceding the matched newline and the
+ * index after the newline char. `null` is returned if no new line is found.
+ *
+ * ```ts
+ * findNewLineIndex('abc\ndef') -> { preceding: 2, index: 3 }
+ * ```
+ */
+function findNewlineIndex(
+  buffer: Uint8Array,
+  startIndex: number | null,
+): { preceding: number; index: number; carriage: boolean } | null {
+  const newline = 0x0a; // \n
+  const carriage = 0x0d; // \r
+
+  for (let i = startIndex ?? 0; i < buffer.length; i++) {
+    if (buffer[i] === newline) {
+      return { preceding: i, index: i + 1, carriage: false };
+    }
+
+    if (buffer[i] === carriage) {
+      return { preceding: i, index: i + 1, carriage: true };
+    }
+  }
+
+  return null;
+}
+
+export function findDoubleNewlineIndex(buffer: Uint8Array): number {
+  // This function searches the buffer for the end patterns (\r\r, \n\n, \r\n\r\n)
+  // and returns the index right after the first occurrence of any pattern,
+  // or -1 if none of the patterns are found.
+  const newline = 0x0a; // \n
+  const carriage = 0x0d; // \r
+
+  for (let i = 0; i < buffer.length - 1; i++) {
+    if (buffer[i] === newline && buffer[i + 1] === newline) {
+      // \n\n
+      return i + 2;
+    }
+    if (buffer[i] === carriage && buffer[i + 1] === carriage) {
+      // \r\r
+      return i + 2;
+    }
+    if (
+      buffer[i] === carriage &&
+      buffer[i + 1] === newline &&
+      i + 3 < buffer.length &&
+      buffer[i + 2] === carriage &&
+      buffer[i + 3] === newline
+    ) {
+      // \r\n\r\n
+      return i + 4;
+    }
+  }
+
+  return -1;
+}

package/src/internal/parse.ts

@@ -1,6 +1,7 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 import type { FinalRequestOptions } from './request-options';
+import { Stream } from '../core/streaming';
 import { type Mobilerun } from '../client';
 import { formatRequestDetails, loggerFor } from './utils/log';
 
@@ -16,6 +17,19 @@ export type APIResponseProps = {
 export async function defaultParseResponse<T>(client: Mobilerun, props: APIResponseProps): Promise<T> {
   const { response, requestLogID, retryOfRequestLogID, startTime } = props;
   const body = await (async () => {
+    if (props.options.stream) {
+      loggerFor(client).debug('response', response.status, response.url, response.headers, response.body);
+
+      // Note: there is an invariant here that isn't represented in the type system
+      // that if you set `stream: true` the response type must also be `Stream<T>`
+
+      if (props.options.__streamClass) {
+        return props.options.__streamClass.fromSSEResponse(response, props.controller, client) as any;
+      }
+
+      return Stream.fromSSEResponse(response, props.controller, client) as any;
+    }
+
     // fetch refuses to read the body when the status code is 204.
     if (response.status === 204) {
       return null as T;

package/src/internal/request-options.ts

@@ -3,6 +3,7 @@
 import { NullableHeaders } from './headers';
 
 import type { BodyInit } from './builtin-types';
+import { Stream } from '../core/streaming';
 import type { HTTPMethod, MergedRequestInit } from './types';
 import { type HeadersLike } from './headers';
 
@@ -76,6 +77,7 @@ export type RequestOptions = {
   defaultBaseURL?: string | undefined;
 
   __binaryResponse?: boolean | undefined;
+  __streamClass?: typeof Stream;
 };
 
 export type EncodedContent = { bodyHeaders: HeadersLike; body: BodyInit };

package/src/resources/apps.ts

@@ -471,7 +471,7 @@ export namespace AppListResponse {
 
       sizeBytes: number | null;
 
-      source: 'user' | 'system' | 'portal' | 'store';
+      source: 'user' | 'system' | 'portal' | 'store' | 'catalog';
 
       status: 'queued' | 'available' | 'failed';
 
@@ -802,7 +802,7 @@ export namespace AppListVersionsResponse {
 
     sizeBytes: number | null;
 
-    source: 'user' | 'system' | 'portal' | 'store';
+    source: 'user' | 'system' | 'portal' | 'store' | 'catalog';
 
     status: 'queued' | 'available' | 'failed';
 

package/src/resources/assistant/assistant.ts

@@ -0,0 +1,59 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import { APIResource } from '../../core/resource';
+import * as ConversationsAPI from './conversations';
+import {
+  ConversationAbortParams,
+  ConversationAbortResponse,
+  ConversationAnswerPermissionParams,
+  ConversationAnswerPermissionResponse,
+  ConversationAnswerQuestionParams,
+  ConversationAnswerQuestionResponse,
+  ConversationCreateParams,
+  ConversationCreateResponse,
+  ConversationHistoryParams,
+  ConversationHistoryResponse,
+  ConversationListParams,
+  ConversationListResponse,
+  ConversationRejectQuestionParams,
+  ConversationRejectQuestionResponse,
+  ConversationSendParams,
+  ConversationSendResponse,
+  ConversationStreamParams,
+  ConversationStreamResponse,
+  ConversationUpdateParams,
+  ConversationUpdateResponse,
+  Conversations,
+} from './conversations';
+
+export class Assistant extends APIResource {
+  conversations: ConversationsAPI.Conversations = new ConversationsAPI.Conversations(this._client);
+}
+
+Assistant.Conversations = Conversations;
+
+export declare namespace Assistant {
+  export {
+    Conversations as Conversations,
+    type ConversationCreateResponse as ConversationCreateResponse,
+    type ConversationUpdateResponse as ConversationUpdateResponse,
+    type ConversationListResponse as ConversationListResponse,
+    type ConversationAbortResponse as ConversationAbortResponse,
+    type ConversationAnswerPermissionResponse as ConversationAnswerPermissionResponse,
+    type ConversationAnswerQuestionResponse as ConversationAnswerQuestionResponse,
+    type ConversationHistoryResponse as ConversationHistoryResponse,
+    type ConversationRejectQuestionResponse as ConversationRejectQuestionResponse,
+    type ConversationSendResponse as ConversationSendResponse,
+    type ConversationStreamResponse as ConversationStreamResponse,
+    type ConversationCreateParams as ConversationCreateParams,
+    type ConversationUpdateParams as ConversationUpdateParams,
+    type ConversationListParams as ConversationListParams,
+    type ConversationAbortParams as ConversationAbortParams,
+    type ConversationAnswerPermissionParams as ConversationAnswerPermissionParams,
+    type ConversationAnswerQuestionParams as ConversationAnswerQuestionParams,
+    type ConversationHistoryParams as ConversationHistoryParams,
+    type ConversationRejectQuestionParams as ConversationRejectQuestionParams,
+    type ConversationSendParams as ConversationSendParams,
+    type ConversationStreamParams as ConversationStreamParams,
+  };
+}