Add metadata

volaradlp · Oct 7, 2024 · 7937e43 · 7937e43
1 parent 3715ab4
commit 7937e43
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/volara_proof/models/proof_response.py b/volara_proof/models/proof_response.py
@@ -12,3 +12,4 @@ class ProofResponse(BaseModel):
     quality: float = 0.0
     uniqueness: float = 0.0
     attributes: Optional[Dict[str, Any]] = {}
+    metadata: Optional[Dict[str, Any]] = {}
diff --git a/volara_proof/proof.py b/volara_proof/proof.py
@@ -11,7 +11,9 @@
 class Proof:
     def __init__(self, config: Dict[str, Any]):
         self.config = ProofConfig(**config)
-        self.proof_response = ProofResponse(dlp_id=config["dlp_id"])
+        self.proof_response = ProofResponse(
+            dlp_id=config["dlp_id"], metadata={"dlp_id": config["dlp_id"]}
+        )
 
     def generate(self) -> ProofResponse:
         """Generate proofs for all input files."""