#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""专题十三 · 步骤 6 · 修改前：只读知识库 retrieval_model（不要 PATCH）。"""
from __future__ import annotations

import json
import sys

import _bootstrap  # noqa: F401
import _config as C
from dify_api import Dify, pp


def main() -> int:
    d = Dify(C.DIFY_EMAIL, C.DIFY_PASSWORD)
    try:
        d.login()
        r = d.get("/datasets/%s" % C.DS_ID)
        pp("dataset before patch", r)
        retr = (r.get("json") or {}).get("retrieval_model_dict") or {}
        print("\n期望修改前关键字段：")
        print(json.dumps({
            "reranking_enable": retr.get("reranking_enable"),
            "top_k": retr.get("top_k"),
            "reranking_mode": retr.get("reranking_mode"),
            "reranking_model": retr.get("reranking_model"),
        }, ensure_ascii=False, indent=2))
        return 0
    finally:
        d.close()


if __name__ == "__main__":
    sys.exit(main())