mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Giles Odigwe
ee56314a26
* Samples Tests * small fixes * job fix * telemetry dependency fix * job error fix * sorting provider specific tests * telemetry fixes * openai file search fix --------- Co-authored-by: Giles Odigwe <gilesodigwe@microsoft.com>
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
import asyncio
|
|
import logging
|
|
from collections.abc import Awaitable, Callable
|
|
from typing import Any
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
async def retry(
|
|
func: Callable[[], Awaitable[Any]],
|
|
retries: int = 3,
|
|
reset: Callable[[], None] | None = None,
|
|
name: str | None = None,
|
|
) -> None:
|
|
"""Retry function with reset capability and proper logging.
|
|
|
|
Args:
|
|
func: The function to retry.
|
|
retries: Number of retries.
|
|
reset: Function to reset the state of any variables used in the function.
|
|
name: Optional name for logging purposes.
|
|
"""
|
|
func_name = name or func.__module__
|
|
logger.info(f"Running {retries} retries with func: {func_name}")
|
|
|
|
for i in range(retries):
|
|
logger.info(f" Try {i + 1} for {func_name}")
|
|
try:
|
|
if reset:
|
|
reset()
|
|
await func()
|
|
return
|
|
except Exception as e:
|
|
logger.warning(f" On try {i + 1} got this error: {e}")
|
|
if i == retries - 1: # Last retry
|
|
raise
|
|
|
|
# Binary exponential backoff like Semantic Kernel
|
|
backoff = 2**i
|
|
logger.info(f" Sleeping for {backoff} seconds before retrying")
|
|
await asyncio.sleep(backoff)
|