sapiopycommons 2025.2.21a446__tar.gz → 2025.2.25a448__tar.gz

{sapiopycommons-2025.2.21a446 → sapiopycommons-2025.2.25a448}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sapiopycommons
-Version: 2025.2.21a446
+Version: 2025.2.25a448
 Summary: Official Sapio Python API Utilities Package
 Project-URL: Homepage, https://github.com/sapiosciences
 Author-email: Jonathan Steck <jsteck@sapiosciences.com>, Yechen Qiao <yqiao@sapiosciences.com>

{sapiopycommons-2025.2.21a446 → sapiopycommons-2025.2.25a448}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "sapiopycommons"
-version='2025.02.21a446'
+version='2025.02.25a448'
 authors = [
     { name="Jonathan Steck", email="jsteck@sapiosciences.com" },
     { name="Yechen Qiao", email="yqiao@sapiosciences.com" },

{sapiopycommons-2025.2.21a446 → sapiopycommons-2025.2.25a448}/src/sapiopycommons/ai/tool_of_tools.py

@@ -190,19 +190,22 @@ class AiHelper:
     # Contextual info.
     user: SapioUser
     exp_id: int
+    timeout: int
 
     # Managers.
     dr_man: DataRecordManager
     eln_man: ElnManager
     dt_man: DataTypeManager
 
-    def __init__(self, user: SapioUser, exp_id: int):
+    def __init__(self, user: SapioUser, exp_id: int, timeout: int = 120):
         """
         :param user: The user to send the requests from.
         :param exp_id: The ID of the experiment to create the entries in.
+        :param timeout: The timeout in seconds to use for requests.
         """
         self.user = user
         self.exp_id = exp_id
+        self.timeout = timeout
 
         self.dr_man = DataRecordManager(self.user)
         self.eln_man = ElnManager(self.user)
@@ -218,7 +221,7 @@ class AiHelper:
         :return: The Response object returned by the endpoint.
         """
         headers = create_tot_headers(self.user.url, self.user.username, self.user.password, self.exp_id, tab_prefix)
-        response = requests.post(url, json=payload, headers=headers)
+        response = requests.post(url, json=payload, headers=headers, timeout=self.timeout)
         response.raise_for_status()
         return response
 
@@ -232,7 +235,7 @@ class AiHelper:
         :return: The Response object returned by the endpoint.
         """
         headers = create_tot_headers(self.user.url, self.user.username, self.user.password, self.exp_id, tab_prefix)
-        response = requests.get(url, params=params, headers=headers)
+        response = requests.get(url, params=params, headers=headers, timeout=self.timeout)
         response.raise_for_status()
         return response
 
@@ -331,8 +334,9 @@ class AiHelper:
             :param n: The unique name of the field.
             """
             link_out: dict[str, str] = {}
-            if v.startswith("https://") or v.startswith("http://"):
-                link_out["Link"] = "[[LINK_OUT]]"
+            if isinstance(v, str):
+	            if v.startswith("https://") or v.startswith("http://"):
+	                link_out["Link"] = "[[LINK_OUT]]"
             return fb.string_field(n, display_name=k, link_out=link_out)
 
         def update_field_length(k: str, v: Any, lengths: dict[str, int]) -> None:
@@ -390,7 +394,7 @@ class AiHelper:
                 if field_type == FieldType.BOOLEAN and isinstance(value, bool):
                     continue
                 # Integers and floats both fit in DOUBLE fields, but floats can't be NaN or infinity.
-                if field_type == FieldType.DOUBLE:
+                if field_type == FieldType.DOUBLE and not isinstance(value, bool):
                     if isinstance(value, int):
                         continue
                     if isinstance(value, float) and not math.isnan(value) and not math.isinf(value):