models

type CreateSessionRequest struct {
	State  map[string]any `json:"state"`
	Events []Event        `json:"events"`
}

type Event struct {
	ID                  string                                      `json:"id"`
	InvocationID        string                                      `json:"invocationId"`
	Branch              string                                      `json:"branch,omitempty"`
	Author              string                                      `json:"author"`
	Partial             bool                                        `json:"partial,omitempty"`
	LongRunningToolIDs  []string                                    `json:"longRunningToolIds,omitempty"`
	Content             *genai.Content                              `json:"content"`
	GroundingMetadata   *genai.GroundingMetadata                    `json:"groundingMetadata"`
	UsageMetadata       *genai.GenerateContentResponseUsageMetadata `json:"usageMetadata"`
	TurnComplete        bool                                        `json:"turnComplete,omitempty"`
	Interrupted         bool                                        `json:"interrupted,omitempty"`
	ErrorCode           string                                      `json:"errorCode,omitempty"`
	ErrorMessage        string                                      `json:"errorMessage,omitempty"`
	AvgLogprobs         float64                                     `json:"avgLogprobs,omitempty"`
	FinishReason        genai.FinishReason                          `json:"finishReason,omitempty"`
	ModelVersion        string                                      `json:"modelVersion,omitempty"`
	InputTranscription  *genai.Transcription                        `json:"inputTranscription,omitempty"`
	OutputTranscription *genai.Transcription                        `json:"outputTranscription,omitempty"`
	Actions             EventActions                                `json:"actions"`
	Output              any                                         `json:"output,omitempty"`
	NodeInfo            *session.NodeInfo                           `json:"nodeInfo,omitempty"`
	IsolationScope      string                                      `json:"isolationScope,omitempty"`
	Routes              []string                                    `json:"routes,omitempty"`
	RequestedInput      *session.RequestInput                       `json:"requestedInput,omitempty"`
}

type EventActions struct {
	StateDelta                 map[string]any                               `json:"stateDelta"`
	ArtifactDelta              map[string]int64                             `json:"artifactDelta"`
	Escalate                   bool                                         `json:"escalate,omitempty"`
	SkipSummarization          bool                                         `json:"skipSummarization,omitempty"`
	TransferToAgent            string                                       `json:"transferToAgent,omitempty"`
	RequestedToolConfirmations map[string]toolconfirmation.ToolConfirmation `json:"requestedToolConfirmations,omitempty"`
}

type EventarcTriggerRequest struct {
	// Unique identifier for the event
	ID string `json:"id"`
	// Identifies the source of the event
	Source string `json:"source"`
	// The type of event data
	Type string `json:"type"`
	// The CloudEvents specification version used for this event
	SpecVersion string `json:"specversion"`
	// Event generation time, in RFC 3339 format (optional)
	Time string `json:"time,omitempty"`
	// In structured mode, “data“ is always present.
	// In binary mode, the entire body is the data (often a Pub/Sub wrapper).
	// But sometimes it can be just a JSON object.
	// E.g. https://googleapis.github.io/google-cloudevents/examples/binary/firestore/DocumentEventData-simple.json
	Data json.RawMessage `json:"data,omitempty"`
}

type LiveRequest struct {
	Content       *genai.Content       `json:"content,omitempty"`
	Blob          *blob                `json:"blob,omitempty"`
	ActivityStart *genai.ActivityStart `json:"activityStart,omitempty"`
	ActivityEnd   *genai.ActivityEnd   `json:"activityEnd,omitempty"`
	Close         bool                 `json:"close,omitempty"`
}

type PubSubMessage struct {
	// The message payload. This will always be a base64-encoded string.
	Data []byte `json:"data"`
	// ID of this message, assigned by the Pub/Sub server.
	MessageID string `json:"messageId"`
	// The time at which the message was published, populated by the server.
	PublishTime string `json:"publishTime"`
	// Optional attributes for this message. An object containing a list of 'key': 'value' string pairs.
	Attributes map[string]string `json:"attributes,omitempty"`
	// If message ordering is enabled, this identifies related messages for which publish order should be respected.
	OrderingKey string `json:"orderingKey,omitempty"`
}

type PubSubTriggerRequest struct {
	// The Pub/Sub message.
	Message PubSubMessage `json:"message"`
	// The subscription this message was published to.
	Subscription string `json:"subscription"`
}

type RunAgentRequest struct {
	AppName string `json:"appName"`

	UserId string `json:"userId"`

	SessionId string `json:"sessionId"`

	NewMessage genai.Content `json:"newMessage"`

	Streaming bool `json:"streaming,omitempty"`

	StateDelta *map[string]any `json:"stateDelta,omitempty"`

	// FunctionCallEventID identifies the event whose function call this
	// request answers (e.g. a human-in-the-loop reply from the web UI).
	// Accepted for adk-python parity but currently ignored; declared so
	// strict decoding (DisallowUnknownFields) does not reject the request.
	FunctionCallEventID *string `json:"functionCallEventId,omitempty"`
}

type Session struct {
	ID        string         `json:"id"`
	AppName   string         `json:"appName"`
	UserID    string         `json:"userId"`
	UpdatedAt int64          `json:"lastUpdateTime"`
	Events    []Event        `json:"events"`
	State     map[string]any `json:"state"`
}

type SessionID struct {
	ID      string `mapstructure:"session_id,optional"`
	AppName string `mapstructure:"app_name,required"`
	UserID  string `mapstructure:"user_id,required"`
}

type TriggerResponse struct {
	// Processing status: 'success' or error message.
	Status string `json:"status"`
}