ob-metaflow-extensions 1.1.72__tar.gz → 1.1.73__tar.gz

{ob-metaflow-extensions-1.1.72 → ob-metaflow-extensions-1.1.73}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-metaflow-extensions
-Version: 1.1.72
+Version: 1.1.73
 Summary: Outerbounds Platform Extensions for Metaflow
 Author: Outerbounds, Inc.
 License: Commercial

{ob-metaflow-extensions-1.1.72 → ob-metaflow-extensions-1.1.73}/metaflow_extensions/outerbounds/plugins/auth_server.py

@@ -52,10 +52,17 @@ def get_token(url_path):
             return token_info
         except requests.exceptions.HTTPError as e:
             raise MetaflowException(repr(e))
-        except requests.exceptions.ConnectionError as e:
+        except (
+            requests.exceptions.ConnectionError,
+            requests.exceptions.ReadTimeout,
+        ) as e:
             # ConnectionErrors are generally temporary errors like DNS resolution failures,
             # timeouts etc.
-            print("received connection error. Retrying...", e, file=sys.stderr)
+            print(
+                "received error of type {}. Retrying...".format(type(e)),
+                e,
+                file=sys.stderr,
+            )
             time.sleep(retry_delay)
             retry_delay *= 2  # Double the delay for the next attempt
             retry_delay += random.uniform(0, 1)  # Add jitter

{ob-metaflow-extensions-1.1.72 → ob-metaflow-extensions-1.1.73}/metaflow_extensions/outerbounds/plugins/nim/nim_manager.py

@@ -174,11 +174,16 @@ class NimChatCompletion(object):
                         self.invocations.append(invocation_id)
                     elif response.status_code == 200:
                         return response.json()
-                except requests.exceptions.ConnectionError as e:
+                except (
+                    requests.exceptions.ConnectionError,
+                    requests.exceptions.ReadTimeout,
+                ) as e:
                     # ConnectionErrors are generally temporary errors like DNS resolution failures,
                     # timeouts etc.
                     print(
-                        "Encountered connection error. Retrying...", e, file=sys.stderr
+                        "received error of type {}. Retrying...".format(type(e)),
+                        e,
+                        file=sys.stderr,
                     )
                     time.sleep(retry_delay)
                     retry_delay *= 2  # Double the delay for the next attempt
@@ -205,11 +210,14 @@ class NimChatCompletion(object):
                             raise Exception(
                                 f"NVCF returned {poll_response.status_code} status code. Please contact Outerbounds."
                             )
-                    except requests.exceptions.ConnectionError as e:
+                    except (
+                        requests.exceptions.ConnectionError,
+                        requests.exceptions.ReadTimeout,
+                    ) as e:
                         # ConnectionErrors are generally temporary errors like DNS resolution failures,
                         # timeouts etc.
                         print(
-                            "Encountered connection error. Retrying...",
+                            "received error of type {}. Retrying...".format(type(e)),
                             e,
                             file=sys.stderr,
                         )

{ob-metaflow-extensions-1.1.72 → ob-metaflow-extensions-1.1.73}/ob_metaflow_extensions.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-metaflow-extensions
-Version: 1.1.72
+Version: 1.1.73
 Summary: Outerbounds Platform Extensions for Metaflow
 Author: Outerbounds, Inc.
 License: Commercial

{ob-metaflow-extensions-1.1.72 → ob-metaflow-extensions-1.1.73}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_namespace_packages
 from pathlib import Path
 
 
-version = "1.1.72"
+version = "1.1.73"
 this_directory = Path(__file__).parent
 long_description = (this_directory / "README.md").read_text()