beamlit 0.0.60rc128__py3-none-any.whl → 0.0.61__py3-none-any.whl

beamlit/functions/common.py

@@ -34,6 +34,22 @@ from beamlit.models import AgentChain
 
 logger = getLogger(__name__)
 
+MAX_RETRIES = 10
+RETRY_DELAY = 1  # 1 second delay between retries
+
+async def initialize_with_retry(toolkit, function_name: str, max_retries: int):
+    for attempt in range(1, max_retries + 1):
+        try:
+            await toolkit.initialize()
+            return await toolkit.get_tools()
+        except Exception as e:
+            if attempt == max_retries:
+                logger.warn(f"Failed to initialize function {function_name} after {max_retries} attempts: {e!s}")
+                raise
+            logger.info(f"Attempt {attempt} failed for {function_name}, retrying...")
+            await asyncio.sleep(RETRY_DELAY)
+    return []
+
 async def get_functions(
     remote_functions: Union[list[str], None] = None,
     local_functions: Union[list[dict], None] = None,
@@ -178,13 +194,12 @@ async def get_functions(
                                                 )
                         except Exception as e:
                             logger.warning(f"Error processing {file_path}: {e!s}")
-
     if remote_functions:
         for function in remote_functions:
             try:
                 toolkit = RemoteToolkit(client, function)
-                await toolkit.initialize()
-                functions.extend(await toolkit.get_tools())
+                tools = await initialize_with_retry(toolkit, function, MAX_RETRIES)
+                functions.extend(tools)
             except Exception as e:
                 if not isinstance(e, RuntimeError):
                     logger.debug(
@@ -196,8 +211,8 @@ async def get_functions(
         for function in local_functions:
             try:
                 toolkit = LocalToolKit(client, function)
-                await toolkit.initialize()
-                functions.extend(await toolkit.get_tools())
+                tools = await initialize_with_retry(toolkit, function["name"], MAX_RETRIES)
+                functions.extend(tools)
             except Exception as e:
                 logger.debug(
                     f"Failed to initialize local function {function}: {e!s}\n"

{beamlit-0.0.60rc128.dist-info → beamlit-0.0.61.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beamlit
-Version: 0.0.60rc128
+Version: 0.0.61
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 License-File: LICENSE

{beamlit-0.0.60rc128.dist-info → beamlit-0.0.61.dist-info}/RECORD

@@ -131,7 +131,7 @@ beamlit/deploy/deploy.py,sha256=AP1FjmoqE3tdKcNKtRZLojTaJ1PR73hwrk6ORRSgRmQ,1165
 beamlit/deploy/format.py,sha256=W3ESUHyFv-iZDjVnHOf9YFDDXZSXYIFFbwCoL1GInE0,1162
 beamlit/deploy/parser.py,sha256=gjRUhOVtfKnc1UNc_FhXsEfj9zrMNuq8W93pNsJBpo0,7586
 beamlit/functions/__init__.py,sha256=Mnoqpa1dm7TXwjodBbF_40JyD78aXsOYWmqjDSnA1lU,317
-beamlit/functions/common.py,sha256=v4nmLP9Wotpb5e6hV30XbItgZjr5lwXcF0EOotxRngI,10127
+beamlit/functions/common.py,sha256=ssCtIR6oTq2JR2FRuZIgW7uH8IwP2DDsZX7AwBgDGdg,10819
 beamlit/functions/decorator.py,sha256=iQbLwUo0K83DFJ3ub8O5jKtkbSINnku6GZcKJ9h7-5E,2292
 beamlit/functions/local/local.py,sha256=qkd-RGObyjZD0KIwaze_TnTuiJ3lyluCcM97nbyPNBk,1897
 beamlit/functions/mcp/client.py,sha256=enLo0dzWMBHJEQf6as3UWM8tN3CjUN1YO3UPn67DLac,4072
@@ -254,8 +254,8 @@ beamlit/serve/app.py,sha256=5XZci-R95Zjl97wMtQd1BRtonnkJJ2AeoTVFPKGAOfA,4283
 beamlit/serve/middlewares/__init__.py,sha256=O7fyfE1DIYmajFY9WWdzxCgeAQWZzJfeUjzHGbpWaAk,309
 beamlit/serve/middlewares/accesslog.py,sha256=lcu33j4epFSHRBaeTpyt8deNb3kaM3K91-andw4fp80,1112
 beamlit/serve/middlewares/processtime.py,sha256=3x5w1yQexB0xFNKK6fgLbINxT-eLLunfZ6UDV0bIIF4,944
-beamlit-0.0.60rc128.dist-info/METADATA,sha256=B3MFxSnd094ye9cY7XsjIi7NU_k-jrm9fwgRYPPGL7E,3547
-beamlit-0.0.60rc128.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.60rc128.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
-beamlit-0.0.60rc128.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-beamlit-0.0.60rc128.dist-info/RECORD,,
+beamlit-0.0.61.dist-info/METADATA,sha256=khTDKpuZB6tg94YRpH_CCmqaW12DZrYzTRU8g4Yf1Vc,3542
+beamlit-0.0.61.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.61.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
+beamlit-0.0.61.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+beamlit-0.0.61.dist-info/RECORD,,