batch

type Config struct {
	MaxConcurrent int           // Max parallel jobs
	RetryCount    int           // Number of retries on failure
	RetryDelay    time.Duration // Delay between retries
	ResultDir     string        // Directory to save results
	ProgressDir   string        // Directory to save progress
	Timeout       time.Duration // Job timeout
}

type Job struct {
	ID             string                 `json:"id"`
	Name           string                 `json:"name"`
	Type           string                 `json:"type"` // trajectory, query, skill
	InputFile      string                 `json:"input_file"`
	OutputFile     string                 `json:"output_file"`
	Params         map[string]interface{} `json:"params"`
	Status         string                 `json:"status"`   // pending, running, completed, failed
	Progress       float64                `json:"progress"` // 0-100
	CreatedAt      time.Time              `json:"created_at"`
	StartedAt      *time.Time             `json:"started_at,omitempty"`
	CompletedAt    *time.Time             `json:"completed_at,omitempty"`
	Error          string                 `json:"error,omitempty"`
	RetryCount     int                    `json:"retry_count"`
	TotalItems     int                    `json:"total_items"`
	ProcessedItems int                    `json:"processed_items"`
}

type JobResult struct {
	JobID       string         `json:"job_id"`
	Success     bool           `json:"success"`
	OutputFile  string         `json:"output_file"`
	Results     []TaskResult   `json:"results,omitempty"`
	Errors      []TaskError    `json:"errors,omitempty"`
	Summary     *ResultSummary `json:"summary,omitempty"`
	Duration    time.Duration  `json:"duration"`
	ProcessedAt time.Time      `json:"processed_at"`
}

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

type QueryJob struct {
	Job
	Tasks []QueryTask `json:"tasks"`
}

type QueryTask struct {
	TaskID   string  `json:"task_id"`
	Query    string  `json:"query"`
	Response string  `json:"response,omitempty"`
	Tokens   int     `json:"tokens"`
	Cost     float64 `json:"cost"`
	Success  bool    `json:"success"`
	Error    string  `json:"error,omitempty"`
}

type ResultSummary struct {
	TotalTasks      int           `json:"total_tasks"`
	Successful      int           `json:"successful"`
	Failed          int           `json:"failed"`
	SuccessRate     float64       `json:"success_rate"`
	TotalDuration   time.Duration `json:"total_duration"`
	AvgTaskDuration time.Duration `json:"avg_task_duration"`
	TotalTokens     int64         `json:"total_tokens"`
	TotalCost       float64       `json:"total_cost"`
}

type TaskError struct {
	TaskID string      `json:"task_id"`
	Error  string      `json:"error"`
	Input  interface{} `json:"input"`
}

type TaskResult struct {
	TaskID   string                 `json:"task_id"`
	Input    interface{}            `json:"input"`
	Output   interface{}            `json:"output"`
	Duration time.Duration          `json:"duration"`
	Success  bool                   `json:"success"`
	Metadata map[string]interface{} `json:"metadata,omitempty"`
}

type ToolCall struct {
	ID        string                 `json:"id"`
	Name      string                 `json:"name"`
	Arguments map[string]interface{} `json:"arguments"`
}

type TrajectoryJob struct {
	Job
	Tasks []TrajectoryTask `json:"tasks"`
}

type TrajectoryStep struct {
	Step       int        `json:"step"`
	Role       string     `json:"role"` // user, assistant, tool
	Content    string     `json:"content"`
	ToolCalls  []ToolCall `json:"tool_calls,omitempty"`
	ToolResult string     `json:"tool_result,omitempty"`
	Timestamp  time.Time  `json:"timestamp"`
}

type TrajectoryTask struct {
	TaskID        string           `json:"task_id"`
	Query         string           `json:"query"`
	SystemPrompt  string           `json:"system_prompt,omitempty"`
	Tools         []string         `json:"tools,omitempty"`
	MaxSteps      int              `json:"max_steps,omitempty"`
	ExpectedTools []string         `json:"expected_tools,omitempty"`
	Trajectory    []TrajectoryStep `json:"trajectory,omitempty"`
	Success       bool             `json:"success"`
	Error         string           `json:"error,omitempty"`
}