remove m.genslot in favor of m.act

Author: jakelorocco

mellea/stdlib/genslot.py

@@ -9,7 +9,7 @@
 from pydantic import BaseModel, Field, create_model
 
 from mellea.stdlib.base import Component, TemplateRepresentation
-from mellea.stdlib.session import get_session
+from mellea.stdlib.session import MelleaSession, get_session
 
 P = ParamSpec("P")
 R = TypeVar("R")
@@ -154,7 +154,7 @@ def __init__(self, func: Callable[P, R]):
 
     def __call__(
         self,
-        m=None,
+        m: MelleaSession | None = None,
         model_options: dict | None = None,
         *args: P.args,
         **kwargs: P.kwargs,
@@ -180,13 +180,11 @@ def __call__(
 
         response_model = create_response_format(self._function._func)
 
-        response = m.genslot(
-            slot_copy, model_options=model_options, format=response_model
-        )
+        response = m.act(slot_copy, format=response_model, model_options=model_options)
 
         function_response: FunctionResponse[R] = response_model.model_validate_json(
-            response.value
-        )  # type: ignore
+            response.value  # type: ignore
+        )
 
         return function_response.result
 

mellea/stdlib/session.py

@@ -525,26 +525,6 @@ def load_default_aloras(self):
             "This model/backend combination does not support any aloras."
         )
 
-    def genslot(
-        self,
-        gen_slot: Component,
-        format: type[BaseModelSubclass] | None = None,
-        model_options: dict | None = None,
-        tool_calls: bool = False,
-    ) -> ModelOutputThunk:
-        """Call generative Slot on a GenerativeSlot Component.
-
-        Args:
-            gen_slot (GenerativeSlot Component): A generative slot
-
-        Returns:
-            ModelOutputThunk: Output thunk
-        """
-        result = self.act(
-            gen_slot, format=format, model_options=model_options, tool_calls=tool_calls
-        )
-        return result
-
     def query(
         self,
         obj: Any,