ai

type AIConfig struct {
	// Endpoint is the base URL for the Hanzo Cloud API.
	// All inference routes through cloud-api natively (no LiteLLM middleman).
	// Default: https://cloud.hanzo.ai/api
	Endpoint string `json:"endpoint"`

	// CloudEndpoint is an alias for Endpoint (kept for backward compatibility).
	CloudEndpoint string `json:"cloudEndpoint"`

	// APIKey is the Bearer token for authentication
	APIKey string `json:"apiKey"`

	// Model is the default model for chat completions
	// Options: gpt-4o, gpt-5, claude-opus-4-6, zen4-pro, zen4-coder-pro
	Model string `json:"model"`

	// EmbeddingModel is the model used for generating embeddings
	// Options: text-embedding-3-small, text-embedding-3-large, text-embedding-ada-002
	EmbeddingModel string `json:"embeddingModel"`

	// Temperature controls randomness in responses (0.0-2.0)
	Temperature float32 `json:"temperature"`

	// MaxTokens limits the response length
	MaxTokens int `json:"maxTokens"`

	// Timeout for API requests
	Timeout time.Duration `json:"timeout"`

	// GRPOEnabled enables Group Relative Policy Optimization for better responses
	GRPOEnabled bool `json:"grpoEnabled"`
}

type APIError struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
	Details string `json:"details,omitempty"`
}

type ChatChoice struct {
	Index        int         `json:"index"`
	Message      ChatMessage `json:"message"`
	FinishReason string      `json:"finish_reason"`
}

type ChatCompletionRequest struct {
	Model       string        `json:"model"`
	Messages    []ChatMessage `json:"messages"`
	Temperature *float32      `json:"temperature,omitempty"`
	MaxTokens   *int          `json:"max_tokens,omitempty"`
	Stream      bool          `json:"stream,omitempty"`
	GRPOEnabled bool          `json:"grpo_enabled,omitempty"`
	Groundtruth *string       `json:"groundtruth,omitempty"`
}

type ChatCompletionResponse struct {
	ID      string       `json:"id"`
	Object  string       `json:"object"`
	Created int64        `json:"created"`
	Model   string       `json:"model"`
	Choices []ChatChoice `json:"choices"`
	Usage   Usage        `json:"usage"`
	// GRPO metadata when enabled
	GRPOMetadata *GRPOMetadata `json:"grpo_metadata,omitempty"`
}

type ChatMessage struct {
	Role    string `json:"role"`    // system, user, assistant
	Content string `json:"content"` // message content
}

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

type CustomerSupportRequest struct {
	UserID       string        `json:"userId,omitempty"`
	Message      string        `json:"message"`
	Context      string        `json:"context,omitempty"`
	History      []ChatMessage `json:"history,omitempty"`
	OrderContext *OrderContext `json:"orderContext,omitempty"`
}

type CustomerSupportResponse struct {
	Response      string   `json:"response"`
	SuggestedNext []string `json:"suggestedNext,omitempty"`
	Sentiment     string   `json:"sentiment,omitempty"`
	RequiresHuman bool     `json:"requiresHuman"`
	Confidence    float64  `json:"confidence"`
}

type EmbeddingData struct {
	Object    string    `json:"object"`
	Index     int       `json:"index"`
	Embedding []float32 `json:"embedding"`
}

type EmbeddingRequest struct {
	Input []string `json:"input"`
	Model string   `json:"model"`
}

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

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

type GRPOMetadata struct {
	ExperiencesUsed []string `json:"experiences_used"`
	GroupSize       int      `json:"group_size"`
	BestReward      float64  `json:"best_reward"`
	AvgReward       float64  `json:"avg_reward"`
}

type ModelInfo struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Created int64  `json:"created"`
	OwnedBy string `json:"owned_by"`
	Premium bool   `json:"premium,omitempty"`
}

type OrderContext struct {
	OrderID     string    `json:"orderId"`
	Status      string    `json:"status"`
	CreatedAt   time.Time `json:"createdAt"`
	TotalAmount float64   `json:"totalAmount"`
	Items       []string  `json:"items"`
}

type ProductInfo struct {
	ID          string   `json:"id"`
	Name        string   `json:"name"`
	Description string   `json:"description"`
	Category    string   `json:"category"`
	Price       float64  `json:"price"`
	Tags        []string `json:"tags"`
	SKU         string   `json:"sku,omitempty"`
}

type Recommendation struct {
	ProductID  string  `json:"productId"`
	Score      float64 `json:"score"`
	Reasoning  string  `json:"reasoning,omitempty"`
	Category   string  `json:"category,omitempty"`
	Similarity float64 `json:"similarity,omitempty"`
}

type RecommendationRequest struct {
	UserID           string        `json:"userId"`
	Products         []ProductInfo `json:"products"`
	PurchaseHistory  []string      `json:"purchaseHistory,omitempty"`
	BrowsingHistory  []string      `json:"browsingHistory,omitempty"`
	MaxResults       int           `json:"maxResults,omitempty"`
	IncludeReasoning bool          `json:"includeReasoning,omitempty"`
}

type RecommendationResponse struct {
	UserID          string           `json:"userId"`
	Recommendations []Recommendation `json:"recommendations"`
	GeneratedAt     time.Time        `json:"generatedAt"`
	Model           string           `json:"model"`
}

type Usage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}