openlayer 0.7.0 → 0.8.0

package/src/index.ts

@@ -1,10 +1,25 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-import * as Errors from './error';
-import * as Uploads from './uploads';
 import { type Agent } from './_shims/index';
 import * as Core from './core';
+import * as Errors from './error';
+import * as Uploads from './uploads';
 import * as API from './resources/index';
+import { Commits } from './resources/commits/commits';
+import {
+  InferencePipelineRetrieveResponse,
+  InferencePipelineUpdateParams,
+  InferencePipelineUpdateResponse,
+  InferencePipelines,
+} from './resources/inference-pipelines/inference-pipelines';
+import {
+  ProjectCreateParams,
+  ProjectCreateResponse,
+  ProjectListParams,
+  ProjectListResponse,
+  Projects,
+} from './resources/projects/projects';
+import { Storage } from './resources/storage/storage';
 
 export interface ClientOptions {
   /**
@@ -150,6 +165,7 @@ export class Openlayer extends Core.APIClient {
   }
 
   static Openlayer = this;
+  static DEFAULT_TIMEOUT = 60000; // 1 minute
 
   static OpenlayerError = Errors.OpenlayerError;
   static APIError = Errors.APIError;
@@ -169,7 +185,35 @@ export class Openlayer extends Core.APIClient {
   static fileFromPath = Uploads.fileFromPath;
 }
 
-export const {
+Openlayer.Projects = Projects;
+Openlayer.Commits = Commits;
+Openlayer.InferencePipelines = InferencePipelines;
+Openlayer.Storage = Storage;
+export declare namespace Openlayer {
+  export type RequestOptions = Core.RequestOptions;
+
+  export {
+    Projects as Projects,
+    type ProjectCreateResponse as ProjectCreateResponse,
+    type ProjectListResponse as ProjectListResponse,
+    type ProjectCreateParams as ProjectCreateParams,
+    type ProjectListParams as ProjectListParams,
+  };
+
+  export { Commits as Commits };
+
+  export {
+    InferencePipelines as InferencePipelines,
+    type InferencePipelineRetrieveResponse as InferencePipelineRetrieveResponse,
+    type InferencePipelineUpdateResponse as InferencePipelineUpdateResponse,
+    type InferencePipelineUpdateParams as InferencePipelineUpdateParams,
+  };
+
+  export { Storage as Storage };
+}
+
+export { toFile, fileFromPath } from './uploads';
+export {
   OpenlayerError,
   APIError,
   APIConnectionError,
@@ -183,28 +227,6 @@ export const {
   InternalServerError,
   PermissionDeniedError,
   UnprocessableEntityError,
-} = Errors;
-
-export import toFile = Uploads.toFile;
-export import fileFromPath = Uploads.fileFromPath;
-
-export namespace Openlayer {
-  export import RequestOptions = Core.RequestOptions;
-
-  export import Projects = API.Projects;
-  export import ProjectCreateResponse = API.ProjectCreateResponse;
-  export import ProjectListResponse = API.ProjectListResponse;
-  export import ProjectCreateParams = API.ProjectCreateParams;
-  export import ProjectListParams = API.ProjectListParams;
-
-  export import Commits = API.Commits;
-
-  export import InferencePipelines = API.InferencePipelines;
-  export import InferencePipelineRetrieveResponse = API.InferencePipelineRetrieveResponse;
-  export import InferencePipelineUpdateResponse = API.InferencePipelineUpdateResponse;
-  export import InferencePipelineUpdateParams = API.InferencePipelineUpdateParams;
-
-  export import Storage = API.Storage;
-}
+} from './error';
 
 export default Openlayer;

package/src/lib/core/openai-monitor.ts

@@ -213,7 +213,7 @@ class OpenAIMonitor {
   public createChatCompletion = async (
     body: ChatCompletionCreateParams,
     options?: RequestOptions,
-    additionalLogs?: Openlayer.RequestOptions<any> | undefined,
+    additionalLogs?: Openlayer.RequestOptions | undefined,
   ): Promise<ChatCompletion | Stream<ChatCompletionChunk>> => {
     // Start a timer to measure latency
     const startTime = Date.now();
@@ -309,7 +309,7 @@ class OpenAIMonitor {
   public createCompletion = async (
     body: CompletionCreateParams,
     options?: RequestOptions,
-    additionalLogs?: Openlayer.RequestOptions<any> | undefined,
+    additionalLogs?: Openlayer.RequestOptions | undefined,
   ): Promise<Completion | Stream<Completion>> => {
     if (!body.prompt) {
       console.error('No prompt provided.');
@@ -407,7 +407,7 @@ class OpenAIMonitor {
    * @param {Openlayer.RequestOptions<any> | undefined} [additionalLogs] - Optional metadata logs to include with the request sent to Openlayer.
    * @returns {Promise<void>} A promise that resolves when the run data has been successfully published to Openlayer.
    */
-  public async monitorThreadRun(run: Run, additionalLogs?: Openlayer.RequestOptions<any> | undefined) {
+  public async monitorThreadRun(run: Run, additionalLogs?: Openlayer.RequestOptions | undefined) {
     if (run.status !== 'completed' || this.openlayerInferencePipelineId.length === 0) {
       return;
     }

package/src/lib/integrations/openAiTracer.ts

@@ -1,5 +1,5 @@
 import OpenAI from 'openai';
-import { Stream } from 'openai/src/streaming';
+import { Stream } from 'openai/streaming';
 import { addChatCompletionStepToTrace } from '../tracing/tracer';
 
 export function traceOpenAI(openai: OpenAI): OpenAI {

package/src/resources/commits/commits.ts

@@ -2,13 +2,18 @@
 
 import { APIResource } from '../../resource';
 import * as TestResultsAPI from './test-results';
+import { TestResultListParams, TestResultListResponse, TestResults } from './test-results';
 
 export class Commits extends APIResource {
   testResults: TestResultsAPI.TestResults = new TestResultsAPI.TestResults(this._client);
 }
 
-export namespace Commits {
-  export import TestResults = TestResultsAPI.TestResults;
-  export import TestResultListResponse = TestResultsAPI.TestResultListResponse;
-  export import TestResultListParams = TestResultsAPI.TestResultListParams;
+Commits.TestResults = TestResults;
+
+export declare namespace Commits {
+  export {
+    TestResults as TestResults,
+    type TestResultListResponse as TestResultListResponse,
+    type TestResultListParams as TestResultListParams,
+  };
 }

package/src/resources/commits/index.ts

@@ -1,4 +1,4 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 export { Commits } from './commits';
-export { TestResultListResponse, TestResultListParams, TestResults } from './test-results';
+export { TestResults, type TestResultListResponse, type TestResultListParams } from './test-results';

package/src/resources/commits/test-results.ts

@@ -3,7 +3,6 @@
 import { APIResource } from '../../resource';
 import { isRequestOptions } from '../../core';
 import * as Core from '../../core';
-import * as TestResultsAPI from './test-results';
 
 export class TestResults extends APIResource {
   /**
@@ -28,34 +27,10 @@ export class TestResults extends APIResource {
 }
 
 export interface TestResultListResponse {
-  _meta: TestResultListResponse._Meta;
-
   items: Array<TestResultListResponse.Item>;
 }
 
 export namespace TestResultListResponse {
-  export interface _Meta {
-    /**
-     * The current page.
-     */
-    page: number;
-
-    /**
-     * The number of items per page.
-     */
-    perPage: number;
-
-    /**
-     * The total number of items.
-     */
-    totalItems: number;
-
-    /**
-     * The total number of pages.
-     */
-    totalPages: number;
-  }
-
   export interface Item {
     /**
      * Project version (commit) id.
@@ -278,7 +253,9 @@ export interface TestResultListParams {
   type?: 'integrity' | 'consistency' | 'performance' | 'fairness' | 'robustness';
 }
 
-export namespace TestResults {
-  export import TestResultListResponse = TestResultsAPI.TestResultListResponse;
-  export import TestResultListParams = TestResultsAPI.TestResultListParams;
+export declare namespace TestResults {
+  export {
+    type TestResultListResponse as TestResultListResponse,
+    type TestResultListParams as TestResultListParams,
+  };
 }

package/src/resources/index.ts

@@ -2,16 +2,16 @@
 
 export { Commits } from './commits/commits';
 export {
-  InferencePipelineRetrieveResponse,
-  InferencePipelineUpdateResponse,
-  InferencePipelineUpdateParams,
   InferencePipelines,
+  type InferencePipelineRetrieveResponse,
+  type InferencePipelineUpdateResponse,
+  type InferencePipelineUpdateParams,
 } from './inference-pipelines/inference-pipelines';
 export {
-  ProjectCreateResponse,
-  ProjectListResponse,
-  ProjectCreateParams,
-  ProjectListParams,
   Projects,
+  type ProjectCreateResponse,
+  type ProjectListResponse,
+  type ProjectCreateParams,
+  type ProjectListParams,
 } from './projects/projects';
 export { Storage } from './storage/storage';

package/src/resources/inference-pipelines/data.ts

@@ -2,7 +2,6 @@
 
 import { APIResource } from '../../resource';
 import * as Core from '../../core';
-import * as DataAPI from './data';
 
 export class Data extends APIResource {
   /**
@@ -279,7 +278,6 @@ export namespace DataStreamParams {
   }
 }
 
-export namespace Data {
-  export import DataStreamResponse = DataAPI.DataStreamResponse;
-  export import DataStreamParams = DataAPI.DataStreamParams;
+export declare namespace Data {
+  export { type DataStreamResponse as DataStreamResponse, type DataStreamParams as DataStreamParams };
 }

package/src/resources/inference-pipelines/index.ts

@@ -1,11 +1,11 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-export { DataStreamResponse, DataStreamParams, Data } from './data';
+export { Data, type DataStreamResponse, type DataStreamParams } from './data';
 export {
-  InferencePipelineRetrieveResponse,
-  InferencePipelineUpdateResponse,
-  InferencePipelineUpdateParams,
   InferencePipelines,
+  type InferencePipelineRetrieveResponse,
+  type InferencePipelineUpdateResponse,
+  type InferencePipelineUpdateParams,
 } from './inference-pipelines';
-export { RowUpdateResponse, RowUpdateParams, Rows } from './rows';
-export { TestResultListResponse, TestResultListParams, TestResults } from './test-results';
+export { Rows, type RowUpdateResponse, type RowUpdateParams } from './rows';
+export { TestResults, type TestResultListResponse, type TestResultListParams } from './test-results';

package/src/resources/inference-pipelines/inference-pipelines.ts

@@ -3,10 +3,12 @@
 import { APIResource } from '../../resource';
 import { isRequestOptions } from '../../core';
 import * as Core from '../../core';
-import * as InferencePipelinesAPI from './inference-pipelines';
 import * as DataAPI from './data';
+import { Data, DataStreamParams, DataStreamResponse } from './data';
 import * as RowsAPI from './rows';
+import { RowUpdateParams, RowUpdateResponse, Rows } from './rows';
 import * as TestResultsAPI from './test-results';
+import { TestResultListParams, TestResultListResponse, TestResults } from './test-results';
 
 export class InferencePipelines extends APIResource {
   data: DataAPI.Data = new DataAPI.Data(this._client);
@@ -235,17 +237,32 @@ export interface InferencePipelineUpdateParams {
   referenceDatasetUri?: string | null;
 }
 
-export namespace InferencePipelines {
-  export import InferencePipelineRetrieveResponse = InferencePipelinesAPI.InferencePipelineRetrieveResponse;
-  export import InferencePipelineUpdateResponse = InferencePipelinesAPI.InferencePipelineUpdateResponse;
-  export import InferencePipelineUpdateParams = InferencePipelinesAPI.InferencePipelineUpdateParams;
-  export import Data = DataAPI.Data;
-  export import DataStreamResponse = DataAPI.DataStreamResponse;
-  export import DataStreamParams = DataAPI.DataStreamParams;
-  export import Rows = RowsAPI.Rows;
-  export import RowUpdateResponse = RowsAPI.RowUpdateResponse;
-  export import RowUpdateParams = RowsAPI.RowUpdateParams;
-  export import TestResults = TestResultsAPI.TestResults;
-  export import TestResultListResponse = TestResultsAPI.TestResultListResponse;
-  export import TestResultListParams = TestResultsAPI.TestResultListParams;
+InferencePipelines.Data = Data;
+InferencePipelines.Rows = Rows;
+InferencePipelines.TestResults = TestResults;
+
+export declare namespace InferencePipelines {
+  export {
+    type InferencePipelineRetrieveResponse as InferencePipelineRetrieveResponse,
+    type InferencePipelineUpdateResponse as InferencePipelineUpdateResponse,
+    type InferencePipelineUpdateParams as InferencePipelineUpdateParams,
+  };
+
+  export {
+    Data as Data,
+    type DataStreamResponse as DataStreamResponse,
+    type DataStreamParams as DataStreamParams,
+  };
+
+  export {
+    Rows as Rows,
+    type RowUpdateResponse as RowUpdateResponse,
+    type RowUpdateParams as RowUpdateParams,
+  };
+
+  export {
+    TestResults as TestResults,
+    type TestResultListResponse as TestResultListResponse,
+    type TestResultListParams as TestResultListParams,
+  };
 }

package/src/resources/inference-pipelines/rows.ts

@@ -2,7 +2,6 @@
 
 import { APIResource } from '../../resource';
 import * as Core from '../../core';
-import * as RowsAPI from './rows';
 
 export class Rows extends APIResource {
   /**
@@ -75,7 +74,6 @@ export namespace RowUpdateParams {
   }
 }
 
-export namespace Rows {
-  export import RowUpdateResponse = RowsAPI.RowUpdateResponse;
-  export import RowUpdateParams = RowsAPI.RowUpdateParams;
+export declare namespace Rows {
+  export { type RowUpdateResponse as RowUpdateResponse, type RowUpdateParams as RowUpdateParams };
 }

package/src/resources/inference-pipelines/test-results.ts

@@ -3,7 +3,6 @@
 import { APIResource } from '../../resource';
 import { isRequestOptions } from '../../core';
 import * as Core from '../../core';
-import * as TestResultsAPI from './test-results';
 
 export class TestResults extends APIResource {
   /**
@@ -28,34 +27,10 @@ export class TestResults extends APIResource {
 }
 
 export interface TestResultListResponse {
-  _meta: TestResultListResponse._Meta;
-
   items: Array<TestResultListResponse.Item>;
 }
 
 export namespace TestResultListResponse {
-  export interface _Meta {
-    /**
-     * The current page.
-     */
-    page: number;
-
-    /**
-     * The number of items per page.
-     */
-    perPage: number;
-
-    /**
-     * The total number of items.
-     */
-    totalItems: number;
-
-    /**
-     * The total number of pages.
-     */
-    totalPages: number;
-  }
-
   export interface Item {
     /**
      * Project version (commit) id.
@@ -273,7 +248,9 @@ export interface TestResultListParams {
   type?: 'integrity' | 'consistency' | 'performance' | 'fairness' | 'robustness';
 }
 
-export namespace TestResults {
-  export import TestResultListResponse = TestResultsAPI.TestResultListResponse;
-  export import TestResultListParams = TestResultsAPI.TestResultListParams;
+export declare namespace TestResults {
+  export {
+    type TestResultListResponse as TestResultListResponse,
+    type TestResultListParams as TestResultListParams,
+  };
 }

package/src/resources/projects/commits.ts

@@ -3,9 +3,19 @@
 import { APIResource } from '../../resource';
 import { isRequestOptions } from '../../core';
 import * as Core from '../../core';
-import * as CommitsAPI from './commits';
 
 export class Commits extends APIResource {
+  /**
+   * Create a new commit (project version) in a project.
+   */
+  create(
+    projectId: string,
+    body: CommitCreateParams,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<CommitCreateResponse> {
+    return this._client.post(`/projects/${projectId}/versions`, { body, ...options });
+  }
+
   /**
    * List the commits (project versions) in a project.
    */
@@ -27,35 +37,162 @@ export class Commits extends APIResource {
   }
 }
 
-export interface CommitListResponse {
-  _meta: CommitListResponse._Meta;
+export interface CommitCreateResponse {
+  /**
+   * The project version (commit) id.
+   */
+  id: string;
 
-  items: Array<CommitListResponse.Item>;
+  /**
+   * The details of a commit (project version).
+   */
+  commit: CommitCreateResponse.Commit;
+
+  /**
+   * The commit archive date.
+   */
+  dateArchived: string | null;
+
+  /**
+   * The project version (commit) creation date.
+   */
+  dateCreated: string;
+
+  /**
+   * The number of tests that are failing for the commit.
+   */
+  failingGoalCount: number;
+
+  /**
+   * The model id.
+   */
+  mlModelId: string | null;
+
+  /**
+   * The number of tests that are passing for the commit.
+   */
+  passingGoalCount: number;
+
+  /**
+   * The project id.
+   */
+  projectId: string;
+
+  /**
+   * The commit status. Initially, the commit is `queued`, then, it switches to
+   * `running`. Finally, it can be `paused`, `failed`, or `completed`.
+   */
+  status: 'queued' | 'running' | 'paused' | 'failed' | 'completed' | 'unknown';
+
+  /**
+   * The commit status message.
+   */
+  statusMessage: string | null;
+
+  /**
+   * The total number of tests for the commit.
+   */
+  totalGoalCount: number;
+
+  /**
+   * The training dataset id.
+   */
+  trainingDatasetId: string | null;
+
+  /**
+   * The validation dataset id.
+   */
+  validationDatasetId: string | null;
+
+  /**
+   * Whether the commit is archived.
+   */
+  archived?: boolean | null;
+
+  /**
+   * The deployment status associated with the commit's model.
+   */
+  deploymentStatus?: string;
+
+  links?: CommitCreateResponse.Links;
 }
 
-export namespace CommitListResponse {
-  export interface _Meta {
+export namespace CommitCreateResponse {
+  /**
+   * The details of a commit (project version).
+   */
+  export interface Commit {
+    /**
+     * The commit id.
+     */
+    id: string;
+
+    /**
+     * The author id of the commit.
+     */
+    authorId: string;
+
+    /**
+     * The size of the commit bundle in bytes.
+     */
+    fileSize: number | null;
+
+    /**
+     * The commit message.
+     */
+    message: string;
+
+    /**
+     * The model id.
+     */
+    mlModelId: string | null;
+
+    /**
+     * The storage URI where the commit bundle is stored.
+     */
+    storageUri: string;
+
+    /**
+     * The training dataset id.
+     */
+    trainingDatasetId: string | null;
+
+    /**
+     * The validation dataset id.
+     */
+    validationDatasetId: string | null;
+
     /**
-     * The current page.
+     * The commit creation date.
      */
-    page: number;
+    dateCreated?: string;
 
     /**
-     * The number of items per page.
+     * The ref of the corresponding git commit.
      */
-    perPage: number;
+    gitCommitRef?: string;
 
     /**
-     * The total number of items.
+     * The SHA of the corresponding git commit.
      */
-    totalItems: number;
+    gitCommitSha?: number;
 
     /**
-     * The total number of pages.
+     * The URL of the corresponding git commit.
      */
-    totalPages: number;
+    gitCommitUrl?: string;
   }
 
+  export interface Links {
+    app: string;
+  }
+}
+
+export interface CommitListResponse {
+  items: Array<CommitListResponse.Item>;
+}
+
+export namespace CommitListResponse {
   export interface Item {
     /**
      * The project version (commit) id.
@@ -208,6 +345,40 @@ export namespace CommitListResponse {
   }
 }
 
+export interface CommitCreateParams {
+  /**
+   * The details of a commit (project version).
+   */
+  commit: CommitCreateParams.Commit;
+
+  /**
+   * The storage URI where the commit bundle is stored.
+   */
+  storageUri: string;
+
+  /**
+   * Whether the commit is archived.
+   */
+  archived?: boolean | null;
+
+  /**
+   * The deployment status associated with the commit's model.
+   */
+  deploymentStatus?: string;
+}
+
+export namespace CommitCreateParams {
+  /**
+   * The details of a commit (project version).
+   */
+  export interface Commit {
+    /**
+     * The commit message.
+     */
+    message: string;
+  }
+}
+
 export interface CommitListParams {
   /**
    * The page to return in a paginated query.
@@ -220,7 +391,11 @@ export interface CommitListParams {
   perPage?: number;
 }
 
-export namespace Commits {
-  export import CommitListResponse = CommitsAPI.CommitListResponse;
-  export import CommitListParams = CommitsAPI.CommitListParams;
+export declare namespace Commits {
+  export {
+    type CommitCreateResponse as CommitCreateResponse,
+    type CommitListResponse as CommitListResponse,
+    type CommitCreateParams as CommitCreateParams,
+    type CommitListParams as CommitListParams,
+  };
 }