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

Fix: restrict SiliconFlow models in CI to 4 free models (#18225)

Browse Source 
### Summary

As title.
This commit is contained in:
Lynn
2026-08-13 18:47:37 +08:00
committed by GitHub
parent c7aca4c97e
commit 34d38b4f80
1 changed files with 24 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
test/testcases/conftest.py
View File

@@ -238,7 +238,30 @@ def add_model_instance(auth):
# and BAAI/bge-reranker-v2-m3@CI@SILICONFLOW).
instance_name = "CI"
add_instance_api = HOST_ADDRESS + f"/api/v1/providers/{provider_name}/instances"
add_instance_response = requests.post(url=add_instance_api, headers=authorization, json={"instance_name": instance_name, "api_key": api_key, "region": "default", "base_url": ""})
# Bind and verify only the free models the suite actually uses.
# Without model_info the server binds/verifies the entire factory
# catalog for the provider, including paid chat models.
if provider_name == "SILICONFLOW":
instance_payload = {
"instance_name": instance_name,
"api_key": api_key,
"region": "default",
"base_url": "",
"model_info": [
{"model_type": ["rerank"], "model_name": "BAAI/bge-reranker-v2-m3", "max_tokens": 8192},
{"model_type": ["embedding"], "model_name": "BAAI/bge-m3", "max_tokens": 8192},
{"model_type": ["embedding"], "model_name": "BAAI/bge-large-en-v1.5", "max_tokens": 512},
{"model_type": ["embedding"], "model_name": "BAAI/bge-large-zh-v1.5", "max_tokens": 512},
],
}
else:
instance_payload = {
"instance_name": instance_name,
"api_key": api_key,
"region": "default",
"base_url": "",
}
add_instance_response = requests.post(url=add_instance_api, headers=authorization, json=instance_payload)
add_instance_res = add_instance_response.json()
if add_instance_res.get("code") != 0:
msg = add_instance_res.get("message", "")