Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Better LLM retry behavior #6557

Open
wants to merge 9 commits into
base: main
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 7 additions & 7 deletions docs/modules/usage/llms/llms.md
Original file line number Diff line number Diff line change
Expand Up @@ -63,22 +63,22 @@ We have a few guides for running OpenHands with specific model providers:
### API retries and rate limits

LLM providers typically have rate limits, sometimes very low, and may require retries. OpenHands will automatically
retry requests if it receives a Rate Limit Error (429 error code), API connection error, or other transient errors.
retry requests if it receives a Rate Limit Error (429 error code).

You can customize these options as you need for the provider you're using. Check their documentation, and set the
following environment variables to control the number of retries and the time between retries:

- `LLM_NUM_RETRIES` (Default of 8)
- `LLM_RETRY_MIN_WAIT` (Default of 15 seconds)
- `LLM_RETRY_MAX_WAIT` (Default of 120 seconds)
- `LLM_NUM_RETRIES` (Default of 4 times)
- `LLM_RETRY_MIN_WAIT` (Default of 5 seconds)
- `LLM_RETRY_MAX_WAIT` (Default of 30 seconds)
- `LLM_RETRY_MULTIPLIER` (Default of 2)

If you are running OpenHands in development mode, you can also set these options in the `config.toml` file:

```toml
[llm]
num_retries = 8
retry_min_wait = 15
retry_max_wait = 120
num_retries = 4
retry_min_wait = 5
retry_max_wait = 30
retry_multiplier = 2
```
44 changes: 43 additions & 1 deletion frontend/src/i18n/translation.json
Original file line number Diff line number Diff line change
Expand Up @@ -3803,6 +3803,37 @@
"pt": "Erro ao autenticar com o provedor LLM. Por favor, verifique sua chave API",
"tr": "LLM sağlayıcısı ile kimlik doğrulama hatası. Lütfen API anahtarınızı kontrol edin"
},
"STATUS$ERROR_LLM_SERVICE_UNAVAILABLE": {
"en": "The LLM provider is currently unavailable. Please try again later.",
"es": "El proveedor LLM no está actualmente disponible. Por favor, inténtelo de nuevo más tarde.",
"zh-CN": "LLM提供商当前不可用",
"zh-TW": "LLM提供商目前無法使用",
"ko-KR": "LLM 공급자가 현재 사용 불가능합니다",
"ja": "LLMプロバイダーが現在利用できません。後でもう一度試してください。",
"no": "LLM-leverandøren er nå ikke tilgjengelig. Vennligst prøv igjen senere.",
"ar": "المزود LLM غير متاح حالياً. يرجى المحاولة مرة أخرى لاحقًا.",
"de": "Der LLM-Anbieter ist derzeit nicht verfügbar. Bitte versuchen Sie es später erneut.",
"fr": "Le fournisseur LLM n'est actuellement pas disponible. Veuillez réessayer plus tard.",
"it": "Il provider LLM non è attualmente disponibile. Per favore, riprova più tardi.",
"pt": "O provedor LLM não está atualmente disponível. Por favor, tente novamente mais tarde.",
"tr": "LLM sağlayıcısı şu anda kullanılamıyor. Lütfen daha sonra tekrar deneyin."
},
"STATUS$ERROR_LLM_INTERNAL_SERVER_ERROR": {
"en": "The request failed with an internal server error.",
"es": "La solicitud falló con un error del servidor interno.",
"zh-CN": "请求失败,请稍后再试",
"zh-TW": "請求失敗,請稍後再試",
"ko-KR": "요청이 실패했습니다. 나중에 다시 시도해주세요.",
"ja": "リクエストが内部サーバーエラーで失敗しました。後でもう一度試してください。",
"no": "Det oppstod en feil ved tilkobling til kjøretidsmiljøet. Vennligst oppdater siden.",
"ar": "حدث خطأ أثناء الاتصال بوقت التشغيل. يرجى تحديث الصفحة.",
"de": "Beim Verbinden mit der Laufzeitumgebung ist ein Fehler aufgetreten. Bitte aktualisieren Sie die Seite.",
"fr": "Une erreur s'est produite lors de la connexion à l'environnement d'exécution. Veuillez rafraîchir la page.",
"it": "Si è verificato un errore durante la connessione al runtime. Aggiorna la pagina.",
"pt": "Ocorreu um erro ao conectar ao ambiente de execução. Por favor, atualize a página.",
"tr": "Çalışma zamanına bağlanırken bir hata oluştu. Lütfen sayfayı yenileyin."
},

"STATUS$ERROR_RUNTIME_DISCONNECTED": {
"en": "There was an error while connecting to the runtime. Please refresh the page.",
"zh-CN": "运行时已断开连接",
Expand All @@ -3820,7 +3851,18 @@
},
"STATUS$LLM_RETRY": {
"en": "Retrying LLM request",
"zh-TW": "重新嘗試 LLM 請求中"
"es": "Reintentando solicitud LLM",
"zh-CN": "重试LLM请求",
"zh-TW": "重試LLM請求",
"ko-KR": "LLM 요청 재시도",
"ja": "LLM リクエストを再試行中",
"no": "Gjenforsøker LLM-forespørsel",
"ar": "يتم إعادة تحميل الطلب LLM",
"de": "LLM-Anfrage erneut versuchen",
"fr": "Réessayer la requête LLM",
"it": "Ritenta la richiesta LLM",
"pt": "Reintentando a solicitação LLM",
"tr": "LLM isteğini yeniden deniyor"
},
"AGENT_ERROR$BAD_ACTION": {
"en": "Agent tried to execute a malformed action.",
Expand Down
11 changes: 11 additions & 0 deletions openhands/controller/agent_controller.py
Original file line number Diff line number Diff line change
Expand Up @@ -214,6 +214,17 @@ async def _react_to_exception(
err_id = ''
if isinstance(e, litellm.AuthenticationError):
err_id = 'STATUS$ERROR_LLM_AUTHENTICATION'
elif isinstance(
e,
(
litellm.ServiceUnavailableError,
litellm.APIConnectionError,
litellm.APIError,
),
):
err_id = 'STATUS$ERROR_LLM_SERVICE_UNAVAILABLE'
elif isinstance(e, litellm.InternalServerError):
err_id = 'STATUS$ERROR_LLM_INTERNAL_SERVER_ERROR'
elif isinstance(e, RateLimitError):
await self.set_agent_state_to(AgentState.RATE_LIMITED)
return
Expand Down
7 changes: 4 additions & 3 deletions openhands/core/config/llm_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,10 +59,11 @@ class LLMConfig(BaseModel):
aws_region_name: str | None = Field(default=None)
openrouter_site_url: str = Field(default='https://docs.all-hands.dev/')
openrouter_app_name: str = Field(default='OpenHands')
num_retries: int = Field(default=8)
# total wait time: 5 + 10 + 20 + 30 = 65 seconds
num_retries: int = Field(default=4)
retry_multiplier: float = Field(default=2)
retry_min_wait: int = Field(default=15)
retry_max_wait: int = Field(default=120)
retry_min_wait: int = Field(default=5)
retry_max_wait: int = Field(default=30)
timeout: int | None = Field(default=None)
max_message_chars: int = Field(
default=30_000
Expand Down
14 changes: 1 addition & 13 deletions openhands/llm/llm.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,11 +18,7 @@
from litellm import completion as litellm_completion
from litellm import completion_cost as litellm_completion_cost
from litellm.exceptions import (
APIConnectionError,
APIError,
InternalServerError,
RateLimitError,
ServiceUnavailableError,
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

503 is a transitory error, we could probably keep it?

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hmm. It's transitory but also unexpected...

I'm open to it but I lean towards telling the user their LLM is flaking out rather than OpenHands looking like it's slow

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I kinda agree with you actually. We always had a problem in understanding our retry settings, because it's a bit weird to figure out a sensible default for "unexpected stuff happened".

And now we do allow the user to continue normally after reporting the error.

eval is the exception, I'd love to hear from Xingyao on that.

)
from litellm.types.utils import CostPerToken, ModelResponse, Usage
from litellm.utils import create_pretrained_tokenizer
Expand All @@ -41,15 +37,7 @@
__all__ = ['LLM']

# tuple of exceptions to retry on
LLM_RETRY_EXCEPTIONS: tuple[type[Exception], ...] = (
APIConnectionError,
# FIXME: APIError is useful on 502 from a proxy for example,
# but it also retries on other errors that are permanent
APIError,
InternalServerError,
RateLimitError,
ServiceUnavailableError,
)
LLM_RETRY_EXCEPTIONS: tuple[type[Exception], ...] = (RateLimitError,)

# cache prompt supporting models
# remove this when we gemini and deepseek are supported
Expand Down
34 changes: 0 additions & 34 deletions tests/unit/test_llm.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,10 +3,7 @@

import pytest
from litellm.exceptions import (
APIConnectionError,
InternalServerError,
RateLimitError,
ServiceUnavailableError,
)

from openhands.core.config import LLMConfig
Expand Down Expand Up @@ -187,21 +184,6 @@ def test_completion_with_mocked_logger(
@pytest.mark.parametrize(
'exception_class,extra_args,expected_retries',
[
(
APIConnectionError,
{'llm_provider': 'test_provider', 'model': 'test_model'},
2,
),
(
InternalServerError,
{'llm_provider': 'test_provider', 'model': 'test_model'},
2,
),
(
ServiceUnavailableError,
{'llm_provider': 'test_provider', 'model': 'test_model'},
2,
),
(RateLimitError, {'llm_provider': 'test_provider', 'model': 'test_model'}, 2),
],
)
Expand Down Expand Up @@ -254,22 +236,6 @@ def test_completion_rate_limit_wait_time(mock_litellm_completion, default_config
), f'Expected wait time between {default_config.retry_min_wait} and {default_config.retry_max_wait} seconds, but got {wait_time}'


@patch('openhands.llm.llm.litellm_completion')
def test_completion_exhausts_retries(mock_litellm_completion, default_config):
mock_litellm_completion.side_effect = APIConnectionError(
'Persistent error', llm_provider='test_provider', model='test_model'
)

llm = LLM(config=default_config)
with pytest.raises(APIConnectionError):
llm.completion(
messages=[{'role': 'user', 'content': 'Hello!'}],
stream=False,
)

assert mock_litellm_completion.call_count == llm.config.num_retries


@patch('openhands.llm.llm.litellm_completion')
def test_completion_operation_cancelled(mock_litellm_completion, default_config):
mock_litellm_completion.side_effect = OperationCancelled('Operation cancelled')
Expand Down
2 changes: 1 addition & 1 deletion tests/unit/test_llm_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -188,7 +188,7 @@ def test_load_from_toml_llm_missing_generic(
assert custom_only.model == 'custom-only-model'
assert custom_only.api_key.get_secret_value() == 'custom-only-api-key'
assert custom_only.embedding_model == 'local' # default value
assert custom_only.num_retries == 8 # default value
assert custom_only.num_retries == 4 # default value


def test_load_from_toml_llm_invalid_config(
Expand Down
Loading