klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-02 21:19:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Retry running code on more request exception

Browse Source
This commit is contained in:
Debanjum
2025-05-17 17:07:31 -07:00
parent 10a5d68a2c
commit 442c7b6153
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/khoj/processor/tools/run_code.py
View File

@@ -9,8 +9,8 @@ from pathlib import Path
from typing import Any, Callable, List, NamedTuple, Optional
import aiohttp
import httpx
from asgiref.sync import sync_to_async
from httpx import RemoteProtocolError
from tenacity import (
before_sleep_log,
retry,
@@ -192,7 +192,9 @@ async def generate_python_code(
| retry_if_exception_type(aiohttp.ClientTimeout)
| retry_if_exception_type(asyncio.TimeoutError)
| retry_if_exception_type(ConnectionError)
| retry_if_exception_type(RemoteProtocolError)
| retry_if_exception_type(httpx.RemoteProtocolError)
| retry_if_exception_type(httpx.NetworkError)
| retry_if_exception_type(httpx.TimeoutException)
),
wait=wait_random_exponential(min=1, max=5),
stop=stop_after_attempt(3),