Fix make_async().

lynxkite-bio/src/lynxkite_bio/nims.py

@@ -3,7 +3,9 @@
 from lynxkite_graph_analytics import Bundle
 from lynxkite.core import ops
 import httpx
+import io
 import pandas as pd
+import rdkit
 import os
 
 from . import k8s
@@ -122,14 +124,15 @@ def view_molecule(
     row_index: int = 0,
 ):
     molecule_data = bundle.dfs[molecule_table][molecule_column].iloc[row_index]
+    if isinstance(molecule_data, rdkit.Chem.Mol):
+        sio = io.StringIO()
+        with rdkit.Chem.SDWriter(sio) as w:
+            w.write(molecule_data)
+        molecule_data = sio.getvalue()
 
     return {
         "data": molecule_data,
-        "format": "pdb"
-        if molecule_data.startswith("ATOM")
-        else "sdf"
-        if molecule_data.startswith("CTfile")
-        else "smiles",
+        "format": "pdb" if molecule_data.startswith("ATOM") else "sdf",
     }
 
 

lynxkite-core/src/lynxkite/core/ops.py

@@ -254,8 +254,8 @@ def op(
             _view = "image"
             func = matplotlib_to_image(func)
         if slow:
-            func = mem.cache(func)
             func = make_async(func)
+            func = mem.cache(func)
         # Positional arguments are inputs.
         inputs = [
             Input(name=name, type=param.annotation)