updated notebooks

lionagi/core/direct/__init__.py

@@ -1,9 +1,11 @@
 from .predict import predict
 from .select import select
 from .score import score
+from .vote import vote
 
 __all__ = [
     "predict",
     "select",
     "score",
+    "vote"
 ]

lionagi/core/direct/select.py

@@ -17,7 +17,6 @@ def __init__(
         self,
         sentence=None,
         choices=None,
-        num_choices=1,
         instruction=None,
         reason=False, 
         confidence_score=False,
@@ -27,7 +26,7 @@ def __init__(
 
         self.sentence = sentence
         self.choices = choices
-        self.task = f"select {num_choices} item(s), from provided choices {choices}."
+        self.task = f"select 1 item, from provided choices {choices}."
         if instruction:
             self.task += f"objetive {instruction}."
 
@@ -41,7 +40,6 @@ def __init__(
 async def select(
     sentence,
     choices=None,
-    num_choices=1,
     instruction=None,
     confidence_score=False,
     reason=False,
@@ -79,7 +77,6 @@ async def select(
     _template = SelectTemplate(
         sentence=sentence,
         choices=choices,
-        num_choices=num_choices,
         instruction=instruction,
         confidence_score=confidence_score,
         reason=reason,
@@ -97,6 +94,7 @@ async def select(
     )
 
     ans = _template.answer
+
     if ans not in _template.choices:
         _template.answer = StringMatch.choose_most_similar(ans, _template.choices)
 

lionagi/core/direct/vote.py

@@ -1,10 +1,11 @@
 from lionagi.libs import func_call
 import numpy as np
+from .predict import predict
 from .score import score
 
 # for example, directive=predict
 
-async def vote(sentence, directive, num_generations=5, num_output=1, num_scorer=5, score_range=(0,100), num_digit=2, scorer_instruction=None, **kwargs):
+async def vote(sentence, directive=predict, num_generations=5, num_output=1, num_scorer=5, score_range=(0,100), num_digit=2, scorer_instruction=None, **kwargs):
 
     async def _inner(i):
         out_ = await directive(sentence, **kwargs)