castorini / rank_llm

Repository for prompt-decoding using LLMs (GPT3.5, GPT4, Vicuna, and Zephyr)

http://rankllm.ai

Apache License 2.0

273 stars 35 forks source link

Init ServiceRetriever Class #118

Closed xpbowler closed 1 month ago

xpbowler commented 1 month ago

Pull Request Checklist

ServiceRetriever class for interacting with Anserini REST API. Currently only supports retrieval from pre-built indexes and k=10 (REST API does not accept k as a parameter currently).

Class variables:

host
dataset name (dataset)
retrieval_mode (only DATASET is valid at the moment)
retrieval_method (e.g., BM25)

from_dataset_with_prebuilt_index() takes in:

request
dataset_name
retrieval_method
k Anserini API currently does not support passing in k
host Defaults to localhost::8081

PR Type

[x] Feature

xpbowler commented 1 month ago

@sahel-sh linted + added error handling