llmprotocol

type Capability string

type CapabilitySet map[Capability]bool

type IRContentPart struct {
	ID           string                 `json:"id,omitempty"`
	Type         IRPartType             `json:"type"`
	Text         string                 `json:"text,omitempty"`
	ToolCall     *IRToolCallPart        `json:"tool_call,omitempty"`
	ToolResult   *IRToolResultPart      `json:"tool_result,omitempty"`
	Reasoning    *IRReasoningPart       `json:"reasoning,omitempty"`
	Refusal      *IRRefusalPart         `json:"refusal,omitempty"`
	Metadata     map[string]string      `json:"metadata,omitempty"`
	CacheControl map[string]interface{} `json:"cache_control,omitempty"`
}

type IRMessage struct {
	ID    string          `json:"id,omitempty"`
	Role  IRRole          `json:"role"`
	Name  string          `json:"name,omitempty"`
	Parts []IRContentPart `json:"parts,omitempty"`
}

type IRPartType string

type IRReasoningPart struct {
	Content          string `json:"content"`
	Signature        string `json:"signature,omitempty"`
	Subtype          string `json:"subtype,omitempty"`
	EncryptedContent string `json:"encrypted_content,omitempty"`
}

type IRRefusalPart struct {
	Text string `json:"text"`
}

type IRRequest struct {
	ID                string                            `json:"id,omitempty"`
	Model             string                            `json:"model"`
	Stream            bool                              `json:"stream,omitempty"`
	User              string                            `json:"user,omitempty"`
	System            string                            `json:"system,omitempty"`
	SystemParts       []IRSystemPart                    `json:"system_parts,omitempty"`
	Instructions      string                            `json:"instructions,omitempty"`
	Messages          []IRMessage                       `json:"messages,omitempty"`
	Tools             []IRToolDecl                      `json:"tools,omitempty"`
	ToolChoice        *IRToolChoice                     `json:"tool_choice,omitempty"`
	ResponseFormat    *IRResponseFormat                 `json:"response_format,omitempty"`
	ReasoningEffort   string                            `json:"reasoning_effort,omitempty"`
	Thinking          *IRThinkingConfig                 `json:"thinking,omitempty"`
	MaxTokens         int                               `json:"max_tokens,omitempty"`
	Temperature       *float64                          `json:"temperature,omitempty"`
	TopP              *float64                          `json:"top_p,omitempty"`
	TopK              *int                              `json:"top_k,omitempty"`
	FrequencyPenalty  *float64                          `json:"frequency_penalty,omitempty"`
	PresencePenalty   *float64                          `json:"presence_penalty,omitempty"`
	Stop              []string                          `json:"stop,omitempty"`
	Seed              *int                              `json:"seed,omitempty"`
	ParallelToolCalls *bool                             `json:"parallel_tool_calls,omitempty"`
	StreamOptions     map[string]interface{}            `json:"stream_options,omitempty"`
	Metadata          map[string]interface{}            `json:"metadata,omitempty"`
	Store             *bool                             `json:"store,omitempty"`
	Include           []string                          `json:"include,omitempty"`
	CacheControl      map[string]interface{}            `json:"cache_control,omitempty"`
	Extensions        map[string]map[string]interface{} `json:"-"` // per-protocol, not serialized
}

type IRResponse struct {
	ID         string                            `json:"id"`
	Model      string                            `json:"model"`
	Created    int64                             `json:"created,omitempty"`
	Content    []IRContentPart                   `json:"content,omitempty"`
	Usage      *IRUsage                          `json:"usage,omitempty"`
	StopReason IRStopReason                      `json:"stop_reason,omitempty"`
	Extensions map[string]map[string]interface{} `json:"-"`
}

type IRResponseFormat struct {
	Type       string                 `json:"type"`
	JSONSchema map[string]interface{} `json:"json_schema,omitempty"`
}

type IRRole string

type IRStopReason string

type IRStreamEvent struct {
	Type          IRStreamEventType `json:"type"`
	ResponseID    string            `json:"response_id,omitempty"`
	ResponseModel string            `json:"response_model,omitempty"`
	Index         int               `json:"index,omitempty"`
	Part          *IRContentPart    `json:"part,omitempty"`
	DeltaText     string            `json:"delta_text,omitempty"`
	DeltaJSON     string            `json:"delta_json,omitempty"`
	DeltaType     string            `json:"delta_type,omitempty"`
	StopReason    IRStopReason      `json:"stop_reason,omitempty"`
	Usage         *IRUsage          `json:"usage,omitempty"`
	ErrorMessage  string            `json:"error_message,omitempty"`
	ErrorType     string            `json:"error_type,omitempty"`
}

type IRStreamEventType string

type IRSystemPart struct {
	Type         string                 `json:"type"`
	Text         string                 `json:"text,omitempty"`
	CacheControl map[string]interface{} `json:"cache_control,omitempty"`
}

type IRThinkingConfig struct {
	Type             string `json:"type,omitempty"`          // enabled, disabled, adaptive
	BudgetTokens     int    `json:"budget_tokens,omitempty"` // explicit token budget
	Effort           string `json:"effort,omitempty"`        // none, minimal, low, medium, high, xhigh, max
	Level            string `json:"level,omitempty"`         // Gemini thinkingLevel
	IncludeThoughts  *bool  `json:"include_thoughts,omitempty"`
	DynamicBudget    bool   `json:"dynamic_budget,omitempty"`
	Display          string `json:"display,omitempty"`
	EncryptedContent string `json:"encrypted_content,omitempty"`
	Subtype          string `json:"subtype,omitempty"`
}

type IRToolCallPart struct {
	ID            string                 `json:"id"`
	Name          string                 `json:"name"`
	ArgumentsRaw  string                 `json:"arguments_raw,omitempty"`
	ArgumentsJSON map[string]interface{} `json:"arguments_json,omitempty"`
	Status        string                 `json:"status,omitempty"` // in_progress, completed, failed
}

type IRToolChoice struct {
	Type string `json:"type"`
	Name string `json:"name,omitempty"`
}

type IRToolDecl struct {
	Type         string                 `json:"type"`
	Name         string                 `json:"name"`
	Description  string                 `json:"description,omitempty"`
	Parameters   interface{}            `json:"parameters,omitempty"` // JSON Schema — acceptable here
	CacheControl map[string]interface{} `json:"cache_control,omitempty"`
}

type IRToolResultPart struct {
	ToolCallID string          `json:"tool_call_id"`
	Content    []IRContentPart `json:"content,omitempty"`
	Status     string          `json:"status,omitempty"`
	Error      string          `json:"error,omitempty"`
}

type IRUsage struct {
	InputTokens  int `json:"input_tokens"`
	OutputTokens int `json:"output_tokens"`
	TotalTokens  int `json:"total_tokens"`

	CacheReadInputTokens     int `json:"cache_read_input_tokens,omitempty"`
	CacheCreationInputTokens int `json:"cache_creation_input_tokens,omitempty"`

	ReasoningTokens int `json:"reasoning_tokens,omitempty"`

	ProviderRaw map[string]interface{} `json:"-"`
}

type ItemType int

type Protocol string

type ProtocolAdapter interface {
	// Protocol returns which protocol this adapter handles.
	Protocol() Protocol

	// DecodeRequest converts a raw protocol-specific request to canonical IR.
	DecodeRequest(raw map[string]interface{}) (*IRRequest, error)

	// EncodeRequest converts canonical IR to protocol-specific request format.
	EncodeRequest(ir *IRRequest) (map[string]interface{}, error)

	// DecodeResponse converts a raw protocol-specific response to canonical IR.
	DecodeResponse(raw map[string]interface{}) (*IRResponse, error)

	// EncodeResponse converts canonical IR to protocol-specific response format.
	EncodeResponse(ir *IRResponse) (map[string]interface{}, error)

	// NewStreamState creates a new opaque stream state for this adapter.
	// The returned value is owned entirely by this adapter.
	NewStreamState() interface{}

	// DecodeStreamEvent converts a raw SSE data payload to IR stream events.
	// May return multiple events from a single raw event (e.g., Gemini multi-part).
	DecodeStreamEvent(raw map[string]interface{}, state interface{}) ([]*IRStreamEvent, error)

	// EncodeStreamEvent converts an IR stream event to protocol-specific SSE payloads.
	// May return multiple payloads from a single IR event (e.g., Responses text.done+part.done+item.done).
	EncodeStreamEvent(ir *IRStreamEvent, state interface{}) ([]map[string]interface{}, error)
}

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

type StreamContext struct {
	ResponseID string
	Model      string
	Usage      *IRUsage
	State      interface{} // adapter-owned opaque state
}

type Warning struct {
	Field   string `json:"field"`
	Message string `json:"message"`
}