Skip to content
Snippets Groups Projects
Unverified Commit 1a688d76 authored by Yue Fei's avatar Yue Fei Committed by GitHub
Browse files

Add retry for batch runner (#12647)

parent f26bbf35
No related branches found
No related tags found
No related merge requests found
import asyncio import asyncio
from tenacity import retry, stop_after_attempt, wait_exponential
from typing import Any, Dict, List, Optional, Sequence, Tuple, cast from typing import Any, Dict, List, Optional, Sequence, Tuple, cast
from llama_index.core.async_utils import asyncio_module from llama_index.core.async_utils import asyncio_module
...@@ -7,6 +8,11 @@ from llama_index.core.base.response.schema import RESPONSE_TYPE, Response ...@@ -7,6 +8,11 @@ from llama_index.core.base.response.schema import RESPONSE_TYPE, Response
from llama_index.core.evaluation.base import BaseEvaluator, EvaluationResult from llama_index.core.evaluation.base import BaseEvaluator, EvaluationResult
@retry(
reraise=True,
stop=stop_after_attempt(3),
wait=wait_exponential(multiplier=1, min=4, max=10),
)
async def eval_response_worker( async def eval_response_worker(
semaphore: asyncio.Semaphore, semaphore: asyncio.Semaphore,
evaluator: BaseEvaluator, evaluator: BaseEvaluator,
...@@ -26,6 +32,11 @@ async def eval_response_worker( ...@@ -26,6 +32,11 @@ async def eval_response_worker(
) )
@retry(
reraise=True,
stop=stop_after_attempt(3),
wait=wait_exponential(multiplier=1, min=4, max=10),
)
async def eval_worker( async def eval_worker(
semaphore: asyncio.Semaphore, semaphore: asyncio.Semaphore,
evaluator: BaseEvaluator, evaluator: BaseEvaluator,
...@@ -46,6 +57,11 @@ async def eval_worker( ...@@ -46,6 +57,11 @@ async def eval_worker(
) )
@retry(
reraise=True,
stop=stop_after_attempt(3),
wait=wait_exponential(multiplier=1, min=4, max=10),
)
async def response_worker( async def response_worker(
semaphore: asyncio.Semaphore, semaphore: asyncio.Semaphore,
query_engine: BaseQueryEngine, query_engine: BaseQueryEngine,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment