diff --git a/tests/manual_test_batches.py b/tests/manual_test_batches.py index 633607a6..c306ca0a 100644 --- a/tests/manual_test_batches.py +++ b/tests/manual_test_batches.py @@ -2,7 +2,8 @@ from portkey_ai import Portkey portkey = Portkey( - api_key="Xgv7++HhSeF70UzdVSsmtJuGYe0=", virtual_key="portkey-welcome-e248d3" + api_key=os.environ.get("PORTKEY_API_KEY"), + virtual_key=os.environ.get("PORTKEY_VIRTUAL_KEY"), )