diff --git a/requirements.txt b/requirements.txt index 254256f6..c6deefab 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,6 +14,7 @@ duckduckgo-search faiss-cpu wget==3.2 +codeinterpreterapi accelerate addict albumentations diff --git a/setup.py b/setup.py index 97a3a7f1..fe59aa78 100644 --- a/setup.py +++ b/setup.py @@ -29,6 +29,7 @@ setup( 'bs4', 'pegasusx', 'oceandb', + 'codeinterpreterapi[all]', 'playwright', 'duckduckgo_search', 'faiss-cpu',