mnds18/agentic-ts-forecasting-system

1# cli_runner.py2"""3Command-line interface to run any individual agent or full pipeline.4"""5 6import argparse7from agentic_ts_forecasting_system import run_pipeline8from agents import (9    requirements_agent,10    data_generation_agent,11    data_scientist_agent,12    peer_review_agent,13    documentation_agent,14    project_manager_agent,15    presentation_agent16)17 18def main():19    parser = argparse.ArgumentParser(description="Agentic AI CLI")20    parser.add_argument("--agent", type=str, help="Which agent to run individually")21    args = parser.parse_args()22 23    if not args.agent:24        run_pipeline()25        return26 27    agent_map = {28        "requirements": requirements_agent.run_requirements_gathering,29        "generation": data_generation_agent.run_data_generation,30        "training": data_scientist_agent.run_model_training,31        "review": peer_review_agent.run_peer_review,32        "report": documentation_agent.generate_report,33        "plan": project_manager_agent.generate_plan_and_gantt,34        "slides": presentation_agent.generate_slide_deck,35    }36 37    agent_fn = agent_map.get(args.agent.lower())38    if agent_fn:39        print(f"Running {args.agent} agent...")40        agent_fn()41    else:42        print("Invalid agent name. Available agents:", ", ".join(agent_map.keys()))43 44if __name__ == "__main__":45    main()46