dream
38 строк · 1.0 Кб
1import requests
2
3
4def test_respond():
5url = "http://0.0.0.0:8140/respond"
6
7test_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}
29gold = []
30
31result = requests.post(url, json=test_data).json()
32
33assert len(result[0][0]) > 0, print(f"Expected: {gold} but got: {result}")
34print("Success")
35
36
37if __name__ == "__main__":
38test_respond()
39