add env variables

app.py

@@ -1,4 +1,5 @@
 import json
+import os
 import invariant.testing.functional as F
 from invariant.testing import TraceFactory, assert_true
 from agent import SantaAgent
@@ -17,14 +18,21 @@ def run_agent(user_prompt, history, invariant_api_key):
 
     agent_params = {"system_prompt": user_prompt}
 
+
+    env={
+        "INVARIANT_API_KEY": invariant_api_key,
+        "OPENAI_API_KEY": os.environ["OPENAI_API_KEY"],
+        "PATH": os.environ["PATH"]
+    }
+    # env = None
+
     # run command invariant test test_agent.py --agent-params '{"system_prompt": "you are santa"}'
     import subprocess
     out = subprocess.run([
-        "INVARIANT_API_KEY=" + invariant_api_key,
         "invariant", "test", "test_agent.py", 
         "--agent-params", json.dumps(agent_params),
         "--push", "--dataset_name", "santa_agent",
-    ], capture_output=True, text=True)
+    ], capture_output=True, text=True, env=env)
     print(out.stdout)
     print(out.stderr)