Aria is a intelligent multi agent system that is designed to help manufacturing engineers and manufacturing plant staff to diagnose machine failures using natural language minimizing the risk of any unhappening by pulling the relevant history from machine records and historic maintenance data.
Ask aria why a machine is failing or an uncertain activity is happening it pulls relevant data from machine records, SAP maintenance data and returns a structured answer with escalation priority.
I have built this project for manufacturing usecase as most of the ai tools in this segment are inefficeint or too expesive.
Aria cosnsit of 6 different agents/ functions orchestrated via LangGraph, each agent has its role and intent as follows:
- classifier : Figures out the user query intent
- Retrieval : makes a hybrid search ( BM25 keyword + semantic ) over machine records
- Sap Connector: custom tools based agent created to fetch real data (workcase demonstrated with syntetic data in mvp) and perform operation/ tool calling based on intent.
- Reasoning: deep analysis for complex root cause queries via user
- Escalationn: Decides the priority of situation based on confidence and stock levels
- Synthesis: Generates the final structured output answer with relevant confidence score
state flows through all agents via shared TypeDict (LangGraph)
- Agentic framework : Langchain, LangGraph
- vector storage: ChromaDB , Sentence-transformers (local embedding)
- LLM Reasoning: Groq llama3-70b (can be swappable to Azure OpenAI in production)
- Dataset: AI4I 2020 Predictive Maintenance ( Kaggle), custom build sap_maintenance.csv dataset
- framework: FastAPi
- Evaluation: Ragas, manual testing
git clone
cd ARIA
python -m venv venv
venv\Scripts\activate
pip install -r requirements.txt
cp .env.example as .env (get api keys)Build Vectorstore (initial setup only):
python src/vectorstore.pyRun pipeline:
python main.pyRun API:
uvicorn src.api:app --reload --port 8000swagger : http://localhost:8000/docs
python evaluation/test_suite.py
python evaluation/ragas_eval.py- Local : ChromaDB, Groq(used)/Gemini , FastAPI
- production: Azure Ai Search, Azure OpenAI, Azure Container Apps
Config change only no code rewrite needed for production migration.