Skip to content

pritpatel2412/StockMind

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

2 Commits
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 

Repository files navigation

๐Ÿ“ˆ StockMind

AI-Powered Multi-Agent Market Simulation & Sentiment Analysis Platform

A neo-brutalist "War Room" where thousands of LLM agents trade in real-time.


What is StockMind?

StockMind is a multi-agent financial simulation system that models hedge funds, retail traders, market makers, and regulators โ€” each with distinct strategies, risk profiles, and AI-driven decision logic.

Watch thousands of agents trade in real-time. Trigger market scenarios. Observe emergent behavior. Understand how sentiment, news, and agent psychology shape market dynamics.

This is not a trading app. It's a simulation engine for market research, strategy testing, and AI behavior analysis.


Agent Types

Agent Strategy Risk Profile Behavior
๐Ÿฆ Hedge Fund Quantitative / Arbitrage High (0.8) Maximize alpha via complex timing
๐Ÿ‘ค Retail Trader Emotional / Momentum Low (0.3) FOMO-driven herding behavior
๐Ÿ“ฐ News Agent Information Broker N/A Classifies & broadcasts sentiment
๐Ÿช Market Maker Delta Neutral / Spread Med (0.2) Provides liquidity, maintains spreads
โš–๏ธ Regulator Market Stability N/A Enforces circuit breakers (20% halt)

Key Features

โšก Real-Time Simulation Engine

  • FastAPI + AsyncIO backend orchestrates thousands of concurrent agent decisions
  • WebSocket streaming delivers live market dynamics to the dashboard
  • Sub-second latency from agent decision โ†’ chart update

๐Ÿง  AI-Driven Sentiment Analysis

  • Groq LLaMA 3.3 70B processes real-time news ingestion
  • Sentiment scores directly influence agent behavior and price movements
  • Heatmaps visualize sentiment distribution across market sectors

๐Ÿ“š RAG-Powered Decision Making

  • NVIDIA NIM embeddings (nv-embedqa-e5-v5) index historical market events
  • Agents query this knowledge base to inform decisions โ€” e.g., "this pattern looks like 2008"
  • Retrieval-augmented reasoning makes agent behavior more realistic

๐ŸŽฎ Scenario Testing

Pre-configured market scenarios:

  • ๐Ÿš€ Bull Run
  • ๐Ÿป Bear Crash
  • ๐Ÿธ Meme Stock Frenzy
  • โšก Flash Crash

๐Ÿ–ฅ๏ธ Neo-Brutalist War Room Dashboard

  • High-contrast dark mode with electric lime, purple, and pink accents
  • Live Recharts visualizations: price charts, volume bars, sentiment heatmaps
  • Agent activity feed โ€” see each agent's decision in real-time

Tech Stack

Layer Technology
Frontend Next.js 15, React 19, TypeScript, Tailwind CSS, Recharts
Backend FastAPI, Python, AsyncIO
Real-time WebSockets
LLM Groq LLaMA 3.3 70B
RAG NVIDIA NIM Embeddings
Market Data yfinance
News/Sentiment NewsAPI
Database Supabase PostgreSQL
Auth JWT with token rotation

Architecture

                    โ”Œโ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”
                    โ”‚   Next.js War Room Dashboard  โ”‚
                    โ”‚  (Neo-brutalist real-time UI) โ”‚
                    โ””โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”ฌโ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”˜
                                   โ”‚ WebSocket
                    โ”Œโ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ–ผโ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”
                    โ”‚      FastAPI Backend           โ”‚
                    โ”‚   Simulation Engine (AsyncIO)  โ”‚
                    โ””โ”€โ”€โ”ฌโ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”ฌโ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”ฌโ”€โ”€โ”€โ”€โ”€โ”˜
                       โ”‚          โ”‚           โ”‚
              โ”Œโ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ–ผโ”€โ”€โ”  โ”Œโ”€โ”€โ”€โ–ผโ”€โ”€โ”€โ”€โ”  โ”Œโ”€โ”€โ–ผโ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”
              โ”‚ Agent Swarmโ”‚  โ”‚  Groq  โ”‚  โ”‚  NVIDIA NIM  โ”‚
              โ”‚(1000s LLM) โ”‚  โ”‚  LLM   โ”‚  โ”‚  RAG Engine  โ”‚
              โ””โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”˜  โ””โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”˜  โ””โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”˜
                       โ”‚
              โ”Œโ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ–ผโ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”
              โ”‚  yfinance + NewsAPI    โ”‚
              โ”‚  (Real market data)    โ”‚
              โ””โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”˜

Getting Started

git clone https://github.com/pritpatel2412/StockMind
cd StockMind

# Frontend
npm install
npm run dev

# Backend
cd stockmind-backend
pip install -r requirements.txt
uvicorn main:app --reload

Required env vars: GROQ_API_KEY, NVIDIA_NIM_API_KEY, NEWS_API_KEY, SUPABASE_URL, SUPABASE_SERVICE_ROLE_KEY


Built By

Prit Patel โ€” B.Tech CSE @ CHARUSAT University GitHub ยท LinkedIn

About

๐Ÿ“ˆ Multi-agent financial simulation โ€” thousands of LLM agents trading in real-time. FastAPI, Groq, NVIDIA NIM RAG, WebSockets, neo-brutalist War Room dashboard.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors