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