vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-11 16:04:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Add error message for contextexceedederror (#1866)

Browse source
This commit is contained in:
Shuchang Zheng 2025-03-02 23:27:20 -05:00 • committed by GitHub
parent b3b548ed38
commit d271a84eff
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 21 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
skyvern/exceptions.py
View file

@ -655,3 +655,9 @@ class NoTOTPVerificationCodeFound(SkyvernHTTPException):
if totp_identifier: if totp_identifier:
msg += f" totp_identifier={totp_identifier}" msg += f" totp_identifier={totp_identifier}"
super().__init__(msg) super().__init__(msg)
class SkyvernContextWindowExceededError(SkyvernException):
def __init__(self) -> None:
message = "Context window exceeded. Please contact support@skyvern.com for help."
super().__init__(message)

15
skyvern/forge/sdk/api/llm/api_handler_factory.py
View file

@ -9,6 +9,7 @@ import structlog
from jinja2 import Template from jinja2 import Template
from skyvern.config import settings from skyvern.config import settings
from skyvern.exceptions import SkyvernContextWindowExceededError
from skyvern.forge import app from skyvern.forge import app
from skyvern.forge.sdk.api.llm.config_registry import LLMConfigRegistry from skyvern.forge.sdk.api.llm.config_registry import LLMConfigRegistry
from skyvern.forge.sdk.api.llm.exceptions import ( from skyvern.forge.sdk.api.llm.exceptions import (
@ -125,6 +126,13 @@ class LLMAPIHandlerFactory:
response = await router.acompletion(model=main_model_group, messages=messages, **parameters) response = await router.acompletion(model=main_model_group, messages=messages, **parameters)
except litellm.exceptions.APIError as e: except litellm.exceptions.APIError as e:
raise LLMProviderErrorRetryableTask(llm_key) from e raise LLMProviderErrorRetryableTask(llm_key) from e
except litellm.exceptions.ContextWindowExceededError as e:
LOG.exception(
"Context window exceeded",
llm_key=llm_key,
model=main_model_group,
)
raise SkyvernContextWindowExceededError() from e
except ValueError as e: except ValueError as e:
LOG.exception( LOG.exception(
"LLM token limit exceeded", "LLM token limit exceeded",
@ -300,6 +308,13 @@ class LLMAPIHandlerFactory:
) )
except litellm.exceptions.APIError as e: except litellm.exceptions.APIError as e:
raise LLMProviderErrorRetryableTask(llm_key) from e raise LLMProviderErrorRetryableTask(llm_key) from e
except litellm.exceptions.ContextWindowExceededError as e:
LOG.exception(
"Context window exceeded",
llm_key=llm_key,
model=llm_config.model_name,
)
raise SkyvernContextWindowExceededError() from e
except CancelledError: except CancelledError:
t_llm_cancelled = time.perf_counter() t_llm_cancelled = time.perf_counter()
LOG.error( LOG.error(