DS_ID = "<dataset_id>"

ds_retrieval = {

    "search_method": "semantic_search",

    "reranking_enable": True,

    "reranking_mode": "reranking_model",

    "reranking_model": {

        "reranking_provider_name": "langgenius/xinference/xinference",

        "reranking_model_name": "bge-reranker-v2-m3",

    },

    "top_k": 5,

    "score_threshold_enabled": False,

    "score_threshold": None,

}

r = d.req("PATCH", "/datasets/%s" % DS_ID, body={"retrieval_model": ds_retrieval})

print(r["code"], r["json"]["retrieval_model_dict"])