clearcast

type Agent struct {
	ID        string
	Prompt    string
	Iteration int
	Model     string
	// contains filtered or unexported fields
}

type AgentChoiceEvent struct {
	Content string `json:"content"`
	Usage   Usage  `json:"usage"`
}

type AgentDef struct {
	ID       string    `yaml:"id"`
	Mode     string    `yaml:"mode"`
	Model    string    `yaml:"model"`
	Prompt   string    `yaml:"prompt"`
	Toolsets []Toolset `yaml:"toolsets,omitempty"` // Tools specific to this agent
}

type AgentOption func(*Agent)

type AgentReasoningEvent struct {
	Content string `json:"content"`
	Usage   Usage  `json:"usage"`
}

type Agents map[string]*Agent

type ChatCompletionChunk struct {
	Delta string // partial text delta
	Done  bool
	Raw   any
	Usage Usage
}

type ChatCompletionRequest struct {
	Model          string
	Messages       []ChatMessage
	Options        map[string]any
	ResponseFormat *ResponseFormat
}

type ChatCompletionResponse struct {
	Content string
	Raw     any // keep raw provider response if needed
	Usage   Usage
}

type ChatMessage struct {
	Role    string
	Content string
}

type CompletionTokenDetails struct {
	ReasoningTokens int `json:"reasoning_tokens"`
}

type CostDetails struct {
	UpstreamInferenceCost float64 `json:"upstream_inference_cost"`
}

type DebugContextKey struct{}

type ErrorEvent struct {
	Type  string `json:"type"`
	Error string `json:"error"`
}

type EventKind string

type ExecuteFunc func(ctx context.Context, params map[string]any) (any, error)

type FinalEvent struct {
	Output string `json:"output"`
}

type JSONSchema struct {
	Name        string         `json:"name"`
	Description string         `json:"description,omitempty"`
	Schema      json.Marshaler `json:"schema"`
	Strict      bool           `json:"strict"`
}

type LLMProvider interface {
	ChatCompletion(ctx context.Context, req ChatCompletionRequest) (ChatCompletionResponse, error)
	ChatCompletionStream(ctx context.Context, req ChatCompletionRequest) (<-chan ChatCompletionChunk, error)
}

type Message struct {
	Role    string
	Content string
	// Summaryy is the summary of all the messages up to this point, act as a compaction
	Summary string
}

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

type OrchestrationFile struct {
	Agents  []AgentDef `yaml:"agents"`
	Session SessionDef `yaml:"session"`
}

type Plan struct {
	// Kind can be either "tool" or "agent"
	// Kind   string         `json:"kind"`
	Tool   string         `json:"tool"`
	Params map[string]any `json:"params"`
}

type PlanResultEvent struct {
	Plans []Plan `json:"plans"`
}

type PromptTokenDetails struct {
	CachedTokens int `json:"cached_tokens"`
}

type RemoteToolset struct {
	URL           string `yaml:"url"`
	TransportType string `yaml:"transport_type"`
}

type ResponseFormat struct {
	Type       ResponseFormatType
	JSONSchema *JSONSchema
}

type ResponseFormatType string

type ResponseOption func(*ChatCompletionRequest)

type Runtime interface {
	Run(ctx context.Context) (string, error)
	RunStream(ctx context.Context) <-chan RuntimeEvent
}

type RuntimeEvent interface {
	// contains filtered or unexported methods
}

type RuntimeOption func(*runtime)

type Session struct {
	RootAgentID   string         `json:"root_agent_id"`
	Workspace     map[string]any `json:"workspace"`
	Mission       string         `json:"mission"`
	Memory        map[string]any `json:"memory"`
	Messages      []Message      `json:"messages"`
	CreatedAt     time.Time      `json:"created_at"`
	MaxIterations int            `json:"max_iterations"`

	InputTokens  int     `json:"input_tokens"`
	OutputTokens int     `json:"output_tokens"`
	Cost         float64 `json:"cost"`
	Tools        []*Tool `json:"tools"`
}

type SessionDef struct {
	RootAgentID string         `yaml:"root_agent_id"`
	Extra       map[string]any `yaml:",inline"` // Captures all other fields like topic, etc.

}

type Tool struct {
	ID          string      `json:"id"`
	Description string      `json:"description"`
	Execute     ExecuteFunc `json:"execute"`
	Usage       string      `json:"usage"`
}

type ToolErrorEvent struct {
	Tool   string `json:"tool"`
	Error  string `json:"error"`
	Params any    `json:"params"`
}

type ToolResultEvent struct {
	Tool   string `json:"tool"`
	Result any    `json:"result"`
	Error  string `json:"error"`
}

type Tools map[string]*Tool

type Toolset struct {
	Type        string         `yaml:"type"`
	ID          string         `yaml:"id"`
	Remote      *RemoteToolset `yaml:"remote,omitempty"`
	Prompt      string         `yaml:"prompt,omitempty"`
	Description string         `yaml:"description,omitempty"`
	Model       string         `yaml:"model,omitempty"`
}

type TransferAgentEvent struct {
	AgentID string `json:"agent_id"`
}

type Usage struct {
	PromptTokens           int                    `json:"prompt_tokens"`
	CompletionTokens       int                    `json:"completion_tokens"`
	CompletionTokenDetails CompletionTokenDetails `json:"completion_token_details"`
	TotalTokens            int                    `json:"total_tokens"`
	Cost                   float64                `json:"cost"`
	CostDetails            CostDetails            `json:"cost_details"`
	PromptTokenDetails     PromptTokenDetails     `json:"prompt_token_details"`
}