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)