dspy.ColBERTv2

dspy.ColBERTv2(url: str = 'http://0.0.0.0', port: Optional[Union[str, int]] = None, post_requests: bool = False)

ColBERTv2 检索器的包装器。

源代码位于 dspy/dsp/colbertv2.py

def __init__(
    self,
    url: str = "http://0.0.0.0",
    port: Optional[Union[str, int]] = None,
    post_requests: bool = False,
):
    self.post_requests = post_requests
    self.url = f"{url}:{port}" if port else url

函数

__call__(query: str, k: int = 10, simplify: bool = False) -> Union[list[str], list[dotdict]]

源代码位于 dspy/dsp/colbertv2.py

def __call__(
    self,
    query: str,
    k: int = 10,
    simplify: bool = False,
) -> Union[list[str], list[dotdict]]:
    if self.post_requests:
        topk: list[dict[str, Any]] = colbertv2_post_request(self.url, query, k)
    else:
        topk: list[dict[str, Any]] = colbertv2_get_request(self.url, query, k)

    if simplify:
        return [psg["long_text"] for psg in topk]

    return [dotdict(psg) for psg in topk]