anthropic

type APIError struct {
	Type    ErrType `json:"type"`
	Message string  `json:"message"`
}

type APIVersion string

type ApiKeyFunc func() string

type BatchId string

type BatchRequest struct {
	Requests []InnerRequests `json:"requests"`
}

type BatchRespCore struct {
	Id                BatchId           `json:"id"`
	Type              BatchResponseType `json:"type"`
	ProcessingStatus  ProcessingStatus  `json:"processing_status"`
	RequestCounts     RequestCounts     `json:"request_counts"`
	EndedAt           *time.Time        `json:"ended_at"`
	CreatedAt         time.Time         `json:"created_at"`
	ExpiresAt         time.Time         `json:"expires_at"`
	ArchivedAt        *time.Time        `json:"archived_at"`
	CancelInitiatedAt *time.Time        `json:"cancel_initiated_at"`
	ResultsUrl        *string           `json:"results_url"`
}

type BatchResponse struct {
	BatchRespCore
	// contains filtered or unexported fields
}

type BatchResponseType string

type BatchResult struct {
	CustomId string          `json:"custom_id"`
	Result   BatchResultCore `json:"result"`
}

type BatchResultCore struct {
	Type   ResultType       `json:"type"`
	Result MessagesResponse `json:"message"`
}

type BetaVersion string

type CacheControlType string

type ChatRole string

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

type ClientAdapter interface {
	// Translate provider specific errors.  Responds with an error and a boolean indicating if the error has been successfully parsed.
	TranslateError(resp *http.Response, body []byte) (error, bool)
	// Prepare the request for the provider and return the full URL
	PrepareRequest(c *Client, method, urlSuffix string, body any) (string, error)
	// Set the request headers for the provider
	SetRequestHeaders(c *Client, req *http.Request) error
}

type ClientConfig struct {
	BaseURL     string
	APIVersion  APIVersion
	BetaVersion []BetaVersion
	HTTPClient  *http.Client

	EmptyMessagesLimit uint

	Adapter ClientAdapter
	// contains filtered or unexported fields
}

type ClientOption func(c *ClientConfig)

type CompleteEvent string

type CompleteRequest struct {
	Model             Model  `json:"model"`
	Prompt            string `json:"prompt"`
	MaxTokensToSample int    `json:"max_tokens_to_sample"`

	StopSequences []string       `json:"stop_sequences,omitempty"`
	Temperature   *float32       `json:"temperature,omitempty"`
	TopP          *float32       `json:"top_p,omitempty"`
	TopK          *int           `json:"top_k,omitempty"`
	MetaData      map[string]any `json:"meta_data,omitempty"`
	Stream        bool           `json:"stream,omitempty"`
}

type CompleteResponse struct {
	Type       string `json:"type"`
	ID         string `json:"id"`
	Completion string `json:"completion"`
	// possible values are: stop_sequence、max_tokens、null
	StopReason string `json:"stop_reason"`
	Model      Model  `json:"model"`
	// contains filtered or unexported fields
}

type CompleteStreamPingData struct {
	Type string `json:"type"`
}

type CompleteStreamRequest struct {
	CompleteRequest

	OnCompletion func(CompleteResponse)       `json:"-"`
	OnPing       func(CompleteStreamPingData) `json:"-"`
	OnError      func(ErrorResponse)          `json:"-"`
}

type CountTokensResponse struct {
	InputTokens int `json:"input_tokens"`
	// contains filtered or unexported fields
}

type DefaultAdapter struct {
}

type ErrType string

type ErrorResponse struct {
	Type  string    `json:"type"`
	Error *APIError `json:"error,omitempty"`
}

type InnerRequests struct {
	CustomId string          `json:"custom_id"`
	Params   MessagesRequest `json:"params"`
}

type ListBatchesRequest struct {
	BeforeId *string `json:"before_id,omitempty"`
	AfterId  *string `json:"after_id,omitempty"`
	Limit    *int    `json:"limit,omitempty"`
}

type ListBatchesResponse struct {
	Data    []BatchRespCore `json:"data"`
	HasMore bool            `json:"has_more"`
	FirstId *BatchId        `json:"first_id"`
	LastId  *BatchId        `json:"last_id"`
	// contains filtered or unexported fields
}

type Message struct {
	Role    ChatRole         `json:"role"`
	Content []MessageContent `json:"content"`
}

type MessageCacheControl struct {
	Type CacheControlType `json:"type"`
}

type MessageContent struct {
	Type MessagesContentType `json:"type"`

	Text *string `json:"text,omitempty"`

	Source *MessageContentSource `json:"source,omitempty"`

	*MessageContentToolResult

	*MessageContentToolUse

	PartialJson *string `json:"partial_json,omitempty"`

	CacheControl *MessageCacheControl `json:"cache_control,omitempty"`

	*MessageContentThinking

	*MessageContentRedactedThinking
}

type MessageContentRedactedThinking struct {
	Data string `json:"data,omitempty"`
}

type MessageContentSource struct {
	Type      MessagesContentSourceType `json:"type"`
	MediaType string                    `json:"media_type"`
	Data      any                       `json:"data"`
}

type MessageContentThinking struct {
	Thinking  string `json:"thinking,omitempty"`
	Signature string `json:"signature,omitempty"`
}

type MessageContentToolResult struct {
	ToolUseID *string          `json:"tool_use_id,omitempty"`
	Content   []MessageContent `json:"content,omitempty"`
	IsError   *bool            `json:"is_error,omitempty"`
}

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

type MessageSystemPart struct {
	Type         string               `json:"type"`
	Text         string               `json:"text"`
	CacheControl *MessageCacheControl `json:"cache_control,omitempty"`
}

type MessagesContentSourceType string

type MessagesContentType string

type MessagesEvent string

type MessagesEventContentBlockDeltaData struct {
	Type  string         `json:"type"`
	Index int            `json:"index"`
	Delta MessageContent `json:"delta"`
}

type MessagesEventContentBlockStartData struct {
	Type         MessagesEvent  `json:"type"`
	Index        int            `json:"index"`
	ContentBlock MessageContent `json:"content_block"`
}

type MessagesEventContentBlockStopData struct {
	Type  string `json:"type"`
	Index int    `json:"index"`
}

type MessagesEventMessageDeltaData struct {
	Type  string           `json:"type"`
	Delta MessagesResponse `json:"delta"`
	Usage MessagesUsage    `json:"usage"`
}

type MessagesEventMessageStartData struct {
	Type    MessagesEvent    `json:"type"`
	Message MessagesResponse `json:"message"`
}

type MessagesEventMessageStopData struct {
	Type string `json:"type"`
}

type MessagesEventPingData struct {
	Type string `json:"type"`
}

type MessagesRequest struct {
	Model            Model     `json:"model,omitempty"`
	AnthropicVersion string    `json:"anthropic_version,omitempty"`
	Messages         []Message `json:"messages"`
	MaxTokens        int       `json:"max_tokens,omitempty"`

	System        string              `json:"-"`
	MultiSystem   []MessageSystemPart `json:"-"`
	Metadata      map[string]any      `json:"metadata,omitempty"`
	StopSequences []string            `json:"stop_sequences,omitempty"`
	Stream        bool                `json:"stream,omitempty"`
	Temperature   *float32            `json:"temperature,omitempty"`
	TopP          *float32            `json:"top_p,omitempty"`
	TopK          *int                `json:"top_k,omitempty"`
	Tools         []ToolDefinition    `json:"tools,omitempty"`
	ToolChoice    *ToolChoice         `json:"tool_choice,omitempty"`
	Thinking      *Thinking           `json:"thinking,omitempty"`
}

type MessagesResponse struct {
	ID           string               `json:"id"`
	Type         MessagesResponseType `json:"type"`
	Role         ChatRole             `json:"role"`
	Content      []MessageContent     `json:"content"`
	Model        Model                `json:"model"`
	StopReason   MessagesStopReason   `json:"stop_reason"`
	StopSequence string               `json:"stop_sequence"`
	Usage        MessagesUsage        `json:"usage"`
	// contains filtered or unexported fields
}

type MessagesResponseType string

type MessagesStopReason string

type MessagesStreamRequest struct {
	MessagesRequest

	OnError             func(ErrorResponse)                                     `json:"-"`
	OnPing              func(MessagesEventPingData)                             `json:"-"`
	OnMessageStart      func(MessagesEventMessageStartData)                     `json:"-"`
	OnContentBlockStart func(MessagesEventContentBlockStartData)                `json:"-"`
	OnContentBlockDelta func(MessagesEventContentBlockDeltaData)                `json:"-"`
	OnContentBlockStop  func(MessagesEventContentBlockStopData, MessageContent) `json:"-"`
	OnMessageDelta      func(MessagesEventMessageDeltaData)                     `json:"-"`
	OnMessageStop       func(MessagesEventMessageStopData)                      `json:"-"`
}

type MessagesUsage struct {
	InputTokens  int `json:"input_tokens"`
	OutputTokens int `json:"output_tokens"`

	// The number of tokens written to the cache when creating a new entry.
	CacheCreationInputTokens int `json:"cache_creation_input_tokens,omitempty"`
	// The number of tokens retrieved from the cache for associated request.
	CacheReadInputTokens int `json:"cache_read_input_tokens,omitempty"`
}

type Model string

type ProcessingStatus string

type RateLimitHeaders struct {
	// The maximum number of requests allowed within the rate limit window.
	RequestsLimit int `json:"anthropic-ratelimit-requests-limit"`
	// The number of requests remaining within the current rate limit window.
	RequestsRemaining int `json:"anthropic-ratelimit-requests-remaining"`
	// The time when the request rate limit window will reset, provided in RFC 3339 format.
	RequestsReset time.Time `json:"anthropic-ratelimit-requests-reset"`
	// The maximum number of tokens allowed within the rate limit window.
	TokensLimit int `json:"anthropic-ratelimit-tokens-limit"`
	// The number of tokens remaining, rounded to the nearest thousand, within the current rate limit window.
	TokensRemaining int `json:"anthropic-ratelimit-tokens-remaining"`
	// The time when the token rate limit window will reset, provided in RFC 3339 format.
	TokensReset time.Time `json:"anthropic-ratelimit-tokens-reset"`
	// The maximum number of input tokens allowed within any rate limit period.
	InputTokensLimit int `json:"anthropic-ratelimit-input-tokens-limit"`
	// The number of input tokens remaining (rounded to the nearest thousand) before being rate limited.
	InputTokensRemaining int `json:"anthropic-ratelimit-input-tokens-remaining"`
	// The time when the input token rate limit will be fully replenished, provided in RFC 3339 format.
	InputTokensReset time.Time `json:"anthropic-ratelimit-input-tokens-reset"`
	// The maximum number of output tokens allowed within any rate limit period.
	OutputTokensLimit int `json:"anthropic-ratelimit-output-tokens-limit"`
	// The number of output tokens remaining (rounded to the nearest thousand) before being rate limited.
	OutputTokensRemaining int `json:"anthropic-ratelimit-output-tokens-remaining"`
	// The time when the output token rate limit will be fully replenished, provided in RFC 3339 format.
	OutputTokensReset time.Time `json:"anthropic-ratelimit-output-tokens-reset"`
	// The number of seconds until the rate limit window resets.
	RetryAfter int `json:"retry-after"`
}

type RequestCounts struct {
	Processing int `json:"processing"`
	Succeeded  int `json:"succeeded"`
	Errored    int `json:"errored"`
	Canceled   int `json:"canceled"`
	Expired    int `json:"expired"`
}

type RequestError struct {
	StatusCode int
	Err        error
	Body       []byte
}

type Response interface {
	SetHeader(http.Header)
}

type ResultType string

type RetrieveBatchResultsResponse struct {
	Responses   []BatchResult
	RawResponse []byte
	// contains filtered or unexported fields
}

type Thinking struct {
	Type ThinkingType `json:"type"`
	// Determines how many tokens Claude can use for its internal reasoning process. Larger budgets can enable more thorough analysis for complex problems, improving response quality.
	// Must be ≥1024 and less than max_tokens.
	BudgetTokens int `json:"budget_tokens"`
}

type ThinkingType string

type ToolChoice struct {
	// oneof: auto(default) any tool
	Type string `json:"type"`
	Name string `json:"name,omitempty"`
}

type ToolDefinition struct {
	Name        string `json:"name"`
	Description string `json:"description,omitempty"`
	// InputSchema is an object describing the tool.
	// You can pass json.RawMessage to describe the schema,
	// or you can pass in a struct which serializes to the proper JSON schema.
	// The jsonschema package is provided for convenience, but you should
	// consider another specialized library if you require more complex schemas.
	InputSchema any `json:"input_schema,omitempty"`

	CacheControl *MessageCacheControl `json:"cache_control,omitempty"`

	// Type is required for Anthropic defined tools.
	Type string `json:"type,omitempty"`
	// DisplayWidthPx is a required parameter of the Computer Use tool.
	DisplayWidthPx int `json:"display_width_px,omitempty"`
	// DisplayHeightPx is a required parameter of the Computer Use tool.
	DisplayHeightPx int `json:"display_height_px,omitempty"`
	// DisplayNumber is an optional parameter of the Computer Use tool.
	DisplayNumber *int `json:"display_number,omitempty"`
}

type VertexAIErrorResponse struct {
	Error *VertexAPIError `json:"error,omitempty"`
}

type VertexAISupport interface {
	GetModel() Model
	SetAnthropicVersion(APIVersion)
	IsStreaming() bool
}

type VertexAPIError struct {
	Code    int                     `json:"code"`
	Message string                  `json:"message"`
	Status  string                  `json:"status"`
	Details []VertexAPIErrorDetails `json:"details"`
}

type VertexAPIErrorDetails struct {
	Type     string `json:"@type"`
	Reason   string `json:"reason"`
	Metadata struct {
		Method  string `json:"method"`
		Service string `json:"service"`
	} `json:"metadata"`
}

type VertexAdapter struct {
}