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 Interaction struct {
	ID    string `yaml:"id" json:"id"`
	User  User   `yaml:"user" json:"user"`
	Agent Agent  `yaml:"agent" json:"agent"`
}

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

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