diff --git a/nix/impure-shell.nix b/nix/impure-shell.nix
index 9b8d7e35e..57a8bf71f 100644
--- a/nix/impure-shell.nix
+++ b/nix/impure-shell.nix
@@ -60,6 +60,7 @@ mkShell {
       pip
       ipdb
       click
+      openai
       pytest
       pytest-asyncio
       syrupy