Revert "tiny remove deprecated endpoint call" (#14533)

This commit is contained in:
Baizhou Zhang
2025-12-05 23:48:54 -08:00
committed by GitHub
parent d257bf87b9
commit 42fcf5438f
36 changed files with 59 additions and 59 deletions
@@ -79,7 +79,7 @@ class TestDeepseekV32CP(CustomTestCase):
metrics = run_eval_few_shot_gsm8k(args)
print(f"{metrics=}")
server_info = requests.get(self.base_url + "/server_info")
server_info = requests.get(self.base_url + "/get_server_info")
avg_spec_accept_length = server_info.json()["internal_states"][0][
"avg_spec_accept_length"
]
@@ -77,7 +77,7 @@ class TestEagleDPAttnServerBase(CustomTestCase):
metrics = run_eval_few_shot_gsm8k(args)
print(f"{metrics=}")
server_info = requests.get(self.base_url + "/server_info")
server_info = requests.get(self.base_url + "/get_server_info")
avg_spec_accept_length = server_info.json()["internal_states"][0][
"avg_spec_accept_length"
]
+10 -10
View File
@@ -3,7 +3,7 @@ Test weight version functionality.
This test suite verifies the weight_version feature implementation including:
1. Default weight_version setting
2. /weight_version endpoint
2. /get_weight_version endpoint
3. /update_weight_version endpoint
4. /generate request meta_info contains weight_version
5. OpenAI API response metadata contains weight_version
@@ -48,13 +48,13 @@ class TestWeightVersion(CustomTestCase):
def test_weight_version_comprehensive(self):
"""Comprehensive test for all weight_version functionality."""
response = requests.get(f"{self.base_url}/model_info")
response = requests.get(f"{self.base_url}/get_model_info")
self.assertEqual(response.status_code, 200)
data = response.json()
self.assertIn("weight_version", data)
self.assertEqual(data["weight_version"], "test_version_1.0")
response = requests.get(f"{self.base_url}/weight_version")
response = requests.get(f"{self.base_url}/get_weight_version")
self.assertEqual(response.status_code, 200)
data = response.json()
self.assertIn("weight_version", data)
@@ -114,7 +114,7 @@ class TestWeightVersion(CustomTestCase):
self.assertTrue(data["success"])
self.assertEqual(data["new_version"], "updated_version_2.0")
response = requests.get(f"{self.base_url}/weight_version")
response = requests.get(f"{self.base_url}/get_weight_version")
self.assertEqual(response.status_code, 200)
data = response.json()
self.assertEqual(data["weight_version"], "updated_version_2.0")
@@ -148,13 +148,13 @@ class TestWeightVersion(CustomTestCase):
self.assertTrue(data["success"])
self.assertEqual(data["new_version"], "final_version_3.0")
# Check /weight_version
response = requests.get(f"{self.base_url}/weight_version")
# Check /get_weight_version
response = requests.get(f"{self.base_url}/get_weight_version")
self.assertEqual(response.status_code, 200)
self.assertEqual(response.json()["weight_version"], "final_version_3.0")
# Check /model_info
response = requests.get(f"{self.base_url}/model_info")
# Check /get_model_info
response = requests.get(f"{self.base_url}/get_model_info")
self.assertEqual(response.status_code, 200)
self.assertEqual(response.json()["weight_version"], "final_version_3.0")
@@ -193,7 +193,7 @@ class TestWeightVersion(CustomTestCase):
print("Testing weight_version update with real weight operations...")
# Get current model info for reference
model_info_response = requests.get(f"{self.base_url}/model_info")
model_info_response = requests.get(f"{self.base_url}/get_model_info")
self.assertEqual(model_info_response.status_code, 200)
current_model_path = model_info_response.json()["model_path"]
@@ -214,7 +214,7 @@ class TestWeightVersion(CustomTestCase):
)
# Verify version was updated
version_response = requests.get(f"{self.base_url}/weight_version")
version_response = requests.get(f"{self.base_url}/get_weight_version")
self.assertEqual(version_response.status_code, 200)
self.assertEqual(
version_response.json()["weight_version"], "disk_update_v2.0.0"