Skip to content

step_2

arai_ai_agents.prompt_chaining.step_2

step_2(ai_model, agent_file_path)

Description

Create a new season for the agent

Parameters:

Name Type Description Default
ai_model

The AI model to use for generating responses

 required
agent_file_path

The path to the agent yaml file

 required

Returns:

Name Type Description
season_file_path

The path to the season yaml file

Raises:

Type Description
Exception

If there's an error creating the season
Example

ai_model = GeminiModel() agent_file_path = "agent_template.yaml" step_2(ai_model, agent_file_path)

Source code in arai_ai_agents/prompt_chaining/step_2.py 
 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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
def step_2(ai_model, agent_file_path):
    '''
    Description:
        Create a new season for the agent

    Args:
        ai_model: The AI model to use for generating responses
        agent_file_path: The path to the agent yaml file

    Returns:
        season_file_path: The path to the season yaml file

    Raises:
        Exception: If there's an error creating the season

    Example:
        >>> ai_model = GeminiModel()
        >>> agent_file_path = "agent_template.yaml"
        >>> step_2(ai_model, agent_file_path)
    '''
    print("Step 2: Create a new season") 

    # step 2.1: load the season template yaml file
    manager = ContentGenerator()
    season_template = manager.create_new_template_yaml(TemplateType.SEASON)

    # step 2.2: load the agent yaml file
    agent_yaml = None    
    with open(agent_file_path, 'r', encoding='utf-8') as file:
        agent_yaml = yaml.safe_load(file)

    # step 2.3: find the previous season 
    # get last season file 
    previous_season = None
    # Update the path to include the agent's specific directory
    agent_seasons_dir = os.path.join(manager.agents_config_dir, agent_yaml['name'])


    # Check if the agent directory exists
    if os.path.exists(agent_seasons_dir):
        # Look for season folders in the agent's directory
        season_folders = [f for f in os.listdir(agent_seasons_dir) if f.startswith("season_")]

        if season_folders:
            # Sort the folders by season number and get the last one
            last_folder = sorted(season_folders, key=lambda x: int(x.split('_')[1]))[-1]

            season_file_path = os.path.join(agent_seasons_dir, last_folder, f"{last_folder}.yaml")
            print(f"Looking for previous season at: {season_file_path}")  # Debug print

            if os.path.exists(season_file_path):
                with open(season_file_path, 'r', encoding='utf-8') as file:
                    previous_season = yaml.safe_load(file)
                print(f"Found previous season: {last_folder}")
        else:
            print(f"No season folders found in {agent_seasons_dir}")
    else:
        print(f"Agent directory not found at {agent_seasons_dir}")



    # step 2.3: Generate a new season name, topic, and communication style with the prompt_2 template
    # prompt 2 Season Creation:
    # note that emojis will be output as unicode characters due to the yaml dump
    prompt_2_vars = {
        "agent_name": agent_yaml["name"],
        "agent_yaml": yaml.dump(agent_yaml),
        "season_yaml": yaml.dump(season_template),
        "previous_season": yaml.dump(previous_season)
    }

    # step 2.4: Run the prompt 
    season_data = manager.run_prompt(
        # prompt_key="prompt_1 (Character Creation)",
        prompt_key="prompt_2 (Season Creation)",
        template_vars=prompt_2_vars, 
        ai_model=ai_model,
    )

    # step 2.5: Add the season data to the season template
    season_template = manager.add_data_to_template(
        current_data=season_template,
        new_data=season_data, 
    )

    # step 2.6: create the file path
    season_file_path = manager.create_filepath(
        agent_name=agent_yaml["name"],
        season_number=season_template["season"]["season_number"],
        episode_number=0,
        template_type=TemplateType.SEASON
    )

    # step 2.7: Save the season data to a file
    manager.save_yaml_file(
        save_path=season_file_path,
        yaml_data=season_template
    )

    return season_file_path