Skip to content

Commit 3c2b9f5

Browse files
[pentest] add extra print
1 parent ac7ff05 commit 3c2b9f5

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/pentest.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,14 @@
1313
api_key = os.environ.get("ZEN_API_KEY")
1414
if not api_key:
1515
raise ValueError("ZEN_API_KEY is not set")
16+
1617
openai_api_key = os.environ.get("OPENAI_API_KEY")
1718
if not openai_api_key:
1819
print("Pentesting ZenGuard only (OpenAI API key not set)")
1920

21+
if api_key and openai_api_key:
22+
print("Pentesting ZenGuard and OpenAI")
23+
2024
config = ZenGuardConfig(credentials=Credentials(api_key=api_key))
2125
if openai_api_key:
2226
config.credentials.llm_api_key = openai_api_key

0 commit comments

Comments
 (0)