@orq-ai/node 3.2.0-rc.11 → 3.2.0-rc.13

package/src/models/operations/updatedataset.ts

@@ -0,0 +1,374 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod";
+import { remap as remap$ } from "../../lib/primitives.js";
+import { safeParse } from "../../lib/schemas.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+export type UpdateDatasetRequestBody = {
+  displayName: string;
+};
+
+export type UpdateDatasetRequest = {
+  datasetId: string;
+  requestBody?: UpdateDatasetRequestBody | undefined;
+};
+
+export type UpdateDatasetMetadata = {
+  totalVersions: number;
+  datapointsCount: number;
+};
+
+/**
+ * Dataset updated.
+ */
+export type UpdateDatasetResponseBody = {
+  /**
+   * The unique identifier of the dataset
+   */
+  id: string;
+  /**
+   * The display name of the dataset
+   */
+  displayName: string;
+  /**
+   * The unique identifier of the project it belongs to
+   */
+  projectId: string;
+  /**
+   * The unique identifier of the workspace it belongs to
+   */
+  workspaceId: string;
+  /**
+   * The unique identifier of the user who created the dataset
+   */
+  createdById?: string | undefined;
+  /**
+   * The unique identifier of the user who last updated the dataset
+   */
+  updatedById?: string | undefined;
+  metadata: UpdateDatasetMetadata;
+  /**
+   * The unique identifier for the parent of the committed version
+   */
+  parentId?: string | undefined;
+  /**
+   * The version of the dataset
+   */
+  version?: string | undefined;
+  /**
+   * The date and time the resource was created
+   */
+  created?: Date | undefined;
+  /**
+   * The date and time the resource was last updated
+   */
+  updated?: Date | undefined;
+};
+
+/** @internal */
+export const UpdateDatasetRequestBody$inboundSchema: z.ZodType<
+  UpdateDatasetRequestBody,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  display_name: z.string(),
+}).transform((v) => {
+  return remap$(v, {
+    "display_name": "displayName",
+  });
+});
+
+/** @internal */
+export type UpdateDatasetRequestBody$Outbound = {
+  display_name: string;
+};
+
+/** @internal */
+export const UpdateDatasetRequestBody$outboundSchema: z.ZodType<
+  UpdateDatasetRequestBody$Outbound,
+  z.ZodTypeDef,
+  UpdateDatasetRequestBody
+> = z.object({
+  displayName: z.string(),
+}).transform((v) => {
+  return remap$(v, {
+    displayName: "display_name",
+  });
+});
+
+/**
+ * @internal
+ * @deprecated This namespace will be removed in future versions. Use schemas and types that are exported directly from this module.
+ */
+export namespace UpdateDatasetRequestBody$ {
+  /** @deprecated use `UpdateDatasetRequestBody$inboundSchema` instead. */
+  export const inboundSchema = UpdateDatasetRequestBody$inboundSchema;
+  /** @deprecated use `UpdateDatasetRequestBody$outboundSchema` instead. */
+  export const outboundSchema = UpdateDatasetRequestBody$outboundSchema;
+  /** @deprecated use `UpdateDatasetRequestBody$Outbound` instead. */
+  export type Outbound = UpdateDatasetRequestBody$Outbound;
+}
+
+export function updateDatasetRequestBodyToJSON(
+  updateDatasetRequestBody: UpdateDatasetRequestBody,
+): string {
+  return JSON.stringify(
+    UpdateDatasetRequestBody$outboundSchema.parse(updateDatasetRequestBody),
+  );
+}
+
+export function updateDatasetRequestBodyFromJSON(
+  jsonString: string,
+): SafeParseResult<UpdateDatasetRequestBody, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => UpdateDatasetRequestBody$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'UpdateDatasetRequestBody' from JSON`,
+  );
+}
+
+/** @internal */
+export const UpdateDatasetRequest$inboundSchema: z.ZodType<
+  UpdateDatasetRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  dataset_id: z.string(),
+  RequestBody: z.lazy(() => UpdateDatasetRequestBody$inboundSchema).optional(),
+}).transform((v) => {
+  return remap$(v, {
+    "dataset_id": "datasetId",
+    "RequestBody": "requestBody",
+  });
+});
+
+/** @internal */
+export type UpdateDatasetRequest$Outbound = {
+  dataset_id: string;
+  RequestBody?: UpdateDatasetRequestBody$Outbound | undefined;
+};
+
+/** @internal */
+export const UpdateDatasetRequest$outboundSchema: z.ZodType<
+  UpdateDatasetRequest$Outbound,
+  z.ZodTypeDef,
+  UpdateDatasetRequest
+> = z.object({
+  datasetId: z.string(),
+  requestBody: z.lazy(() => UpdateDatasetRequestBody$outboundSchema).optional(),
+}).transform((v) => {
+  return remap$(v, {
+    datasetId: "dataset_id",
+    requestBody: "RequestBody",
+  });
+});
+
+/**
+ * @internal
+ * @deprecated This namespace will be removed in future versions. Use schemas and types that are exported directly from this module.
+ */
+export namespace UpdateDatasetRequest$ {
+  /** @deprecated use `UpdateDatasetRequest$inboundSchema` instead. */
+  export const inboundSchema = UpdateDatasetRequest$inboundSchema;
+  /** @deprecated use `UpdateDatasetRequest$outboundSchema` instead. */
+  export const outboundSchema = UpdateDatasetRequest$outboundSchema;
+  /** @deprecated use `UpdateDatasetRequest$Outbound` instead. */
+  export type Outbound = UpdateDatasetRequest$Outbound;
+}
+
+export function updateDatasetRequestToJSON(
+  updateDatasetRequest: UpdateDatasetRequest,
+): string {
+  return JSON.stringify(
+    UpdateDatasetRequest$outboundSchema.parse(updateDatasetRequest),
+  );
+}
+
+export function updateDatasetRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<UpdateDatasetRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => UpdateDatasetRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'UpdateDatasetRequest' from JSON`,
+  );
+}
+
+/** @internal */
+export const UpdateDatasetMetadata$inboundSchema: z.ZodType<
+  UpdateDatasetMetadata,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  total_versions: z.number(),
+  datapoints_count: z.number(),
+}).transform((v) => {
+  return remap$(v, {
+    "total_versions": "totalVersions",
+    "datapoints_count": "datapointsCount",
+  });
+});
+
+/** @internal */
+export type UpdateDatasetMetadata$Outbound = {
+  total_versions: number;
+  datapoints_count: number;
+};
+
+/** @internal */
+export const UpdateDatasetMetadata$outboundSchema: z.ZodType<
+  UpdateDatasetMetadata$Outbound,
+  z.ZodTypeDef,
+  UpdateDatasetMetadata
+> = z.object({
+  totalVersions: z.number(),
+  datapointsCount: z.number(),
+}).transform((v) => {
+  return remap$(v, {
+    totalVersions: "total_versions",
+    datapointsCount: "datapoints_count",
+  });
+});
+
+/**
+ * @internal
+ * @deprecated This namespace will be removed in future versions. Use schemas and types that are exported directly from this module.
+ */
+export namespace UpdateDatasetMetadata$ {
+  /** @deprecated use `UpdateDatasetMetadata$inboundSchema` instead. */
+  export const inboundSchema = UpdateDatasetMetadata$inboundSchema;
+  /** @deprecated use `UpdateDatasetMetadata$outboundSchema` instead. */
+  export const outboundSchema = UpdateDatasetMetadata$outboundSchema;
+  /** @deprecated use `UpdateDatasetMetadata$Outbound` instead. */
+  export type Outbound = UpdateDatasetMetadata$Outbound;
+}
+
+export function updateDatasetMetadataToJSON(
+  updateDatasetMetadata: UpdateDatasetMetadata,
+): string {
+  return JSON.stringify(
+    UpdateDatasetMetadata$outboundSchema.parse(updateDatasetMetadata),
+  );
+}
+
+export function updateDatasetMetadataFromJSON(
+  jsonString: string,
+): SafeParseResult<UpdateDatasetMetadata, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => UpdateDatasetMetadata$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'UpdateDatasetMetadata' from JSON`,
+  );
+}
+
+/** @internal */
+export const UpdateDatasetResponseBody$inboundSchema: z.ZodType<
+  UpdateDatasetResponseBody,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  _id: z.string(),
+  display_name: z.string(),
+  project_id: z.string(),
+  workspace_id: z.string(),
+  created_by_id: z.string().optional(),
+  updated_by_id: z.string().optional(),
+  metadata: z.lazy(() => UpdateDatasetMetadata$inboundSchema),
+  parent_id: z.string().optional(),
+  version: z.string().optional(),
+  created: z.string().datetime({ offset: true }).transform(v => new Date(v))
+    .optional(),
+  updated: z.string().datetime({ offset: true }).default(
+    "2025-02-25T11:29:25.365Z",
+  ).transform(v => new Date(v)),
+}).transform((v) => {
+  return remap$(v, {
+    "_id": "id",
+    "display_name": "displayName",
+    "project_id": "projectId",
+    "workspace_id": "workspaceId",
+    "created_by_id": "createdById",
+    "updated_by_id": "updatedById",
+    "parent_id": "parentId",
+  });
+});
+
+/** @internal */
+export type UpdateDatasetResponseBody$Outbound = {
+  _id: string;
+  display_name: string;
+  project_id: string;
+  workspace_id: string;
+  created_by_id?: string | undefined;
+  updated_by_id?: string | undefined;
+  metadata: UpdateDatasetMetadata$Outbound;
+  parent_id?: string | undefined;
+  version?: string | undefined;
+  created?: string | undefined;
+  updated: string;
+};
+
+/** @internal */
+export const UpdateDatasetResponseBody$outboundSchema: z.ZodType<
+  UpdateDatasetResponseBody$Outbound,
+  z.ZodTypeDef,
+  UpdateDatasetResponseBody
+> = z.object({
+  id: z.string(),
+  displayName: z.string(),
+  projectId: z.string(),
+  workspaceId: z.string(),
+  createdById: z.string().optional(),
+  updatedById: z.string().optional(),
+  metadata: z.lazy(() => UpdateDatasetMetadata$outboundSchema),
+  parentId: z.string().optional(),
+  version: z.string().optional(),
+  created: z.date().transform(v => v.toISOString()).optional(),
+  updated: z.date().default(() => new Date("2025-02-25T11:29:25.365Z"))
+    .transform(v => v.toISOString()),
+}).transform((v) => {
+  return remap$(v, {
+    id: "_id",
+    displayName: "display_name",
+    projectId: "project_id",
+    workspaceId: "workspace_id",
+    createdById: "created_by_id",
+    updatedById: "updated_by_id",
+    parentId: "parent_id",
+  });
+});
+
+/**
+ * @internal
+ * @deprecated This namespace will be removed in future versions. Use schemas and types that are exported directly from this module.
+ */
+export namespace UpdateDatasetResponseBody$ {
+  /** @deprecated use `UpdateDatasetResponseBody$inboundSchema` instead. */
+  export const inboundSchema = UpdateDatasetResponseBody$inboundSchema;
+  /** @deprecated use `UpdateDatasetResponseBody$outboundSchema` instead. */
+  export const outboundSchema = UpdateDatasetResponseBody$outboundSchema;
+  /** @deprecated use `UpdateDatasetResponseBody$Outbound` instead. */
+  export type Outbound = UpdateDatasetResponseBody$Outbound;
+}
+
+export function updateDatasetResponseBodyToJSON(
+  updateDatasetResponseBody: UpdateDatasetResponseBody,
+): string {
+  return JSON.stringify(
+    UpdateDatasetResponseBody$outboundSchema.parse(updateDatasetResponseBody),
+  );
+}
+
+export function updateDatasetResponseBodyFromJSON(
+  jsonString: string,
+): SafeParseResult<UpdateDatasetResponseBody, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => UpdateDatasetResponseBody$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'UpdateDatasetResponseBody' from JSON`,
+  );
+}

package/src/models/operations/updateprompt.ts

@@ -221,6 +221,10 @@ export type UpdatePromptModelParameters = {
    * Constrains effort on reasoning for reasoning models. Reducing reasoning effort can result in faster responses and fewer tokens used on reasoning in a response.
    */
   reasoningEffort?: UpdatePromptReasoningEffort | undefined;
+  /**
+   * Gives the model enhanced reasoning capabilities for complex tasks. A value of 0 disables thinking. The minimum budget tokens for thinking are 1024. The Budget Tokens should never exceed the Max Tokens parameter. Only supported by `Anthropic`
+   */
+  budgetTokens?: number | undefined;
 };
 
 export const UpdatePromptProvider = {
@@ -665,6 +669,10 @@ export type UpdatePromptPromptsModelParameters = {
    * Constrains effort on reasoning for reasoning models. Reducing reasoning effort can result in faster responses and fewer tokens used on reasoning in a response.
    */
   reasoningEffort?: UpdatePromptPromptsReasoningEffort | undefined;
+  /**
+   * Gives the model enhanced reasoning capabilities for complex tasks. A value of 0 disables thinking. The minimum budget tokens for thinking are 1024. The Budget Tokens should never exceed the Max Tokens parameter. Only supported by `Anthropic`
+   */
+  budgetTokens?: number | undefined;
 };
 
 export const UpdatePromptPromptsProvider = {
@@ -1344,6 +1352,7 @@ export const UpdatePromptModelParameters$inboundSchema: z.ZodType<
   photoRealVersion: UpdatePromptPhotoRealVersion$inboundSchema.optional(),
   encoding_format: UpdatePromptEncodingFormat$inboundSchema.optional(),
   reasoningEffort: UpdatePromptReasoningEffort$inboundSchema.optional(),
+  budgetTokens: z.number().optional(),
 }).transform((v) => {
   return remap$(v, {
     "encoding_format": "encodingFormat",
@@ -1372,6 +1381,7 @@ export type UpdatePromptModelParameters$Outbound = {
   photoRealVersion?: string | undefined;
   encoding_format?: string | undefined;
   reasoningEffort?: string | undefined;
+  budgetTokens?: number | undefined;
 };
 
 /** @internal */
@@ -1401,6 +1411,7 @@ export const UpdatePromptModelParameters$outboundSchema: z.ZodType<
   photoRealVersion: UpdatePromptPhotoRealVersion$outboundSchema.optional(),
   encodingFormat: UpdatePromptEncodingFormat$outboundSchema.optional(),
   reasoningEffort: UpdatePromptReasoningEffort$outboundSchema.optional(),
+  budgetTokens: z.number().optional(),
 }).transform((v) => {
   return remap$(v, {
     encodingFormat: "encoding_format",
@@ -2867,6 +2878,7 @@ export const UpdatePromptPromptsModelParameters$inboundSchema: z.ZodType<
     .optional(),
   encoding_format: UpdatePromptPromptsEncodingFormat$inboundSchema.optional(),
   reasoningEffort: UpdatePromptPromptsReasoningEffort$inboundSchema.optional(),
+  budgetTokens: z.number().optional(),
 }).transform((v) => {
   return remap$(v, {
     "encoding_format": "encodingFormat",
@@ -2895,6 +2907,7 @@ export type UpdatePromptPromptsModelParameters$Outbound = {
   photoRealVersion?: string | undefined;
   encoding_format?: string | undefined;
   reasoningEffort?: string | undefined;
+  budgetTokens?: number | undefined;
 };
 
 /** @internal */
@@ -2925,6 +2938,7 @@ export const UpdatePromptPromptsModelParameters$outboundSchema: z.ZodType<
     .optional(),
   encodingFormat: UpdatePromptPromptsEncodingFormat$outboundSchema.optional(),
   reasoningEffort: UpdatePromptPromptsReasoningEffort$outboundSchema.optional(),
+  budgetTokens: z.number().optional(),
 }).transform((v) => {
   return remap$(v, {
     encodingFormat: "encoding_format",

package/src/sdk/datasets.ts

@@ -0,0 +1,222 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { datasetsClear } from "../funcs/datasetsClear.js";
+import { datasetsCreate } from "../funcs/datasetsCreate.js";
+import { datasetsCreateDatapoint } from "../funcs/datasetsCreateDatapoint.js";
+import { datasetsCreateDatapoints } from "../funcs/datasetsCreateDatapoints.js";
+import { datasetsDelete } from "../funcs/datasetsDelete.js";
+import { datasetsDeleteDatapoint } from "../funcs/datasetsDeleteDatapoint.js";
+import { datasetsList } from "../funcs/datasetsList.js";
+import { datasetsListDatapoints } from "../funcs/datasetsListDatapoints.js";
+import { datasetsRetrieve } from "../funcs/datasetsRetrieve.js";
+import { datasetsRetrieveDatapoint } from "../funcs/datasetsRetrieveDatapoint.js";
+import { datasetsUpdate } from "../funcs/datasetsUpdate.js";
+import { datasetsUpdateDatapoint } from "../funcs/datasetsUpdateDatapoint.js";
+import { ClientSDK, RequestOptions } from "../lib/sdks.js";
+import * as operations from "../models/operations/index.js";
+import { unwrapAsync } from "../types/fp.js";
+
+export class Datasets extends ClientSDK {
+  /**
+   * List datasets
+   *
+   * @remarks
+   * Retrieves a paginated list of datasets for the current workspace. Results can be paginated using cursor-based pagination.
+   */
+  async list(
+    request?: operations.ListDatasetsRequest | undefined,
+    options?: RequestOptions,
+  ): Promise<operations.ListDatasetsResponseBody> {
+    return unwrapAsync(datasetsList(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Create a dataset
+   *
+   * @remarks
+   * Creates a new dataset in the specified project.
+   */
+  async create(
+    request?: operations.CreateDatasetRequestBody | undefined,
+    options?: RequestOptions,
+  ): Promise<operations.CreateDatasetResponseBody> {
+    return unwrapAsync(datasetsCreate(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Retrieve a dataset
+   *
+   * @remarks
+   * Retrieves a specific dataset by its unique identifier
+   */
+  async retrieve(
+    request: operations.RetrieveDatasetRequest,
+    options?: RequestOptions,
+  ): Promise<operations.RetrieveDatasetResponseBody> {
+    return unwrapAsync(datasetsRetrieve(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Update a dataset
+   *
+   * @remarks
+   * Update a dataset
+   */
+  async update(
+    request: operations.UpdateDatasetRequest,
+    options?: RequestOptions,
+  ): Promise<operations.UpdateDatasetResponseBody> {
+    return unwrapAsync(datasetsUpdate(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Delete a dataset
+   *
+   * @remarks
+   * Permanently deletes a dataset and all its datapoints. This action is irreversible.
+   */
+  async delete(
+    request: operations.DeleteDatasetRequest,
+    options?: RequestOptions,
+  ): Promise<void> {
+    return unwrapAsync(datasetsDelete(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * List datapoints
+   *
+   * @remarks
+   * Retrieves a paginated list of datapoints from a specific dataset.
+   */
+  async listDatapoints(
+    request: operations.ListDatasetDatapointsRequest,
+    options?: RequestOptions,
+  ): Promise<operations.ListDatasetDatapointsResponseBody> {
+    return unwrapAsync(datasetsListDatapoints(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Create a datapoint
+   *
+   * @remarks
+   * Creates a new datapoint in the specified dataset.
+   */
+  async createDatapoint(
+    request: operations.CreateDatasetItemRequest,
+    options?: RequestOptions,
+  ): Promise<operations.CreateDatasetItemResponseBody> {
+    return unwrapAsync(datasetsCreateDatapoint(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Retrieve a datapoint
+   *
+   * @remarks
+   * Retrieves a datapoint object
+   */
+  async retrieveDatapoint(
+    request: operations.RetrieveDatapointRequest,
+    options?: RequestOptions,
+  ): Promise<operations.RetrieveDatapointResponseBody> {
+    return unwrapAsync(datasetsRetrieveDatapoint(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Update a datapoint
+   */
+  async updateDatapoint(
+    request: operations.UpdateDatapointRequest,
+    options?: RequestOptions,
+  ): Promise<operations.UpdateDatapointResponseBody> {
+    return unwrapAsync(datasetsUpdateDatapoint(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Delete a datapoint
+   *
+   * @remarks
+   * Permanently deletes a specific datapoint from a dataset.
+   */
+  async deleteDatapoint(
+    request: operations.DeleteDatapointRequest,
+    options?: RequestOptions,
+  ): Promise<void> {
+    return unwrapAsync(datasetsDeleteDatapoint(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Create multiple datapoints
+   *
+   * @remarks
+   * Create multiple datapoints at once.
+   */
+  async createDatapoints(
+    request: operations.BulkCreateDatapointsRequest,
+    options?: RequestOptions,
+  ): Promise<Array<operations.ResponseBody>> {
+    return unwrapAsync(datasetsCreateDatapoints(
+      this,
+      request,
+      options,
+    ));
+  }
+
+  /**
+   * Delete all datapoints
+   *
+   * @remarks
+   * Delete all datapoints from a dataset. This action is irreversible.
+   */
+  async clear(
+    request: operations.ClearDatasetRequest,
+    options?: RequestOptions,
+  ): Promise<void> {
+    return unwrapAsync(datasetsClear(
+      this,
+      request,
+      options,
+    ));
+  }
+}

package/src/sdk/sdk.ts

@@ -4,6 +4,7 @@
 
 import { ClientSDK } from "../lib/sdks.js";
 import { Contacts } from "./contacts.js";
+import { Datasets } from "./datasets.js";
 import { Deployments } from "./deployments.js";
 import { Feedback } from "./feedback.js";
 import { Files } from "./files.js";
@@ -40,4 +41,9 @@ export class Orq extends ClientSDK {
   get remoteconfigs(): Remoteconfigs {
     return (this._remoteconfigs ??= new Remoteconfigs(this._options));
   }
+
+  private _datasets?: Datasets;
+  get datasets(): Datasets {
+    return (this._datasets ??= new Datasets(this._options));
+  }
 }