ai

type AIError struct {
	Type       string                  `json:"type"`    // "invalid_request_error", "authentication_error", etc.
	Message    string                  `json:"message"` // Human-readable error message
	StatusCode int                     `json:"-"`       // HTTP status code
	Provider   string                  `json:"provider,omitempty"`
	Param      optional.Option[string] `json:"param"`
	Code       optional.Option[string] `json:"code"`
}

type AdapterFactory func(opts LLMAdapterOptions) (LLMAdapter, error)

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

type ChatRequest struct {
	Model             string         `json:"model"`
	Messages          []Message      `json:"messages"`
	Stream            bool           `json:"stream,omitempty"`
	StreamOptions     *StreamOptions `json:"stream_options,omitempty"`
	Temperature       *float64       `json:"temperature,omitempty"`
	TopP              *float64       `json:"top_p,omitempty"`
	MaxTokens         *int           `json:"max_tokens,omitempty"`
	Stop              []string       `json:"stop,omitempty"`
	Tools             []Tool         `json:"tools,omitempty"`
	ToolChoice        any            `json:"tool_choice,omitempty"`
	ParallelToolCalls *bool          `json:"parallel_tool_calls,omitempty"`
	Reasoning         *Reasoning     `json:"reasoning,omitempty"`
	ResponseFormat    any            `json:"response_format,omitempty"`
	UnknownFields     map[string]any `json:"-"`               // Collects unmapped fields for passthrough
	Extra             *ExtraOptions  `json:"extra,omitempty"` // Internal metadata
}

type ChatResponse struct {
	ID                string   `json:"id"`
	Object            string   `json:"object"` // "chat.completion"
	Created           int64    `json:"created"`
	Model             string   `json:"model"`
	SystemFingerprint string   `json:"system_fingerprint,omitempty"`
	Choices           []Choice `json:"choices"`
	Usage             Usage    `json:"usage"`
}

type Choice struct {
	Index        int     `json:"index"`
	Message      Message `json:"message"`
	FinishReason string  `json:"finish_reason"` // "stop", "length", "tool_calls", etc.
	Logprobs     any     `json:"logprobs,omitempty"`
}

type ClientAdapter interface {
	// Name returns the unique identifier for this client protocol (e.g., "openai-chat", "anthropic").
	Name() string

	// ToChatRequest translates a raw client JSON body into a canonical ChatRequest.
	ToChatRequest(body []byte) (*ChatRequest, error)

	// ToResponsesRequest translates a raw client JSON body into a canonical ResponsesRequest.
	ToResponsesRequest(body []byte) (*ResponsesRequest, error)

	// ToClientChatResponse translates a canonical ChatResponse back into the client's expected format.
	ToClientChatResponse(resp *ChatResponse) (any, error)

	// ToClientResponsesResponse translates a canonical ResponsesResponse back into the client's expected format.
	ToClientResponsesResponse(resp *ResponsesResponse) (any, error)

	// StreamConverter wraps a canonical OpenAI-format SSE stream with a protocol-specific
	// translator to re-encode chunks into the format expected by the client SDK.
	// The input stream must contain valid OpenAI chat.completion.chunk SSE lines.
	// The returned stream emits client-format SSE lines and must be closed by the caller.
	StreamConverter(stream io.ReadCloser) io.ReadCloser

	// ToClientError translates a canonical AIError into the client's expected format.
	ToClientError(err *AIError) (any, error)
}

type ClientAdapterFactory func() ClientAdapter

type CompletionTokensDetails struct {
	ReasoningTokens int `json:"reasoning_tokens"`
}

type ContentPart struct {
	Type     string    `json:"type"` // "text" or "image_url"
	Text     string    `json:"text,omitempty"`
	ImageURL *ImageURL `json:"image_url,omitempty"`
}

type ExtraOptions struct {
	AIFamily string `json:"ai_family"` // "chat" or "responses"
}

type FunctionCall struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"` // JSON string
}

type FunctionDesc struct {
	Name        string          `json:"name"`
	Description string          `json:"description,omitempty"`
	Parameters  json.RawMessage `json:"parameters,omitempty"` // JSON Schema
}

type ImageURL struct {
	URL    string `json:"url"`
	Detail string `json:"detail,omitempty"` // "auto", "low", "high"
}

type LLMAdapter interface {
	// Name returns the unique identifier for this adapter (e.g., "openai-chat", "anthropic").
	Name() string

	// Chat executes a non-streaming chat completion request.
	// It handles the translation from canonical ChatRequest to native payload,
	// performs the network request, and translates the native response back to ChatResponse.
	Chat(ctx context.Context, req *ChatRequest) (*ChatResponse, error)

	// StreamChat executes a streaming chat completion request.
	// It returns an io.ReadCloser that yields a stream of translated canonical SSE chunks.
	// The caller is responsible for closing the stream.
	StreamChat(ctx context.Context, req *ChatRequest) (io.ReadCloser, error)

	// Responses executes a non-streaming OpenAI-compatible Responses API request.
	Responses(ctx context.Context, req *ResponsesRequest) (*ResponsesResponse, error)

	// StreamResponses executes a streaming OpenAI-compatible Responses API request.
	// It returns an io.ReadCloser yielding canonical SSE chunks for the Responses family.
	StreamResponses(ctx context.Context, req *ResponsesRequest) (io.ReadCloser, error)
}

type LLMAdapterOptions struct {
	HTTPClient *client.Client
	APIKey     string
	BaseURL    string
}

type Message struct {
	Role       string     `json:"role"`
	Content    any        `json:"content"` // string OR []ContentPart for multi-modal
	Name       string     `json:"name,omitempty"`
	ToolCalls  []ToolCall `json:"tool_calls,omitempty"`
	ToolCallID string     `json:"tool_call_id,omitempty"` // Required for "tool" role
}

type ObservedStream struct {
	io.ReadCloser
	// contains filtered or unexported fields
}

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

type OpenAIChatClientAdapter struct{}

type OpenAIErrorDetail struct {
	Message string                  `json:"message"`
	Type    string                  `json:"type"`
	Param   optional.Option[string] `json:"param"`
	Code    optional.Option[string] `json:"code"`
}

type OpenAIErrorResponse struct {
	Error OpenAIErrorDetail `json:"error"`
}

type PromptTokensDetails struct {
	CachedTokens int `json:"cached_tokens"`
}

type Reasoning struct {
	Effort string `json:"effort,omitempty"` // "low", "medium", "high"
}

type ResponsesRequest struct {
	Model        string    `json:"model"`
	Instructions string    `json:"instructions,omitempty"`
	Input        []Message `json:"input"`
}

type ResponsesResponse struct {
	ID     string `json:"id"`
	Status string `json:"status"`
	Model  string `json:"model"`
	Usage  Usage  `json:"usage"`
}

type StreamChoice struct {
	Index        int         `json:"index"`
	Delta        StreamDelta `json:"delta"`
	FinishReason *string     `json:"finish_reason"`
}

type StreamChunk struct {
	ID      string         `json:"id"`
	Object  string         `json:"object"` // "chat.completion.chunk"
	Created int64          `json:"created"`
	Model   string         `json:"model"`
	Choices []StreamChoice `json:"choices"`
	Usage   *Usage         `json:"usage,omitempty"`
}

type StreamDelta struct {
	Role             string     `json:"role,omitempty"`
	Content          string     `json:"content,omitempty"`
	ReasoningContent string     `json:"reasoning_content,omitempty"` // For reasoning models
	ToolCalls        []ToolCall `json:"tool_calls,omitempty"`
}

type StreamOptions struct {
	IncludeUsage bool `json:"include_usage,omitempty"`
}

type Tool struct {
	Type     string       `json:"type"` // Currently only "function"
	Function FunctionDesc `json:"function"`
}

type ToolCall struct {
	ID       string       `json:"id"`
	Type     string       `json:"type"`
	Function FunctionCall `json:"function"`
}

type Usage struct {
	PromptTokens            int                      `json:"prompt_tokens"`
	CompletionTokens        int                      `json:"completion_tokens"`
	TotalTokens             int                      `json:"total_tokens"`
	PromptTokensDetails     *PromptTokensDetails     `json:"prompt_tokens_details,omitempty"`
	CompletionTokensDetails *CompletionTokensDetails `json:"completion_tokens_details,omitempty"`
	InputCost               float64                  `json:"input_cost,omitempty"`
	OutputCost              float64                  `json:"output_cost,omitempty"`
}

type UsageMetadata struct {
	Model    string `json:"model"`    // Logical model name (e.g., "gpt-4o")
	UserID   string `json:"user_id"`  // User identifier for billing/quota
	RouteID  string `json:"route_id"` // Bifrost route ID
	Provider string `json:"provider"` // Target provider ID (e.g., "openai-official")
}

type UsageObserver interface {
	// OnUsage is called with the usage metrics and associated metadata.
	// Context is provided for potential database or remote logging operations.
	OnUsage(ctx context.Context, metadata UsageMetadata, usage Usage)
}