Spaces:
Runtime error
Runtime error
File size: 668 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.base import ResponseFormatter, Source
@dataclass
class MarkdownResponseFormatter(ResponseFormatter):
"""Format the answer in markdown."""
source_template: str = """[🔗 {source.title}]({source.url}), 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(f"{ind}. {item}" for ind, item in enumerate(items, 1))
|