shuttle

type AudioGenRequest struct {
	//required
	Input string `json:"input"`

	//required
	Model string `json:"model"`

	//The voice to use for the generation > Voices can be found https://api.shuttleai.app/v1/models
	Voice *string `json:"voice,omitempty"`
}

type AudioGenResponse struct {
	Data struct {
		URL string `json:"url"`
	} `json:"data"`
	ExpiresIn int    `json:"expiresIn"`
	Model     string `json:"model"`
	Chars     int    `json:"chars"`
}

type AudioTranscribeRequest struct {
	File []byte `json:"file"`
}

type AudioTranscribeResponse struct {
	Text string `json:"text"`
}

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

type ChatRequest struct {
	//Required
	Messages []ChatMessage `json:"messages"`

	//The id of the model to use for the generation
	Model string `json:"model"`

	//Whether to include citations in the generation > For use in owned_by: bing/openai models only, defaults to true.
	Citations *bool `json:"citations,omitempty"`

	Image *string `json:"image,omitempty"`

	//Whether to use the internet for the generation > Model limitations apply, defaults to true for bing models.
	Internet *bool `json:"internet,omitempty"`

	//The maximum number of tokens to generate
	MaxToken *int `json:"max_token,omitempty"`

	//Whether to return the raw response > For use in owned_by: bing/openai models only! Returns Bing AI Suggestions and Search Results, defaults to false.
	Raw *bool `json:"raw,omitempty"`

	//The temperature of the sampling distribution
	Temperature *int `json:"temperature,omitempty"`

	//The tone of the generation > For use in owned_by: bing/openai models only, defaults to precise.
	Tone *string `json:"tone,omitempty"`

	//The cumulative probability of the top tokens to keep in the nucleus of the distribution
	TopP *int `json:"top_p,omitempty"`
}

type ChatResponse struct {
	Choices []struct {
		FinishReason string `json:"finish_reason"`
		Index        int    `json:"index"`
		Logprobs     struct {
		} `json:"logprobs"`
		Message struct {
			Content string `json:"content"`
			Role    string `json:"role"`
		} `json:"message"`
	} `json:"choices"`
	Created           int    `json:"created"`
	ID                string `json:"id"`
	Model             string `json:"model"`
	Object            string `json:"object"`
	SystemFingerprint string `json:"system_fingerprint"`
	Usage             struct {
		CompletionTokens int `json:"completion_tokens"`
		PromptTokens     int `json:"prompt_tokens"`
		TotalTokens      int `json:"total_tokens"`
	} `json:"usage"`
}

type ErrorResponse struct {
	Error string `json:"error"`
}

type HTTPClient interface {
	Do(req *http.Request) (*http.Response, error)
}

type ImageRequest struct {
	Model  string `json:"model"`
	Prompt string `json:"prompt"`
}

type ImageResponse struct {
	Created int `json:"created"`
	Data    []struct {
		URL string `json:"url"`
	} `json:"data"`
}

type ModelResponse []struct {
	Data []struct {
		Cost    int    `json:"cost"`
		Created int    `json:"created"`
		ID      string `json:"id"`
		Object  string `json:"object"`
		OwnedBy string `json:"owned_by"`
		Premium bool   `json:"premium"`
		Tokens  int    `json:"tokens"`
	} `json:"data"`
	Object string `json:"object"`
}

type ShuttleClient struct {
	Httpclient HTTPClient
	ApiKey     string
	Baseurl    string
}