Spaces:
Runtime error
Runtime error
marondeau
commited on
Remove clever booleanization of Response
Browse files- buster/formatter/base.py +3 -6
buster/formatter/base.py
CHANGED
@@ -17,9 +17,6 @@ class Response:
|
|
17 |
error: bool = False
|
18 |
error_msg: str | None = None
|
19 |
|
20 |
-
def __bool__(self) -> bool:
|
21 |
-
return not self.error
|
22 |
-
|
23 |
|
24 |
@dataclass
|
25 |
class Formatter:
|
@@ -58,6 +55,6 @@ class Formatter:
|
|
58 |
|
59 |
def __call__(self, response: Response, sources: Iterable[Source]) -> str:
|
60 |
"""Format an answer and its sources, or an error message."""
|
61 |
-
if response:
|
62 |
-
return self.
|
63 |
-
return self.
|
|
|
17 |
error: bool = False
|
18 |
error_msg: str | None = None
|
19 |
|
|
|
|
|
|
|
20 |
|
21 |
@dataclass
|
22 |
class Formatter:
|
|
|
55 |
|
56 |
def __call__(self, response: Response, sources: Iterable[Source]) -> str:
|
57 |
"""Format an answer and its sources, or an error message."""
|
58 |
+
if response.error:
|
59 |
+
return self.error(response)
|
60 |
+
return self.answer(response, sources)
|