dream

1
import requests
2

3
use_context = True
4

5

6
def main():
7
    url = "http://0.0.0.0:8075/model"
8

9
    request_data = [
10
        {
11
            "entity_substr": [["форрест гамп"]],
12
            "entity_tags": [["film"]],
13
            "context": [["кто снял фильм форрест гамп?"]],
14
        },
15
        {
16
            "entity_substr": [["роберт левандовский"]],
17
            "entity_tags": [["per"]],
18
            "context": [["за какую команду играет роберт левандовский?"]],
19
        },
20
    ]
21

22
    gold_results = [["Q134773"], ["Q151269"]]
23

24
    count = 0
25
    for data, gold_result in zip(request_data, gold_results):
26
        result = requests.post(url, json=data).json()
27
        entity_ids = result[0][0]["entity_ids"]
28
        if entity_ids == gold_result:
29
            count += 1
30
        else:
31
            print(f"Got {result}, but expected: {gold_result}")
32

33
    assert count == len(request_data)
34
    print("Success")
35

36

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