dream

1
import requests
2

3

4
def main():
5
    url = "http://0.0.0.0:8114/respond"
6

7
    result = requests.post(
8
        url,
9
        json={
10
            "dialogs": [
11
                {
12
                    "human_utterances": [{"text": "i like my grandma", "annotations": {}}],
13
                    "bot_utterances": [],
14
                    "utterances": [{"text": "i like my grandma", "annotations": {}}],
15
                    "human": {"attributes": {}},
16
                },
17
                {
18
                    "human_utterances": [
19
                        {
20
                            "text": "fuck off",
21
                            "annotations": {
22
                                "badlisted_words": {
23
                                    "profanity": True,
24
                                    "inappropriate": True,
25
                                    "restricted_topics": False,
26
                                }
27
                            },
28
                        }
29
                    ],
30
                    "bot_utterances": [],
31
                    "utterances": [
32
                        {
33
                            "text": "fuck off",
34
                            "annotations": {
35
                                "badlisted_words": {
36
                                    "profanity": True,
37
                                    "inappropriate": True,
38
                                    "restricted_topics": False,
39
                                }
40
                            },
41
                        }
42
                    ],
43
                    "human": {"attributes": {}},
44
                },
45
            ]
46
        },
47
    ).json()
48
    gold_result = [{"human_attributes": {"age_group": "kid"}}, {"human_attributes": {"age_group": "adult"}}]
49
    assert result == gold_result, print(f"Got\n{result}\n, but expected:\n{gold_result}")
50

51
    print("Success")
52

53

54
if __name__ == "__main__":
55
    main()
56