Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 1 addition & 19 deletions src/resources/datasets.ts
Original file line number Diff line number Diff line change
Expand Up @@ -35,25 +35,7 @@ export class Datasets extends APIResource {
}

export interface ListDatasetsResponse {
data: Array<ListDatasetsResponse.Data>;
}

export namespace ListDatasetsResponse {
export interface Data {
dataset_schema: Record<string, Shared.ParamType>;

identifier: string;

metadata: Record<string, boolean | number | string | Array<unknown> | unknown | null>;

provider_id: string;

provider_resource_id: string;

type: 'dataset';

url: Shared.URL;
}
data: DatasetListResponse;
}

export interface DatasetRetrieveResponse {
Expand Down
2 changes: 1 addition & 1 deletion src/resources/eval-tasks.ts
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ export interface EvalTask {
}

export interface ListEvalTasksResponse {
data: Array<EvalTask>;
data: EvalTaskListResponse;
}

export type EvalTaskListResponse = Array<EvalTask>;
Expand Down
2 changes: 1 addition & 1 deletion src/resources/models.ts
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ export class Models extends APIResource {
}

export interface ListModelsResponse {
data: Array<Model>;
data: ModelListResponse;
}

export interface Model {
Expand Down
7 changes: 5 additions & 2 deletions src/resources/post-training/job.ts
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,14 @@

import { APIResource } from '../../resource';
import * as Core from '../../core';
import { ListPostTrainingJobsResponse } from './post-training';

export class Job extends APIResource {
list(options?: Core.RequestOptions): Core.APIPromise<JobListResponse> {
list(options?: Core.RequestOptions): Core.APIPromise<Array<ListPostTrainingJobsResponse.Data>> {
return (
this._client.get('/v1/post-training/jobs', options) as Core.APIPromise<{ data: JobListResponse }>
this._client.get('/v1/post-training/jobs', options) as Core.APIPromise<{
data: Array<ListPostTrainingJobsResponse.Data>;
}>
)._thenUnwrap((obj) => obj.data);
}

Expand Down
2 changes: 1 addition & 1 deletion src/resources/providers.ts
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ export class Providers extends APIResource {
}

export interface ListProvidersResponse {
data: Array<InspectAPI.ProviderInfo>;
data: ProviderListResponse;
}

export type ProviderListResponse = Array<InspectAPI.ProviderInfo>;
Expand Down
2 changes: 1 addition & 1 deletion src/resources/routes.ts
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ export class Routes extends APIResource {
}

export interface ListRoutesResponse {
data: Array<InspectAPI.RouteInfo>;
data: RouteListResponse;
}

export type RouteListResponse = Array<InspectAPI.RouteInfo>;
Expand Down
2 changes: 1 addition & 1 deletion src/resources/scoring-functions.ts
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ export class ScoringFunctions extends APIResource {
}

export interface ListScoringFunctionsResponse {
data: Array<ScoringFn>;
data: ScoringFunctionListResponse;
}

export interface ScoringFn {
Expand Down
2 changes: 1 addition & 1 deletion src/resources/shields.ts
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ export class Shields extends APIResource {
}

export interface ListShieldsResponse {
data: Array<Shield>;
data: ShieldListResponse;
}

export interface Shield {
Expand Down
20 changes: 1 addition & 19 deletions src/resources/telemetry.ts
Original file line number Diff line number Diff line change
Expand Up @@ -165,25 +165,7 @@ export interface QueryCondition {
}

export interface QuerySpansResponse {
data: Array<QuerySpansResponse.Data>;
}

export namespace QuerySpansResponse {
export interface Data {
name: string;

span_id: string;

start_time: string;

trace_id: string;

attributes?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;

end_time?: string;

parent_span_id?: string;
}
data: TelemetryQuerySpansResponse;
}

export interface SpanWithStatus {
Expand Down
2 changes: 1 addition & 1 deletion src/resources/toolgroups.ts
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ export class Toolgroups extends APIResource {
}

export interface ListToolGroupsResponse {
data: Array<ToolGroup>;
data: ToolgroupListResponse;
}

export interface ToolGroup {
Expand Down
2 changes: 1 addition & 1 deletion src/resources/tools.ts
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ export class Tools extends APIResource {
}

export interface ListToolsResponse {
data: Array<Tool>;
data: ToolListResponse;
}

export interface Tool {
Expand Down
18 changes: 1 addition & 17 deletions src/resources/vector-dbs.ts
Original file line number Diff line number Diff line change
Expand Up @@ -33,23 +33,7 @@ export class VectorDBs extends APIResource {
}

export interface ListVectorDBsResponse {
data: Array<ListVectorDBsResponse.Data>;
}

export namespace ListVectorDBsResponse {
export interface Data {
embedding_dimension: number;

embedding_model: string;

identifier: string;

provider_id: string;

provider_resource_id: string;

type: 'vector_db';
}
data: VectorDBListResponse;
}

export interface VectorDBRetrieveResponse {
Expand Down