Fix lint issue

This commit is contained in:
Thiago Moretto
2024-08-05 13:34:03 -03:00
parent 67e4433dc2
commit f3b3d321e5

View File

@@ -17,7 +17,7 @@ class TokenCalcHandler(BaseCallbackHandler):
self.token_cost_process = token_cost_process
try:
self.encoding = tiktoken.encoding_for_model(self.model_name)
except KeyError as e:
except KeyError:
self.encoding = tiktoken.get_encoding("cl100k_base")
def on_llm_start(