protocolv2

type Block struct {
	Type       BlockType   `json:"type"`
	Text       *Text       `json:"text,omitempty"`
	Thinking   *Thinking   `json:"thinking,omitempty"`
	ToolUse    *ToolUse    `json:"tool_use,omitempty"`
	ToolResult *ToolResult `json:"tool_result,omitempty"`
}

type BlockType string

type ContextEntity struct {
	EntityType ContextEntityType `json:"entity_type"`
	EntityID   string            `json:"entity_id"`
}

type ContextEntityType string

type Message struct {
	Role       Role        `json:"role"`
	Content    []Block     `json:"content"`
	Model      string      `json:"model,omitempty"`
	StopReason *StopReason `json:"stop_reason,omitempty"`
}

type Request struct {
	ChatProtocolVersion string          `json:"chat_protocol_version"`
	ConversationID      string          `json:"conversation_id"`
	Messages            []Message       `json:"messages"`
	ContextEntities     []ContextEntity `json:"context_entities,omitempty"`
	Tools               []Tool          `json:"tools,omitempty"`
}

type Role string

type StopReason string

type Text struct {
	Content string `json:"content"`
}

type Thinking struct {
	Content string `json:"content"`
}

type Tool struct {
	Name        string         `json:"name"`
	Description string         `json:"description"`
	InputSchema map[string]any `json:"input_schema"`
}

type ToolResult struct {
	ToolUseID string          `json:"tool_use_id"`
	IsError   *bool           `json:"is_error"`
	Error     string          `json:"error,omitempty"`
	Content   json.RawMessage `json:"content,omitempty"`
}

type ToolUse struct {
	ID    string          `json:"id"`
	Name  string          `json:"name"`
	Input json.RawMessage `json:"input"`
}