Skip to content

Commit 6d8e51d

Browse files
committed
rename methods
Signed-off-by: jonoillar <jonirvingrubio@gmail.com>
1 parent be3cc30 commit 6d8e51d

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

src/vllm_router/service_discovery.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -504,7 +504,7 @@ def remove_sleep_label(self, pod_name):
504504
except client.rest.ApiException as e:
505505
logger.error(f"Error removing sleeping label: {e}")
506506

507-
async def _get_model_names_async(self, pod_ip) -> List[str]:
507+
async def _get_model_names(self, pod_ip) -> List[str]:
508508
"""
509509
Get the model names of the serving engine pod by querying the pod's
510510
'/v1/models' endpoint.
@@ -540,7 +540,7 @@ async def _get_model_names_async(self, pod_ip) -> List[str]:
540540
logger.error(f"Failed to get model names from {url}: {e}")
541541
return []
542542

543-
async def _get_model_info_async(self, pod_ip) -> Dict[str, ModelInfo]:
543+
async def _get_model_info(self, pod_ip) -> Dict[str, ModelInfo]:
544544
"""
545545
Get detailed model information from the serving engine pod.
546546
@@ -704,7 +704,7 @@ async def _preprocess_event(self, pod, pod_ip: str) -> dict:
704704
is_pod_ready = is_container_ready and not is_pod_terminating
705705

706706
if is_pod_ready:
707-
model_names = await self._get_model_names_async(pod_ip)
707+
model_names = await self._get_model_names(pod_ip)
708708
model_label = self._get_model_label(pod)
709709
else:
710710
model_names = []
@@ -731,7 +731,7 @@ async def _add_engine_async(
731731
)
732732

733733
# Get detailed model information
734-
model_info = await self._get_model_info_async(engine_ip)
734+
model_info = await self._get_model_info(engine_ip)
735735

736736
# Check if engine is enabled with sleep mode and set engine sleep status
737737
if self._check_engine_sleep_mode(engine_name):

0 commit comments

Comments
 (0)