common

type Adaptor interface {
	// Init initializes the adaptor with the relay info.
	Init(info *RelayInfo)

	// GetRequestURL returns the upstream URL for this request.
	GetRequestURL(info *RelayInfo) (string, error)

	// SetupRequestHeader sets auth and content-type headers on the request.
	SetupRequestHeader(ctx context.Context, header *http.Header, info *RelayInfo) error

	// ConvertOpenAIRequest converts an OpenAI-format chat request to the
	// provider's native format.
	ConvertOpenAIRequest(ctx context.Context, info *RelayInfo, request *dto.GeneralOpenAIRequest) (any, error)

	// ConvertClaudeRequest converts a Claude Messages request to the
	// provider's native format.
	ConvertClaudeRequest(ctx context.Context, info *RelayInfo, request *dto.ClaudeRequest) (any, error)

	// ConvertGeminiRequest converts a Gemini generateContent request to the
	// provider's native format.
	ConvertGeminiRequest(ctx context.Context, info *RelayInfo, request *dto.GeminiChatRequest) (any, error)

	// ConvertImageRequest converts an image generation request.
	ConvertImageRequest(ctx context.Context, info *RelayInfo, request dto.ImageRequest) (any, error)

	// ConvertEmbeddingRequest converts an embedding request.
	ConvertEmbeddingRequest(ctx context.Context, info *RelayInfo, request dto.EmbeddingRequest) (any, error)

	// ConvertAudioRequest converts an audio (TTS/ASR) request.
	ConvertAudioRequest(ctx context.Context, info *RelayInfo, request dto.AudioRequest) (io.Reader, error)

	// ConvertRerankRequest converts a rerank request.
	ConvertRerankRequest(ctx context.Context, relayMode int, request dto.RerankRequest) (any, error)

	// ConvertOpenAIResponsesRequest converts an OpenAI Responses request.
	ConvertOpenAIResponsesRequest(ctx context.Context, info *RelayInfo, request dto.OpenAIResponsesRequest) (any, error)

	// DoRequest sends the HTTP request to the upstream provider.
	DoRequest(ctx context.Context, info *RelayInfo, requestBody io.Reader) (*http.Response, error)

	// DoResponse parses the upstream response and returns usage info.
	// For streaming responses, this writes to the provided http.ResponseWriter.
	DoResponse(ctx context.Context, resp *http.Response, info *RelayInfo, w http.ResponseWriter) (usage any, err *types.NewAPIError)

	// GetModelList returns the models supported by this provider.
	GetModelList() []string

	// GetChannelName returns the provider name.
	GetChannelName() string
}

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

type ChannelMeta struct {
	ChannelType          int            // constant.ChannelType*
	ApiType              int            // constant.APIType*
	ApiVersion           string         // Azure API version, Vertex region, etc.
	ApiKey               string         // upstream API key
	Organization         string         // OpenAI org ID, etc.
	ChannelBaseUrl       string         // override default endpoint
	ParamOverride        map[string]any // parameter overrides
	HeadersOverride      map[string]any // header overrides
	UpstreamModelName    string         // model name after mapping
	IsModelMapped        bool
	SupportStreamOptions bool
}

type HTTPClient struct {
	Client  *http.Client
	Headers map[string]string
}

type RelayInfo struct {
	IsStream           bool
	RelayMode          int    // relaymode.RelayMode*
	OriginModelName    string // original model name from client
	RequestURLPath     string // e.g. "/v1/chat/completions"
	RequestHeaders     map[string]string
	ShouldIncludeUsage bool
	RelayFormat        types.RelayFormat // input format (openai/claude/gemini)
	StartTime          time.Time

	// Channel/provider metadata.
	*ChannelMeta

	// The parsed request DTO (GeneralOpenAIRequest, ClaudeRequest, etc.)
	Request dto.Request

	// Request format conversion chain, e.g. ["openai", "claude"].
	RequestConversionChain []types.RelayFormat
	// Final request format sent to upstream.
	FinalRequestRelayFormat types.RelayFormat

	// API type for adaptor dispatch.
	ApiType int

	// Reasoning effort ("low", "medium", "high", "").
	ReasoningEffort string

	// Action is the task action type (e.g. constant.TaskActionGenerate).
	// Used by async task providers.
	Action string

	// PublicTaskID is the client-facing task ID.
	PublicTaskID string

	// OriginTaskID is the original task ID for remix operations.
	OriginTaskID string

	// SendResponseCount / ReceivedResponseCount for streaming.
	SendResponseCount     int
	ReceivedResponseCount int
	// contains filtered or unexported fields
}

type TaskAdaptor interface {
	Init(info *RelayInfo)

	// ValidateRequestAndSetAction validates the request and sets the action type.
	ValidateRequestAndSetAction(ctx context.Context, info *RelayInfo) *TaskError

	BuildRequestURL(info *RelayInfo) (string, error)
	BuildRequestHeader(ctx context.Context, req *http.Request, info *RelayInfo) error
	BuildRequestBody(ctx context.Context, info *RelayInfo) (io.Reader, error)

	DoRequest(ctx context.Context, info *RelayInfo, requestBody io.Reader) (*http.Response, error)
	DoResponse(ctx context.Context, resp *http.Response, info *RelayInfo) (taskID string, taskData []byte, err *TaskError)

	GetModelList() []string
	GetChannelName() string

	FetchTask(baseURL, key string, body map[string]any, proxy string) (*http.Response, error)
	ParseTaskResult(respBody []byte) (*TaskInfo, error)
}

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

type TaskError struct {
	Code       string `json:"code"`
	Message    string `json:"message"`
	Data       any    `json:"data"`
	StatusCode int    `json:"-"`
	LocalError bool   `json:"-"`
	Err        error  `json:"-"`
}

type TaskInfo struct {
	Code             int    `json:"code"`
	TaskID           string `json:"task_id"`
	Status           string `json:"status"` // "PENDING", "RUNNING", "SUCCESS", "FAILURE"
	Reason           string `json:"reason,omitempty"`
	Url              string `json:"url,omitempty"`
	RemoteUrl        string `json:"remote_url,omitempty"`
	Progress         string `json:"progress,omitempty"`
	CompletionTokens int    `json:"completion_tokens,omitempty"`
	TotalTokens      int    `json:"total_tokens,omitempty"`
}

type TaskSubmitReq struct {
	Prompt         string         `json:"prompt"`
	Images         []string       `json:"images,omitempty"`
	Duration       int            `json:"duration,omitempty"`
	Seconds        string         `json:"seconds,omitempty"`
	Size           string         `json:"size,omitempty"`
	Metadata       map[string]any `json:"metadata,omitempty"`
	Model          string         `json:"model,omitempty"`
	Mode           string         `json:"mode,omitempty"`
	Image          string         `json:"image,omitempty"`
	InputReference string         `json:"input_reference,omitempty"`
}