瀏覽代碼

Update abtest models: warn duplicated params

StrayWarrior 4 天之前
父節點
當前提交
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