From f91dc9a9a3da4490d445574783f9e2d99b5ea9ad Mon Sep 17 00:00:00 2001
From: Michael Sekamanya <86433807+mawandm@users.noreply.github.com>
Date: Mon, 17 Jun 2024 19:41:10 -0700
Subject: [PATCH] fix(api)(rag): upgrade werkzeug for security (#125)

Upgrade Werkzueg for fix the security advisory.
---
 nesis/api/requirements.txt  | 2 +-
 nesis/rag/requirements.txt  | 2 +-
 nesis/rag/tests/__init__.py | 4 ----
 3 files changed, 2 insertions(+), 6 deletions(-)

diff --git a/nesis/api/requirements.txt b/nesis/api/requirements.txt
index dddf43d..9c50d44 100644
--- a/nesis/api/requirements.txt
+++ b/nesis/api/requirements.txt
@@ -21,7 +21,7 @@ smbprotocol==1.12.0
 flask==3.0.0
 boto3==1.34.75
 
-Werkzeug~=3.0.1
+Werkzeug>=3.0.3
 dropbox~=11.36.2
 
 # Document datasources
diff --git a/nesis/rag/requirements.txt b/nesis/rag/requirements.txt
index 2e1a89d..b5599c1 100644
--- a/nesis/rag/requirements.txt
+++ b/nesis/rag/requirements.txt
@@ -13,7 +13,7 @@ gevent==23.9.1
 PyPDF2==3.0.1
 
 chromadb==0.4.24
-Werkzeug==3.0.1
+Werkzeug>=3.0.3
 pandas==2.2.1
 
 injector==0.21.0
diff --git a/nesis/rag/tests/__init__.py b/nesis/rag/tests/__init__.py
index dc93b53..3904514 100644
--- a/nesis/rag/tests/__init__.py
+++ b/nesis/rag/tests/__init__.py
@@ -4,9 +4,7 @@
 import os
 
 from nesis.api.core.models.entities import (
-    Rule,
     Prediction,
-    Model,
     Setting,
     Role,
     User,
@@ -78,8 +76,6 @@ def get_header(token=None):
 
 
 def clear_database(session):
-    session.query(Rule).delete()
-    session.query(Model).delete()
     session.query(Prediction).delete()
     session.query(Setting).delete()
     session.query(User).delete()