ai

type AzureRawRequestOptions struct {
	UseJSON bool
}

type BedRockClaudeChatMessage struct {
	Role    string                        `json:"role"`
	Content []BedRockClaudeMessageContent `json:"content"`
}

type BedRockClaudeMessageContent struct {
	Type string `json:"type"`
	Text string `json:"text,omitempty"`
}

type BedrockClaudeResponse struct {
	ID           string                        `json:"id"`
	Type         string                        `json:"type"`
	Role         string                        `json:"role"`
	Model        string                        `json:"model"`
	Content      []BedRockClaudeMessageContent `json:"content"`
	StopReason   string                        `json:"stop_reason"`
	StopSequence interface{}                   `json:"stop_sequence"`
	Usage        map[string]int                `json:"usage"`
}

type ChainParams struct {
	Format           string
	Steps            []ChainParamsStep
	RawRequestParams map[string]any
}

type ChainParamsStep struct {
	Input       string
	Instruction string
	Options     any
}

type Config struct {
	// openai
	OpenAIAPIBase        string
	OpenAIAPIKey         string
	OpenAIModel          string
	OpenAIEmbeddingModel string

	// azure openai
	AzureOpenAIAPIKey         string
	AzureOpenAIEndpoint       string
	AzureOpenAIModel          string
	AzureOpenAIEmbeddingModel string

	// aws bedrock
	AwsKey                      string
	AwsSecret                   string
	AwsBedrockModelArn          string
	AwsBedrockEmbeddingModelArn string

	// susanoo
	SusanooAPIBase string
	SusanooAPIKey  string

	Provider string

	Debug bool
}

type GeneralChatCompletionMessage struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

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

type OpenAIRawRequestOptions struct {
	UseJSON bool
}

type Result struct {
	Text string
	Json map[string]any
}

type SusanoParams struct {
	Format     string                 `json:"format"`
	Search     SusanoParamsSearch     `json:"search"`
	Conditions SusanoParamsConditions `json:"conditions"`
}

type SusanoParamsConditions struct {
	PreferredProvider string `json:"preferred_provider"`
	PreferredModel    string `json:"preferred_model"`
}

type SusanoParamsSearch struct {
	Enabled bool `json:"enabled"`
	Limit   int  `json:"limit"`
}

type SusanooTaskRequest struct {
	Messages []GeneralChatCompletionMessage `json:"messages"`
	Params   map[string]any                 `json:"params"`
}

type SusanooTaskResponse struct {
	Data struct {
		Code    int    `json:"code"`
		TraceID string `json:"trace_id"`
	} `json:"data"`
	Ts int `json:"ts"`
}

type SusanooTaskResultResponse struct {
	Data struct {
		ID           int            `json:"id"`
		ProxyID      int            `json:"proxy_id"`
		Result       map[string]any `json:"result"`
		Status       int            `json:"status"`
		TraceID      string         `json:"trace_id"`
		ScheduledAt  string         `json:"scheduled_at"`
		CreatedAt    string         `json:"created_at"`
		UpdatedAt    string         `json:"updated_at"`
		PendingCount int            `json:"pending_count"`
	} `json:"data"`
	Ts int `json:"ts"`
}