Ruff check and format

Author: XkunW

vec_inf/cli/_cli.py

@@ -1,7 +1,7 @@
 """Command line interface for Vector Inference."""
 
 import time
-from typing import Optional, Union, cast
+from typing import Optional, Union
 
 import click
 from rich.console import Console
@@ -184,8 +184,10 @@ def metrics(slurm_job_id: int, log_dir: Optional[str] = None) -> None:
 
     # Check if metrics URL is ready
     if not helper.metrics_url.startswith("http"):
-        table = utils.create_table("Metric", "Value")    
-        helper.display_failed_metrics(table, f"Metrics endpoint unavailable - {helper.metrics_url}")
+        table = utils.create_table("Metric", "Value")
+        helper.display_failed_metrics(
+            table, f"Metrics endpoint unavailable - {helper.metrics_url}"
+        )
         CONSOLE.print(table)
         return
 

vec_inf/cli/_helper.py

@@ -340,7 +340,7 @@ def _get_status_info(self) -> dict[str, Union[str, None]]:
         status_helper = StatusHelper(self.slurm_job_id, output, self.log_dir)
         return status_helper.status_info
 
-    def _build_metrics_url(self) -> Optional[str]:
+    def _build_metrics_url(self) -> str:
         """Construct metrics endpoint URL from base URL with version stripping."""
         if self.status_info.get("state") == "PENDING":
             return "Pending resources for server initialization"
@@ -467,9 +467,7 @@ def _parse_metrics(self, metrics_text: str) -> dict[str, float]:
         return parsed
 
     def display_failed_metrics(self, table: Table, metrics: str) -> None:
-        table.add_row(
-            "Server State", self.status_info["state"], style="yellow"
-        )
+        table.add_row("Server State", self.status_info["state"], style="yellow")
         table.add_row("Message", metrics)
 
     def display_metrics(self, table: Table, metrics: dict[str, float]) -> None:
@@ -528,6 +526,7 @@ def display_metrics(self, table: Table, metrics: dict[str, float]) -> None:
             f"{metrics.get('successful_requests_total', 0):.0f} reqs",
         )
 
+
 class ListHelper:
     """Helper class for handling model listing functionality."""