community[patch]: callbacks mypy fixes (#17058)

Related to #17048
This commit is contained in:
Bagatur
2024-02-05 12:37:27 -08:00
committed by GitHub
parent 75b6fa1134
commit af5ae24af2
12 changed files with 49 additions and 59 deletions

View File

@@ -162,7 +162,7 @@ class ArthurCallbackHandler(BaseCallbackHandler):
def on_llm_end(self, response: LLMResult, **kwargs: Any) -> None:
"""On LLM end, send data to Arthur."""
try:
import pytz # type: ignore[import]
import pytz
except ImportError as e:
raise ImportError(
"Could not import pytz. Please install it with 'pip install pytz'."