gatewaysdk 0.3.2__py3-none-any.whl

gatewaysdk/algorithm/__init__.py

@@ -0,0 +1,39 @@
+# Copyright (c) Microsoft. All rights reserved.
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
+from .base import Algorithm
+from .decorator import algo
+from .fast import Baseline, FastAlgorithm
+
+if TYPE_CHECKING:
+    from .apo import APO as APOType
+    from .gepa import GEPA as GEPAType
+    from .verl import VERL as VERLType
+
+__all__ = ["Algorithm", "algo", "FastAlgorithm", "Baseline", "APO", "GEPA", "VERL", "GEPAAdapter"]
+
+# Re-export GEPAAdapter from gepa submodule
+from .gepa import GEPAAdapter
+
+# Shortcuts for usages like algo.APO(...)
+
+
+def APO(*args: Any, **kwargs: Any) -> APOType[Any]:
+    from .apo import APO as APOImplementation
+
+    return APOImplementation(*args, **kwargs)
+
+
+def GEPA(*args: Any, **kwargs: Any) -> GEPAType[Any]:
+    from .gepa import GEPA as GEPAImplementation
+
+    return GEPAImplementation(*args, **kwargs)
+
+
+def VERL(*args: Any, **kwargs: Any) -> VERLType:
+    from .verl import VERL as VERLImplementation
+
+    return VERLImplementation(*args, **kwargs)

gatewaysdk/algorithm/apo/__init__.py

@@ -0,0 +1,5 @@
+# Copyright (c) Microsoft. All rights reserved.
+
+from .apo import APO
+
+__all__ = ["APO"]