tests

type Agent struct {
	Validate []Validation `yaml:"validate" json:"validate"`
}

type AgentTest struct {
	Goal                       string                      `yaml:"goal" json:"goal"`
	Persona                    string                      `yaml:"persona" json:"persona"`
	MaxSteps                   int                         `yaml:"maxSteps" json:"maxSteps"`
	UserInformation            []UserInfo                  `yaml:"userInformation,omitempty" json:"userInformation,omitempty"`
	OpenAIConfig               *OpenAIConfig               `yaml:"openAIConfig,omitempty" json:"openAIConfig,omitempty"`
	VoiceflowAgentTesterConfig *VoiceflowAgentTesterConfig `yaml:"voiceflowAgentTesterConfig,omitempty" json:"voiceflowAgentTesterConfig,omitempty"`
}

type Button struct {
	Name    string        `json:"name"`
	Request ButtonRequest `json:"request"`
}

type ButtonRequest struct {
	Type    string                 `json:"type"`
	Payload map[string]interface{} `json:"payload"`
}

type IntentEntity struct {
	Name  string `yaml:"name" json:"name"`
	Value string `yaml:"value" json:"value"`
}

type IntentRequest struct {
	Name     string         `yaml:"name" json:"name"`
	Entities []IntentEntity `yaml:"entities,omitempty" json:"entities,omitempty"`
}

type Interaction struct {
	ID      string   `yaml:"id" json:"id"`
	User    User     `yaml:"user" json:"user"`
	Agent   Agent    `yaml:"agent" json:"agent"`
	Buttons []Button `yaml:"-" json:"-"` // Stores buttons from previous response, not serialized
}

type OpenAIConfig struct {
	Model       string   `yaml:"model,omitempty" json:"model,omitempty"`
	Temperature *float64 `yaml:"temperature,omitempty" json:"temperature,omitempty"`
}

type SimilarityConfig struct {
	Provider            string  `yaml:"provider" json:"provider"`
	Model               string  `yaml:"model" json:"model"`
	Temperature         float64 `yaml:"temperature" json:"temperature"`
	SimilarityThreshold float64 `yaml:"similarityThreshold" json:"similarityThreshold"`
	TopK                int     `yaml:"top_k" json:"top_k"`
	TopP                float64 `yaml:"top_p" json:"top_p"`
}

type Suite struct {
	Name              string        `yaml:"name" json:"name"`
	Description       string        `yaml:"description" json:"description"`
	EnvironmentName   string        `yaml:"environmentName" json:"environmentName"`
	NewSessionPerTest bool          `yaml:"newSessionPerTest,omitempty" json:"newSessionPerTest,omitempty"`
	Tests             []TestFile    `yaml:"tests" json:"tests"`
	OpenAIConfig      *OpenAIConfig `yaml:"openAIConfig,omitempty" json:"openAIConfig,omitempty"`
}

type Test struct {
	Name         string        `yaml:"name" json:"name"`
	Description  string        `yaml:"description" json:"description"`
	Interactions []Interaction `yaml:"interactions,omitempty" json:"interactions,omitempty"`
	Agent        *AgentTest    `yaml:"agent,omitempty" json:"agent,omitempty"`
}

type TestFile struct {
	ID   string `yaml:"id" json:"id"`
	File string `yaml:"file" json:"file"`
}

type User struct {
	Type   string         `yaml:"type" json:"type"`
	Text   string         `yaml:"text,omitempty" json:"text,omitempty"`
	Value  string         `yaml:"value,omitempty" json:"value,omitempty"` // Used for button label
	Event  string         `yaml:"event,omitempty" json:"event,omitempty"`
	Intent *IntentRequest `yaml:"intent,omitempty" json:"intent,omitempty"`
}

type UserInfo struct {
	Name  string `yaml:"name" json:"name"`
	Value string `yaml:"value" json:"value"`
}

type Validation struct {
	ID               string            `yaml:"id" json:"id,omitempty"`
	Type             string            `yaml:"type" json:"type"`
	Value            string            `yaml:"value,omitempty" json:"value,omitempty"`
	Values           []string          `yaml:"values,omitempty" json:"values,omitempty"`
	SimilarityConfig *SimilarityConfig `yaml:"similarityConfig,omitempty" json:"similarityConfig,omitempty"`
	VariableConfig   *VariableConfig   `yaml:"variableConfig,omitempty" json:"variableConfig,omitempty"`
}

type VariableConfig struct {
	Name     string `yaml:"name" json:"name"`
	JsonPath string `yaml:"jsonPath,omitempty" json:"jsonPath,omitempty"`
}

type VoiceflowAgentTesterConfig struct {
	EnvironmentName string `yaml:"environmentName" json:"environmentName"`
	APIKey          string `yaml:"apiKey" json:"apiKey"`
}