setup

type AgentConfig struct {
	Runs            int              `yaml:"runs"`
	Concurrent      bool             `yaml:"concurrent"`
	ConcurrentCount int              `yaml:"concurrent_count"`
	Tools           AgentToolsConfig `yaml:"tools"`
	Subagents       SubagentsConfig  `yaml:"subagents"`
}

type AgentToolsConfig struct {
	Count     int    `yaml:"count"`
	Execution string `yaml:"execution"`
}

type AgentTree struct {
	RootPrompt string
	SubAgents  []*agent.Agent
	Created    []*agent.Agent
}

type Config struct {
	Runtime  string         `yaml:"runtime"`
	Temporal TemporalConfig `yaml:"temporal"`
	LLM      LLMConfig      `yaml:"llm"`
	Tool     ToolConfig     `yaml:"tool"`
	Agent    AgentConfig    `yaml:"agent"`
	Memory   MemoryConfig   `yaml:"memory"`
	Logger   LoggerConfig   `yaml:"logger"`
	Output   OutputConfig   `yaml:"output"`
}

type LLMConfig struct {
	LatencyMs  int `yaml:"latency_ms"`
	JitterMs   int `yaml:"jitter_ms"`
	MockTokens int `yaml:"mock_tokens"`
}

type LLMStats struct {
	TotalInputTokens  int
	TotalOutputTokens int
	// contains filtered or unexported fields
}

type LoggerConfig struct {
	Enabled bool   `yaml:"enabled"`
	Dir     string `yaml:"dir"`
	Level   string `yaml:"level"`
}

type MemoryConfig struct {
	Enabled   bool   `yaml:"enabled"`
	StoreMode string `yaml:"store_mode"`
	UserID    string `yaml:"user_id"`
}

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

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

type OutputConfig struct {
	Console bool   `yaml:"console"`
	File    bool   `yaml:"file"`
	Dir     string `yaml:"dir"`
	Format  string `yaml:"format"`
}

type SubagentsConfig struct {
	Count  int `yaml:"count"`
	Levels int `yaml:"levels"`
}

type TemporalConfig struct {
	Host         string `yaml:"host"`
	Port         int    `yaml:"port"`
	Namespace    string `yaml:"namespace"`
	TaskQueue    string `yaml:"task_queue"`
	WorkersCount int    `yaml:"workers_count"`
}

type ToolConfig struct {
	LatencyMs int `yaml:"latency_ms"`
	JitterMs  int `yaml:"jitter_ms"`
}