responses

type Annotation struct {
	Type       string `json:"type"` // Any of "file_citation", "url_citation", "container_file_citation", "file_path".
	Title      string `json:"title"`
	URL        string `json:"url"`
	StartIndex int    `json:"start_index"`
	EndIndex   int    `json:"end_index"`

	ExtraParams map[string]any `json:"extra_params"`
}

type ChunkContentPart[T any] struct {
	Type           T                  `json:"type"`
	SequenceNumber int                `json:"sequence_number"`
	ItemId         string             `json:"item_id"`
	OutputIndex    int                `json:"output_index"`
	ContentIndex   int                `json:"content_index"`
	Part           OutputContentUnion `json:"part"`
}

type ChunkFunctionCall[T any] struct {
	Type T `json:"type"`

	SequenceNumber int    `json:"sequence_number"`
	ItemId         string `json:"item_id"`
	OutputIndex    int    `json:"output_index"`
	Delta          string `json:"delta"`
	Arguments      string `json:"arguments"`
	Obfuscation    string `json:"obfuscation"`
}

type ChunkImageGenerationCall[T any] struct {
	Type T `json:"type"`

	SequenceNumber int    `json:"sequence_number"`
	ItemId         string `json:"item_id"`
	OutputIndex    int    `json:"output_index"`

	// Only on response.image_generation_call.partial_image
	PartialImageIndex  int     `json:"partial_image_index"`
	PartialImageBase64 string  `json:"partial_image_b64"`
	Background         *string `json:"background,omitempty"`    // "opaque"
	OutputFormat       *string `json:"output_format,omitempty"` // "png"
	Quality            *string `json:"quality,omitempty"`       // "medium"
	Size               *string `json:"size,omitempty"`          // "1024x1024"
}

type ChunkOutputItem[T any] struct {
	Type           T                   `json:"type"`
	SequenceNumber int                 `json:"sequence_number"`
	OutputIndex    int                 `json:"output_index"`
	Item           ChunkOutputItemData `json:"item"`
}

type ChunkOutputItemData struct {
	Type string `json:"type"` // "function_call" , "message", "reasoning", "image_generation_call", "web_search_call"

	// Common fields
	Id     string `json:"id"`
	Status string `json:"status"`

	// For output_item of type "message"
	Content OutputContent  `json:"content"`
	Role    constants.Role `json:"role"`

	// For output_item of type "function_call"
	CallID           *string `json:"call_id,omitempty"`
	Name             *string `json:"name,omitempty"`
	Arguments        *string `json:"arguments,omitempty"`
	ThoughtSignature *string `json:"thought_signature,omitempty"` // Exception: Gemini function calls can have though signature

	// For "reasoning"
	EncryptedContent *string              `json:"encrypted_content,omitempty"`
	Summary          []SummaryTextContent `json:"summary,omitempty"`

	// For "image_generation_call"
	Background   *string `json:"background,omitempty"`    // "opaque"
	OutputFormat *string `json:"output_format,omitempty"` // "png"
	Quality      *string `json:"quality,omitempty"`       // "medium"
	Result       *string `json:"result,omitempty"`        // base64 image
	Size         *string `json:"size,omitempty"`          // "1024x1024"

	// For "web_search_call"
	Action *WebSearchCallActionUnion `json:"action,omitempty"`
}

type ChunkOutputText[T any] struct {
	Type           T             `json:"type"`
	SequenceNumber int           `json:"sequence_number"`
	ItemId         string        `json:"item_id"`
	OutputIndex    int           `json:"output_index"`
	ContentIndex   int           `json:"content_index"`
	Delta          string        `json:"delta"`
	Logprobs       []interface{} `json:"logprobs"`
	Obfuscation    string        `json:"obfuscation"`

	// Only on content.output_text.done (contains the accumulated content)
	Text *string `json:"text,omitempty"`

	// Only on response.output_text.annotation.added
	Annotation      Annotation `json:"annotation,omitempty"`
	AnnotationIndex int        `json:"annotation_index"`
}

type ChunkReasoningSummaryPart[T any] struct {
	Type           T                  `json:"type"`
	SequenceNumber int                `json:"sequence_number"`
	ItemId         string             `json:"item_id"`
	OutputIndex    int                `json:"output_index"`
	SummaryIndex   int                `json:"summary_index"`
	Part           SummaryTextContent `json:"part"`
}

type ChunkReasoningSummaryText[T any] struct {
	Type           T      `json:"type"`
	SequenceNumber int    `json:"sequence_number"`
	ItemId         string `json:"item_id"`
	OutputIndex    int    `json:"output_index"`
	SummaryIndex   int    `json:"summary_index"`

	// Only on response.reasoning_summary_text.delta
	Delta string `json:"delta"`

	// Only on response.reasoning_summary_text.done
	Text *string `json:"text,omitempty"`

	// Helpers - Anthropic sends signature as a separate delta with different type "signature_delta"
	// We would have been using "Delta" for storing the reasoning summary delta, and since we need another field
	// to store the signature, we use this field.
	// We don't use "Text" to avoid confusion.
	EncryptedContent *string `json:"encrypted_content,omitempty"`
}

type ChunkResponse[T any] struct {
	Type           T                 `json:"type"`
	SequenceNumber int               `json:"sequence_number"`
	Response       ChunkResponseData `json:"response"`
}

type ChunkResponseData struct {
	Id                string               `json:"id"`
	Object            string               `json:"object"`
	CreatedAt         int                  `json:"created_at"`
	Status            string               `json:"status"`
	Background        bool                 `json:"background"`
	Error             interface{}          `json:"error"`
	IncompleteDetails interface{}          `json:"incomplete_details"`
	Output            []OutputMessageUnion `json:"output"`
	Usage             Usage                `json:"usage"`
	Request
}

type ChunkResponseUsage struct {
	InputTokens        int `json:"input_tokens"`
	InputTokensDetails struct {
		CachedTokens int `json:"cached_tokens"`
	} `json:"input_tokens_details"`
	OutputTokens        int `json:"output_tokens"`
	OutputTokensDetails struct {
		ReasoningTokens int `json:"reasoning_tokens"`
	} `json:"output_tokens_details"`
	TotalTokens int `json:"total_tokens"`
}

type ChunkRun[T any] struct {
	Type           T            `json:"type"`
	SequenceNumber int          `json:"sequence_number"`
	RunState       ChunkRunData `json:"run_state"`
}

type ChunkRunData struct {
	Id               string                `json:"id"`
	Object           string                `json:"object"` // "run"
	Status           string                `json:"status"` // "created", "in_progress", "paused", "resumed", "completed", "aborted"
	PendingToolCalls []FunctionCallMessage `json:"pending_tool_calls"`
	Usage            Usage                 `json:"usage"`
	TraceID          string                `json:"traceid"`
}

type ChunkWebSearchCall[T any] struct {
	Type T `json:"type"`

	SequenceNumber int    `json:"sequence_number"`
	ItemId         string `json:"item_id"`
	OutputIndex    int    `json:"output_index"`
}

type EasyInputContentUnion struct {
	OfString           *string      `json:",omitempty"`
	OfInputMessageList InputContent `json:",omitempty"`
}

type EasyMessage struct {
	Type    constants.MessageTypeMessage `json:"type"` // Always "message".
	ID      string                       `json:"id,omitempty"`
	Role    constants.Role               `json:"role,omitempty"`
	Content EasyInputContentUnion        `json:"content"`
}

type Error struct {
	Type    string `json:"type"`
	Message string `json:"message"`
	Param   string `json:"param"`
	Code    string `json:"code"`
}

type FunctionCallApprovalResponseMessage struct {
	Type            constants.MessageTypeFunctionCallApprovalResponse `json:"type"`
	ID              string                                            `json:"id"`
	ApprovedCallIds []string                                          `json:"approved_call_ids"`
	RejectedCallIds []string                                          `json:"rejected_call_ids"`
}

type FunctionCallMessage struct {
	Type             constants.MessageTypeFunctionCall `json:"type"`
	ID               string                            `json:"id,omitempty"`
	CallID           string                            `json:"call_id,omitempty"`
	Name             string                            `json:"name"`
	Arguments        string                            `json:"arguments"`
	ThoughtSignature *string                           `json:"thought_signature,omitempty"` // Only for gemini
}

type FunctionCallOutputContentUnion struct {
	OfString *string      `json:",omitempty"`
	OfList   InputContent `json:",omitempty"`
}

type FunctionCallOutputMessage struct {
	Type   constants.MessageTypeFunctionCallOutput `json:"type"`
	ID     string                                  `json:"id,omitempty"`
	CallID string                                  `json:"call_id"`
	Output FunctionCallOutputContentUnion          `json:"output"`
}

type FunctionTool struct {
	Type        constants.ToolTypeFunction `json:"type"` // "function"
	Name        string                     `json:"name"`
	Description *string                    `json:"description,omitempty"`
	Parameters  map[string]any             `json:"parameters,omitempty"`
	Strict      *bool                      `json:"strict,omitempty"`
}

type ImageGenerationCallMessage struct {
	Type         constants.MessageTypeImageGenerationCall `json:"type"`
	ID           string                                   `json:"id"`            // "ig_" prefixed
	Status       string                                   `json:"status"`        // "generating"
	Background   string                                   `json:"background"`    // "opaque"
	OutputFormat string                                   `json:"output_format"` // "png"
	Quality      string                                   `json:"quality"`       // "medium"
	Size         string                                   `json:"size"`          // 100x100
	Result       string                                   `json:"result"`        // Base64 image
}

type ImageGenerationTool struct {
	Type constants.ToolTypeImageGeneration `json:"type"` // image_generation
}

type Includable string

type InputContent []InputContentUnion

type InputContentUnion struct {
	OfInputText  *InputTextContent  `json:",omitempty"`
	OfOutputText *OutputTextContent `json:",omitempty"`
	OfInputImage *InputImageContent `json:",omitempty,inline"`
}

type InputImageContent struct {
	Type     constants.ContentTypeInputImage `json:"type"`
	ImageURL *string                         `json:"image_url,omitempty"` // A fully qualified URL or base64 encoded image in a data URL.
	FileID   *string                         `json:"file_id,omitempty"`
	Detail   string                          `json:"detail"` // "low", "high", "auto"
}

type InputMessage struct {
	Type    constants.MessageTypeMessage `json:"type"` // Always "message".
	ID      string                       `json:"id,omitempty"`
	Role    constants.Role               `json:"role,omitempty"` // Any of "user", "system", "developer".
	Content InputContent                 `json:"content,omitempty"`
}

type InputMessageList []InputMessageUnion

type InputMessageUnion struct {
	OfEasyInput                    *EasyMessage                         `json:",omitempty"`
	OfInputMessage                 *InputMessage                        `json:",omitempty"`
	OfOutputMessage                *OutputMessage                       `json:",omitempty"`
	OfFunctionCall                 *FunctionCallMessage                 `json:",omitempty"`
	OfFunctionCallApprovalResponse *FunctionCallApprovalResponseMessage `json:",omitempty"`
	OfFunctionCallOutput           *FunctionCallOutputMessage           `json:",omitempty"`
	OfReasoning                    *ReasoningMessage                    `json:",omitempty"`
	OfImageGenerationCall          *ImageGenerationCallMessage          `json:",omitempty,inline"`
	OfWebSearchCall                *WebSearchCallMessage                `json:",omitempty,inline"`
}

type InputTextContent struct {
	Type constants.ContentTypeInputText `json:"type"`
	Text string                         `json:"text"`
}

type InputUnion struct {
	OfString           *string          `json:",omitempty,inline"`
	OfInputMessageList InputMessageList `json:",omitempty,inline"`
}

type OutputContent []OutputContentUnion

type OutputContentUnion struct {
	OfOutputText *OutputTextContent `json:",omitempty,inline"`
}

type OutputMessage struct {
	ID      string                       `json:"id"`
	Type    constants.MessageTypeMessage `json:"type,omitempty"`          // Always "message".
	Role    constants.Role               `json:"role,omitempty,required"` // Any of "user", "system", "developer".
	Content OutputContent                `json:"content,omitempty,required"`
}

type OutputMessageUnion struct {
	OfOutputMessage       *OutputMessage              `json:",omitempty"`
	OfFunctionCall        *FunctionCallMessage        `json:",omitempty"`
	OfReasoning           *ReasoningMessage           `json:",omitempty"`
	OfImageGenerationCall *ImageGenerationCallMessage `json:",omitempty"`
	OfWebSearchCall       *WebSearchCallMessage       `json:",omitempty"`
}

type OutputTextContent struct {
	Type        constants.ContentTypeOutputText `json:"type"`
	Text        string                          `json:"text"`
	Annotations []Annotation                    `json:"annotations"`
}

type Parameters struct {
	Temperature     *float64          `json:"temperature,omitempty"`
	MaxOutputTokens *int              `json:"max_output_tokens,omitempty"`
	TopP            *float64          `json:"top_p,omitempty"`
	TopLogprobs     *int64            `json:"top_logprobs,omitempty"`
	Text            *TextFormat       `json:"text,omitempty"`
	Background      *bool             `json:"background,omitempty"`
	Reasoning       *ReasoningParam   `json:"reasoning,omitempty"`
	Store           *bool             `json:"store,omitempty"`
	Include         []Includable      `json:"include,omitempty"`
	Metadata        map[string]string `json:"metadata,omitempty"`
	Stream          *bool             `json:"stream,omitempty"`

	MaxToolCalls      *int  `json:"max_tool_calls,omitempty"`
	ParallelToolCalls *bool `json:"parallel_tool_calls,omitempty"`
}

type ReasoningMessage struct {
	Type             constants.MessageTypeReasoning `json:"type"`
	ID               string                         `json:"id"`
	Summary          []SummaryTextContent           `json:"summary"`
	EncryptedContent *string                        `json:"encrypted_content,omitempty"`
}

type ReasoningParam struct {
	Summary      *string `json:"summary"` // "auto", "concise", "detailed"
	Effort       *string `json:"effort"`  // "none", "minimal", "low", "medium", "high", "xhigh"
	BudgetTokens *int    `json:",omitempty"`
}

type Request struct {
	Parameters
	Model        string      `json:"model"`
	Input        InputUnion  `json:"input,omitempty,omitzero"`
	Instructions *string     `json:"instructions,omitempty"`
	Tools        []ToolUnion `json:"tools,omitempty"`
}

type Response struct {
	ID          string                 `json:"id"`
	Model       string                 `json:"model"`
	Output      []OutputMessageUnion   `json:"output"`
	Usage       *Usage                 `json:"usage"`
	Error       *Error                 `json:"error"`
	ServiceTier string                 `json:"service_tier"`
	Metadata    map[string]interface{} `json:"metadata"`
}

type ResponseChunk struct {
	OfResponseCreated    *ChunkResponse[constants.ChunkTypeResponseCreated]    `json:",omitempty"`
	OfResponseInProgress *ChunkResponse[constants.ChunkTypeResponseInProgress] `json:",omitempty"`
	OfResponseCompleted  *ChunkResponse[constants.ChunkTypeResponseCompleted]  `json:",omitempty"`

	OfOutputItemAdded *ChunkOutputItem[constants.ChunkTypeOutputItemAdded] `json:",omitempty"`
	OfOutputItemDone  *ChunkOutputItem[constants.ChunkTypeOutputItemDone]  `json:",omitempty"`

	// For output item of type "message"
	OfContentPartAdded          *ChunkContentPart[constants.ChunkTypeContentPartAdded]         `json:",omitempty"`
	OfContentPartDone           *ChunkContentPart[constants.ChunkTypeContentPartDone]          `json:",omitempty"`
	OfOutputTextDelta           *ChunkOutputText[constants.ChunkTypeOutputTextDelta]           `json:",omitempty"`
	OfOutputTextAnnotationAdded *ChunkOutputText[constants.ChunkTypeOutputTextAnnotationAdded] `json:",omitempty"`
	OfOutputTextDone            *ChunkOutputText[constants.ChunkTypeOutputTextDone]            `json:",omitempty"`

	// For output item of type "function_call"
	OfFunctionCallArgumentsDelta *ChunkFunctionCall[constants.ChunkTypeFunctionCallArgumentsDelta] `json:",omitempty"`
	OfFunctionCallArgumentsDone  *ChunkFunctionCall[constants.ChunkTypeFunctionCallArgumentsDone]  `json:",omitempty"`

	// For output item of type "reasoning"
	OfReasoningSummaryPartAdded *ChunkReasoningSummaryPart[constants.ChunkTypeReasoningSummaryPartAdded] `json:",omitempty"`
	OfReasoningSummaryPartDone  *ChunkReasoningSummaryPart[constants.ChunkTypeReasoningSummaryPartDone]  `json:",omitempty"`
	OfReasoningSummaryTextDelta *ChunkReasoningSummaryText[constants.ChunkTypeReasoningSummaryTextDelta] `json:",omitempty"`
	OfReasoningSummaryTextDone  *ChunkReasoningSummaryText[constants.ChunkTypeReasoningSummaryTextDone]  `json:",omitempty"`

	// For output item of type "image_generation_call"
	OfImageGenerationCallInProgress   *ChunkImageGenerationCall[constants.ChunkTypeImageGenerationCallInProgress]   `json:",omitempty"`
	OfImageGenerationCallGenerating   *ChunkImageGenerationCall[constants.ChunkTypeImageGenerationCallGenerating]   `json:",omitempty"`
	OfImageGenerationCallPartialImage *ChunkImageGenerationCall[constants.ChunkTypeImageGenerationCallPartialImage] `json:",omitempty"`

	// For output item of type "web_search_call"
	OfWebSearchCallInProgress *ChunkWebSearchCall[constants.ChunkTypeWebSearchCallInProgress] `json:",omitempty"`
	OfWebSearchCallSearching  *ChunkWebSearchCall[constants.ChunkTypeWebSearchCallSearching]  `json:",omitempty"`
	OfWebSearchCallCompleted  *ChunkWebSearchCall[constants.ChunkTypeWebSearchCallCompleted]  `json:",omitempty"`

	// Custom Chunks
	OfRunCreated         *ChunkRun[constants.ChunkTypeRunCreated]    `json:",omitempty"`
	OfRunInProgress      *ChunkRun[constants.ChunkTypeRunInProgress] `json:",omitempty"`
	OfRunPaused          *ChunkRun[constants.ChunkTypeRunPaused]     `json:",omitempty"`
	OfRunCompleted       *ChunkRun[constants.ChunkTypeRunCompleted]  `json:",omitempty"`
	OfFunctionCallOutput *FunctionCallOutputMessage                  `json:",omitempty"`
}

type SummaryTextContent struct {
	Type constants.ContentTypeSummaryText `json:"type"`
	Text string                           `json:"text"`
}

type TextFormat struct {
	Format map[string]any `json:"format,omitempty"`
}

type ToolUnion struct {
	OfFunction        *FunctionTool        `json:",omitempty"`
	OfImageGeneration *ImageGenerationTool `json:",omitempty"`
	OfWebSearch       *WebSearchTool       `json:",omitempty"`
}

type Usage struct {
	InputTokens        int `json:"input_tokens"`
	InputTokensDetails struct {
		CachedTokens int `json:"cached_tokens"`
	} `json:"input_tokens_details"`
	OutputTokens        int `json:"output_tokens"`
	OutputTokensDetails struct {
		ReasoningTokens int `json:"reasoning_tokens"`
	} `json:"output_tokens_details"`
	TotalTokens int `json:"total_tokens"`
}

type WebSearchCallActionOfFind struct {
	Type    constants.WebSearchActionTypeFind `json:"type"`
	URL     string                            `json:"url"`
	Pattern string                            `json:"pattern"`
}

type WebSearchCallActionOfOpenPage struct {
	Type constants.WebSearchActionTypeOpenPage `json:"type"`
	URL  string                                `json:"url"`
}

type WebSearchCallActionOfSearch struct {
	Type    constants.WebSearchActionTypeSearch `json:"type"`
	Queries []string                            `json:"queries"`
	Query   string                              `json:"query"`
	Sources []WebSearchCallActionOfSearchSource `json:"sources"`
}

type WebSearchCallActionOfSearchSource struct {
	Type        string         `json:"type"` // always "url"
	URL         string         `json:"url"`
	ExtraParams map[string]any `json:"extra_params"`
}

type WebSearchCallActionUnion struct {
	OfSearch   *WebSearchCallActionOfSearch   `json:",omitempty"`
	OfOpenPage *WebSearchCallActionOfOpenPage `json:",omitempty"`
	OfFind     *WebSearchCallActionOfFind     `json:",omitempty"`
}

type WebSearchCallMessage struct {
	Type   constants.MessageTypeWebSearchCall `json:"type"`
	ID     string                             `json:"id"`
	Action WebSearchCallActionUnion           `json:"action"`
	Status string                             `json:"status"` // "in_progress", "searching", "completed", "failed"
}

type WebSearchTool struct {
	Type              constants.ToolTypeWebSearch `json:"type"` // web_search
	Filters           *WebSearchToolFilters       `json:"filters,omitempty"`
	UserLocation      *WebSearchToolUserLocation  `json:"user_location,omitempty"`
	ExternalWebAccess *bool                       `json:"external_web_access,omitempty"`
	SearchContextSize *string                     `json:"search_context_size,omitempty"` // "low", "medium", "high" - "medium" is default
}

type WebSearchToolFilters struct {
	AllowedDomains []string `json:"allowed_domains"`
}

type WebSearchToolUserLocation struct {
	Type     string `json:"type"`
	Country  string `json:"country"`
	City     string `json:"city"`
	Region   string `json:"region"`
	Timezone string `json:"timezone"`
}