File tree 2 files changed +10
-6
lines changed
llama-index-packs/llama-index-packs-zenguard
2 files changed +10
-6
lines changed Original file line number Diff line number Diff line change @@ -44,7 +44,7 @@ version = "0.3.0"
44
44
45
45
[tool .poetry .dependencies ]
46
46
python = " >=3.9,<4.0"
47
- zenguard = " ^0.1.13 "
47
+ zenguard = " ^0.2.1 "
48
48
llama-index-core = " ^0.12.0"
49
49
50
50
[tool .poetry .group .dev .dependencies ]
Original file line number Diff line number Diff line change 1
- import pytest
2
-
1
+ import os
3
2
from typing import Dict
4
3
4
+ import pytest
5
5
from llama_index .core .llama_pack import BaseLlamaPack
6
6
from llama_index .packs .zenguard import (
7
- ZenGuardPack ,
8
- ZenGuardConfig ,
9
7
Credentials ,
10
8
Detector ,
9
+ ZenGuardConfig ,
10
+ ZenGuardPack ,
11
+ )
12
+
13
+ pytestmark = pytest .mark .skipif (
14
+ os .environ .get ("ZEN_API_KEY" ) is None , reason = "ZEN_API_KEY not set"
11
15
)
12
16
13
17
14
18
@pytest .fixture ()
15
19
def zenguard_pack ():
16
- api_key = "3Ev_DGvELv7EnlgWMTlpmWTo82tpstyz4Li_R7kTDQw" # mock key. whitelisted only for LlamaIndex tests.
20
+ api_key = os . environ . get ( "ZEN_API_KEY" )
17
21
config = ZenGuardConfig (credentials = Credentials (api_key = api_key ))
18
22
return ZenGuardPack (config )
19
23
You can’t perform that action at this time.
0 commit comments