aicontext

type AnthropicConfig struct {
	// Token limits configuration
	MinTokensLimit int
	MaxTokensLimit int

	// Model mapping configuration
	SmallModel  string
	MiddleModel string
	BigModel    string
}

type ClaudeContentBlock struct {
	Type      string                 `json:"type"`
	Text      string                 `json:"text,omitempty"`
	Source    map[string]interface{} `json:"source,omitempty"`
	ID        string                 `json:"id,omitempty"`
	Name      string                 `json:"name,omitempty"`
	Input     map[string]interface{} `json:"input,omitempty"`
	ToolUseID string                 `json:"tool_use_id,omitempty"`
	Content   interface{}            `json:"content,omitempty"`
}

type ClaudeContentBlockImage struct {
	Type   string                 `json:"type"`
	Source map[string]interface{} `json:"source"`
}

type ClaudeContentBlockText struct {
	Type string `json:"type"`
	Text string `json:"text"`
}

type ClaudeContentBlockToolResult struct {
	Type      string      `json:"type"`
	ToolUseID string      `json:"tool_use_id"`
	Content   interface{} `json:"content"` // Can be string, []map[string]interface{}, or map[string]interface{}
}

type ClaudeContentBlockToolUse struct {
	Type  string                 `json:"type"`
	ID    string                 `json:"id"`
	Name  string                 `json:"name"`
	Input map[string]interface{} `json:"input"`
}

type ClaudeMessage struct {
	Role    string      `json:"role"`    // "user" or "assistant"
	Content interface{} `json:"content"` // Can be string or []ClaudeContentBlock
}

type ClaudeMessagesRequest struct {
	Model         string                 `json:"model"`
	MaxTokens     int                    `json:"max_tokens"`
	Messages      []ClaudeMessage        `json:"messages"`
	System        interface{}            `json:"system,omitempty"` // Can be string or []ClaudeSystemContent
	StopSequences []string               `json:"stop_sequences,omitempty"`
	Stream        *bool                  `json:"stream,omitempty"`      // Default: false in Python
	Temperature   *float64               `json:"temperature,omitempty"` // Default: 1.0 in Python
	TopP          *float64               `json:"top_p,omitempty"`       // Default: None in Python
	TopK          *int                   `json:"top_k,omitempty"`       // Default: None in Python
	Metadata      map[string]interface{} `json:"metadata,omitempty"`
	Tools         []ClaudeTool           `json:"tools,omitempty"`
	ToolChoice    map[string]interface{} `json:"tool_choice,omitempty"`
	Thinking      *ClaudeThinkingConfig  `json:"thinking,omitempty"`
}

type ClaudeSystemContent struct {
	Type string `json:"type"`
	Text string `json:"text"`
}

type ClaudeThinkingConfig struct {
	Enabled bool `json:"enabled"` // Default: true in Python
}

type ClaudeTokenCountRequest struct {
	Model      string                 `json:"model"`
	Messages   []ClaudeMessage        `json:"messages"`
	System     interface{}            `json:"system,omitempty"` // Can be string or []ClaudeSystemContent
	Tools      []ClaudeTool           `json:"tools,omitempty"`
	Thinking   *ClaudeThinkingConfig  `json:"thinking,omitempty"`
	ToolChoice map[string]interface{} `json:"tool_choice,omitempty"`
}

type ClaudeTool struct {
	Name        string                 `json:"name"`
	Description *string                `json:"description,omitempty"`
	InputSchema map[string]interface{} `json:"input_schema"`
}

type Context struct {
	Ctx      *context.Context
	Provider *ProviderSpec

	// Req is the original request from the user. It's body is in ReqBody.
	Req                   *httpprot.Request
	ReqBody               []byte
	ReqInfo               *protocol.GeneralRequest
	OpenAIReq             map[string]any
	RespType              ResponseType
	ClaudeMessagesRequest *ClaudeMessagesRequest

	// ParseMetricFn is a function that parses the response body to a metric.
	// If it is sent, it will be called to parse the response body to a metric.
	// Otherwise, default ParseMetricFn will be used.
	ParseMetricFn func(fc *FinishContext) *metricshub.Metric
	// contains filtered or unexported fields
}

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

type FinishContext struct {
	StatusCode int
	Header     http.Header
	// RespBody is the final response body that sent to the user.
	RespBody []byte
	Duration int64
}

type ModelManager interface {
	MapClaudeModelToOpenAI(claudeModel string) string
}

type ProviderSpec struct {
	Name         string            `json:"name"`
	ProviderType string            `json:"providerType"`
	BaseURL      string            `json:"baseURL"`
	APIKey       string            `json:"apiKey"`
	Headers      map[string]string `json:"headers,omitempty"`
	// Optional parameters for specific providers, such as Azure.
	Endpoint     string `json:"endpoint,omitempty"`     // It is used for Azure OpenAI.
	DeploymentID string `json:"deploymentID,omitempty"` // It is used for Azure OpenAI.
	APIVersion   string `json:"apiVersion,omitempty"`   // It is used for Azure OpenAI.
}

type Response struct {
	StatusCode int
	// ContentLength records the length of the associated content. The
	// value -1 indicates that the length is unknown. Unless Request.Method
	// is "HEAD", values >= 0 indicate that the given number of bytes may
	// be read from Body.
	ContentLength int64
	Header        http.Header
	// BodyReader is the response body reader. Only one of BodyReader or BodyBytes should be set.
	BodyReader io.Reader
	// BodyBytes is the response body bytes. Only one of BodyReader or BodyBytes should be set.
	BodyBytes []byte
}

type ResponseType string

type ResultError string

type StreamingConversionState struct {
	MessageID        string
	TextBlockIndex   int
	ToolBlockCounter int
	CurrentToolCalls map[int]*ToolCallState
	FinalStopReason  string
}

type ToolCallState struct {
	ID          string
	Name        string
	ArgsBuffer  string
	JSONSent    bool
	ClaudeIndex *int
	Started     bool
}