File size: 609 Bytes
0e4a27a
 
5a55b5b
5b7d0e6
0e4a27a
 
 
5b7d0e6
0e4a27a
 
c6dd20e
0e4a27a
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
from dataclasses import dataclass
from typing import Iterable

from buster.formatter import ResponseFormatter, Source


@dataclass
class SlackResponseFormatter(ResponseFormatter):
    """Format the answer for Slack."""

    source_template: str = """<{source.url}|🔗 {source.title}>, relevance: {source.question_similarity:2.3f}"""

    def sources_list(self, sources: Iterable[Source]) -> str | None:
        """Format sources into a list."""
        items = [self.source_item(source) for source in sources]
        if not items:
            return None  # No list needed.

        return "\n".join(items)