File size: 1,135 Bytes
2004c79
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
"""
Factory module for data creation - both algorithmic and AI-powered.

This module contains all data creation, generation, and formatting logic
for the Yuga Planner scheduling system, organized into:
- data: Algorithmic data generation and formatting
- agents: AI-powered task composition
"""

# Import from data submodule
from .data.formatters import schedule_to_dataframe, employees_to_dataframe
from .data.generators import (
    generate_employees,
    generate_employee_availability,
    earliest_monday_on_or_after,
)
from .data.provider import generate_agent_data, generate_mcp_data

# Import from agents submodule
from .agents.task_composer_agent import TaskComposerAgent

__all__ = [
    # Data formatters - convert domain objects to DataFrames
    "schedule_to_dataframe",
    "employees_to_dataframe",
    # Data generators - create domain objects
    "generate_employees",
    "generate_employee_availability",
    "earliest_monday_on_or_after",
    # Data providers - orchestrate data creation
    "generate_agent_data",
    "generate_mcp_data",
    # AI agents - intelligent task composition
    "TaskComposerAgent",
]