diff --git a/pyproject.toml b/pyproject.toml
index 9d4ca4b0..ea378e6e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -53,6 +53,7 @@ docstring_parser = "0.16"
 fastapi = "*"
 openai = ">=1.30.1,<2.0"
 termcolor = "*"
+chromadb = "*"
 
 
 [tool.poetry.group.lint.dependencies]
diff --git a/requirements.txt b/requirements.txt
index 09a4fc1d..97ab8704 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -27,4 +27,5 @@ mypy-protobuf>=3.0.0
 pytest>=8.1.1
 termcolor>=2.4.0
 pandas>=2.2.2
-fastapi>=0.110.1
\ No newline at end of file
+fastapi>=0.110.1
+chromadb
\ No newline at end of file