chore: add validation for limit value

Signed-off-by: TomeHirata <tomu.hirata@gmail.com>

mlflow/gateway/config.py

@@ -8,6 +8,7 @@
 
 import pydantic
 import yaml
+from limits import parse
 from packaging import version
 from packaging.version import Version
 from pydantic import ConfigDict, Field, ValidationError, root_validator, validator
@@ -398,6 +399,20 @@ def validate_route_type(cls, value):
             return value
         raise MlflowException.invalid_parameter_value(f"The route_type '{value}' is not supported.")
 
+    @validator("limit", pre=True)
+    def validate_limit(cls, value: Optional[Limit]):
+        if value:
+            try:
+                parse(f"{value.calls}/{value.renewal_period}")
+            except ValueError:
+                raise MlflowException.invalid_parameter_value(
+                    "Failed to parse the rate limit configuration."
+                    "Please make sure limit.calls is a positive number and"
+                    "limit.renewal_period is a right granularity"
+                )
+
+        return value
+
     def to_route(self) -> "Route":
         return Route(
             name=self.name,

tests/gateway/tools.py

@@ -174,7 +174,7 @@ class UvicornGateway:
     # this module which executes the uvicorn server through gunicorn as a process manager.
     def __init__(self, config_path: Union[str, Path], *args, **kwargs):
         self.port = get_safe_port()
-        self.host = "localhost"
+        self.host = "127.0.0.1"
         self.url = f"http://{self.host}:{self.port}"
         self.config_path = config_path
         self.server = None