notdiamond 2.0.0-rc1 → 2.0.0-rc11

package/src/resources/prompt-adaptation.ts

@@ -3,7 +3,6 @@
 import { APIResource } from '../core/resource';
 import * as PromptAdaptationAPI from './prompt-adaptation';
 import { APIPromise } from '../core/api-promise';
-import { buildHeaders } from '../internal/headers';
 import { RequestOptions } from '../internal/request-options';
 import { path } from '../internal/utils/path';
 
@@ -34,7 +33,7 @@ export class PromptAdaptation extends APIResource {
    *
    * **Dataset Requirements:**
    *
-   * - Minimum 5 examples in train_goldens (more examples = better adaptation)
+   * - Minimum 25 examples in train_goldens (more examples = better adaptation)
    * - Each example must have fields matching your template placeholders
    * - Supervised evaluation requires 'answer' field in each golden record
    * - Unsupervised evaluation can work without answers
@@ -55,7 +54,7 @@ export class PromptAdaptation extends APIResource {
    * **Best Practices:**
    *
    * 1. Use diverse, representative examples from your production workload
-   * 2. Include 10-20 examples for best results (5 minimum)
+   * 2. Include examples for best results (25 minimum)
    * 3. Ensure consistent evaluation across all examples
    * 4. Test both train_goldens and test_goldens split for validation
    * 5. Use the same model versions you'll use in production
@@ -75,16 +74,16 @@ export class PromptAdaptation extends APIResource {
    *
    * @example
    * ```ts
-   * const response = await client.promptAdaptation.adapt({
+   * const promptAdaptation = await client.promptAdaptation.create({
    *   fields: ['question'],
-   *   origin_model: { provider: 'openai', model: 'gpt-4o' },
    *   system_prompt: 'You are a helpful assistant that answers questions accurately.',
    *   target_models: [
-   *     { provider: 'anthropic', model: 'claude-3-5-sonnet-20241022' },
-   *     { provider: 'google', model: 'gemini-1.5-pro' },
+   *     { provider: 'anthropic', model: 'claude-sonnet-4-5-20250929' },
+   *     { provider: 'google', model: 'gemini-2.5-flash' },
    *   ],
    *   template: 'Question: {question}\nAnswer:',
    *   evaluation_metric: 'LLMaaJ:Sem_Sim_3',
+   *   origin_model: { provider: 'openai', model: 'gpt-4o' },
    *   test_goldens: [
    *     {
    *       fields: { ... },
@@ -120,10 +119,10 @@ export class PromptAdaptation extends APIResource {
    * });
    * ```
    */
-  adapt(
-    body: PromptAdaptationAdaptParams,
+  create(
+    body: PromptAdaptationCreateParams,
     options?: RequestOptions,
-  ): APIPromise<PromptAdaptationAdaptResponse> {
+  ): APIPromise<PromptAdaptationCreateResponse> {
     return this._client.post('/v2/prompt/adapt', { body, ...options });
   }
 
@@ -187,61 +186,17 @@ export class PromptAdaptation extends APIResource {
    *
    * @example
    * ```ts
-   * const adaptationRunResults =
+   * const response =
    *   await client.promptAdaptation.getAdaptResults(
    *     'adaptation_run_id',
    *   );
    * ```
    */
-  getAdaptResults(adaptationRunID: string, options?: RequestOptions): APIPromise<AdaptationRunResults> {
-    return this._client.get(path`/v2/prompt/adaptResults/${adaptationRunID}`, options);
-  }
-
-  /**
-   * Get Adapt Run Results
-   *
-   * @example
-   * ```ts
-   * const adaptationRunResults =
-   *   await client.promptAdaptation.getAdaptRunResults(
-   *     'adaptation_run_id',
-   *     { user_id: 'user_id', 'x-token': 'x-token' },
-   *   );
-   * ```
-   */
-  getAdaptRunResults(
+  getAdaptResults(
     adaptationRunID: string,
-    params: PromptAdaptationGetAdaptRunResultsParams,
     options?: RequestOptions,
-  ): APIPromise<AdaptationRunResults> {
-    const { user_id, 'x-token': xToken } = params;
-    return this._client.get(path`/v2/prompt/frontendAdaptRunResults/${user_id}/${adaptationRunID}`, {
-      ...options,
-      headers: buildHeaders([{ 'x-token': xToken }, options?.headers]),
-    });
-  }
-
-  /**
-   * Get Adapt Runs
-   *
-   * @example
-   * ```ts
-   * const adaptationRunResults =
-   *   await client.promptAdaptation.getAdaptRuns('user_id', {
-   *     'x-token': 'x-token',
-   *   });
-   * ```
-   */
-  getAdaptRuns(
-    userID: string,
-    params: PromptAdaptationGetAdaptRunsParams,
-    options?: RequestOptions,
-  ): APIPromise<PromptAdaptationGetAdaptRunsResponse> {
-    const { 'x-token': xToken } = params;
-    return this._client.get(path`/v2/prompt/frontendAdaptRuns/${userID}`, {
-      ...options,
-      headers: buildHeaders([{ 'x-token': xToken }, options?.headers]),
-    });
+  ): APIPromise<PromptAdaptationGetAdaptResultsResponse> {
+    return this._client.get(path`/v2/prompt/adaptResults/${adaptationRunID}`, options);
   }
 
   /**
@@ -292,85 +247,219 @@ export class PromptAdaptation extends APIResource {
   }
 
   /**
-   * Get LLM costs for a specific adaptation run
+   * Get LLM usage costs for a specific prompt adaptation run.
+   *
+   * This endpoint returns the total cost and detailed usage records for all LLM
+   * requests made during a prompt adaptation run. Use this to track costs associated
+   * with optimizing prompts for different target models.
+   *
+   * **Cost Breakdown:**
+   *
+   * - Total cost across all models used in the adaptation
+   * - Individual usage records with provider, model, tokens, and costs
+   * - Timestamps for each LLM request
+   * - Task type (e.g., optimization, evaluation)
+   *
+   * **Access Control:**
+   *
+   * - Only accessible by the user who created the adaptation run
+   * - Requires prompt adaptation access
    *
    * @example
    * ```ts
-   * const response =
-   *   await client.promptAdaptation.retrieveCosts(
-   *     'adaptation_run_id',
-   *   );
+   * const response = await client.promptAdaptation.getCosts(
+   *   'adaptation_run_id',
+   * );
    * ```
    */
-  retrieveCosts(
-    adaptationRunID: string,
-    options?: RequestOptions,
-  ): APIPromise<PromptAdaptationRetrieveCostsResponse> {
-    return this._client.get(path`/v1/adaptation-runs/${adaptationRunID}/costs`, options);
+  getCosts(adaptationRunID: string, options?: RequestOptions): APIPromise<PromptAdaptationGetCostsResponse> {
+    return this._client.get(path`/v2/prompt/adapt/${adaptationRunID}/costs`, options);
   }
 }
 
 /**
- * Complete results for a prompt adaptation run including all target models.
+ * A training or test example for prompt adaptation.
  */
-export interface AdaptationRunResults {
+export interface GoldenRecord {
   /**
-   * Unique ID for this adaptation run
+   * Dictionary mapping field names to their values. Keys must match the fields
+   * specified in the template
    */
-  id: string;
+  fields: { [key: string]: string };
 
   /**
-   * Timestamp when this adaptation run was created
+   * Expected answer for supervised evaluation. Required for supervised metrics,
+   * optional for unsupervised
    */
-  created_at: string;
+  answer?: string | null;
+}
 
+/**
+ * Status enum for asynchronous jobs (prompt adaptation, custom router training,
+ * etc.).
+ *
+ * Represents the current state of a long-running operation:
+ *
+ * - **created**: Job has been initialized but not yet queued
+ * - **queued**: Job is waiting in the queue to be processed
+ * - **processing**: Job is currently being executed
+ * - **completed**: Job finished successfully and results are available
+ * - **failed**: Job encountered an error and did not complete
+ */
+export type JobStatus = 'created' | 'queued' | 'processing' | 'completed' | 'failed';
+
+/**
+ * Model for specifying an LLM provider in API requests.
+ */
+export interface RequestProvider {
   /**
-   * Overall status of the adaptation run
+   * Model name (e.g., 'gpt-4o', 'claude-sonnet-4-5-20250929')
    */
-  job_status: JobStatus;
+  model: string;
 
   /**
-   * Results for the origin model (baseline performance)
+   * Provider name (e.g., 'openai', 'anthropic', 'google')
    */
-  origin_model: AdaptationRunResults.OriginModel;
+  provider: string;
 
   /**
-   * Results for each target model with optimized prompts
+   * Maximum context length for the model (required for custom models)
    */
-  target_models: Array<AdaptationRunResults.TargetModel>;
+  context_length?: number | null;
 
   /**
-   * Timestamp of last update to this adaptation run
+   * Input token price per million tokens in USD (required for custom models)
    */
-  updated_at: string | null;
+  input_price?: number | null;
 
-  evaluation_config?: string | null;
+  /**
+   * Whether this is a custom model not in Not Diamond's supported model list
+   */
+  is_custom?: boolean;
 
-  evaluation_metric?: string | null;
+  /**
+   * Average latency in seconds (required for custom models)
+   */
+  latency?: number | null;
+
+  /**
+   * Output token price per million tokens in USD (required for custom models)
+   */
+  output_price?: number | null;
 }
 
-export namespace AdaptationRunResults {
+/**
+ * Response model for POST /v2/prompt/adapt endpoint.
+ *
+ * Returned immediately after submitting a prompt adaptation request. The
+ * adaptation process runs asynchronously, so use the returned adaptation_run_id to
+ * track progress and retrieve results when complete.
+ *
+ * **Next steps:**
+ *
+ * 1. Store the adaptation_run_id
+ * 2. Poll GET /v2/prompt/adaptStatus/{adaptation_run_id} to check progress
+ * 3. When status is 'completed', retrieve optimized prompts from GET
+ *    /v2/prompt/adaptResults/{adaptation_run_id}
+ * 4. Use the optimized prompts with your target models
+ */
+export interface PromptAdaptationCreateResponse {
   /**
-   * Results for the origin model (baseline performance)
+   * Unique identifier for this adaptation run. Use this to poll status and retrieve
+   * optimized prompts when complete
    */
-  export interface OriginModel {
-    cost: number | null;
+  adaptation_run_id: string;
+}
 
-    evals: { [key: string]: unknown } | null;
+/**
+ * Response model for GET /v2/prompt/adaptResults/{adaptation_run_id} endpoint.
+ *
+ * Contains the complete results of a prompt adaptation run, including optimized
+ * prompts and evaluation metrics for all target models. Use this to retrieve your
+ * adapted prompts after the adaptation status is 'completed'.
+ *
+ * The response includes:
+ *
+ * - Baseline performance of your original prompt on the origin model
+ * - Optimized prompts for each target model with pre/post optimization scores
+ * - Evaluation metrics and cost information for each model
+ */
+export interface PromptAdaptationGetAdaptResultsResponse {
+  /**
+   * Unique ID for this adaptation run
+   */
+  id: string;
 
-    model_name: string;
+  /**
+   * Timestamp when this adaptation run was created
+   */
+  created_at: string;
 
-    result_status: PromptAdaptationAPI.JobStatus | null;
+  /**
+   * Overall status of the adaptation run (queued, running, completed, failed)
+   */
+  job_status: JobStatus;
 
-    score: number | null;
+  /**
+   * Results for each target model with optimized prompts and improvement scores
+   */
+  target_models: Array<PromptAdaptationGetAdaptResultsResponse.TargetModel>;
 
-    system_prompt: string | null;
+  /**
+   * Timestamp of last update to this adaptation run
+   */
+  updated_at: string | null;
 
-    user_message_template: string | null;
-  }
+  evaluation_config?: string | null;
+
+  evaluation_metric?: string | null;
+
+  /**
+   * Metrics for the LLM requests made during the adaptation run (e.g.,
+   * total_requests, avg_latency)
+   */
+  llm_request_metrics?: { [key: string]: number };
+
+  /**
+   * Baseline results for the origin model in prompt adaptation.
+   *
+   * Part of AdaptationRunResultsResponse. Contains the performance metrics and
+   * prompt configuration for your original prompt on the origin model. This serves
+   * as the baseline to compare against optimized prompts for target models.
+   *
+   * **Fields include:**
+   *
+   * - Original system prompt and user message template
+   * - Baseline performance score and evaluation metrics
+   * - Cost of running the baseline evaluation
+   * - Job status for the origin model evaluation
+   */
+  origin_model?: PromptAdaptationGetAdaptResultsResponse.OriginModel | null;
+}
 
+export namespace PromptAdaptationGetAdaptResultsResponse {
   /**
-   * Results for a single target model adaptation.
+   * Optimized prompt results for a single target model in prompt adaptation.
+   *
+   * Part of AdaptationRunResultsResponse. Contains the optimized system prompt and
+   * user message template for a specific target model, along with performance scores
+   * before and after optimization. Use these optimized prompts with the target model
+   * to achieve better performance than the original prompt.
+   *
+   * **Key metrics:**
+   *
+   * - **pre_optimization_score**: Performance with original prompt on this target
+   *   model
+   * - **post_optimization_score**: Performance with optimized prompt on this target
+   *   model
+   * - **Score improvement**: post - pre shows how much optimization helped
+   *
+   * **Usage:**
+   *
+   * 1. Extract the optimized system_prompt and user_message_template
+   * 2. Replace placeholders in user_message_template using fields from your data
+   * 3. Use these prompts when calling this target model
+   * 4. Compare pre/post scores to see improvement gained
    */
   export interface TargetModel {
     cost: number | null;
@@ -385,113 +474,236 @@ export namespace AdaptationRunResults {
 
     pre_optimization_score: number | null;
 
+    task_type: string | null;
+
     /**
-     * Status of this specific target model adaptation
+     * Status enum for asynchronous jobs (prompt adaptation, custom router training,
+     * etc.).
+     *
+     * Represents the current state of a long-running operation:
+     *
+     * - **created**: Job has been initialized but not yet queued
+     * - **queued**: Job is waiting in the queue to be processed
+     * - **processing**: Job is currently being executed
+     * - **completed**: Job finished successfully and results are available
+     * - **failed**: Job encountered an error and did not complete
      */
-    result_status: PromptAdaptationAPI.JobStatus | null;
+    result_status?: PromptAdaptationAPI.JobStatus | null;
 
     /**
-     * Optimized system prompt for this target model
+     * Optimized system prompt for this target model. Use this as the system message in
+     * your LLM calls
      */
-    system_prompt: string | null;
+    system_prompt?: string | null;
 
     /**
-     * Optimized user message template for this target model
+     * Optimized user message template with placeholders. Substitute fields using your
+     * data before calling the LLM
      */
-    user_message_template: string | null;
+    user_message_template?: string | null;
 
     /**
-     * Field names used in the optimized template
+     * List of field names to substitute in the template (e.g., ['question',
+     * 'context']). These match the curly-brace placeholders in user_message_template
      */
-    user_message_template_fields: Array<string> | null;
+    user_message_template_fields?: Array<string> | null;
   }
-}
 
-export type JobStatus = 'created' | 'queued' | 'processing' | 'completed' | 'failed';
-
-/**
- * Response from prompt adaptation request.
- */
-export interface PromptAdaptationAdaptResponse {
   /**
-   * Unique ID for this adaptation run. Use this to check status and retrieve results
+   * Baseline results for the origin model in prompt adaptation.
+   *
+   * Part of AdaptationRunResultsResponse. Contains the performance metrics and
+   * prompt configuration for your original prompt on the origin model. This serves
+   * as the baseline to compare against optimized prompts for target models.
+   *
+   * **Fields include:**
+   *
+   * - Original system prompt and user message template
+   * - Baseline performance score and evaluation metrics
+   * - Cost of running the baseline evaluation
+   * - Job status for the origin model evaluation
    */
-  adaptation_run_id: string;
-}
+  export interface OriginModel {
+    cost: number | null;
+
+    evals: { [key: string]: unknown } | null;
+
+    model_name: string | null;
+
+    score: number | null;
 
-export type PromptAdaptationGetAdaptRunsResponse = Array<AdaptationRunResults>;
+    /**
+     * Status enum for asynchronous jobs (prompt adaptation, custom router training,
+     * etc.).
+     *
+     * Represents the current state of a long-running operation:
+     *
+     * - **created**: Job has been initialized but not yet queued
+     * - **queued**: Job is waiting in the queue to be processed
+     * - **processing**: Job is currently being executed
+     * - **completed**: Job finished successfully and results are available
+     * - **failed**: Job encountered an error and did not complete
+     */
+    result_status?: PromptAdaptationAPI.JobStatus | null;
+
+    /**
+     * Original system prompt used for the origin model
+     */
+    system_prompt?: string | null;
+
+    /**
+     * Original user message template used for the origin model
+     */
+    user_message_template?: string | null;
+  }
+}
 
 /**
- * Status response for a prompt adaptation run.
+ * Response model for GET /v2/prompt/adaptStatus/{adaptation_run_id} endpoint.
+ *
+ * Returns the current status of an asynchronous prompt adaptation job. Poll this
+ * endpoint periodically to track progress. When status is 'completed', you can
+ * retrieve the optimized prompts using the /adaptResults endpoint.
+ *
+ * **Status values:**
+ *
+ * - **created**: Job has been initialized
+ * - **queued**: Waiting in queue (check queue_position for your place in line)
+ * - **processing**: Currently running optimization
+ * - **completed**: Finished successfully, results available via /adaptResults
+ * - **failed**: Encountered an error during processing
+ *
+ * **Polling recommendations:**
+ *
+ * - Poll every 30-60 seconds while status is incomplete
+ * - Stop polling once status is 'completed' or 'failed'
+ * - Adaptation typically takes 10-30 minutes total
  */
 export interface PromptAdaptationGetAdaptStatusResponse {
   /**
-   * Unique ID for this adaptation run. Use this to check status and retrieve results
+   * Unique identifier for this adaptation run. Use this to poll status and retrieve
+   * optimized prompts when complete
    */
   adaptation_run_id: string;
 
   /**
-   * Current status of the adaptation run (created, queued, processing, completed, or
-   * failed)
+   * Current status of the adaptation run. Poll until this is 'completed' or 'failed'
    */
   status: JobStatus;
 
   /**
-   * Position in queue if status is 'queued'. Lower numbers mean earlier processing
+   * Position in queue when status is 'queued'. Lower numbers process sooner. Null
+   * when not queued
    */
   queue_position?: number | null;
 }
 
-export interface PromptAdaptationRetrieveCostsResponse {
+/**
+ * Response model for GET /v2/prompt/adapt/{adaptation_run_id}/costs endpoint.
+ *
+ * Contains the total LLM costs and detailed usage records for a prompt adaptation
+ * run. Use this to track costs associated with optimizing prompts for different
+ * target models.
+ */
+export interface PromptAdaptationGetCostsResponse {
+  /**
+   * Unique identifier for the adaptation run
+   */
   adaptation_run_id: string;
 
+  /**
+   * Total cost in USD across all LLM requests in this adaptation run
+   */
   total_cost: number;
 
-  usage_records: Array<PromptAdaptationRetrieveCostsResponse.UsageRecord>;
+  /**
+   * Detailed usage records for each LLM request made during the adaptation
+   */
+  usage_records: Array<PromptAdaptationGetCostsResponse.UsageRecord>;
 }
 
-export namespace PromptAdaptationRetrieveCostsResponse {
+export namespace PromptAdaptationGetCostsResponse {
+  /**
+   * Individual LLM usage record with token counts and cost breakdown.
+   *
+   * Returned by GET /llm-usage endpoint and included in AdaptationRunCostResponse.
+   * Each record represents a single LLM API call with detailed usage metrics.
+   */
   export interface UsageRecord {
+    /**
+     * Unique identifier for this usage record
+     */
     id: string;
 
+    /**
+     * Adaptation run ID this usage is associated with
+     */
     adaptation_run_id: string;
 
+    /**
+     * Cost of input tokens in USD
+     */
     input_cost: number;
 
+    /**
+     * Number of input tokens consumed
+     */
     input_tokens: number;
 
+    /**
+     * Model name (e.g., 'gpt-4', 'claude-3-opus-20240229')
+     */
     model: string;
 
+    /**
+     * Organization ID associated with the request
+     */
     organization_id: string;
 
+    /**
+     * Cost of output tokens in USD
+     */
     output_cost: number;
 
+    /**
+     * Number of output tokens generated
+     */
     output_tokens: number;
 
+    /**
+     * LLM provider (e.g., 'openai', 'anthropic', 'google')
+     */
     provider: string;
 
+    /**
+     * Type of task (e.g., 'optimization', 'evaluation')
+     */
     task_type: string;
 
+    /**
+     * Unix timestamp when the request was made
+     */
     timestamp: number;
 
+    /**
+     * Total cost (input + output) in USD
+     */
     total_cost: number;
 
+    /**
+     * User ID who made the request
+     */
     user_id: string;
   }
 }
 
-export interface PromptAdaptationAdaptParams {
+export interface PromptAdaptationCreateParams {
   /**
    * List of field names that will be substituted into the template. Must match keys
    * in golden records
    */
   fields: Array<string>;
 
-  /**
-   * The model your current prompt is optimized for
-   */
-  origin_model: PromptAdaptationAdaptParams.OriginModel;
-
   /**
    * System prompt to use with the origin model. This sets the context and role for
    * the LLM
@@ -500,9 +712,9 @@ export interface PromptAdaptationAdaptParams {
 
   /**
    * List of models to adapt the prompt for. Maximum count depends on your
-   * subscription tier
+   * subscription tier (Free: 1, Starter: 3, Startup: 5, Enterprise: 10)
    */
-  target_models: Array<PromptAdaptationAdaptParams.TargetModel>;
+  target_models: Array<RequestProvider>;
 
   /**
    * User message template with placeholders for fields. Use curly braces for field
@@ -516,185 +728,43 @@ export interface PromptAdaptationAdaptParams {
 
   /**
    * Training examples (legacy parameter). Use train_goldens and test_goldens for
-   * better control
+   * better control. Minimum 25 examples
    */
-  goldens?: Array<PromptAdaptationAdaptParams.Golden> | null;
-
-  /**
-   * Optional baseline score for the origin model
-   */
-  origin_model_evaluation_score?: number | null;
-
-  /**
-   * Test examples for evaluation. Required if train_goldens is provided
-   */
-  test_goldens?: Array<PromptAdaptationAdaptParams.TestGolden> | null;
-
-  /**
-   * Training examples for prompt optimization. Minimum 5 examples required
-   */
-  train_goldens?: Array<PromptAdaptationAdaptParams.TrainGolden> | null;
-}
-
-export namespace PromptAdaptationAdaptParams {
-  /**
-   * The model your current prompt is optimized for
-   */
-  export interface OriginModel {
-    /**
-     * Model name (e.g., 'gpt-4o', 'claude-3-5-sonnet-20241022')
-     */
-    model: string;
-
-    /**
-     * Provider name (e.g., 'openai', 'anthropic', 'google')
-     */
-    provider: string;
-
-    /**
-     * Maximum context length for the model (required for custom models)
-     */
-    context_length?: number | null;
-
-    /**
-     * Input token price per million tokens in USD (required for custom models)
-     */
-    input_price?: number | null;
-
-    /**
-     * Whether this is a custom model not in Not Diamond's supported model list
-     */
-    is_custom?: boolean;
-
-    /**
-     * Average latency in seconds (required for custom models)
-     */
-    latency?: number | null;
-
-    /**
-     * Output token price per million tokens in USD (required for custom models)
-     */
-    output_price?: number | null;
-  }
+  goldens?: Array<GoldenRecord> | null;
 
   /**
    * Model for specifying an LLM provider in API requests.
    */
-  export interface TargetModel {
-    /**
-     * Model name (e.g., 'gpt-4o', 'claude-3-5-sonnet-20241022')
-     */
-    model: string;
-
-    /**
-     * Provider name (e.g., 'openai', 'anthropic', 'google')
-     */
-    provider: string;
-
-    /**
-     * Maximum context length for the model (required for custom models)
-     */
-    context_length?: number | null;
-
-    /**
-     * Input token price per million tokens in USD (required for custom models)
-     */
-    input_price?: number | null;
-
-    /**
-     * Whether this is a custom model not in Not Diamond's supported model list
-     */
-    is_custom?: boolean;
-
-    /**
-     * Average latency in seconds (required for custom models)
-     */
-    latency?: number | null;
-
-    /**
-     * Output token price per million tokens in USD (required for custom models)
-     */
-    output_price?: number | null;
-  }
-
-  /**
-   * A training or test example for prompt adaptation.
-   */
-  export interface Golden {
-    /**
-     * Dictionary mapping field names to their values. Keys must match the fields
-     * specified in the template
-     */
-    fields: { [key: string]: string };
-
-    /**
-     * Expected answer for supervised evaluation. Required for supervised metrics,
-     * optional for unsupervised
-     */
-    answer?: string | null;
-  }
+  origin_model?: RequestProvider | null;
 
   /**
-   * A training or test example for prompt adaptation.
+   * Optional baseline score for the origin model. If provided, can skip origin model
+   * evaluation
    */
-  export interface TestGolden {
-    /**
-     * Dictionary mapping field names to their values. Keys must match the fields
-     * specified in the template
-     */
-    fields: { [key: string]: string };
-
-    /**
-     * Expected answer for supervised evaluation. Required for supervised metrics,
-     * optional for unsupervised
-     */
-    answer?: string | null;
-  }
-
-  /**
-   * A training or test example for prompt adaptation.
-   */
-  export interface TrainGolden {
-    /**
-     * Dictionary mapping field names to their values. Keys must match the fields
-     * specified in the template
-     */
-    fields: { [key: string]: string };
-
-    /**
-     * Expected answer for supervised evaluation. Required for supervised metrics,
-     * optional for unsupervised
-     */
-    answer?: string | null;
-  }
-}
+  origin_model_evaluation_score?: number | null;
 
-export interface PromptAdaptationGetAdaptRunResultsParams {
   /**
-   * Path param:
+   * Test examples for evaluation. Required if train_goldens is provided. Used to
+   * measure final performance on held-out data
    */
-  user_id: string;
+  test_goldens?: Array<GoldenRecord> | null;
 
   /**
-   * Header param:
+   * Training examples for prompt optimization. Minimum 25 examples required. Cannot
+   * be used with 'goldens' parameter
    */
-  'x-token': string;
-}
-
-export interface PromptAdaptationGetAdaptRunsParams {
-  'x-token': string;
+  train_goldens?: Array<GoldenRecord> | null;
 }
 
 export declare namespace PromptAdaptation {
   export {
-    type AdaptationRunResults as AdaptationRunResults,
+    type GoldenRecord as GoldenRecord,
     type JobStatus as JobStatus,
-    type PromptAdaptationAdaptResponse as PromptAdaptationAdaptResponse,
-    type PromptAdaptationGetAdaptRunsResponse as PromptAdaptationGetAdaptRunsResponse,
+    type RequestProvider as RequestProvider,
+    type PromptAdaptationCreateResponse as PromptAdaptationCreateResponse,
+    type PromptAdaptationGetAdaptResultsResponse as PromptAdaptationGetAdaptResultsResponse,
     type PromptAdaptationGetAdaptStatusResponse as PromptAdaptationGetAdaptStatusResponse,
-    type PromptAdaptationRetrieveCostsResponse as PromptAdaptationRetrieveCostsResponse,
-    type PromptAdaptationAdaptParams as PromptAdaptationAdaptParams,
-    type PromptAdaptationGetAdaptRunResultsParams as PromptAdaptationGetAdaptRunResultsParams,
-    type PromptAdaptationGetAdaptRunsParams as PromptAdaptationGetAdaptRunsParams,
+    type PromptAdaptationGetCostsResponse as PromptAdaptationGetCostsResponse,
+    type PromptAdaptationCreateParams as PromptAdaptationCreateParams,
   };
 }