lionagi 0.0.315__py3-none-any.whl → 0.0.316__py3-none-any.whl

lionagi/core/direct/__init__.py

@@ -3,5 +3,8 @@ from .select import select
 from .score import score
 from .react import react
 from .vote import vote
+from .plan import plan
+from .cot import chain_of_thoughts, chain_of_react
 
-__all__ = ["predict", "select", "score", "vote", "react"]
+
+__all__ = ["predict", "select", "score", "vote", "react", "plan", "chain_of_thoughts", "chain_of_react"]

lionagi/core/direct/cot.py

@@ -1 +1,88 @@
-# TODO: chain of thoughts
+from typing import Callable
+from lionagi.libs import convert
+from ..tool import func_to_tool
+from ..schema import Tool
+from .predict import predict
+from .plan import plan
+from .react import react
+
+from .utils import _process_tools
+
+
+async def chain_of_thoughts(
+    sentence=None,
+    branch=None,
+    instruction=None,
+    reason=False,
+    confidence_score=False,
+    num_steps=3,
+    directive_kwargs={},
+    return_branch=False,
+    **kwargs
+):
+    
+    out_, outs, answer, reasons, confidence_score = "", [], "", [], 0
+    if branch is not None:
+        out_ = await plan(sentence, branch=branch, instruction=instruction, num_steps=num_steps, **kwargs)
+    else:
+        out_, branch = await plan(sentence, instruction=instruction, branch=branch, num_steps=num_steps, return_branch=True, **kwargs)
+    
+    for i in range(len(out_.plan)):
+        _out = await predict(branch=branch, instruction=out_.plan[f"step_{i+1}"], reason=reason, confidence_score=confidence_score, **directive_kwargs)
+        answer += _out.answer
+        if reason:
+            reasons.append(_out.reason)
+        if confidence_score:
+            confidence_score += _out.confidence_score
+        outs.append(_out)
+    
+    setattr(out_, "chain_output", outs)
+    setattr(out_, "chain_answer", answer)
+    
+    if reason:
+        setattr(out_, "chain_reasons", reasons)
+    if confidence_score:
+        setattr(out_, "chain_confidence_score", confidence_score/len(outs))
+    
+    if return_branch:
+        return out_, branch
+    
+    return out_
+    
+
+async def chain_of_react(
+    sentence=None,
+    branch=None,
+    instruction=None,
+    num_steps=3,
+    tools=None,
+    directive_system=None,
+    directive_kwargs={},
+    return_branch=False,
+    **kwargs
+):
+    out_, outs, reasons, actions, action_responses = "", [], [], [], []
+    if branch is not None:
+        out_ = await plan(sentence, branch=branch, instruction=instruction, num_steps=num_steps, **kwargs)
+    else:
+        out_, branch = await plan(sentence, instruction=instruction, branch=branch, num_steps=num_steps, return_branch=True, **kwargs)
+    
+    _process_tools(tools, branch)
+    
+    for i in range(len(out_.plan)):
+        _out = await react(branch=branch, system=directive_system, instruction=out_.plan[f"step_{i+1}"], **directive_kwargs)
+        outs.append(_out)
+        reasons.append(_out.reason)
+        actions.append(_out.actions)
+        if _out.action_needed:
+            action_responses.append(_out.action_response)
+    
+    setattr(out_, "chain_output", convert.to_list(outs))
+    setattr(out_, "chain_reason", convert.to_list(reasons))
+    setattr(out_, "chain_actions", convert.to_list(actions))
+    setattr(out_, "chain_action_response", convert.to_list(action_responses))
+
+    if return_branch:
+        return out_, branch
+    
+    return out_

lionagi/core/direct/plan.py

@@ -0,0 +1,162 @@
+# plan.py
+
+from lionagi.libs import func_call, ParseUtil
+from lionagi.integrations.bridge.pydantic_.pydantic_bridge import Field
+from ..prompt.scored_template import ScoredTemplate
+from ..branch import Branch
+
+
+class PlanTemplate(ScoredTemplate):
+    template_name: str = "default_plan"
+    sentence: str | list | dict = Field(
+        default_factory=str,
+        description="the given sentence(s) or context to generate a plan for",
+    )
+    plan: dict | str= Field(
+        default_factory=dict, description="the generated step by step plan, return as a dictionary following {step_n: {plan: ..., reason: ...}} format")
+    signature: str = "sentence -> plan"
+
+    def __init__(
+        self,
+        sentence=None,
+        instruction=None,
+        confidence_score=False,
+        reason=False,
+        num_step=3,
+        **kwargs,
+    ):
+        super().__init__(**kwargs)
+
+        self.sentence = sentence
+        self.task = f"Generate a {num_step}_step plan based on the given context. Instruction: {instruction}."
+
+        if reason:
+            self.output_fields.append("reason")
+
+        if confidence_score:
+            self.output_fields.append("confidence_score")
+
+
+async def _plan(
+    sentence,
+    *,
+    instruction=None,
+    branch=None,
+    confidence_score=False,
+    reason=False,
+    retries=2,
+    delay=0.5,
+    backoff_factor=2,
+    default_value=None,
+    timeout=None,
+    branch_name=None,
+    system=None,
+    messages=None,
+    service=None,
+    sender=None,
+    llmconfig=None,
+    tools=None,
+    datalogger=None,
+    persist_path=None,
+    tool_manager=None,
+    return_branch=False,
+    **kwargs,
+):
+    if "temperature" not in kwargs:
+        kwargs["temperature"] = 0.1
+
+    instruction = instruction or ""
+
+    branch = branch or Branch(
+        name=branch_name,
+        system=system,
+        messages=messages,
+        service=service,
+        sender=sender,
+        llmconfig=llmconfig,
+        tools=tools,
+        datalogger=datalogger,
+        persist_path=persist_path,
+        tool_manager=tool_manager,
+    )
+
+    _template = PlanTemplate(
+        sentence=sentence,
+        instruction=instruction,
+        confidence_score=confidence_score,
+        reason=reason,
+    )
+
+    await func_call.rcall(
+        branch.chat,
+        prompt_template=_template,
+        retries=retries,
+        delay=delay,
+        backoff_factor=backoff_factor,
+        default=default_value,
+        timeout=timeout,
+        **kwargs,
+    )
+    
+    _template.plan = ParseUtil.fuzzy_parse_json(_template.plan)
+
+    return (_template, branch) if return_branch else _template
+
+
+async def plan(
+    sentence,
+    *,
+    instruction=None,
+    num_instances=1,
+    branch=None,
+    confidence_score=False,
+    reason=False,
+    retries=2,
+    delay=0.5,
+    backoff_factor=2,
+    default_value=None,
+    timeout=None,
+    branch_name=None,
+    system=None,
+    messages=None,
+    service=None,
+    sender=None,
+    llmconfig=None,
+    tools=None,
+    datalogger=None,
+    persist_path=None,
+    tool_manager=None,
+    return_branch=False,
+    **kwargs,
+):
+    async def _inner(i=0):
+        return await _plan(
+            sentence=sentence,
+            instruction=instruction,
+            branch=branch,
+            confidence_score=confidence_score,
+            reason=reason,
+            retries=retries,
+            delay=delay,
+            backoff_factor=backoff_factor,
+            default_value=default_value,
+            timeout=timeout,
+            branch_name=branch_name,
+            system=system,
+            messages=messages,
+            service=service,
+            sender=sender,
+            llmconfig=llmconfig,
+            tools=tools,
+            datalogger=datalogger,
+            persist_path=persist_path,
+            tool_manager=tool_manager,
+            return_branch=return_branch,
+            **kwargs,
+        )
+
+    if num_instances == 1:
+        return await _inner()
+
+    elif num_instances > 1:
+        return await func_call.alcall(range(num_instances), _inner)

lionagi/core/direct/predict.py

@@ -43,14 +43,15 @@ class PredictTemplate(ScoredTemplate):
         default_factory=int, description="the number of sentences to predict"
     )
     answer: str | list = Field(
-        default_factory=str, description="the predicted sentence(s)"
+        default_factory=str, description="the predicted sentence(s) or desired output"
     )
     signature: str = "sentence -> answer"
 
     def __init__(
         self,
         sentence=None,
-        num_sentences=None,
+        instruction=None,
+        num_sentences=1,
         confidence_score=False,
         reason=False,
         **kwargs,
@@ -67,9 +68,9 @@ class PredictTemplate(ScoredTemplate):
         """
         super().__init__(**kwargs)
 
-        self.sentence = sentence
+        self.sentence = sentence or ''
         self.num_sentences = num_sentences
-        self.task = f"predict the next {self.num_sentences} sentence(s)"
+        self.task = f"follow instruction to predict the next {self.num_sentences} sentence(s). Instruction: {instruction}."
 
         if reason:
             self.output_fields.append("reason")
@@ -82,6 +83,8 @@ async def predict(
     sentence=None,
     num_sentences=1,
     confidence_score=False,
+    instruction=None,
+    branch=None,
     reason=False,
     retries=2,
     delay=0.5,
@@ -128,7 +131,7 @@ async def predict(
     Returns:
         PredictTemplate: The predict template with the predicted sentence(s).
     """
-    branch = Branch(
+    branch = branch or Branch(
         name=branch_name,
         system=system,
         messages=messages,
@@ -142,6 +145,7 @@ async def predict(
     )
 
     predict_template = PredictTemplate(
+        instruction=instruction,
         sentence=sentence,
         num_sentences=num_sentences,
         confidence_score=confidence_score,

lionagi/core/direct/react.py

@@ -3,11 +3,12 @@ from lionagi.libs import func_call, convert, AsyncUtil
 from lionagi.integrations.bridge.pydantic_.pydantic_bridge import Field
 from ..prompt.action_template import ActionedTemplate
 from ..branch import Branch
+from .utils import _process_tools
 
 
 class ReactTemplate(ActionedTemplate):
     template_name: str = "default_react"
-    sentence: str | list | dict = Field(
+    sentence: str | list | dict | None= Field(
         default_factory=str,
         description="the given sentence(s) to reason and take actions on",
     )
@@ -29,7 +30,7 @@ class ReactTemplate(ActionedTemplate):
 
 
 async def _react(
-    sentence,
+    sentence=None,
     *,
     instruction=None,
     branch=None,
@@ -58,6 +59,9 @@ async def _react(
 
     instruction = instruction or ""
 
+    if branch and tools:
+        _process_tools(tools, branch)
+        
     branch = branch or Branch(
         name=branch_name,
         system=system,
@@ -109,7 +113,7 @@ async def _react(
 
 
 async def react(
-    sentence,
+    sentence=None,
     *,
     instruction=None,
     num_instances=1,

lionagi/core/direct/utils.py

@@ -1,3 +1,6 @@
+from typing import Callable
+from ..tool import func_to_tool
+from ..schema import Tool
 # import contextlib
 # from lionagi.libs import ParseUtil, StringMatch, convert, func_call
 
@@ -85,3 +88,20 @@
 #         return _out
 
 #     return out_ if len(out_) > 1 else out_[0]
+
+
+def _process_tools(tool_obj, branch):
+    if isinstance(tool_obj, Callable):
+        _process_tool(tool_obj, branch)
+    else:
+        for i in tool_obj:
+            _process_tool(i, branch)
+
+
+def _process_tool(tool_obj, branch):
+    if isinstance(tool_obj, Tool) and tool_obj.schema_["function"]["name"] not in branch.tool_manager.registry:
+        branch.register_tools(tool_obj)
+    if isinstance(tool_obj, Callable):
+        tool = func_to_tool(tool_obj)[0]
+        if tool.schema_["function"]["name"] not in branch.tool_manager.registry:
+            branch.register_tools(tool)

lionagi/version.py

@@ -1 +1 @@
-__version__ = "0.0.315"
+__version__ = "0.0.316"

{lionagi-0.0.315.dist-info → lionagi-0.0.316.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lionagi
-Version: 0.0.315
+Version: 0.0.316
 Summary: Towards automated general intelligence.
 Author: HaiyangLi
 Author-email: Haiyang Li <ocean@lionagi.ai>

{lionagi-0.0.315.dist-info → lionagi-0.0.316.dist-info}/RECORD

@@ -1,5 +1,5 @@
 lionagi/__init__.py,sha256=i6Ci7FebU2s4EVVnBFj1Dsi5RvP80JqeSqW-iripRPg,418
-lionagi/version.py,sha256=Zazlk4sxt5cxFTrUeqVNVrVkGcIAkFTm-b9a6VLDqkw,24
+lionagi/version.py,sha256=Qy3TjUfD4-wZt6o0kQq4iMofgfH5V1ldDvANr5qAYN8,24
 lionagi/core/__init__.py,sha256=M5YXmJJiLcR5QB1VRmYvec14cHT6pKvxZOEs737BmP8,322
 lionagi/core/agent/__init__.py,sha256=IVcw9yn_QMBJGBou1Atck98Us9uwPGFs-gERTv0RWew,59
 lionagi/core/agent/base_agent.py,sha256=CRUpl7Zc5d2H9uCa17nMiFAnhKM_UH5Ujo1NHo3JAxg,3371
@@ -10,15 +10,15 @@ lionagi/core/branch/branch_flow_mixin.py,sha256=yXEfpxTaJ1aoDQQnCBYx5wShn9zt1ki8
 lionagi/core/branch/executable_branch.py,sha256=Yi0t4fDNMa5UaHo15sX-zBchr5auvXOtSc0RnSpG2a8,12151
 lionagi/core/branch/util.py,sha256=os7Qp7HpDfyyCvdkbBTyIQ3AYHfzUP0M684W4XMDHN4,11813
 lionagi/core/branch/base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lionagi/core/direct/__init__.py,sha256=P17UfY3tLFgu0ncxMy4FRoVDlvOGUc7jzeowN41akBk,188
-lionagi/core/direct/cot.py,sha256=3hz0CjFN2Bw5IW1tOh26fzd1UVrV_41KKIS7pzCd6ok,26
-lionagi/core/direct/plan.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lionagi/core/direct/predict.py,sha256=tkxvN9m_XOf3SW8xTi5yanXylV8xVNRn9a8DeGd9xgs,6355
-lionagi/core/direct/react.py,sha256=IJ6sKgajCjhQ_UpJHf-j71tnVehEtIXFnyeB6bNlZwk,4196
+lionagi/core/direct/__init__.py,sha256=nnA1v-pXDiypH9TvnhKW2igwHTYvnNFvu9l5PMupZwk,310
+lionagi/core/direct/cot.py,sha256=gfitm365i8m7Ov2YXWJyF0HbAceloLfUqEaj7Wkw45k,2819
+lionagi/core/direct/plan.py,sha256=1qK19Q63pGMcaByxNoxjQifg3WtpuyEtzcU8ptoZKfU,4022
+lionagi/core/direct/predict.py,sha256=xjO0o2OsibL_FEHg8aC70gUanqp514AJ6L0yQwQIwLs,6535
+lionagi/core/direct/react.py,sha256=rd2iXn6_VF5WBLOruQk9saUnafpo3qgu2Kc3pFvSpjU,4318
 lionagi/core/direct/score.py,sha256=QHO11WtAUfMEdfa1K-SRyn5uqf6_N0UmyCbEJsiqcQw,10328
 lionagi/core/direct/select.py,sha256=pPwesq29C3JZ5J3piwjBHqjOCsEM4uChPKMGBRxtSTE,6127
 lionagi/core/direct/sentiment.py,sha256=rNwBs-I2XICOwsXxFvfM1Tlc_afsVcRCNCXCxfxm_2k,27
-lionagi/core/direct/utils.py,sha256=yqu4qv9aaU4qzUD9QovtN2m21QySzdMLmcBp5recWC0,2333
+lionagi/core/direct/utils.py,sha256=Jszxx7QFIgC97YOwEhDStZC1g_9Zj-RhAxWaRqfeeJ8,3003
 lionagi/core/direct/vote.py,sha256=tjs-EYDGlGB3J6d_nSl1oIuJYHtxncjustBbU_pXDqQ,2449
 lionagi/core/flow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/core/flow/base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -114,8 +114,8 @@ lionagi/tests/test_libs/test_func_call.py,sha256=xvs19YBNxqh3RbWLjQXY19L06b1_uZY
 lionagi/tests/test_libs/test_nested.py,sha256=eEcE4BXJEkjoPZsd9-0rUxOJHjmu8W2hgVClUTwXEFY,13106
 lionagi/tests/test_libs/test_parse.py,sha256=aa74kfOoJwDU7L7-59EcgBGYc5-OtafPIP2oGTI3Zrk,6814
 lionagi/tests/test_libs/test_sys_util.py,sha256=Y-9jxLGxgbFNp78Z0PJyGUjRROMuRAG3Vo3i5LAH8Hs,7849
-lionagi-0.0.315.dist-info/LICENSE,sha256=vfczrx-xFNkybZ7Ef-lGUnA1Vorky6wL4kwb1Fd5o3I,1089
-lionagi-0.0.315.dist-info/METADATA,sha256=FVnSivifINUlYoYjEh7s01WKZ3h1Hn1AW_uKW3KfdLg,7934
-lionagi-0.0.315.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-lionagi-0.0.315.dist-info/top_level.txt,sha256=szvch_d2jE1Lu9ZIKsl26Ll6BGfYfbOgt5lm-UpFSo4,8
-lionagi-0.0.315.dist-info/RECORD,,
+lionagi-0.0.316.dist-info/LICENSE,sha256=vfczrx-xFNkybZ7Ef-lGUnA1Vorky6wL4kwb1Fd5o3I,1089
+lionagi-0.0.316.dist-info/METADATA,sha256=StvgG7rD-1_VdC1-ftaT_DKrO4VZP8QvRl29xPnnIAM,7934
+lionagi-0.0.316.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+lionagi-0.0.316.dist-info/top_level.txt,sha256=szvch_d2jE1Lu9ZIKsl26Ll6BGfYfbOgt5lm-UpFSo4,8
+lionagi-0.0.316.dist-info/RECORD,,