rocketcode

type ActiveTurnCheckpoint struct {
	TurnID                   string                     `json:"turn_id"`
	ConversationKey          string                     `json:"conversation_key"`
	Agent                    string                     `json:"agent"`
	Model                    string                     `json:"model"`
	DisplayModel             string                     `json:"display_model"`
	ReplayInput              []json.RawMessage          `json:"replay_input,omitempty"`
	OutputTrace              []json.RawMessage          `json:"output_trace,omitempty"`
	TokenUsage               *TokenUsage                `json:"token_usage,omitempty"`
	ResponseID               string                     `json:"response_id,omitempty"`
	OpenFunctionCalls        []FunctionCallCheckpoint   `json:"open_function_calls,omitempty"`
	CompletedFunctionOutputs []FunctionOutputCheckpoint `json:"completed_function_outputs,omitempty"`
}

type Agent struct {
	Name            string
	Description     string
	Model           string
	ReasoningEffort string
	Verbosity       string
	MaxRecursion    *int
	Guardrail       string
	Prompt          string
	Location        string
	Permission      PermissionSet
	Frontmatter     map[string]any
	FileMode        fs.FileMode
}

type AgentLoadResult struct {
	Agents Agents
	Errors []error
}

type AgentMaxRecursionError struct {
	Location string
	Err      error
}

type Agents struct {
	Items map[string]Agent
}

type Attachment struct {
	MIME     string `json:"mime"`
	Filename string `json:"filename,omitempty"`
	URL      string `json:"url"`
}

type BashCommand struct {
	Command     string
	Timeout     int
	Workdir     string
	Description string
}

type BashResult struct {
	Output  string
	Success bool
}

type ChatResponse struct {
	Kind     string
	Text     string
	Tool     *ToolDiagnostic
	Subagent *SubagentDiagnostic
	Provider *ProviderDiagnostic
}

type CheckpointSink interface {
	StartActiveTurn(context.Context, *ActiveTurnCheckpoint) error
	RecordProviderResponse(context.Context, *ActiveTurnCheckpoint) error
	RecordCompletedToolOutput(context.Context, *ActiveTurnCheckpoint) error
	RecordRecoveredReplay(context.Context, *ActiveTurnCheckpoint) error
	ClearCompletedTurn(context.Context, string) error
}

type ChildRunEvent struct {
	Kind  ChildRunKind
	Stage ChildRunStage
	Agent string
	Item  ChatResponse
}

type ChildRunKind string

type ChildRunLogger func(*ChildRunEvent)

type ChildRunStage string

type Config struct {
	Model                      shared.ResponsesModel
	AutoApproverModel          shared.ResponsesModel
	ReasoningEffort            shared.ReasoningEffort
	Diagnostics                bool
	ExperimentalStrongerSkills bool
	ExpandPromptShellCommands  PromptShellCommandExpansion
	CompactThreshold           int64
	CompactionSteering         string
	ParallelToolCalls          int
	ShellOutputDir             string
	SandboxedBash              bool
	AutoApprovePermissions     bool
	Observability              ObservabilityConfig
	ChildRunLogger             ChildRunLogger
	CheckpointSink             CheckpointSink
	CustomTools                []Tool
	ShellEnv                   map[string]string
}

type FunctionCallCheckpoint struct {
	CallID    string          `json:"call_id"`
	Name      string          `json:"name"`
	Arguments json.RawMessage `json:"arguments,omitempty"`
}

type FunctionOutputCheckpoint struct {
	CallID      string            `json:"call_id"`
	Name        string            `json:"name,omitempty"`
	ReplayInput []json.RawMessage `json:"replay_input,omitempty"`
}

type InertCheckpointSink struct{}

type Looper interface {
	Loop(ctx context.Context, input <-chan PromptInput, sessionIn iter.Seq2[SessionEntry, error], sessionOut func(SessionEntry) error, interrupts <-chan os.Signal) error
}

type ObservabilityConfig struct {
	Enabled     bool
	Tracer      trace.Tracer
	TraceConfig instrumentation.TraceConfig
}

type Operation string

type OperationError struct {
	Operation Operation
	Err       error
}

type PermissionAction string

type PermissionBucket struct {
	Name  string
	Rules []PermissionRule
}

type PermissionRule struct {
	Pattern  string
	Action   PermissionAction
	Reviewer string
}

type PermissionSet struct {
	Buckets []PermissionBucket
}

type PromptInput struct {
	// Role defaults to PromptInputRoleUser when empty.
	Role        PromptInputRole         `json:"role,omitempty"`
	Text        string                  `json:"text"`
	Attachments []Attachment            `json:"attachments,omitempty"`
	DirectSkill *PromptInputDirectSkill `json:"directSkill,omitempty"`
	// Responses receives user-visible response items for this prompt. The runtime
	// closes it after the prompt's turn reaches a terminal state.
	Responses chan<- ChatResponse `json:"-"`
}

type PromptInputDirectSkill struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments,omitempty"`
}

type PromptInputRole string

type PromptShellCommandExpansion struct {
	PrimaryPrompts  bool
	SubagentPrompts bool
	SkillPrompts    bool
	InputPrompts    bool
}

type ProviderDiagnostic struct {
	Phase          string            `json:"phase"`
	HTTPStatus     int               `json:"http_status,omitempty"`
	ResponseStatus string            `json:"response_status,omitempty"`
	Code           string            `json:"code,omitempty"`
	Type           string            `json:"type,omitempty"`
	Message        string            `json:"message,omitempty"`
	Attempt        int               `json:"attempt,omitempty"`
	RetryAfter     string            `json:"retry_after,omitempty"`
	ResponseID     string            `json:"response_id,omitempty"`
	Headers        map[string]string `json:"headers,omitempty"`
}

type Providers struct {
	OpenAI *openai.Client
}

type ReplayDecodeError struct {
	EntryIndex int
	ItemIndex  int
	Kind       string
	Cause      error
}

type Runtime = looper

type SessionEntry struct {
	Version     int               `json:"version"`
	Type        string            `json:"type"`
	Timestamp   time.Time         `json:"timestamp"`
	ResponseID  string            `json:"response_id,omitempty"`
	Model       string            `json:"model,omitempty"`
	TokenUsage  *TokenUsage       `json:"token_usage,omitempty"`
	ReplayInput []json.RawMessage `json:"replay_input,omitempty"`
	OutputTrace []json.RawMessage `json:"output_trace,omitempty"`
}

type Skill struct {
	Name          string
	Description   string
	License       string
	Compatibility string
	Metadata      map[string]any
	Location      string
	Content       string
}

type SkillLoadResult struct {
	Skills Skills
	Errors []error
}

type Skills struct {
	Root  string
	Items map[string]Skill
	Dirs  []string
	// contains filtered or unexported fields
}

type SubagentDiagnostic struct {
	Name     string              `json:"name"`
	Label    string              `json:"label"`
	Index    int                 `json:"index,omitempty"`
	Total    int                 `json:"total,omitempty"`
	Text     string              `json:"text,omitempty"`
	Tool     *ToolDiagnostic     `json:"tool,omitempty"`
	Subagent *SubagentDiagnostic `json:"subagent,omitempty"`
	Provider *ProviderDiagnostic `json:"provider,omitempty"`
}

type TokenUsage struct {
	PromptTokens              int64 `json:"prompt_tokens,omitempty"`
	CompletionTokens          int64 `json:"completion_tokens,omitempty"`
	TotalTokens               int64 `json:"total_tokens,omitempty"`
	PromptCacheReadTokens     int64 `json:"prompt_cache_read_tokens,omitempty"`
	PromptCacheWriteTokens    int64 `json:"prompt_cache_write_tokens,omitempty"`
	CompletionReasoningTokens int64 `json:"completion_reasoning_tokens,omitempty"`
}

type Tool struct {
	Name               string
	Description        string
	Parameters         map[string]any
	Permission         string
	VisibilitySubjects []string
	Subjects           func(json.RawMessage) ([]string, error)
	Call               func(context.Context, json.RawMessage, chan<- ChatResponse) (ToolResult, error)
}

type ToolDiagnostic struct {
	Phase     string          `json:"phase"`
	Name      string          `json:"name"`
	Arguments json.RawMessage `json:"arguments,omitempty"`
	Result    string          `json:"result,omitempty"`
	Status    string          `json:"status,omitempty"`
	Action    json.RawMessage `json:"action,omitempty"`
}

type ToolResult struct {
	Output      string
	Attachments []Attachment
}