diff --git a/ee/recommendation/core/recommendation_model.py b/ee/recommendation/core/recommendation_model.py
index 823fd5093..2e7805222 100644
--- a/ee/recommendation/core/recommendation_model.py
+++ b/ee/recommendation/core/recommendation_model.py
@@ -57,6 +57,17 @@ def preprocess(X):
     return x, transform
 
 
+class RecommendationSystem(mlflow.pyfunc.PythonModel):
+    def __init__(self):
+        ...
+
+    def fit(self, X, y):
+        ...
+
+    def predict(self, X):
+        return None
+
+
 class SVM_recommendation(mlflow.pyfunc.PythonModel):
 
     def __init__(self, test=False, **params):
diff --git a/ee/recommendation/ml_service/requirements.txt b/ee/recommendation/ml_service/requirements.txt
index 0eb819bf7..71d0aeb4a 100644
--- a/ee/recommendation/ml_service/requirements.txt
+++ b/ee/recommendation/ml_service/requirements.txt
@@ -1,4 +1,3 @@
-fastapi==0.95.2
-apscheduler==3.10.1
-uvicorn==0.22.0
-SQLAlchemy==2.0.15
+fastapi==0.104.0
+apscheduler==3.10.4
+uvicorn==0.23.2
diff --git a/ee/recommendation/ml_trainer/entrypoint.sh b/ee/recommendation/ml_trainer/entrypoint.sh
index af93b034b..cfc96e85e 100755
--- a/ee/recommendation/ml_trainer/entrypoint.sh
+++ b/ee/recommendation/ml_trainer/entrypoint.sh
@@ -7,6 +7,7 @@ find airflow/ -type f -name "*.cfg" -exec sed -i "s/{{pg_dbname_airflow}}/${pg_d
 find airflow/ -type f -name "*.cfg" -exec sed -i "s#{{airflow_secret_key}}#${airflow_secret_key}#g" {} \;
 export MLFLOW_TRACKING_URI=postgresql+psycopg2://${pg_user_ml}:${pg_password_ml}@${pg_host_ml}:${pg_port_ml}/${pg_dbname_ml}
 git init airflow/dags
+airflow db upgrade
 # Airflow setup
 # airflow db init
 # airflow users create \
diff --git a/ee/recommendation/ml_trainer/requirements.txt b/ee/recommendation/ml_trainer/requirements.txt
index 7efa93d01..0a2b6cd6a 100644
--- a/ee/recommendation/ml_trainer/requirements.txt
+++ b/ee/recommendation/ml_trainer/requirements.txt
@@ -1,3 +1,3 @@
-argcomplete==3.0.8
-apache-airflow==2.6.2
-airflow-code-editor==7.2.1
+argcomplete==3.1.2
+apache-airflow==2.7.2
+airflow-code-editor==7.4.1
diff --git a/ee/recommendation/requirements_base.txt b/ee/recommendation/requirements_base.txt
index e378cc7f0..4e0150aa1 100644
--- a/ee/recommendation/requirements_base.txt
+++ b/ee/recommendation/requirements_base.txt
@@ -1,19 +1,19 @@
 requests==2.31.0
-urllib3==1.26.16
+urllib3==2.0.7
 pyjwt==2.8.0
-SQLAlchemy==2.0.20
-alembic==1.11.1
-psycopg2-binary==2.9.7
+SQLAlchemy==2.0.22
+alembic==1.12.0
+psycopg2-binary==2.9.9
 
 joblib==1.3.2
-scipy==1.11.2
-scikit-learn==1.3.0
-mlflow==2.5
+scipy==1.11.3
+scikit-learn==1.3.1
+mlflow==2.7.1
 
 clickhouse-driver==0.2.6
-python3-saml==1.15.0
+python3-saml==1.16.0
 python-multipart==0.0.6
 python-decouple==3.8
-pydantic==1.10.12
+pydantic==2.4.2
 
-boto3==1.28.29
+boto3==1.28.67