zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-15 20:27:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor: reformat all code for lefthook using ruff and gofmt (#16585)

Browse Source
This commit is contained in:
Wang Qi
2026-07-03 12:53:39 +08:00
committed by GitHub
parent 19fcb4a981
commit 6a4b9be426
588 changed files with 11123 additions and 15412 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
sdk/python/test/conftest.py
View File

@@ -113,12 +113,7 @@ def add_model_instance(auth):
pytest.exit(f"Critical error in add model provider: {add_provider_res.get('message')}")
add_instance_api = HOST_ADDRESS + "/api/v1/providers/ZHIPU-AI/instances"
add_instance_response = requests.post(url=add_instance_api, headers=authorization, json={
"instance_name": "CI",
"api_key": ZHIPU_AI_API_KEY,
"region": "default",
"base_url": ""
})
add_instance_response = requests.post(url=add_instance_api, headers=authorization, json={"instance_name": "CI", "api_key": ZHIPU_AI_API_KEY, "region": "default", "base_url": ""})
add_instance_res = add_instance_response.json()
if add_instance_res.get("code") != 0:
pytest.exit(f"Critical error in add model instance: {add_instance_res.get('message')}")
@@ -139,41 +134,19 @@ def set_tenant_info(get_auth):
url = HOST_ADDRESS + "/api/v1/models/default"
authorization = {"Authorization": get_auth}
# set chat model
set_default_llm_response = requests.patch(
url=url,
headers=authorization,
json={
"model_provider": "ZHIPU-AI",
"model_instance": "CI",
"model_type": "chat",
"model_name": "glm-4-flash"
})
set_default_llm_response = requests.patch(url=url, headers=authorization, json={"model_provider": "ZHIPU-AI", "model_instance": "CI", "model_type": "chat", "model_name": "glm-4-flash"})
llm_res = set_default_llm_response.json()
if llm_res.get("code") != 0:
raise Exception(llm_res.get("message"))
# set embedding model
set_default_embedding_response = requests.patch(
url=url,
headers=authorization,
json={
"model_provider": "Builtin",
"model_instance": "Local",
"model_type": "embedding",
"model_name": "BAAI/bge-small-en-v1.5"
})
url=url, headers=authorization, json={"model_provider": "Builtin", "model_instance": "Local", "model_type": "embedding", "model_name": "BAAI/bge-small-en-v1.5"}
)
embd_res = set_default_embedding_response.json()
if embd_res.get("code") != 0:
raise Exception(embd_res.get("message"))
# set image to text model
set_default_img2txt_response = requests.patch(
url=url,
headers=authorization,
json={
"model_provider": "ZHIPU-AI",
"model_instance": "CI",
"model_type": "vision",
"model_name": "glm-4v"
})
set_default_img2txt_response = requests.patch(url=url, headers=authorization, json={"model_provider": "ZHIPU-AI", "model_instance": "CI", "model_type": "vision", "model_name": "glm-4v"})
img2txt_res = set_default_img2txt_response.json()
if img2txt_res.get("code") != 0:
raise Exception(img2txt_res.get("message"))