Skip to content

regex

RegexAnswer

Extract answers from the text using regular expressions.

Pattern is the regular expression used to extract the answer. Stopping pattern is a string used to split the answer.

Example: r = RegexAnswer("<ANSWER>: (.*)", "[,.;]")

Source code in ragfit/processing/answer_processors/regex.py 
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
class RegexAnswer:
    """
    Extract answers from the text using regular expressions.

    Pattern is the regular expression used to extract the answer.
    Stopping pattern is a string used to split the answer.

    Example:
    `r = RegexAnswer("<ANSWER>: (.*)", "[,.;]")`
    """

    def __init__(self, capture_pattern=None, stopping_pattern=None):
        self.capture_pattern = capture_pattern
        self.stopping_pattern = stopping_pattern

    def __call__(self, text: str):
        """
        Extract the answer from the text.
        """
        if (capture := self.capture_pattern) and capture != "":
            match = re.search(capture, text, re.MULTILINE | re.DOTALL)
            if match:
                text = match.group(1)

        if (stopping := self.stopping_pattern) and stopping != "":
            text = re.split(stopping, text)[0]

        return text

__call__(text: str)

Extract the answer from the text.

Source code in ragfit/processing/answer_processors/regex.py 
19
20
21
22
23
24
25
26
27
28
29
30
31
def __call__(self, text: str):
    """
    Extract the answer from the text.
    """
    if (capture := self.capture_pattern) and capture != "":
        match = re.search(capture, text, re.MULTILINE | re.DOTALL)
        if match:
            text = match.group(1)

    if (stopping := self.stopping_pattern) and stopping != "":
        text = re.split(stopping, text)[0]

    return text