feat: add quality, ownership, and authenticity fields into proof result

SocialTensor · Nov 22, 2024 · 461ebfd · 461ebfd
1 parent 0681bc8
commit 461ebfd
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/my_proof/models/proof_response.py b/my_proof/models/proof_response.py
@@ -32,3 +32,6 @@ class ProofResponse(BaseModel):
     model_distribution: float = 0.0
     poison_data: float = 0.0
     uniqueness: float = 0.0
+    quality: float = 0.0
+    ownership: float = 1.0
+    authenticity: float = 1.0
diff --git a/my_proof/proof.py b/my_proof/proof.py
@@ -58,6 +58,9 @@ def generate(self) -> ProofResponse:
         self.proof_response.both_sides = qualityRes['Both_Sides']['score']
         self.proof_response.model_distribution = qualityRes['Model_Distribution']['score']
         self.proof_response.poison_data = qualityRes['Poisin_Data']['score']
+        self.proof_response.quality = qualityRes['score']
+        self.proof_response.ownership = 1.0
+        self.proof_response.authenticity = 1.0
 
         self.proof_response.uniqueness = Uniqueness(input_data, self.aws_access_key_id, self.aws_secret_access_key)