Inspiration

The inspiration for this project came from a simple question:
What if a user could generate a complete machine learning project just by describing their idea?

Most ML workflows require repetitive manual steps like data preparation, model selection, training, and API deployment. I wanted to automate this entire pipeline using multiple collaborating AI agents. These help people with non tech background to create a lot of projects on their own.

What it does

I built a Multi-Agent AI Orchestrator System that takes a single user prompt and automatically generates a complete machine learning project.

The system consists of specialized agents:

  • Requirements Agent → understands and structures the problem
  • Planning Agent → designs ML architecture and workflow
  • Code Agent → generates full working Python ML project
  • Dataset Agent → creates dataset schema and sample data
  • Documentation Agent → generates professional README documentation

All agents are powered by Groq LLM (Llama 3.1)for fast and reliable execution.

How we built it

The system is built in Python using a modular agent-based architecture.

Each agent performs a specific task and passes structured output to the next stage.

The pipeline works as follows:

[ \text{User Prompt} \rightarrow \text{Requirements Agent} \rightarrow \text{Planning Agent} \rightarrow \text{Code Agent} \rightarrow \text{Dataset Agent} \rightarrow \text{Documentation Agent} ]

Finally, all outputs are saved into a fully runnable ML project inside the output_project/ folder.

Challenges we ran into

Some major challenges I faced include:

  • Handling API rate limits and failures from LLM providers especially with gemini-2.0-flash. I used gemini-2.0-flash,gemini-2.0-pro,gemini-1.5-flash etc but due to its limited count I was not able to continue further.
  • Designing a stable fallback system when APIs fail
  • Parsing structured outputs from LLMs reliably
  • Ensuring consistent file generation across multiple agents -A lot of trial and error in different agents. -It was difficult to generate files but finally it was a success. The project generates files, instructions, code inside project_output.

I solved these by simplifying the system and making Groq the primary execution engine for stability.


Accomplishments that we're proud of

-Completing a hackathon on my own. -Finding a problem and fully implementing the solution and that too in a fully working environment. -Help a lot of people to generate projects based on the same.

What we learned

The final system can generate:

  • Complete ML training pipeline
  • Prediction scripts
  • Flask API for deployment
  • Dataset generation
  • Documentation

from just a single user prompt.

What's next for AI-Agent Orchestrator

In the future, I plan to:

  • Add memory-based agents
  • Introduce LangGraph orchestration
  • Add UI dashboard for interaction
  • Support OpenAI + Gemini + Groq dynamic routing -And a lot more.

Built With

Share this project:

Updates