feat: add download event (#22)

* feat: add download event

* fix: wrong download state entity

* feat: add download event

* change rest api

* refactor

---------

Co-authored-by: Louis <louis@jan.ai>

Author: marknguyen1302

src/core.ts

@@ -62,6 +62,7 @@ async function defaultParseResponse<T>(props: APIResponseProps): Promise<T> {
   }
 
   const contentType = response.headers.get('content-type');
+
   const isJSON =
     contentType?.includes('application/json') || contentType?.includes('application/vnd.api+json');
   if (isJSON) {

src/index.ts

@@ -159,6 +159,7 @@ export class OpenAI extends Core.APIClient {
   moderations: API.Moderations = new API.Moderations(this);
   models: API.Models = new API.Models(this);
   engines: API.Engines = new API.Engines(this);
+  events: API.Events = new API.Events(this);
   fineTuning: API.FineTuning = new API.FineTuning(this);
   beta: API.Beta = new API.Beta(this);
   batches: API.Batches = new API.Batches(this);
@@ -300,6 +301,12 @@ export namespace OpenAI {
   export import Engine = API.Engine;
   export import EnginesPage = API.EnginesPage;
 
+  export import Events = API.Events;
+  export import DownloadStateEvent = API.DownloadStateEvent;
+  export import DownloadStatus = API.DownloadStatus;
+  export import DownloadType = API.DownloadType;
+  export import DownloadItem = API.DownloadItem;
+
   export import FineTuning = API.FineTuning;
 
   export import Beta = API.Beta;

src/resources/engines.ts

@@ -5,6 +5,15 @@ import * as Core from '../core';
 import { APIResource } from '../resource';
 import * as EnginesAPI from './engines';
 
+export interface EngineInitRequestParams {
+  runMode?: 'CPU' | 'GPU';
+  gpuType?: 'Nvidia' | 'Others (Vulkan)';
+  instructions?: 'AVX' | 'AVX2' | 'AVX512' | undefined;
+  cudaVersion?: '11' | '12';
+  silent?: boolean;
+  vulkan?: boolean;
+}
+
 export class Engines extends APIResource {
   /**
    * Retrieves an engine instance, providing basic information about the engine.
@@ -24,8 +33,15 @@ export class Engines extends APIResource {
    * Initializes an engine instance with the given name.
    * It will download the engine if it is not available locally.
    */
-  init(engine: string, options?: Core.RequestOptions): Core.APIPromise<Engine> {
-    return this._client.post(`/engines/${engine}/init`, options);
+  init(
+    engine: string,
+    initOptions: EngineInitRequestParams = {},
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<Engine> {
+    return this._client.post(`/engines/${engine}/init`, {
+      body: initOptions,
+      ...options,
+    });
   }
 
   /**

src/resources/events.ts

@@ -0,0 +1,96 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import * as Core from '../core';
+import { APIResource } from '../resource';
+import * as EventsAPI from './events';
+import { Stream } from '../streaming';
+
+export class Events extends APIResource {
+  /**
+   * Get download events.
+   */
+  downloadEvent(): Core.APIPromise<Stream<DownloadStateEvent>> {
+    return this._client.get(`/events/download`, {
+      stream: true,
+    }) as Core.APIPromise<Stream<DownloadStateEvent>>;
+  }
+}
+
+export enum DownloadStatus {
+  Pending = 'pending',
+  Downloading = 'downloading',
+  Error = 'error',
+  Downloaded = 'downloaded',
+}
+
+export enum DownloadType {
+  Model = 'model',
+  Miscelanous = 'miscelanous',
+  Engine = 'engine',
+}
+
+export interface DownloadItem {
+  /**
+   * Filename of the download.
+   */
+  id: string;
+
+  time: {
+    elapsed: number;
+    remaining: number;
+  };
+
+  size: {
+    total: number;
+    transferred: number;
+  };
+
+  checksum?: string;
+
+  status: DownloadStatus;
+
+  error?: string;
+
+  metadata?: Record<string, unknown>;
+}
+
+export interface DownloadState {
+  /**
+   * The id of a particular download. Being used to prevent duplication of downloads.
+   */
+  id: string;
+
+  /**
+   * For displaying purposes.
+   */
+  title: string;
+
+  /**
+   * The type of download.
+   */
+  type: DownloadType;
+
+  /**
+   * The status of the download.
+   */
+  status: DownloadStatus;
+
+  /**
+   * Explanation of the error if the download failed.
+   */
+  error?: string;
+
+  /**
+   * The actual downloads. [DownloadState] is just a group to supporting for download multiple files.
+   */
+  children: DownloadItem[];
+}
+
+export type DownloadStateEvent = DownloadState[];
+
+export namespace Events {
+  export import DownloadStateEvent = EventsAPI.DownloadStateEvent;
+  export import DownloadStatus = EventsAPI.DownloadStatus;
+  export import DownloadType = EventsAPI.DownloadType;
+  export import DownloadItem = EventsAPI.DownloadItem;
+}

src/resources/index.ts

@@ -45,3 +45,4 @@ export { Model, ModelDeleted, ModelsPage, Models } from './models';
 export { Moderation, ModerationCreateResponse, ModerationCreateParams, Moderations } from './moderations';
 
 export { Engine, Engines, EnginesPage } from './engines';
+export { DownloadStateEvent, DownloadStatus, DownloadType, DownloadItem, Events } from './events';

src/resources/models.ts

@@ -48,14 +48,14 @@ export class Models extends APIResource {
    * Download a model.
    */
   download(model: string) {
-    return this._client.get(`/models/download/${model}`);
+    return this._client.post(`/models/pull/${model}`);
   }
 
   /**
    * Abort a model download.
    */
   abortDownload(downloadId: string) {
-    return this._client.get(`/models/abort-download/${downloadId}`);
+    return this._client.delete(`/models/${downloadId}/pull`);
   }
 
   /**