fix(api)(rag): upgrade werkzeug for security (#125)

Upgrade Werkzueg for fix the security advisory.
ametnes · Jun 18, 2024 · f91dc9a · f91dc9a
1 parent 4a46ed7
commit f91dc9a
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 6 deletions.
diff --git 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
@@ -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
@@ -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()