move the utilities from __init__ to a new file

textgrad/engine/__init__.py

@@ -51,21 +51,3 @@ def get_engine(engine_name: str, **kwargs) -> EngineLM:
         return ChatCohere(model_string=engine_name, **kwargs)
     else:
         raise ValueError(f"Engine {engine_name} not supported")
-
-
-def is_jpeg(data):
-    jpeg_signature = b'\xFF\xD8\xFF'
-    return data.startswith(jpeg_signature)
-
-def is_png(data):
-    png_signature = b'\x89\x50\x4E\x47\x0D\x0A\x1A\x0A'
-    return data.startswith(png_signature)
-
-
-def get_image_type_from_bytes(data):
-    if is_jpeg(data):
-        return "jpeg"
-    elif is_png(data):
-        return "png"
-    else:
-        raise ValueError("Image type not supported, only jpeg and png supported.")

textgrad/engine/anthropic.py

@@ -14,7 +14,7 @@
 import json
 from typing import List, Union
 from .base import EngineLM, CachedEngine
-from textgrad.engine import get_image_type_from_bytes
+from .engine_utils import get_image_type_from_bytes
 
 class ChatAnthropic(EngineLM, CachedEngine):
     SYSTEM_PROMPT = "You are a helpful, creative, and smart assistant."

textgrad/engine/engine_utils.py

@@ -0,0 +1,16 @@
+def is_jpeg(data):
+    jpeg_signature = b'\xFF\xD8\xFF'
+    return data.startswith(jpeg_signature)
+
+def is_png(data):
+    png_signature = b'\x89\x50\x4E\x47\x0D\x0A\x1A\x0A'
+    return data.startswith(png_signature)
+
+
+def get_image_type_from_bytes(data):
+    if is_jpeg(data):
+        return "jpeg"
+    elif is_png(data):
+        return "png"
+    else:
+        raise ValueError("Image type not supported, only jpeg and png supported.")

textgrad/engine/openai.py

@@ -13,9 +13,10 @@
     wait_random_exponential,
 )
 from typing import List, Union
-from textgrad.engine import get_image_type_from_bytes
 
 from .base import EngineLM, CachedEngine
+from .engine_utils import get_image_type_from_bytes
+
 
 class ChatOpenAI(EngineLM, CachedEngine):
     DEFAULT_SYSTEM_PROMPT = "You are a helpful, creative, and smart assistant."