From 1ce9ef38280677cd5904ac29475bcb24a7cdd94f Mon Sep 17 00:00:00 2001 From: Bagatur <22008038+baskaryan@users.noreply.github.com> Date: Fri, 30 Jun 2023 09:24:01 -0700 Subject: [PATCH] Rm pytz dep (#6978) --- langchain/callbacks/arthur_callback.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/langchain/callbacks/arthur_callback.py b/langchain/callbacks/arthur_callback.py index 9e72958b9b..c7c8c2317a 100644 --- a/langchain/callbacks/arthur_callback.py +++ b/langchain/callbacks/arthur_callback.py @@ -9,7 +9,6 @@ from time import time from typing import TYPE_CHECKING, Any, DefaultDict, Dict, List, Optional, Union import numpy as np -import pytz from langchain.callbacks.base import BaseCallbackHandler from langchain.schema import AgentAction, AgentFinish, LLMResult @@ -162,6 +161,12 @@ 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] + except ImportError as e: + raise ImportError( + "Could not import pytz. Please install it with 'pip install pytz'." + ) from e run_id = kwargs["run_id"]