浏览代码

Add abtest/utils

StrayWarrior 4 天之前
父节点
当前提交
35e04ca3b8
共有 1 个文件被更改,包括 7 次插入0 次删除
  1. 7 0
      pqai_agent/abtest/utils.py

+ 7 - 0
pqai_agent/abtest/utils.py

@@ -0,0 +1,7 @@
+from pqai_agent.abtest.models import ExperimentContext
+from pqai_agent.abtest.client import get_client
+
+def get_abtest_info(uid: str):
+    client = get_client()
+    exp_ctx = ExperimentContext(uid=uid)
+    return client.match_experiment('PQAgent', exp_ctx)