dream

1
import argparse
2
import json
3
import pathlib
4

5
import numpy as np
6
import requests
7
import tqdm
8

9
# TGT_URL = os.getenv("TGT_URL", "http://localhost:8029/convert_reddit")
10
# N_REQUESTS = int(os.getenv("N_REQUESTS", 5))
11
# OUT_FILE = str(os.getenv("OUT_FILE", "confidences.npy"))
12

13
parser = argparse.ArgumentParser()
14
parser.add_argument(
15
    "--url",
16
    help="skill_url",
17
    default="http://localhost:8029/convert_reddit",
18
)
19
parser.add_argument(
20
    "--questions_json_file",
21
    type=pathlib.Path,
22
    help="path to a json file",
23
    default="tests/test_question_tasks.json",
24
)
25
parser.add_argument(
26
    "-o",
27
    "--npy_file_path",
28
    type=pathlib.Path,
29
    help="path to npy file",
30
    default="confidences.npy",
31
)
32
args = parser.parse_args()
33
data = json.load(args.questions_json_file.open())
34

35

36
def history_gen(dialogs):
37
    for dialog in dialogs:
38
        for i in range(1, len(dialog) + 1):
39
            history = dialog[:i]
40
            yield history
41

42

43
confidences = []
44
for task in tqdm.tqdm(data["tasks"]):
45
    response = {}
46
    for _ in range(1):
47
        res = requests.post(
48
            args.url,
49
            json={"personality": [data["personality"]], "utterances_histories": [task["utterances_histories"]]},
50
        ).json()[0]
51
        response[res[0]] = res[1]
52
    confidences.extend(response.values())
53

54
np.save(str(args.npy_file_path), confidences)
55