dream

1
import requests
2

3

4
def test_respond():
5
    url = "http://0.0.0.0:8131/respond"
6

7
    test_data = {
8
        "last_annotated_utterances": [
9
            {
10
                "text": "What should I cook for a dinner?",
11
                "annotations": {
12
                    "relative_persona_extractor": {
13
                        "persona": ["I like ice-cream.", "I hate onions."],
14
                        "max_similarity": 0.8,
15
                    },
16
                    "midas_classification": [{"open_question_personal": 1.0}],
17
                },
18
            }
19
        ],
20
        "utterances_histories": [
21
            [
22
                "What are you doing?",
23
                "I am planning what to cook.",
24
                "Sounds interesting.",
25
                "What should I cook for a dinner?",
26
            ]
27
        ],
28
    }
29
    gold = []
30

31
    result = requests.post(url, json=test_data).json()
32

33
    assert len(result[0][0]) > 0, print(f"Expected: {gold} but got: {result}")
34
    print("Success")
35

36

37
if __name__ == "__main__":
38
    test_respond()
39