diff --git a/pyproject.toml b/pyproject.toml
index 16d01d66..a304d064 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -76,7 +76,7 @@ scikit-image = "*"
 pinecone-client = "*"
 roboflow = "*"
 python-dotenv = "*"
-
+pyyaml = "*"
 
 
 [tool.poetry.group.lint.dependencies]
diff --git a/requirements.txt b/requirements.txt
index d674eecf..ef1bd15f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -53,4 +53,5 @@ anthropic
 pinecone-client
 roboflow
 rich==13.7.0
-python-dotenv
\ No newline at end of file
+python-dotenv
+pyyaml
\ No newline at end of file