feat: Enhance AuthAgentInvocation with step and last activity tracking

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 89
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/kernel%2Fkernel-13214b99e392aab631aa1ca99b6a51a58df81e34156d21b8d639bea779566123.yml
-openapi_spec_hash: a88d175fc3980de3097ac1411d8dcbff
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/kernel%2Fkernel-2cf104c4b88159c6a50713b019188f83983748b9cacec598089cf9068dc5b1cd.yml
+openapi_spec_hash: 84ea30ae65ad7ebcc04d2f3907d1e73b
 config_hash: 179f33af31ece83563163d5b3d751d13

agentauth.go

@@ -166,13 +166,19 @@ type AgentAuthInvocationResponse struct {
 	//
 	// Any of "IN_PROGRESS", "SUCCESS", "EXPIRED", "CANCELED".
 	Status AgentAuthInvocationResponseStatus `json:"status,required"`
+	// Current step in the invocation workflow
+	//
+	// Any of "initialized", "discovering", "awaiting_input", "submitting",
+	// "completed", "expired".
+	Step AgentAuthInvocationResponseStep `json:"step,required"`
 	// Target domain for authentication
 	TargetDomain string `json:"target_domain,required"`
 	// JSON contains metadata for fields, check presence with [respjson.Field.Valid].
 	JSON struct {
 		AppName      respjson.Field
 		ExpiresAt    respjson.Field
 		Status       respjson.Field
+		Step         respjson.Field
 		TargetDomain respjson.Field
 		ExtraFields  map[string]respjson.Field
 		raw          string
@@ -195,6 +201,18 @@ const (
 	AgentAuthInvocationResponseStatusCanceled   AgentAuthInvocationResponseStatus = "CANCELED"
 )
 
+// Current step in the invocation workflow
+type AgentAuthInvocationResponseStep string
+
+const (
+	AgentAuthInvocationResponseStepInitialized   AgentAuthInvocationResponseStep = "initialized"
+	AgentAuthInvocationResponseStepDiscovering   AgentAuthInvocationResponseStep = "discovering"
+	AgentAuthInvocationResponseStepAwaitingInput AgentAuthInvocationResponseStep = "awaiting_input"
+	AgentAuthInvocationResponseStepSubmitting    AgentAuthInvocationResponseStep = "submitting"
+	AgentAuthInvocationResponseStepCompleted     AgentAuthInvocationResponseStep = "completed"
+	AgentAuthInvocationResponseStepExpired       AgentAuthInvocationResponseStep = "expired"
+)
+
 // Response from submit endpoint matching SubmitResult schema
 type AgentAuthSubmitResponse struct {
 	// Whether submission succeeded
@@ -359,7 +377,7 @@ func (r *AuthAgentInvocationCreateRequestParam) UnmarshalJSON(data []byte) error
 //
 // Use the methods beginning with 'As' to cast the union to one of its variants.
 type AuthAgentInvocationCreateResponseUnion struct {
-	// Any of "already_authenticated", "invocation_created".
+	// Any of "ALREADY_AUTHENTICATED", "INVOCATION_CREATED".
 	Status string `json:"status"`
 	// This field is from variant [AuthAgentInvocationCreateResponseInvocationCreated].
 	ExpiresAt time.Time `json:"expires_at"`
@@ -401,9 +419,9 @@ func (AuthAgentInvocationCreateResponseInvocationCreated) implAuthAgentInvocatio
 //	}
 func (u AuthAgentInvocationCreateResponseUnion) AsAny() anyAuthAgentInvocationCreateResponse {
 	switch u.Status {
-	case "already_authenticated":
+	case "ALREADY_AUTHENTICATED":
 		return u.AsAlreadyAuthenticated()
-	case "invocation_created":
+	case "INVOCATION_CREATED":
 		return u.AsInvocationCreated()
 	}
 	return nil
@@ -526,7 +544,7 @@ const (
 type ReauthResponse struct {
 	// Result of the re-authentication attempt
 	//
-	// Any of "reauth_started", "already_authenticated", "cannot_reauth".
+	// Any of "REAUTH_STARTED", "ALREADY_AUTHENTICATED", "CANNOT_REAUTH".
 	Status ReauthResponseStatus `json:"status,required"`
 	// ID of the re-auth invocation if one was created
 	InvocationID string `json:"invocation_id"`
@@ -552,9 +570,9 @@ func (r *ReauthResponse) UnmarshalJSON(data []byte) error {
 type ReauthResponseStatus string
 
 const (
-	ReauthResponseStatusReauthStarted        ReauthResponseStatus = "reauth_started"
-	ReauthResponseStatusAlreadyAuthenticated ReauthResponseStatus = "already_authenticated"
-	ReauthResponseStatusCannotReauth         ReauthResponseStatus = "cannot_reauth"
+	ReauthResponseStatusReauthStarted        ReauthResponseStatus = "REAUTH_STARTED"
+	ReauthResponseStatusAlreadyAuthenticated ReauthResponseStatus = "ALREADY_AUTHENTICATED"
+	ReauthResponseStatusCannotReauth         ReauthResponseStatus = "CANNOT_REAUTH"
 )
 
 type AgentAuthNewParams struct {

agentauthinvocation.go

@@ -47,8 +47,8 @@ func (r *AgentAuthInvocationService) New(ctx context.Context, body AgentAuthInvo
 	return
 }
 
-// Returns invocation details including app_name and target_domain. Uses the JWT
-// returned by the exchange endpoint, or standard API key or JWT authentication.
+// Returns invocation details including status, app_name, and target_domain.
+// Supports both API key and JWT (from exchange endpoint) authentication.
 func (r *AgentAuthInvocationService) Get(ctx context.Context, invocationID string, opts ...option.RequestOption) (res *AgentAuthInvocationResponse, err error) {
 	opts = slices.Concat(r.Options, opts)
 	if invocationID == "" {
@@ -62,7 +62,7 @@ func (r *AgentAuthInvocationService) Get(ctx context.Context, invocationID strin
 
 // Inspects the target site to detect logged-in state or discover required fields.
 // Returns 200 with success: true when fields are found, or 4xx/5xx for failures.
-// Requires the JWT returned by the exchange endpoint.
+// Supports both API key and JWT (from exchange endpoint) authentication.
 func (r *AgentAuthInvocationService) Discover(ctx context.Context, invocationID string, body AgentAuthInvocationDiscoverParams, opts ...option.RequestOption) (res *AgentAuthDiscoverResponse, err error) {
 	opts = slices.Concat(r.Options, opts)
 	if invocationID == "" {
@@ -88,7 +88,8 @@ func (r *AgentAuthInvocationService) Exchange(ctx context.Context, invocationID
 }
 
 // Submits field values for the discovered login form and may return additional
-// auth fields or success. Requires the JWT returned by the exchange endpoint.
+// auth fields or success. Supports both API key and JWT (from exchange endpoint)
+// authentication.
 func (r *AgentAuthInvocationService) Submit(ctx context.Context, invocationID string, body AgentAuthInvocationSubmitParams, opts ...option.RequestOption) (res *AgentAuthSubmitResponse, err error) {
 	opts = slices.Concat(r.Options, opts)
 	if invocationID == "" {

shared/constant/constants.go

@@ -18,22 +18,22 @@ func ValueOf[T Constant[T]]() T {
 	return t.Default()
 }
 
-type AlreadyAuthenticated string // Always "already_authenticated"
+type AlreadyAuthenticated string // Always "ALREADY_AUTHENTICATED"
 type AppVersionSummary string    // Always "app_version_summary"
 type AwsUsEast1a string          // Always "aws.us-east-1a"
 type DeploymentState string      // Always "deployment_state"
 type Error string                // Always "error"
-type InvocationCreated string    // Always "invocation_created"
+type InvocationCreated string    // Always "INVOCATION_CREATED"
 type InvocationState string      // Always "invocation_state"
 type Log string                  // Always "log"
 type SseHeartbeat string         // Always "sse_heartbeat"
 
-func (c AlreadyAuthenticated) Default() AlreadyAuthenticated { return "already_authenticated" }
+func (c AlreadyAuthenticated) Default() AlreadyAuthenticated { return "ALREADY_AUTHENTICATED" }
 func (c AppVersionSummary) Default() AppVersionSummary       { return "app_version_summary" }
 func (c AwsUsEast1a) Default() AwsUsEast1a                   { return "aws.us-east-1a" }
 func (c DeploymentState) Default() DeploymentState           { return "deployment_state" }
 func (c Error) Default() Error                               { return "error" }
-func (c InvocationCreated) Default() InvocationCreated       { return "invocation_created" }
+func (c InvocationCreated) Default() InvocationCreated       { return "INVOCATION_CREATED" }
 func (c InvocationState) Default() InvocationState           { return "invocation_state" }
 func (c Log) Default() Log                                   { return "log" }
 func (c SseHeartbeat) Default() SseHeartbeat                 { return "sse_heartbeat" }