ai-trustgraph 0.1.1

AI Trust Infrastructure for Enterprise Data and AI Agents

TrustGraph API

AI data trust layer for enterprise data platforms. Connects to Snowflake and Databricks, registers data sources, stores metadata in PostgreSQL, and exposes a FastAPI backend.

Quick Start

cd trustgraph-api
python -m venv .venv
source .venv/bin/activate
pip install -r requirements.txt
docker compose up -d postgres
cp .env.example .env
uvicorn app.main:app --reload

Open http://localhost:8000/docs for Swagger UI.

Endpoints

Method	Path	Description
GET	/health	Health check
POST	/auth/register	Register new user
POST	/auth/login	Login, get JWT
GET	/auth/me	Current user info
POST	/sources	Create data source
GET	/sources	List all sources
GET	/sources/{id}	Get source by ID
DELETE	/sources/{id}	Delete source
POST	/sources/snowflake/test	Test Snowflake connection
POST	/sources/databricks/test	Test Databricks connection

Running Tests

python -m pytest tests/ -v

Docker

docker compose up --build

Project Structure

trustgraph-api/
├── app/
│   ├── api/           # Route handlers
│   │   ├── auth.py
│   │   ├── health.py
│   │   ├── source_tests.py
│   │   └── sources.py
│   ├── auth/          # JWT and security
│   │   ├── dependencies.py
│   │   └── security.py
│   ├── connectors/    # DB connectors
│   │   ├── databricks_connector.py
│   │   └── snowflake_connector.py
│   ├── db/            # Database engine/session
│   │   ├── base.py
│   │   ├── engine.py
│   │   └── session.py
│   ├── models/        # ORM models
│   │   ├── data_source.py
│   │   ├── dataset.py
│   │   └── user.py
│   ├── schemas/       # Pydantic schemas
│   │   ├── auth.py
│   │   ├── dataset.py
│   │   └── source.py
│   ├── services/      # Business logic
│   │   ├── metadata_service.py
│   │   └── source_service.py
│   ├── main.py
│   └── settings.py
├── tests/
├── Dockerfile
├── docker-compose.yml
├── requirements.txt
└── .env.example