scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-10-08 03:10:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(llm): enhance rate limit error detection in base LLM implementation

Browse source
This commit is contained in:
jinli.yl 2026-01-20 20:47:24 +08:00
parent ff7d25e313
commit c3c7b5a4a0
1 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
reme_ai/core/llm/base_llm.py
View file

@ -402,7 +402,11 @@ class BaseLLM(ABC):
# Check if this is an inappropriate content error
error_message = str(e.args[0]) if e.args else str(e)
is_inappropriate_content = "inappropriate content" in error_message.lower()
is_rate_limit_error = "request rate increased too quickly" in error_message.lower()
is_rate_limit_error = (
"request rate increased too quickly" in error_message.lower() or
"exceeded your current quota" in error_message.lower() or
"insufficient_quota" in error_message.lower()
)
if is_inappropriate_content:
logger.error(f"chat with model={effective_model} detected inappropriate content error")
@ -475,7 +479,11 @@ class BaseLLM(ABC):
# Check if this is an inappropriate content error
error_message = str(e.args[0]) if e.args else str(e)
is_inappropriate_content = "inappropriate content" in error_message.lower()
is_rate_limit_error = "request rate increased too quickly" in error_message.lower()
is_rate_limit_error = (
"request rate increased too quickly" in error_message.lower() or
"exceeded your current quota" in error_message.lower() or
"insufficient_quota" in error_message.lower()
)
if is_inappropriate_content:
logger.error(f"chat sync with model={effective_model} detected inappropriate content error")