restate

type AgentEventLog struct {
	// contains filtered or unexported fields
}

type AgentLoop struct {
	// contains filtered or unexported fields
}

type AgentLoopInput struct {

	// StreamingEnabled requests token-level LLM streaming.
	// Only active when IsStreamHandler is also true (the Stream handler enables both).
	StreamingEnabled bool
	// IsStreamHandler is true when this input is for the Stream handler path.
	// Stream runs publish events to AgentEventLog; Run runs publish only approval events.
	IsStreamHandler bool
	// Tools is the resolved tool list for this run (not on the Restate wire payload).
	Tools []interfaces.Tool
	// contains filtered or unexported fields
}

type AgentLoopRequest struct {
	AgentName        string `json:"agent_name,omitempty"`
	LLMStreamEnabled bool   `json:"llm_stream_enabled"`
	StreamHandler    bool   `json:"stream_handler"`
	// contains filtered or unexported fields
}

type AgentLoopResponse struct {
	Result *types.AgentRunResult `json:"result,omitempty"`
}

type AgentLoopResult struct {
	Content   string
	LLMUsage  *interfaces.LLMUsage
	Telemetry *types.AgentTelemetry
}

type CancelRequest struct {
	RunID        string `json:"run_id,omitempty"`
	InvocationID string `json:"invocation_id"`
}

type EndpointConfig struct {
	// ListenAddress is the bind address for the SDK endpoint (e.g. ":9080").
	// Empty defaults to ":9080".
	ListenAddress string
	// IdentityPublicKeys are optional Restate request-identity public keys
	// used to verify inbound requests from Restate.
	IdentityPublicKeys []string
	// AdminURL is the Restate admin API base URL (e.g. "http://localhost:9070").
	// When set, NewRestateRuntime registers this endpoint automatically after startup.
	AdminURL string
	// DeploymentURL is the URL Restate uses to call back into this process.
	// Empty defaults to http://127.0.0.1 + ListenAddress port. Override when Restate
	// runs in a container with a different network (e.g. "http://host.docker.internal:9080").
	DeploymentURL string
}

type EventLogConfig struct {
	// DisableClear skips scheduling AgentEventLog/Clear after a root run.
	// Default false means cleanup is on.
	DisableClear bool
	// TTL is how long to wait after a root Run/Stream completes before Clear.
	// Used only when DisableClear is false. Zero defaults to 90 seconds.
	TTL time.Duration
}

type IngressConfig struct {
	// URL is the Restate ingress base URL (e.g. "http://localhost:8080").
	URL string
	// AuthKey is an optional bearer token for authenticated ingress (e.g. Restate Cloud).
	AuthKey string
	// HTTPTimeout is the per-attempt timeout for short ingress RPCs.
	// Zero defaults to 30s. Long-running Attach is not bounded by this field.
	HTTPTimeout time.Duration
	// HTTPMaxAttempts is the retry budget for transient failures (network, 429/5xx).
	// Zero defaults to 3.
	HTTPMaxAttempts int
}

type Option func(*RestateRuntime)

type PullRequest struct {
	// Offset is the next message index to read. Nil means wait for new messages at tail
	// (no catch-up). Zero is a valid start-of-stream offset.
	Offset *int64 `json:"offset,omitempty"`
	// Wait, when true, long-polls up to pubsubPullTimeout (Restate PubSub convention).
	// SDK stream/approval listeners use Wait=false so Pull always completes quickly and
	// is never left suspended when the process exits.
	Wait bool `json:"wait,omitempty"`
}

type PullResponse struct {
	Messages   []json.RawMessage `json:"messages"`
	NextOffset int64             `json:"nextOffset"`
}

type RestateConfig struct {
	Ingress  IngressConfig
	Endpoint EndpointConfig
	// EventLog controls post-run cleanup of the per-run [AgentEventLog] virtual object.
	EventLog EventLogConfig
}

type RestateRuntime struct {
	base.Runtime
	// contains filtered or unexported fields
}

type RuntimeFactory struct {
	Config *RestateConfig
}

type SubAgentRoute struct {
	Name        string                   `json:"name"`
	ToolName    string                   `json:"tool_name"`
	ServiceName string                   `json:"service_name,omitempty"`
	ChildRoutes map[string]SubAgentRoute `json:"child_routes,omitempty"`
}

type ToolsResolver func(ctx context.Context) ([]interfaces.Tool, error)