ai

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
}

type ChainParamsStep struct {
	Input       string
	Instruction string
	Options     any
}

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

type Config struct {
	// openai
	OpenAIApiKey         string
	OpenAIGptModel       string
	OpenAIEmbeddingModel string

	// azure openai
	AzureOpenAIApiKey                string
	AzureOpenAIEndpoint              string
	AzureOpenAIGptDeploymentID       string
	AzureOpenAIEmbeddingDeploymentID string

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

	// susanoo
	SusanooEndpoint 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 RawRequestResponse struct {
	Text string
	Json map[string]any
}

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"`
}