6
6
import yaml
7
7
8
8
if TYPE_CHECKING :
9
- from yaml import Dumper , Loader
9
+ from yaml import Dumper
10
10
else :
11
11
try :
12
- from yaml import CDumper as Dumper , CLoader as Loader
13
- except ImportError :
14
- from yaml import Dumper , Loader
12
+ from yaml import CDumper as Dumper
13
+ except ImportError : # pragma: no cover
14
+ from yaml import Dumper
15
15
16
16
FILTERED_HEADER_PREFIXES = ['anthropic-' , 'cf-' , 'x-' ]
17
17
FILTERED_HEADERS = {'authorization' , 'date' , 'request-id' , 'server' , 'user-agent' , 'via' , 'set-cookie' , 'api-key' }
@@ -35,7 +35,7 @@ def str_presenter(dumper: Dumper, data: str):
35
35
36
36
37
37
def deserialize (cassette_string : str ):
38
- cassette_dict = yaml .load (cassette_string , Loader = Loader )
38
+ cassette_dict = yaml .safe_load (cassette_string )
39
39
for interaction in cassette_dict ['interactions' ]:
40
40
for kind , data in interaction .items ():
41
41
parsed_body = data .pop ('parsed_body' , None )
@@ -45,7 +45,7 @@ def deserialize(cassette_string: str):
45
45
return cassette_dict
46
46
47
47
48
- def serialize (cassette_dict : Any ):
48
+ def serialize (cassette_dict : Any ): # pragma: lax no cover
49
49
for interaction in cassette_dict ['interactions' ]:
50
50
for _kind , data in interaction .items ():
51
51
headers : dict [str , list [str ]] = data .get ('headers' , {})
0 commit comments