Skip to content

dspy.ColBERTv2

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

Wrapper for the ColBERTv2 Retrieval.

Source code in dspy/dsp/colbertv2.py 
15
16
17
18
19
20
21
22
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

Functions

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

Source code in dspy/dsp/colbertv2.py 
24
25
26
27
28
29
30
31
32
33
34
35
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]