sapiopycommons 2025.2.14a434__tar.gz → 2025.2.14a436__tar.gz

{sapiopycommons-2025.2.14a434 → sapiopycommons-2025.2.14a436}/PKG-INFO

@@ -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 → sapiopycommons-2025.2.14a436}/pyproject.toml

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

{sapiopycommons-2025.2.14a434/src/sapiopycommons/elain → sapiopycommons-2025.2.14a436/src/sapiopycommons/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:
     """