diff --git a/environment.yml b/environment.yml
index 6f8ed31a..e56ebe26 100644
--- a/environment.yml
+++ b/environment.yml
@@ -15,3 +15,4 @@ dependencies:
   - torchvision=0.*
   - openai=0.*
   - pydantic=1.*
+  - jinja2=3.0.*