Function load_gold

src/retrieval/server.py:116–127

(trec_path: str)

    return passage_embedding2id

def load_gold(trec_path: str):
    qid2goldid = {}
    with open(trec_path, 'r') as f:
        for line in f:
            line = line.split()
            qid = line[0]
            goldid = line[2]
            if qid in qid2goldid:
                qid2goldid[qid].append(goldid)
            else:
                qid2goldid[qid] = [goldid]
    return qid2goldid

class QueryRequest(BaseModel):
    queries: List[str]