EMS Analyst Agent 🤖💬

Energy analyst agent with persistent memory, agentic RAG using vector DB, Neo4j knowledge graph powered by Graphiti, and external APIs for comprehensive energy market analysis

Architecture

rectangle agent {
  database vector_chat_history
  rectangle conversational_llm
  rectangle tool_api
}
rectangle domain_mcp_server {
    database vector_knowledge_base
    database knowledge_graph
    rectangle query_logic
}


cloud openweather 
cloud yes_energy 
cloud permutable 
person user
query_logic -u- knowledge_graph
query_logic -u- vector_knowledge_base
vector_chat_history -l- tool_api
tool_api -u- query_logic
tool_api -d- openweather
tool_api -d- yes_energy
tool_api -d- permutable
conversational_llm -r- tool_api
conversational_llm -l- user

Tools

Memory Tools

• Conversation History: Persistent message storage and retrieval

• Context Awareness: Access to previous conversations and analysis

• Session Management: Multi-turn conversation capabilities

Vector Search Tools

• Semantic Search: Vector similarity search across energy domain documents

• Hybrid Search: Combined vector and keyword search

• Document Retrieval: Full document access with chunking

Knowledge Graph Tools

• Entity Relationships: Explore connections between energy concepts

• Timeline Queries: Historical analysis of energy events

• Graph Search: Semantic queries across knowledge relationships

External API Tools

• Weather Analysis: OpenWeather API for weather impact on energy systems

• Market Data: YES Energy API for real-time energy market information

• Geopolitical Intelligence: Permutable AI for events affecting energy markets

Prompts

System Prompts

• system_analyst.md: Core energy analyst personality and capabilities

• system_safety.md: Safety guidelines for energy recommendations

Task Prompts

• market_analysis.md: Energy market analysis and forecasting

• weather_impact.md: Weather correlation with energy demand/supply

• geopolitical_analysis.md: Geopolitical event impact assessment

• technical_analysis.md: Power systems and grid analysis

Response Prompts

• explanation.md: Detailed technical explanations

• summary.md: Executive summary format

• recommendation.md: Actionable recommendations

Domain Knowledge

Vector Database & Knowledge Graph Content

Both systems are trained on the same energy domain books for consistency:

• Energy Trading Books: Market fundamentals and trading strategies

• Power Systems Engineering Book: Grid operations and electrical engineering

• **The BESS Book **: Battery energy storage system book

• Historical Market Data: Past energy market events and patterns

Knowledge Graph Entities

• Market Participants: Utilities, ISOs, traders, generators

• Infrastructure: Power plants, transmission lines, storage systems

• Regulations: FERC orders, state policies, market rules

• Events: Outages, weather events, policy changes

Project Structure

├── pyproject.toml              # Dependencies and build config
├── src/
│   ├── main.py                 # Agent entry point
│   ├── agent.py                # Pydantic AI agent definition
│   ├── agent_test.py           # Agent unit tests
│   ├── memory/
│   │   ├── __init__.py
│   │   ├── history.py          # Message history management
│   │   └── history_test.py
│   ├── rag/
│   │   ├── __init__.py
│   │   ├── vector_search.py    # Vector search implementation
│   │   ├── vector_search_test.py
│   │   ├── graph_utils.py      # Knowledge graph utilities
│   │   └── graph_utils_test.py
│   ├── tools/
│   │   ├── __init__.py
│   │   ├── weather.py          # OpenWeather API integration
│   │   ├── weather_test.py
│   │   ├── markets.py          # YES Energy API integration
│   │   ├── markets_test.py
│   │   ├── geopolitical.py     # Permutable AI integration
│   │   └── geopolitical_test.py
│   └── utils/
│       ├── __init__.py
│       ├── embeddings.py       # Embedding utilities
│       └── embeddings_test.py
├── prompts/
│   ├── system/
│   │   ├── system_analyst.md
│   │   └── system_safety.md
│   ├── tasks/
│   │   ├── market_analysis.md
│   │   ├── weather_impact.md
│   │   ├── geopolitical_analysis.md
│   │   └── technical_analysis.md
│   └── responses/
│       ├── explanation.md
│       ├── summary.md
│       └── recommendation.md
├── tests/
│   └── test_integration.py     # End-to-end agent tests
└── README.md                   # This file