diff --git a/model_server/pyproject.toml b/model_server/pyproject.toml
index aef566a0..376c84fc 100644
--- a/model_server/pyproject.toml
+++ b/model_server/pyproject.toml
@@ -35,6 +35,7 @@ opentelemetry-sdk = "^1.28.0"
 opentelemetry-exporter-otlp = "^1.28.0"
 opentelemetry-instrumentation-fastapi = "^0.49b0"
 overrides = "^7.7.0"
+pytest-retry = "^1.6.3"
 
 [tool.poetry.scripts]
 archgw_modelserver = "src.cli:run_server"