src/pqc_content_provenance/assertions/__init__.py
688 B · 23 lines · python Raw
1 """Assertion registry -- maps label -> Assertion subclass."""
2
3 from __future__ import annotations
4
5 from pqc_content_provenance.assertions.ai_generated import AIGeneratedAssertion
6 from pqc_content_provenance.assertions.base import Assertion
7 from pqc_content_provenance.assertions.training import TrainingAssertion
8 from pqc_content_provenance.assertions.usage import UsageAssertion
9
10 ASSERTION_REGISTRY: dict[str, type[Assertion]] = {
11 "c2pa.ai_generated": AIGeneratedAssertion,
12 "c2pa.training": TrainingAssertion,
13 "c2pa.usage": UsageAssertion,
14 }
15
16 __all__ = [
17 "Assertion",
18 "AIGeneratedAssertion",
19 "TrainingAssertion",
20 "UsageAssertion",
21 "ASSERTION_REGISTRY",
22 ]
23