dream

1
import requests
2

3

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

7
    dialog_contexts = [
8
        [
9
            {"speaker": "human", "text": "Привет, как день прошел?"},
10
            {"speaker": "bot", "text": "Хорошо, а у тебя как?"},
11
            {"speaker": "human", "text": "Нормально, посоветуй фильм посмотреть"},
12
        ]
13
    ]
14

15
    request_data = {"dialog_contexts": dialog_contexts, "num_return_sequences": 5}
16
    result = requests.post(url, json=request_data).json()[0]
17
    print(result)
18
    assert len(result) == 5 and len(result[0]) > 0, f"Got\n{result}"
19
    print("Success!")
20

21

22
if __name__ == "__main__":
23
    test_respond()
24