diff --git a/libs/python/agent/agent/loops/__init__.py b/libs/python/agent/agent/loops/__init__.py
index ab23ac27..5006c102 100644
--- a/libs/python/agent/agent/loops/__init__.py
+++ b/libs/python/agent/agent/loops/__init__.py
@@ -17,6 +17,7 @@ from . import (
     opencua,
     qwen,
     uitars,
+    gelato,
 )
 
 __all__ = [
@@ -33,4 +34,5 @@ __all__ = [
     "moondream3",
     "gemini",
     "qwen",
+    "gelato",
 ]