Procházet zdrojové kódy

Update abtest client: better logs

StrayWarrior před 4 dny
rodič
revize
82cc06d6c1
1 změnil soubory, kde provedl 8 přidání a 7 odebrání
  1. 8 7
      pqai_agent/abtest/client.py

+ 8 - 7
pqai_agent/abtest/client.py

@@ -50,7 +50,7 @@ class ExperimentClient:
 
         for project_data in projects:
             project = Project(name=project_data.name, project_id=project_data.project_id)
-            logger.debug(f"[Project] {project_data}")
+            # logger.debug(f"[Project] {project_data}")
 
             # 获取项目的域
             list_domain_req = ListDomainsRequest()
@@ -66,7 +66,7 @@ class ExperimentClient:
                                 is_default_domain=domain_data.is_default_domain,
                                 exp_layer_id=domain_data.layer_id,
                                 debug_users=domain_data.debug_users)
-                logger.debug(f"[Domain] {domain_data}")
+                # logger.debug(f"[Domain] {domain_data}")
                 if domain.is_default_domain:
                     project.set_default_domain(domain)
                 domain.init()
@@ -84,7 +84,7 @@ class ExperimentClient:
                 list_layer_req.domain_id = str(domain.id)
                 layers_response = self.client.list_layers(list_layer_req)
                 for layer_data in layers_response.body.layers:
-                    logger.debug(f'[Layer] {layer_data}')
+                    # logger.debug(f'[Layer] {layer_data}')
                     layer = Layer(id=int(layer_data.layer_id), name=layer_data.name)
                     project.add_layer(layer)
 
@@ -96,7 +96,7 @@ class ExperimentClient:
                     experiments_response = self.client.list_experiments(list_experiment_req)
 
                     for experiment_data in experiments_response.body.experiments:
-                        logger.debug(f'[Experiment] {experiment_data}')
+                        # logger.debug(f'[Experiment] {experiment_data}')
                         # FIXME: Java SDK中有特殊处理
                         crowd_ids = experiment_data.crowd_ids if experiment_data.crowd_ids else ""
                         experiment = Experiment(id=int(experiment_data.experiment_id), bucket_type=experiment_data.bucket_type,
@@ -111,9 +111,8 @@ class ExperimentClient:
                         list_exp_ver_req = ListExperimentVersionsRequest()
                         list_exp_ver_req.experiment_id = int(experiment.id)
                         versions_response = self.client.list_experiment_versions(list_exp_ver_req)
-                        logger.debug(versions_response)
                         for version_data in versions_response.body.experiment_versions:
-                            logger.debug(f'[ExperimentVersion] {version_data}')
+                            # logger.debug(f'[ExperimentVersion] {version_data}')
                             version = ExperimentVersion(exp_version_id=version_data.experiment_version_id,
                                                         exp_id=experiment.id,
                                                         flow=int(version_data.flow),
@@ -150,6 +149,8 @@ class ExperimentClient:
         experiment_result = ExperimentResult(project=project, experiment_context=experiment_context)
 
         self._match_domain(project.default_domain, experiment_result)
+        matched_versions = [str(ver.id) for ver in experiment_result.experiment_versions]
+        logger.debug(f"Matched experiment, uid[{experiment_context.uid}], versions[{','.join(matched_versions)}], params: {experiment_result.params}")
         experiment_result.init()
         return experiment_result
 
@@ -176,7 +177,7 @@ class ExperimentClient:
 
         for domain in layer.domains:
             if domain.match_debug_users(experiment_result.experiment_context):
-                logger.debug(f"Matched debug user for domain: {domain.id}")
+                # logger.debug(f"Matched debug user for domain: {domain.id}")
                 self._match_domain(domain, experiment_result)
 
         hash_key = f"{experiment_result.experiment_context.uid}_LAYER{layer.id}"