sapiopycommons 2025.2.14a434__py3-none-any.whl → 2025.2.14a436__py3-none-any.whl

sapiopycommons/{elain → ai}/tool_of_tools.py

@@ -1,4 +1,5 @@
 import base64
+import io
 import math
 import re
 from typing import Final, Mapping, Any
@@ -27,6 +28,8 @@ from sapiopycommons.callbacks.field_builder import FieldBuilder
 from sapiopycommons.general.aliases import AliasUtil, SapioRecord
 from sapiopycommons.general.exceptions import SapioException
 from sapiopycommons.general.time_util import TimeUtil
+from sapiopycommons.multimodal.multimodal import MultiModalManager
+from sapiopycommons.multimodal.multimodal_data import ImageDataRequestPojo
 
 CREDENTIALS_HEADER: Final[str] = "SAPIO_APP_API_KEY"
 API_URL_HEADER: Final[str] = "SAPIO_APP_API_URL"
@@ -388,6 +391,32 @@ class AiHelper:
             file_contents: bytes = f.read()
             return self.create_attachment_entry(tab, entry_name, file_path, file_contents)
 
+    def smiles_to_svg(self, smiles: list[str]) -> list[bytes]:
+        """
+        Given a list of SMILES strings, return a list of the corresponding images in SVG format.
+
+        :param smiles: The SMILES strings to retrieve images for.
+        :return: The images in SVG format. The indices of the returned list correspond to the indices of the input
+            SMILES.
+        """
+        if not smiles:
+            return []
+        reg_man = MultiModalManager(self.user)
+        image_list: list[str] = reg_man.load_image_data(ImageDataRequestPojo(smiles, False))
+        return [x.encode() for x in image_list]
+
+    def update_record_images(self, records: list[SapioRecord], images: list[bytes]) -> None:
+        """
+        Update the images of the given records with the given images.
+
+        :param records: The records to update the images of.
+        :param images: The images to update the records with. Records will be match with the image in the matching
+            index of this list.
+        """
+        for record, image in zip(AliasUtil.to_data_records(records), images):
+            with io.BytesIO(image) as bytes_io:
+                self.dr_man.set_record_image(record, bytes_io)
+
 
 class ToolOfToolsHelper:
     """

{sapiopycommons-2025.2.14a434.dist-info → sapiopycommons-2025.2.14a436.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sapiopycommons
-Version: 2025.2.14a434
+Version: 2025.2.14a436
 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.14a434.dist-info → sapiopycommons-2025.2.14a436.dist-info}/RECORD

@@ -1,4 +1,6 @@
 sapiopycommons/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sapiopycommons/ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sapiopycommons/ai/tool_of_tools.py,sha256=zdQIgZV0ISyE75GhWY-YHy6V9ZM6FM4xXZvEmlJ0lME,33737
 sapiopycommons/callbacks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sapiopycommons/callbacks/callback_util.py,sha256=68lPK8GtE6zd8C0fQaMLxT57PtU-6HnO37Zk-u56Mrs,129933
 sapiopycommons/callbacks/field_builder.py,sha256=p2XacN99MuKk3ite8GAqstUMpixqugul2CsC4gB83-o,38620
@@ -14,8 +16,6 @@ sapiopycommons/datatype/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG
 sapiopycommons/datatype/attachment_util.py,sha256=_l2swuP8noIGAl4bwzBUEhr6YlN_OVZl3-gi1XqFHYA,3364
 sapiopycommons/datatype/data_fields.py,sha256=C6HpqtEuF0KsxhlBUprfyv1XguaXql3EYWVbh8y-IFU,4064
 sapiopycommons/datatype/pseudo_data_types.py,sha256=6TG7aJxgmUZ8FQkWBcgmbK5oy7AFFNtKOPpi1w1OOYA,27657
-sapiopycommons/elain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sapiopycommons/elain/tool_of_tools.py,sha256=ciVans-wV5yuUjCImgAgqJsiWN3g89OrtGoR2Y6IdLE,32379
 sapiopycommons/eln/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sapiopycommons/eln/experiment_handler.py,sha256=8hmR7sawDo9K6iBwB44QSoxlH1M91inor7dfuXQ4LKs,69323
 sapiopycommons/eln/experiment_report_util.py,sha256=EA2Iq8gW17bSEI6lPoHYQQ-fDvG4O28RWOoTPXpOlUw,36640
@@ -58,7 +58,7 @@ sapiopycommons/webhook/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3
 sapiopycommons/webhook/webhook_context.py,sha256=D793uLsb1691SalaPnBUk3rOSxn_hYLhdvkaIxjNXss,1909
 sapiopycommons/webhook/webhook_handlers.py,sha256=L0HetSm43NvA5KyW3xbLpGFh2DbAaeZJVtXIEl2fvV8,39689
 sapiopycommons/webhook/webservice_handlers.py,sha256=Y5dHx_UFWFuSqaoPL6Re-fsKYRuxvCWZ8bj6KSZ3jfM,14285
-sapiopycommons-2025.2.14a434.dist-info/METADATA,sha256=F8Gim4X_8zcHuP5NJt8PT49HyDnR-s0koyDRGVHav0Q,3143
-sapiopycommons-2025.2.14a434.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-sapiopycommons-2025.2.14a434.dist-info/licenses/LICENSE,sha256=HyVuytGSiAUQ6ErWBHTqt1iSGHhLmlC8fO7jTCuR8dU,16725
-sapiopycommons-2025.2.14a434.dist-info/RECORD,,
+sapiopycommons-2025.2.14a436.dist-info/METADATA,sha256=8bjTtK1h5j5TnmkppQddbsbMOj5Bz9E5wO60JoVfnEs,3143
+sapiopycommons-2025.2.14a436.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+sapiopycommons-2025.2.14a436.dist-info/licenses/LICENSE,sha256=HyVuytGSiAUQ6ErWBHTqt1iSGHhLmlC8fO7jTCuR8dU,16725
+sapiopycommons-2025.2.14a436.dist-info/RECORD,,