models

type ChatCompletionRequest struct {
	Model       string    `json:"model"`
	Messages    []Message `json:"messages"`
	Temperature *float32  `json:"temperature,omitempty"` // Optional field (pointer to float32)
	Stream      bool      `json:"stream"`
}

type ChatCompletionResponse struct {
	Choices []Choice `json:"choices"`
}

type Choice struct {
	Delta Delta `json:"delta"`
}

type Data struct {
	Object    string    `json:"object"`
	Embedding []float64 `json:"embedding"` // Embedding is an array of 1536 floats
	Index     int       `json:"index"`
}

type Delta struct {
	Content string `json:"content"`
}

type EmbeddingRequest struct {
	Input          string `json:"input"`           // The input text to be embedded
	Model          string `json:"model"`           // The model used for generating embeddings
	EncodingFormat string `json:"encoding_format"` // The encoding format (in this case, "float")
}

type EmbeddingResponse struct {
	Object string    `json:"object"`
	Data   []Data    `json:"data"`
	Model  string    `json:"model"`
	Usage  UsageInfo `json:"usage"`
}

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

type StreamResponse struct {
	Content string // Holds content chunks
	Err     error  // Holds error details
	Done    bool   // Signals end of stream
}

type UsageInfo struct {
	PromptTokens int `json:"prompt_tokens"`
	TotalTokens  int `json:"total_tokens"`
}