Skip to main content

SightRAG: Image and Video RAG. See. Search. Retrieve.

Project description

SightRAG Banner

SightRAG

See. Search. Retrieve.

PyPI License Python Open In Colab

Search your images and videos with plain English. Three lines of code.


Quick Start

from sightrag import SightRAG

rag = SightRAG()
rag.index("./photos/")
results = rag.query("find empty shelf")

Install

pip install sightrag

For REST API:

pip install sightrag[api]

What SightRAG Does

Point SightRAG at any image folder, video file, or camera. Ask in plain English. Get back matched images with bounding boxes, timestamps, and confidence scores.

SightRAG is not a model. Not a wrapper. Not a framework plugin. It is a complete retrieval system that handles detection, embedding, indexing, and search — so you don't have to.

All models and indexes are stored in ~/.sightrag/ — your project folder stays clean.

Project Structure

sightrag/
├── sightrag/                    ← main library (pip install)
│   ├── core.py                  ← SightRAG class
│   ├── detector.py              ← YOLO detection
│   ├── embedder.py              ← CLIP embeddings
│   ├── indexer.py               ← image/video/camera indexing
│   ├── retriever.py             ← text + reference queries
│   ├── api.py                   ← REST API (FastAPI)
│   ├── store/
│   │   ├── base.py              ← abstract store interface
│   │   ├── sqlite_store.py      ← SQLite fallback
│   │   └── chroma_store.py      ← ChromaDB (default)
│   └── utils/
│       ├── image.py             ← image loading
│       ├── video.py             ← frame extraction
│       └── camera.py            ← webcam capture
│
├── demo_sightrag/               ← test locally — run the scripts
│   ├── sightrag_images.py       ← demo: image folder indexing
│   ├── sightrag_video.py        ← demo: video file indexing
│   ├── sightrag_livecam.py      ← demo: live webcam capture
│   ├── sightrag_restapi.py      ← demo: REST API usage
│   ├── input_images/            ← sample images to index
│   ├── reference_images/        ← sample reference query images
│   ├── camera_captures/         ← webcam frames stored here
│   └── video_samples/           ← put your videos here
│
├── notebooks/                   ← test on Google Colab
│   └── SightRAG_Colab_Demo.ipynb
│
├── examples/                    ← code examples
│   ├── basic_usage.py
│   ├── camera_example.py
│   ├── custom_domain_example.py
│   └── rest_api_example.py
│
├── tests/                       ← unit tests
│   └── test_core.py
│
├── docs/                        ← documentation
│   └── DOCKER.md
│
├── assets/                      ← banner image
├── setup.py                     ← PyPI packaging
├── pyproject.toml               ← build config
├── requirements.txt             ← dependencies
├── Dockerfile                   ← container
├── docker-compose.yml           ← one command deploy
├── LICENSE                      ← Apache 2.0
└── test_sightrag.py             ← quick test script

How To Test

Quick Test (terminal)

python test_sightrag.py

Demo Scripts (test each mode)

python demo_sightrag/sightrag_images.py     # image folder
python demo_sightrag/sightrag_video.py       # video files
python demo_sightrag/sightrag_livecam.py     # live webcam
python demo_sightrag/sightrag_restapi.py     # REST API

Google Colab

Upload notebooks/SightRAG_Colab_Demo.ipynb to Google Colab → Run All

Unit Tests

pip install pytest
python -m pytest tests/ -v

Usage

Image Folder

from sightrag import SightRAG

rag = SightRAG()
rag.index("./shelf_photos/")
results = rag.query("find empty shelf near dairy")

for r in results:
    print(f"{r['image_path']} — score: {r['score']}{r['label']}")

Video File

rag = SightRAG()
rag.index("./cctv_footage.mp4")
results = rag.query("person near exit door")

for r in results:
    print(f"Timestamp: {r['timestamp']} — score: {r['score']}")

Mixed Folder (images + videos)

rag = SightRAG()
rag.index("./my_data/")  # automatically detects images AND videos

Live Camera

rag = SightRAG()
rag.index(source="camera")              # default webcam
rag.index(source="camera", camera_id=1) # specific camera
results = rag.query("find person")

Reference Image Query

results = rag.query(reference="sample_shelf.jpg")

Custom Domain (medical, industrial, satellite)

rag = SightRAG(domain_hint="pcb defect solder joint circuit board")
rag.index("./circuit_boards/")
results = rag.query("find defective solder joint")

SQLite Fallback (lightweight)

rag = SightRAG(store="sqlite")
rag.index("./small_dataset/")

REST API

pip install sightrag[api]

Start the server:

# Command line
sightrag-server

# Or from Python
from sightrag import serve
serve(port=8000)

# Or Docker (API starts automatically)
docker-compose up

API available at http://localhost:8000

Endpoints

Method Endpoint Description
GET / API info and available endpoints
GET /status Index count, store type, domain hint
POST /index/folder Index all images and videos in a folder
POST /index/video Index a video file
POST /index/upload Upload and index images directly
POST /query/text Search with plain English text
POST /query/reference Search with a reference image
DELETE /index Clear all indexed data

Examples

# Index a folder
curl -X POST http://localhost:8000/index/folder \
     -F "path=./data/"

# Search with text
curl -X POST http://localhost:8000/query/text \
     -F "text=find empty shelf" \
     -F "top_k=5"

# Upload and search with reference image
curl -X POST http://localhost:8000/query/reference \
     -F "file=@new_photo.jpg" \
     -F "top_k=5"

# Check status
curl http://localhost:8000/status

Interactive API docs available at http://localhost:8000/docs (Swagger UI).

Result Format

[
    {
        "image_path":  "./photos/shelf_042.jpg",
        "score":       0.9134,
        "label":       "bottle",
        "confidence":  0.8721,
        "bbox":        [120, 45, 380, 290],
        "timestamp":   "",
        "source_type": "image"
    }
]

Storage

SightRAG uses a built-in SQLite vector store by default — zero extra dependencies, works everywhere.

Store Scale Cost Usage
SQLite (default) Up to 100k images Free SightRAG()
ChromaDB (optional) Large scale Free SightRAG(store="chroma")

Enterprise connectors (Qdrant, Pinecone, Azure) coming in v2.

Where SightRAG Stores Data

~/.sightrag/
├── models/      ← YOLO weights (auto-downloaded once)
└── index/       ← vector database (ChromaDB/SQLite)

Your project folder stays clean. No random .pt files or sightrag_index/ folders appearing.

Docker

docker-compose up

This starts the REST API server on port 8000. See Docker Guide for details.

Architecture

Input (images / video / camera / reference image)
        ↓
   Preprocessor (resize, validate, keyframe extract)
        ↓
   YOLO Detection (80 COCO classes + whole-image fallback)
        ↓
   CLIP Embedding (domain_hint enrichment for custom domains)
        ↓
   Vector Store (ChromaDB default / SQLite fallback)
        ↓
   Retrieval + Ranking (cosine similarity, confidence scoring)
        ↓
Output (matched images, timestamps, bounding boxes, scores)

Honest Limitations (v0.1.0)

  • Indexing ~500 images takes ~2 minutes on CPU (one-time cost — search is instant after)
  • Custom domains (medical, satellite) use whole-image CLIP without region grounding
  • Single webcam only (multiple cameras in v2)
  • SQLite vector store loads all vectors into memory for search

Roadmap

Version Features
v0.1 (current) Image + Video + Camera + REST API + ChromaDB
v0.2 C++ core, CLI, Grounding DINO, SAM
v0.3 Person Re-ID, scene graph, edge deployment
v1.0 Jetson Orin, compliance modes (GDPR/HIPAA/DPDP)

Three Library Ecosystem

SightRAG is part of the VK-Ant AI ecosystem:

Library Purpose Status
SightRAG Image & Video RAG v0.1
adaptive-intelligence RL-based RAG orchestration v4.0.7
llmevalkit LLM evaluation (97+ tests) Stable

License

Apache 2.0

Author

Built by Venkatkumar Rajan

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

sightrag-0.1.5.tar.gz (24.1 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

sightrag-0.1.5-py3-none-any.whl (22.9 kB view details)

Uploaded Python 3

File details

Details for the file sightrag-0.1.5.tar.gz.

File metadata

  • Download URL: sightrag-0.1.5.tar.gz
  • Upload date:
  • Size: 24.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.14.2

File hashes

Hashes for sightrag-0.1.5.tar.gz
Algorithm Hash digest
SHA256 f33004460786a79fb539c73e92a8bfd2f831c364a3661ae2222c70d06766e445
MD5 fda25d9daa7c0baf1c94ac8576022f3f
BLAKE2b-256 5265c14a073514625f9c33287445b7ff31f7e3b44ee3bc65b2e31215748b400c

See more details on using hashes here.

File details

Details for the file sightrag-0.1.5-py3-none-any.whl.

File metadata

  • Download URL: sightrag-0.1.5-py3-none-any.whl
  • Upload date:
  • Size: 22.9 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.14.2

File hashes

Hashes for sightrag-0.1.5-py3-none-any.whl
Algorithm Hash digest
SHA256 aa881e93ce1ff3c60249d5c3f274672750edef120cf0c5e5519c09f592388662
MD5 770a7860b639b51aa9c5f39828b59df6
BLAKE2b-256 8144db2891e7373a743ceed82b49ed258027e8f7730c7e4778096b14f3a20605

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page