omni-proof 0.0.1__tar.gz

omni_proof-0.0.1/.claude/scheduled_tasks.lock

@@ -0,0 +1 @@
+{"sessionId":"2ca994de-85ae-4b26-8b4d-cbdaa5f3ac43","pid":22746,"acquiredAt":1773525213679}

omni_proof-0.0.1/.env.example

@@ -0,0 +1,6 @@
+OMNI_PROOF_GEMINI_API_KEY=your-gemini-api-key
+OMNI_PROOF_PINECONE_API_KEY=your-pinecone-api-key
+OMNI_PROOF_PINECONE_INDEX_HOST=your-index-host.svc.pinecone.io
+OMNI_PROOF_DATABASE_URL=sqlite+aiosqlite:///./omni_proof.db
+OMNI_PROOF_EMBEDDING_DIMENSIONS=3072
+OMNI_PROOF_LOG_LEVEL=INFO

omni_proof-0.0.1/.github/workflows/ci.yml

@@ -0,0 +1,47 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  lint:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+      - run: pip install -e ".[dev]"
+      - name: Ruff check
+        run: ruff check src/ tests/
+      - name: Mypy
+        run: mypy src/omni_proof/ --ignore-missing-imports
+
+  test:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.11", "3.12"]
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+      - run: pip install -e ".[dev]"
+      - name: Unit tests with coverage
+        run: pytest tests/unit/ -v --tb=short --cov=omni_proof --cov-report=term-missing --cov-fail-under=70
+
+  integration:
+    runs-on: ubuntu-latest
+    needs: test
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+      - run: pip install -e ".[dev]"
+      - name: Integration tests
+        run: pytest tests/integration/ -v --tb=short

omni_proof-0.0.1/.github/workflows/release.yml

@@ -0,0 +1,65 @@
+name: Release
+
+on:
+  push:
+    tags: ["v*"]
+
+permissions:
+  contents: write
+  id-token: write
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    # Only release from main branch
+    if: github.ref_type == 'tag' && startsWith(github.ref, 'refs/tags/v') && github.event.base_ref == 'refs/heads/main' || github.sha == github.event.repository.default_branch
+    steps:
+      - uses: actions/checkout@v4
+      - name: Verify tag is on main
+        run: |
+          git fetch origin main
+          if ! git merge-base --is-ancestor "$GITHUB_SHA" origin/main; then
+            echo "ERROR: Tag is not on the main branch. Aborting release."
+            exit 1
+          fi
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+
+      - name: Install build tools
+        run: pip install build
+
+      - name: Build package
+        run: python -m build
+
+      - name: Upload artifacts
+        uses: actions/upload-artifact@v4
+        with:
+          name: dist
+          path: dist/
+
+  publish:
+    needs: build
+    runs-on: ubuntu-latest
+    environment: pypi
+    steps:
+      - name: Download artifacts
+        uses: actions/download-artifact@v4
+        with:
+          name: dist
+          path: dist/
+
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+
+  github-release:
+    needs: publish
+    runs-on: ubuntu-latest
+    permissions:
+      contents: write
+    steps:
+      - uses: actions/checkout@v4
+      - name: Create GitHub Release
+        env:
+          GH_TOKEN: ${{ github.token }}
+        run: gh release create "${{ github.ref_name }}" --generate-notes

omni_proof-0.0.1/.gitignore

@@ -0,0 +1,15 @@
+__pycache__/
+*.py[cod]
+*$py.class
+*.egg-info/
+dist/
+build/
+.env
+*.db
+.mypy_cache/
+.pytest_cache/
+.ruff_cache/
+.venv/
+venv/
+*.log
+.DS_Store

omni_proof-0.0.1/CHANGELOG.md

@@ -0,0 +1,48 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
+and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
+
+## [0.0.1] - 2026-03-15
+
+### Added
+- Initial project structure with src-layout and pyproject.toml configuration
+- Multimodal ingestion pipeline with Gemini Embedding 2 support
+  - Preprocessing for images, video, audio, PDF, and text
+  - GeminiClient with retry logic and rate limiting
+  - IngestPipeline for end-to-end asset processing
+- Vector storage layer with Pinecone integration
+  - Abstract VectorStore interface
+  - PineconeVectorStore implementation with namespaces
+- Relational storage with SQLAlchemy 2.0 async
+  - Campaign, creative, and performance models
+  - Alembic migrations support
+- Causal inference pipeline using DoWhy and EconML
+  - DAG builder for causal graph construction
+  - DML estimator (LinearDML, CausalForestDML)
+  - Refutation tests for sensitivity analysis
+  - DICE-DML for visual embedding counterfactuals
+- Orchestration layer
+  - ComplianceChain for brand guideline validation
+  - InsightSynthesizer for generating actionable recommendations
+- FastAPI-based dashboard API
+  - Campaign ingestion endpoint
+  - Brand compliance endpoints
+  - Causal analysis endpoints
+  - Generative prompt builder integration
+- Comprehensive test suite
+  - 140 unit tests and 17 integration tests with pytest and pytest-asyncio
+  - Integration test scaffolding
+- Development tooling
+  - Ruff for linting and formatting
+  - Type hints throughout codebase
+  - Structured logging with structlog
+
+### Dependencies
+- google-genai for Gemini API access
+- pinecone for vector storage
+- dowhy and econml for causal inference
+- fastapi and uvicorn for API layer
+- sqlalchemy and alembic for relational storage

omni_proof-0.0.1/CLAUDE.md

@@ -0,0 +1,81 @@
+# OmniProof — AI Assistant Context
+
+## Project
+Causal-Multimodal Engine for Creative Performance Attribution using Gemini Embedding 2.
+
+## Tech Stack
+- Python 3.11+, src-layout, pyproject.toml (hatchling)
+- Gemini Embedding 2 (gemini-embedding-2-preview) + Gemini 3.1 Flash Lite
+- Pinecone serverless (vector DB)
+- SQLAlchemy 2.0 async + Alembic (relational DB)
+- DoWhy + EconML (causal inference)
+- FastAPI (API layer)
+
+## Commands
+```bash
+pip install -e ".[dev]"                          # Install
+pytest -v --tb=short                             # Run all tests (157)
+pytest tests/unit/ -v                            # Unit tests (140)
+pytest tests/integration/ -v                     # Integration tests (17)
+ruff check src/ tests/                           # Lint
+ruff format src/ tests/                          # Format
+mypy src/omni_proof/ --ignore-missing-imports    # Type check
+python -m build                                  # Build wheel + sdist
+docker-compose up -d                             # Start API + PostgreSQL
+```
+
+## Environment Setup
+Required env vars (prefix `OMNI_PROOF_`):
+```bash
+OMNI_PROOF_GEMINI_API_KEY=AIza...              # Gemini API
+OMNI_PROOF_PINECONE_API_KEY=pcsk_...           # Pinecone
+OMNI_PROOF_PINECONE_INDEX_HOST=https://...     # Pinecone index URL
+OMNI_PROOF_DATABASE_URL=sqlite+aiosqlite:///./omni_proof.db  # Default
+```
+Causal analysis layer needs no API keys — works with local data only.
+
+## Architecture (5 Layers)
+1. **Ingestion** (`src/omni_proof/ingestion/`): AssetPreprocessor + GeminiClient + IngestPipeline
+2. **Storage** (`src/omni_proof/storage/`): PineconeVectorStore + InMemoryVectorStore + RelationalStore
+3. **Causal** (`src/omni_proof/causal/`): DAG builder + DML estimator + refuter + DICE-DML
+4. **Brand & Orchestration** (`src/omni_proof/brand_extraction/`, `src/omni_proof/orchestration/`): BrandExtractor + ComplianceChain + InsightSynthesizer
+5. **API** (`src/omni_proof/api/`): FastAPI routes + GenerativePromptBuilder
+
+## Key Patterns
+- All Gemini calls go through GeminiClient (retry + rate limiting)
+- Vector store uses abstract VectorStore interface (Pinecone + InMemory implementations)
+- Causal pipeline: DAG -> Identify -> Estimate (DML) -> Refute
+- DICE-DML for visual embeddings: counterfactual pairs -> disentangle -> estimate
+- Brand compliance: embed asset -> retrieve guidelines (RAG) -> evaluate -> report
+- BrandExtractor: extract structured brand identity from multimodal assets with conflict detection
+
+## Gotchas
+- Use `async_sessionmaker` (not `sessionmaker`) for SQLAlchemy async — mypy rejects the sync overload with AsyncEngine
+- `Counter` variables need explicit type annotations (`Counter[str]`) for mypy
+- Gemini `generate_embedding` config dict must be typed `dict[str, Any]` because it mixes `int` and `str` values
+- DAG template values from dicts are `Sequence[str]` — cast with `str()` / `list()` before passing to typed methods
+- `generate_embedding` retry loop needs an explicit `raise` after the loop for mypy return-type satisfaction
+
+## Releasing to PyPI
+Tags must be on the `main` branch — the release workflow rejects tags on other branches.
+```bash
+# 1. Update version in pyproject.toml
+# 2. Update CHANGELOG.md (change "Unreleased" to date)
+# 3. Commit, push to main
+git tag v0.1.0
+git push origin v0.1.0
+```
+This triggers `.github/workflows/release.yml` which:
+1. Verifies the tag is on `main`
+2. Builds wheel + sdist
+3. Publishes to PyPI (trusted publisher — no token needed)
+4. Creates a GitHub Release with auto-generated notes
+
+**Prerequisites:** Configure [PyPI trusted publishing](https://docs.pypi.org/trusted-publishers/adding-a-publisher/) with owner `navidgh66`, repo `omni_proof`, workflow `release.yml`, environment `pypi`. Also create a `pypi` environment in GitHub repo settings.
+
+## Gemini Embedding 2 Limits
+- Video: 80s (with audio) / 120s (without)
+- Audio: 80s max
+- Images: 6 per request
+- PDF: 1 doc, 6 pages
+- Output: 3072 dims (Matryoshka: 128/768/1536/3072)

omni_proof-0.0.1/CONTRIBUTING.md

@@ -0,0 +1,82 @@
+# Contributing to OmniProof
+
+Thank you for your interest in contributing to OmniProof!
+
+## Development Setup
+
+1. Clone the repository:
+   ```bash
+   git clone https://github.com/your-org/omni_proof.git
+   cd omni_proof
+   ```
+
+2. Install in development mode:
+   ```bash
+   pip install -e ".[dev]"
+   ```
+
+3. Set up environment variables:
+   ```bash
+   cp .env.example .env
+   # Edit .env with your API keys (GEMINI_API_KEY, PINECONE_API_KEY, etc.)
+   ```
+
+## Running Tests
+
+Run all tests:
+```bash
+pytest -v --tb=short
+```
+
+Run only unit tests:
+```bash
+pytest tests/unit/ -v
+```
+
+Run only integration tests:
+```bash
+pytest tests/integration/ -v
+```
+
+Run with coverage:
+```bash
+pytest --cov=src/omni_proof --cov-report=term-missing
+```
+
+## Code Quality
+
+Lint your code:
+```bash
+ruff check src/ tests/
+```
+
+Format your code:
+```bash
+ruff format src/ tests/
+```
+
+Type checking (optional):
+```bash
+mypy src/
+```
+
+## Pull Request Guidelines
+
+1. **Keep PRs focused**: One feature or fix per PR
+2. **Write tests**: Add unit tests for new functionality
+3. **Follow existing patterns**: Match the project's architecture and style
+4. **Update documentation**: Add docstrings and update README if needed
+5. **Run tests locally**: Ensure all tests pass before submitting
+6. **Write clear commit messages**: Use descriptive commit messages
+
+## Code Style
+
+- Python 3.11+ features are encouraged
+- Follow PEP 8 (enforced by Ruff)
+- Use type hints where possible
+- Keep line length to 100 characters
+- Write docstrings for public APIs
+
+## Questions?
+
+Open an issue for discussion before starting major changes.

omni_proof-0.0.1/Dockerfile

@@ -0,0 +1,18 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+
+COPY pyproject.toml .
+COPY src/ src/
+
+RUN pip install --no-cache-dir . \
+    && useradd --create-home appuser
+
+USER appuser
+
+EXPOSE 8000
+
+HEALTHCHECK --interval=30s --timeout=5s --retries=3 \
+    CMD python -c "import urllib.request; urllib.request.urlopen('http://localhost:8000/health')" || exit 1
+
+CMD ["uvicorn", "omni_proof.api.app:create_app", "--factory", "--host", "0.0.0.0", "--port", "8000"]

omni_proof-0.0.1/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 OmniProof Contributors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

omni_proof-0.0.1/Makefile

@@ -0,0 +1,27 @@
+.PHONY: install test lint format build clean publish-test publish
+
+install:
+	pip install -e ".[dev]"
+
+test:
+	pytest -v --tb=short
+
+lint:
+	ruff check src/ tests/
+	mypy src/
+
+format:
+	ruff format src/ tests/
+	ruff check --fix src/ tests/
+
+build:
+	pip install build && python -m build
+
+clean:
+	rm -rf dist/ build/ *.egg-info src/*.egg-info
+
+publish-test:
+	pip install twine && twine upload --repository testpypi dist/*
+
+publish:
+	pip install twine && twine upload dist/*