feat: MelleaSession.register for functional interface and MelleaSession.powerup for dynamic mixin (register all methods in a class)

mellea/stdlib/genslot.py

@@ -182,7 +182,7 @@ def __call__(
 
         response_model = create_response_format(self._function._func)
 
-        response = m.act(slot_copy, format=response_model, model_options=model_options)
+        response = m.act(slot_copy, format=response_model, model_options=model_options)  # type:ignore
 
         function_response: FunctionResponse[R] = response_model.model_validate_json(
             response.value  # type: ignore
@@ -243,7 +243,7 @@ def __call__(
         # they must return a coroutine object.
         async def __async_call__() -> R:
             # Use the async act func so that control flow doesn't get stuck here in async event loops.
-            response = await m.aact(
+            response = await m.aact(  # type: ignore
                 slot_copy, format=response_model, model_options=model_options
             )
 

mellea/stdlib/sampling/base.py

@@ -5,9 +5,9 @@
 
 import tqdm
 
+import mellea.stdlib.functional as mfuncs
 from mellea.backends import Backend, BaseModelSubclass
 from mellea.helpers.fancy_logger import FancyLogger
-from mellea.stdlib import funcs as mfuncs
 from mellea.stdlib.base import CBlock, ChatContext, Component, Context, ModelOutputThunk
 from mellea.stdlib.chat import Message
 from mellea.stdlib.instruction import Instruction

mellea/stdlib/sampling/best_of_n.py

@@ -4,10 +4,10 @@
 
 import tqdm
 
+import mellea.stdlib.functional as mfuncs
 from mellea.backends import Backend, BaseModelSubclass
 from mellea.helpers.async_helpers import wait_for_all_mots
 from mellea.helpers.fancy_logger import FancyLogger
-from mellea.stdlib import funcs as mfuncs
 from mellea.stdlib.base import CBlock, ChatContext, Component, Context, ModelOutputThunk
 from mellea.stdlib.instruction import Instruction
 from mellea.stdlib.requirement import Requirement, ScorerRequirement, ValidationResult