Spaces:
Runtime error
Runtime error
File size: 2,303 Bytes
0e4a27a |
1 2 3 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 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
from dataclasses import dataclass
from typing import Iterable, NamedTuple
# Should be from the `documents` module.
class Source(NamedTuple):
name: str
url: str
question_similarity: float
# TODO Add answer similarity.
# answer_similarity: float
# Should be from the `nlp` module.
@dataclass(slots=True)
class Response:
text: str
error: bool = False
error_msg: str | None = None
def __bool__(self) -> bool:
return not self.error
@dataclass
class Formatter:
source_template: str = "{source.name} (relevance: {source.question_similarity:2.3f})"
error_msg_template: str = "Something went wrong: {response.error_msg}"
error_fallback_template: str = "Something went very wrong."
sourced_answer_template: str = "{response.text}\n\nSources:\n{sources}\n\nBut what do I know, I'm a chatbot."
unsourced_answer_template: str = "{response.text}\n\nBut what do I know, I'm a chatbot."
def source_item(self, source: Source) -> str:
"""Format a single source item."""
return self.source_template.format(source=source)
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))
def error(self, response: Response) -> str:
"""Format an error message."""
if response.error_msg:
return self.error_msg_template.format(response=response)
return self.error_fallback_template.format(response=response)
def answer(self, response: Response, sources: Iterable[Source]) -> str:
"""Format an answer and its sources."""
sources_list = self.sources_list(sources)
if not sources_list:
return self.sourced_answer_template.format(response=response, sources=sources_list)
return self.unsourced_answer_template.format(response=response)
def __call__(self, response: Response, sources: Iterable[Source]) -> str:
"""Format an answer and its sources, or an error message."""
if response:
return self.answer(response, sources)
return self.error(response)
|