fix on using loguru, but not logging

This commit is contained in:
王同贵 2023-08-08 20:43:55 +08:00
parent ec4c773c80
commit 90c687bbea

View file

@ -6,7 +6,6 @@
@File : openai.py
"""
import asyncio
import logging
import time
from typing import NamedTuple
@ -214,7 +213,7 @@ class OpenAIGPTAPI(BaseGPTAPI, RateLimiter):
@retry(
stop=stop_after_attempt(3),
wait=wait_fixed(1),
after=after_log(logger, logging.WARNING),
after=after_log(logger, logger.level('WARNING').name),
retry=retry_if_exception_type(APIConnectionError),
retry_error_callback=log_and_reraise,
)