SagemakerJobRuntimeClient

Agentic RFT Runtime Service - manages trajectory and transition data for reinforcement fine-tuning jobs.

Properties

Link copied to clipboard

SagemakerJobRuntimeClient's configuration

Functions

Link copied to clipboard

Marks a rollout as complete, indicating that no further turns will be appended to the trajectory. After calling this operation, the trajectory is sealed and eligible for reward submission via the UpdateReward operation.

Link copied to clipboard
abstract suspend fun sample(input: SampleRequest): SampleResponse

Sends an inference request to the model during a job execution. The request and response bodies are forwarded to and from the model without modification. Each turn (prompt and response) is captured for later use.

Link copied to clipboard

Sends a streaming inference request to the model during a job execution. Returns the response as a stream of payload chunks. Each turn is captured for later use.

Link copied to clipboard

Updates the reward values for a trajectory and transitions it to reward-received status, signaling that it is eligible for processing. Call this operation after CompleteRollout to provide the computed reward scores.

Inherited functions

Link copied to clipboard
expect abstract fun close()
Link copied to clipboard

Marks a rollout as complete, indicating that no further turns will be appended to the trajectory. After calling this operation, the trajectory is sealed and eligible for reward submission via the UpdateReward operation.

Link copied to clipboard
inline suspend fun SagemakerJobRuntimeClient.sample(crossinline block: SampleRequest.Builder.() -> Unit): SampleResponse

Sends an inference request to the model during a job execution. The request and response bodies are forwarded to and from the model without modification. Each turn (prompt and response) is captured for later use.

Link copied to clipboard

Updates the reward values for a trajectory and transitions it to reward-received status, signaling that it is eligible for processing. Call this operation after CompleteRollout to provide the computed reward scores.

Link copied to clipboard

Create a copy of the client with one or more configuration values overridden. This method allows the caller to perform scoped config overrides for one or more client operations.