diff --git a/poetry.lock b/poetry.lock
index 6e2defe..b76a972 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1140,13 +1140,13 @@ omegaconf = ">=2.3.0,<3.0.0"
 
 [[package]]
 name = "mlflow"
-version = "2.9.0"
+version = "2.9.2"
 description = "MLflow: A Platform for ML Development and Productionization"
 optional = false
 python-versions = ">=3.8"
 files = [
-    {file = "mlflow-2.9.0-py3-none-any.whl", hash = "sha256:a321b020eba5aaf5714adc946c7048f75f8bb794c9f04dede93d366000517374"},
-    {file = "mlflow-2.9.0.tar.gz", hash = "sha256:937ef806f46d5ad5a0e32367954ec86eb0d301c3dccb4b3ef01d7156274f73b8"},
+    {file = "mlflow-2.9.2-py3-none-any.whl", hash = "sha256:5c1aac70238636bfc798f41925b93c6cce3a8835287311e97b7dbc90d07a8777"},
+    {file = "mlflow-2.9.2.tar.gz", hash = "sha256:25f2ad3d0b0e78bd32296058fb4c315dff4379515cd669ac6137f39507b9dfe9"},
 ]
 
 [package.dependencies]
@@ -1185,8 +1185,8 @@ waitress = {version = "<3", markers = "platform_system == \"Windows\""}
 aliyun-oss = ["aliyunstoreplugin"]
 databricks = ["azure-storage-file-datalake (>12)", "boto3 (>1)", "google-cloud-storage (>=1.30.0)"]
 extras = ["azureml-core (>=1.2.0)", "boto3", "google-cloud-storage (>=1.30.0)", "kubernetes", "mlserver (>=1.2.0,!=1.3.1)", "mlserver-mlflow (>=1.2.0,!=1.3.1)", "prometheus-flask-exporter", "pyarrow", "pysftp", "requests-auth-aws-sigv4", "virtualenv"]
-gateway = ["aiohttp (<4)", "boto3 (>=1.28.56,<2)", "fastapi (<1)", "pydantic (>=1.0,<3)", "uvicorn[standard] (<1)", "watchfiles (<1)"]
-genai = ["aiohttp (<4)", "boto3 (>=1.28.56,<2)", "fastapi (<1)", "pydantic (>=1.0,<3)", "uvicorn[standard] (<1)", "watchfiles (<1)"]
+gateway = ["aiohttp (<4)", "boto3 (>=1.28.56,<2)", "fastapi (<1)", "pydantic (>=1.0,<3)", "tiktoken (<1)", "uvicorn[standard] (<1)", "watchfiles (<1)"]
+genai = ["aiohttp (<4)", "boto3 (>=1.28.56,<2)", "fastapi (<1)", "pydantic (>=1.0,<3)", "tiktoken (<1)", "uvicorn[standard] (<1)", "watchfiles (<1)"]
 sqlserver = ["mlflow-dbstore"]
 xethub = ["mlflow-xethub"]
 
@@ -2516,4 +2516,4 @@ testing = ["big-O", "jaraco.functools", "jaraco.itertools", "more-itertools", "p
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.10"
-content-hash = "0e7dd3fc181287edeffbc915485510fc90d09143bee724b540245075c3ad76d6"
+content-hash = "36c8906f348cbfa07c1c2956e86bb0ea5e1abae90f6e1d72c334ede168deb85a"
diff --git a/pyproject.toml b/pyproject.toml
index 1a5a921..3e1246e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -12,7 +12,7 @@ torchmetrics = "^1.2.0"
 tqdm = "^4.66.1"
 loguru = "^0.7.2"
 mlconfig = "^0.2.0"
-mlflow = "^2.9.0"
+mlflow = "^2.9.2"
 
 [tool.poetry.group.dev.dependencies]
 black = "^23.11.0"