diff --git a/log10/load.py b/log10/load.py index 32d1af66..bde1d6d6 100644 --- a/log10/load.py +++ b/log10/load.py @@ -74,10 +74,9 @@ def post_request(url: str, json_payload: dict = {}) -> requests.Response: headers = {"x-log10-token": token, "Content-Type": "application/json"} json_payload["organization_id"] = org_id try: - # todo: set timeout - res = requests.post( - url, headers=headers, json=json_payload, timeout=os.environ.get("LOG10_REQUESTS_TIMEOUT", 10) - ) + timeout = int(os.environ.get("LOG10_REQUESTS_TIMEOUT", 10)) + assert isinstance(timeout, int) + res = requests.post(url, headers=headers, json=json_payload, timeout=timeout) # raise_for_status() will raise an exception if the status is 4xx, 5xxx res.raise_for_status() diff --git a/pyproject.toml b/pyproject.toml index ce587dad..6f5ce5ab 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api" [tool.poetry] name = "log10-io" -version = "0.6.3" +version = "0.6.4" authors = ["log10 team"] license = "MIT" description = "Unified LLM data management" diff --git a/setup.py b/setup.py index 9c27d690..57c9fc5d 100644 --- a/setup.py +++ b/setup.py @@ -5,7 +5,7 @@ setup( name="Log10", - version="0.6.3", + version="0.6.4", description="Log10 LLM data management", author="Log10 team", author_email="team@log10.io",