diff --git a/requirements.txt b/requirements.txt
index 73d7f7bf74..463484f881 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,5 @@
 openai
 cohere
 func_timeout
-anthropic
\ No newline at end of file
+anthropic
+pytest
\ No newline at end of file
diff --git a/setup.py b/setup.py
index f4de6c1816..aed2c27472 100644
--- a/setup.py
+++ b/setup.py
@@ -12,5 +12,6 @@ setup(
         'openai',
         'cohere',
         'func_timeout'
+        'pytest'
     ],
 )