convenient array to base64 function

Author: TLSDC

src/agentlab/llm/llm_utils.py

@@ -381,6 +381,17 @@ def image_to_jpg_base64_url(image: np.ndarray | Image.Image):
     return f"data:image/jpeg;base64,{image_base64}"
 
 
+def image_to_png_base64_url(image: np.ndarray | Image.Image):
+    if isinstance(image, np.ndarray):
+        image = Image.fromarray(image)
+    if image.mode in ("RGBA", "LA"):
+        image = image.convert("RGB")
+    buffered = io.BytesIO()
+    image.save(buffered, "PNG")
+    image_base64 = base64.b64encode(buffered.getvalue()).decode()
+    return f"data:image/png;base64,{image_base64}"
+
+
 class BaseMessage(dict):
     def __init__(self, role: str, content: Union[str, list[dict]], **kwargs):
         allowed_attrs = {"log_probs"}