소스 검색

Update abtest models: warn duplicated params

StrayWarrior 5 일 전
부모
커밋
fe8ba7aaf7
2개의 변경된 파일7개의 추가작업 그리고 1개의 파일을 삭제
  1. 7 1
      pqai_agent/abtest/models.py
  2. 0 0
      pqai_agent/abtest/utils.py

+ 7 - 1
pqai_agent/abtest/models.py

@@ -3,6 +3,9 @@ import json
 from dataclasses import dataclass, field
 import hashlib
 
+from pqai_agent.logging_service import logger
+
+
 class FNV:
     INIT64 = int("cbf29ce484222325", 16)
     PRIME64 = int("100000001b3", 16)
@@ -232,7 +235,10 @@ class ExperimentResult:
         self.exp_id = ""
 
     def add_params(self, params: Dict[str, str]):
-        self.params.update(params)
+        for key, value in params.items():
+            if key in self.params:
+                logger.warning(f"Duplicate key '{key}' in params, overwriting value: {self.params[key]} with {value}")
+            self.params[key] = value
 
     def add_experiment_version(self, version):
         self.experiment_versions.append(version)

+ 0 - 0
pqai_agent/abtest/utils.py