prodigy 0.4.4

Turn ad-hoc Claude sessions into reproducible development pipelines with parallel AI agents
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

//! Tests for cook orchestrator core functionality
//!
//! Separated from core.rs to keep the main module under 500 LOC.

use super::core::{CookConfig, DefaultCookOrchestrator};
use super::workflow_classifier::WorkflowType;
use crate::config::WorkflowConfig;
use crate::cook::command::CookCommand;
use std::path::PathBuf;
use std::sync::Arc;

fn create_test_cook_command() -> CookCommand {
    CookCommand {
        playbook: PathBuf::from("test.yaml"),
        path: None,
        max_iterations: 1,
        map: vec![],
        args: vec![],
        fail_fast: false,
        auto_accept: false,
        resume: None,
        verbosity: 0,
        quiet: false,
        dry_run: false,
        params: std::collections::HashMap::new(),
    }
}

#[test]
fn test_classify_workflow_type_standard() {
    let config = CookConfig {
        command: create_test_cook_command(),
        project_path: Arc::new(PathBuf::from("/test")),
        workflow: Arc::new(WorkflowConfig {
            name: None,
            commands: vec![],
            env: None,
            secrets: None,
            env_files: None,
            profiles: None,
            merge: None,
        }),
        mapreduce_config: None,
    };

    assert_eq!(
        DefaultCookOrchestrator::classify_workflow_type(&config),
        WorkflowType::Standard
    );
}

#[test]
fn test_classify_workflow_type_with_arguments() {
    let mut command = create_test_cook_command();
    command.args = vec!["arg1".to_string()];

    let config = CookConfig {
        command,
        project_path: Arc::new(PathBuf::from("/test")),
        workflow: Arc::new(WorkflowConfig {
            name: None,
            commands: vec![],
            env: None,
            secrets: None,
            env_files: None,
            profiles: None,
            merge: None,
        }),
        mapreduce_config: None,
    };

    assert_eq!(
        DefaultCookOrchestrator::classify_workflow_type(&config),
        WorkflowType::WithArguments
    );
}

#[test]
fn test_classify_workflow_type_with_map_patterns() {
    let mut command = create_test_cook_command();
    command.map = vec!["*.rs".to_string()];

    let config = CookConfig {
        command,
        project_path: Arc::new(PathBuf::from("/test")),
        workflow: Arc::new(WorkflowConfig {
            name: None,
            commands: vec![],
            env: None,
            secrets: None,
            env_files: None,
            profiles: None,
            merge: None,
        }),
        mapreduce_config: None,
    };

    assert_eq!(
        DefaultCookOrchestrator::classify_workflow_type(&config),
        WorkflowType::WithArguments
    );
}