lionagi 0.8.2__py3-none-any.whl → 0.8.4__py3-none-any.whl

lionagi/operations/ReAct/ReAct.py

@@ -11,6 +11,8 @@ from lionagi.operatives.types import Instruct
 from lionagi.service.imodel import iModel
 from lionagi.utils import copy
 
+from .utils import ReActAnalysis
+
 if TYPE_CHECKING:
     from lionagi.session.branch import Branch
 
@@ -60,11 +62,9 @@ async def ReAct(
     kwargs_for_operate["actions"] = True
     kwargs_for_operate["reason"] = True
 
-    # We'll pass the refined instruct_dict plus the user's other kwargs
-    from .utils import ReActAnalysis
-
     # Step 1: Generate initial ReAct analysis
     analysis: ReActAnalysis = await branch.operate(
+        instruct=instruct_dict,
         response_format=ReActAnalysis,
         tools=tools,
         tool_schemas=tool_schemas,
@@ -83,7 +83,7 @@ async def ReAct(
     while (
         extension_allowed
         and analysis.extension_needed
-        and (extensions if extensions else 1) > 0
+        and (extensions if max_extensions else 0) > 0
     ):
         new_instruction = None
         if extensions == max_extensions:
@@ -95,17 +95,13 @@ async def ReAct(
                 extensions=extensions
             )
 
-        # Each expansion uses a fresh copy of instruct_dict + forcibly "reason" + "actions"
-        expanded_kwargs = copy(instruct_dict)
-        expanded_kwargs["instruction"] = new_instruction
-        expanded_kwargs["reason"] = True
-        expanded_kwargs["actions"] = True
-
         analysis = await branch.operate(
+            instruction=new_instruction,
             response_format=ReActAnalysis,
             tools=tools,
             tool_schemas=tool_schemas,
-            **expanded_kwargs,
+            reason=True,
+            actions=True,
         )
         analyses.append(analysis)
 

lionagi/operations/ReAct/utils.py

@@ -10,19 +10,19 @@ from pydantic import BaseModel, Field
 class ReActAnalysis(BaseModel):
 
     FIRST_EXT_PROMPT: ClassVar[str] = (
-        "You are provided with another round to perform reason action to provide an accurate final answer. you have max another {extensions} rounds, set extension_needed to False if you are done and ready to provide final answer."
+        "You are provided with additional rounds to perform reason action to provide an accurate final answer. you have max another {extensions} rounds. Pleasen continue."
     )
 
     CONTINUE_EXT_PROMPT: ClassVar[str] = (
-        "You are provided with another round, you have max another {extensions} rounds"
+        "You are provided with another round, you have max another {extensions} rounds. Please continue."
     )
 
     ANSWER_PROMPT: ClassVar[str] = (
-        "given above reason and actions, please provide final answer to the original user request {instruction}"
+        "given above reason and actions, please provide final answer to the original user request:\n\n {instruction}"
     )
 
     analysis: str
     extension_needed: bool = Field(
         False,
-        description="Set to True if more steps are needed to provide an accurate answer. If True, additional rounds are allowed.",
+        description="Set to True if more steps are needed to provide an accurate answer. If True, additional rounds are allowed. Typically should be set to true if more actions should be taken or planned to be taken. If false, will proceed to provide final answer next.",
     )

lionagi/operatives/action/function_calling.py

@@ -5,7 +5,7 @@
 import asyncio
 from typing import Any
 
-from pydantic import Field, model_validator
+from pydantic import BaseModel, Field, field_validator, model_validator
 from typing_extensions import Self
 
 from lionagi.protocols.generic.event import Event, EventStatus
@@ -27,12 +27,22 @@ class FunctionCalling(Event):
         exclude=True,
     )
 
-    arguments: dict[str, Any] = Field(
+    arguments: dict[str, Any] | BaseModel = Field(
         ..., description="Dictionary of arguments to pass to the function"
     )
 
+    @field_validator("arguments", mode="before")
+    def _validate_argument(cls, value):
+        if isinstance(value, BaseModel):
+            return value.model_dump(exclude_unset=True)
+        return value
+
     @model_validator(mode="after")
     def _validate_strict_tool(self) -> Self:
+        if self.func_tool.request_options:
+            args: BaseModel = self.func_tool.request_options(**self.arguments)
+            self.arguments = args.model_dump(exclude_unset=True)
+
         if self.func_tool.strict_func_call is True:
             if (
                 not set(self.arguments.keys())

lionagi/operatives/action/tool.py

@@ -49,6 +49,11 @@ class Tool(Element):
         description="Schema describing the function's parameters and structure",
     )
 
+    request_options: type | None = Field(
+        default=None,
+        description="Optional Pydantic model for validating the function's input",
+    )
+
     preprocessor: Callable[[Any], Any] | None = Field(
         default=None,
         description="Optional function for preprocessing inputs before execution",
@@ -88,6 +93,11 @@ class Tool(Element):
     def _validate_tool_schema(self) -> Self:
         if self.tool_schema is None:
             self.tool_schema = function_to_schema(self.func_callable)
+        if self.request_options is not None:
+            schema_ = self.request_options.model_json_schema()
+            schema_.pop("title", None)
+            self.tool_schema["function"]["parameters"] = schema_
+
         return self
 
     @property

lionagi/operatives/forms/flow.py

@@ -1,5 +1,4 @@
 # forms/flow.py
-from typing import List
 
 from pydantic import BaseModel, ConfigDict, Field
 

lionagi/operatives/forms/form.py

@@ -1,6 +1,6 @@
 # forms/form.py
 
-from typing import Any, Optional
+from typing import Any
 
 from pydantic import ConfigDict, Field, model_validator
 from typing_extensions import Self

lionagi/protocols/messages/instruction.py

@@ -81,68 +81,121 @@ def format_text_item(item: Any) -> str:
 
 def format_text_content(content: dict) -> str:
     """
-    Convert a dictionary with keys like 'guidance', 'instruction', 'context', etc.
-    into a readable text block.
+    Convert a content dictionary into a minimal textual summary for LLM consumption.
 
-    Args:
-        content (dict): The content dictionary.
-
-    Returns:
-        str: A textual summary.
+    Emphasizes brevity and clarity:
+      - Skips empty or None fields.
+      - Bullet-points for lists.
+      - Key-value pairs for dicts.
+      - Minimal headings for known fields (guidance, instruction, etc.).
     """
-    if "plain_content" in content and isinstance(
-        content["plain_content"], str
-    ):
-        return content["plain_content"]
 
-    msg = "\n---\n # Task\n"
+    if isinstance(content.get("plain_content"), str):
+        return content["plain_content"]
 
-    for k in [
+    lines = []
+    # We only want minimal headings for certain known fields:
+    known_field_order = [
         "guidance",
         "instruction",
         "context",
         "tool_schemas",
         "respond_schema_info",
         "request_response_format",
-    ]:
-        if k in content:
-            v = content[k]
-
-            if k == "tool_schemas":
-                if "tools" in v:
-                    v = v["tools"]
-
-                if isinstance(v, list):
-                    z = []
-                    for idx, z_ in enumerate(v):
-                        if isinstance(z_, dict) and "function" in z_:
-                            z.append({f"Tool {idx+1}": z_["function"]})
-                    v = z
-
-            if k == "request_response_format":
-                k = "response format"
-
-            if v not in [None, [], {}, UNDEFINED]:
-                if isinstance(v, list):
-                    msg += f"## - **{k}**\n"
-                    for i in v:
-                        if (
-                            len(format_text_item(v).replace("\n", "").strip())
-                            > 0
-                        ):
-                            msg += format_text_item(i).strip()
-                    msg += "\n"
-                else:
-                    if len(format_text_item(v).replace("\n", "").strip()) > 0:
-                        msg += (
-                            f"## - **{k}**\n{format_text_item(v).strip()}\n\n"
-                        )
-
-    if not msg.endswith("\n\n"):
-        msg += "\n\n---\n"
+    ]
+
+    # Render known fields in that order
+    for field in known_field_order:
+        if field in content:
+            val = content[field]
+            if _is_not_empty(val):
+                if field == "request_response_format":
+                    field = "response format"
+                elif field == "respond_schema_info":
+                    field = "response schema info"
+                lines.append(f"\n## {field.upper()}:\n")
+                rendered = _render_value(val)
+                # Indent or bullet the rendered result if multiline
+                # We'll keep it minimal: each line is prefixed with "  ".
+                lines.extend(
+                    f"  {line}"
+                    for line in rendered.split("\n")
+                    if line.strip()
+                )
+
+    # Join all lines into a single string
+    return "\n".join(lines).strip()
+
+
+def _render_value(val) -> str:
+    """
+    Render an arbitrary value (scalar, list, dict) in minimal form:
+    - Lists become bullet points.
+    - Dicts become key-value lines.
+    - Strings returned directly.
+    """
+    if isinstance(val, dict):
+        return _render_dict(val)
+    elif isinstance(val, list):
+        return _render_list(val)
     else:
-        msg += "---\n"
-    return msg
+        return str(val).strip()
+
+
+def _render_dict(dct: dict) -> str:
+    """
+    Minimal bullet list for dictionary items:
+      key: rendered subvalue
+    """
+    lines = []
+    for k, v in dct.items():
+        if not _is_not_empty(v):
+            continue
+        subrendered = _render_value(v)
+        # Indent subrendered if multiline
+        sublines = subrendered.split("\n")
+        if len(sublines) == 1:
+            if sublines[0].startswith("- "):
+                lines.append(f"- {k}: {sublines[0][2:]}")
+            else:
+                lines.append(f"- {k}: {sublines[0]}")
+        else:
+            lines.append(f"- {k}:")
+            for s in sublines:
+                lines.append(f"  {s}")
+    return "\n".join(lines)
+
+
+def _render_list(lst: list) -> str:
+    """
+    Each item in the list gets a bullet. Nested structures are recursed.
+    """
+    lines = []
+    for idx, item in enumerate(lst, 1):
+        sub = _render_value(item)
+        sublines = sub.split("\n")
+        if len(sublines) == 1:
+            if sublines[0].startswith("- "):
+                lines.append(f"- {sublines[0][2:]}")
+            else:
+                lines.append(f"- {sublines[0]}")
+        else:
+            lines.append("-")
+            lines.extend(f"  {s}" for s in sublines)
+    return "\n".join(lines)
+
+
+def _is_not_empty(x) -> bool:
+    """
+    Returns True if x is neither None, nor empty string/list/dict.
+    """
+    if x is None:
+        return False
+    if isinstance(x, (list, dict)) and not x:
+        return False
+    if isinstance(x, str) and not x.strip():
+        return False
+    return True
 
 
 def format_image_content(

lionagi/session/branch.py

@@ -345,6 +345,20 @@ class Branch(Element, Communicatable, Relational):
 
         return branch_clone
 
+    def register_tools(
+        self, tools: FuncTool | list[FuncTool], update: bool = False
+    ):
+        """
+        Registers one or more tools in the ActionManager.
+
+        Args:
+            tools (FuncTool | list[FuncTool]):
+                A single tool or a list of tools to register.
+            update (bool, optional):
+                If `True`, updates existing tools with the same name.
+        """
+        self._action_manager.register_tools(tools, update=update)
+
     # -------------------------------------------------------------------------
     # Conversion / Serialization
     # -------------------------------------------------------------------------
@@ -532,6 +546,33 @@ class Branch(Element, Communicatable, Relational):
         for key in self.mailbox.pending_ins:
             self.receive(key)
 
+    def connect(
+        self,
+        name: str,
+        imodel: iModel,
+        request_options: type[BaseModel],
+        description: str = None,
+        update: bool = False,
+    ):
+        if not update and name in self.tools:
+            raise ValueError(f"Tool with name '{name}' already exists.")
+
+        async def _connect(**kwargs):
+            """connect to an api endpoint"""
+            api_call = await imodel.invoke(**kwargs)
+            self._log_manager.log(Log.create(api_call))
+            return api_call.response
+
+        _connect.__name__ = name
+        if description:
+            _connect.__doc__ = description
+
+        tool = Tool(
+            func_callable=_connect,
+            request_options=request_options,
+        )
+        self._action_manager.register_tools(tool, update=update)
+
     # -------------------------------------------------------------------------
     # Dictionary Conversion
     # -------------------------------------------------------------------------

lionagi/version.py

@@ -1 +1 @@
-__version__ = "0.8.2"
+__version__ = "0.8.4"

{lionagi-0.8.2.dist-info → lionagi-0.8.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lionagi
-Version: 0.8.2
+Version: 0.8.4
 Summary: An Intelligence Operating System.
 Author-email: HaiyangLi <quantocean.li@gmail.com>
 License:            Apache License

{lionagi-0.8.2.dist-info → lionagi-0.8.4.dist-info}/RECORD

@@ -4,7 +4,7 @@ lionagi/_errors.py,sha256=wNKdnVQvE_CHEstK7htrrj334RA_vbGcIds-3pUiRkc,455
 lionagi/_types.py,sha256=9g7iytvSj3UjZxD-jL06_fxuNfgZyWT3Qnp0XYp1wQU,63
 lionagi/settings.py,sha256=k9zRJXv57TveyfHO3Vr9VGiKrSwlRUUVKt5zf6v9RU4,1627
 lionagi/utils.py,sha256=QbF4E1PG-BaRcEVH3kJIYCJVNq-oRNoTxjda5k8NYW4,73177
-lionagi/version.py,sha256=B7GiO0rd49YwtLYjvPg4lmCZEDlMTonslQKdSImaMJk,22
+lionagi/version.py,sha256=jhHEJFZWhkQDemoZMomBYq-RNrKXknYzUaeIU9A6XsI,22
 lionagi/libs/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/libs/parse.py,sha256=tpEbmIRGuHhLCJlUlm6fjmqm_Z6XJLAXGNFHNuk422I,1011
 lionagi/libs/file/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
@@ -48,9 +48,9 @@ lionagi/operations/__init__.py,sha256=O7nV0tedpUe7_OlUWmCcduGPFtqtzWZcR_SIOnjLsr
 lionagi/operations/manager.py,sha256=H7UY86PIxvxKdzJY9YVsWyJcqlwLWhVyvm4sYePH_uY,565
 lionagi/operations/types.py,sha256=LIa68xcyKLVafof-DSFwKtSkneuYPFqrtGyClohYI6o,704
 lionagi/operations/utils.py,sha256=Twy6L_UFt9JqJFRYuKKTKVZIXsePidNl5ipcYcCbesI,1220
-lionagi/operations/ReAct/ReAct.py,sha256=tAZ-3Ya68tVUa112wgOMUJpBVw-RWBSYTfgicbInRuQ,3954
+lionagi/operations/ReAct/ReAct.py,sha256=07F_VhxNnOhOInEZkbY9nT2YiCcnApSibxjz2yAa7Wk,3696
 lionagi/operations/ReAct/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
-lionagi/operations/ReAct/utils.py,sha256=0OhZhoc8QsTsMFo2Jmys1mgpnMwHsldKuLVSbev9uZM,994
+lionagi/operations/ReAct/utils.py,sha256=yBsbaZm3NNb-LhdjdK3jVmxSYbp0enWzl8d09iv8oSo,1099
 lionagi/operations/_act/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/operations/_act/act.py,sha256=HBp-sNwNigLDNkuEZqGU_98UCaJXPZsaokkFAXwOMn0,2454
 lionagi/operations/brainstorm/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
@@ -82,15 +82,15 @@ lionagi/operatives/operative.py,sha256=Pahhjlav-Y1rCYwUZkWNrw6nIOpR9zrqXwuHTYcbj
 lionagi/operatives/step.py,sha256=DevwisZc2q88ynUiiSu7VBEY2A_G4Q5iRLrVgVLHNJU,9843
 lionagi/operatives/types.py,sha256=X0Pz0aqVg20D3lfnKOZ0GtLj01M5WoYagVmu_aArkoo,1784
 lionagi/operatives/action/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
-lionagi/operatives/action/function_calling.py,sha256=dHWxnWj-YD2PkYoRO_iOI1dryb8tThuPDX1OMI0y8GM,4631
+lionagi/operatives/action/function_calling.py,sha256=dFNldEuV77quaM2M4VGXPQRhh3yGCAA5WucyipRu13M,5067
 lionagi/operatives/action/manager.py,sha256=FXfWhQMSFE5qJNZPpgdz4ePvthLafZfnmak2PImCrsc,8824
 lionagi/operatives/action/request_response_model.py,sha256=mCyKub_WoEttJ_mqLhGoOoPVBQHOhr7sswy_jN6-620,3378
-lionagi/operatives/action/tool.py,sha256=CVCNd154XDxRvinmfO_2y72RsCn3UfZEecY7QG2qIjE,5217
+lionagi/operatives/action/tool.py,sha256=39Z2beUUdG0ngoK1r2ynmX4yIiDIyHv_dxb4X53In24,5584
 lionagi/operatives/action/utils.py,sha256=vUe7Aysuzbg16rAfe2Ttp5QUz5_L6mMedBVAWzGAHwk,4330
 lionagi/operatives/forms/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/operatives/forms/base.py,sha256=hitr0eKk7Yts2VfpuBGl0YxpClMJEpsy7xWodhg2AhQ,2720
-lionagi/operatives/forms/flow.py,sha256=JRHY_t8yZNN2p_spPvoFG1hD--2YVtAiA8h3hrK7Vt0,2556
-lionagi/operatives/forms/form.py,sha256=GJTE_PrlHT_bgWf-zOQaIFdEkx1CUDTzgAyRofPXoOE,2934
+lionagi/operatives/forms/flow.py,sha256=Em0AIiSQT5r45wKjoq57J_wuDZ9fecKZxf6FzmmamQM,2532
+lionagi/operatives/forms/form.py,sha256=j5Mf9DCblJYtN_uCgBhXzE4IkrZHWc7jyzog3tXqeW4,2924
 lionagi/operatives/forms/report.py,sha256=ZQsM5nDnmF3WdCSurDUBTMI-PEy8yTnRA_uN6tyRneA,1488
 lionagi/operatives/instruct/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/operatives/instruct/base.py,sha256=Vw9u3BGzjNRGcPfW1ROr2uCZcZsjrm4WU6491P6ecCI,1895
@@ -149,7 +149,7 @@ lionagi/protocols/messages/action_request.py,sha256=Nsh6Nm6M1BvBOw3TzI5DvSuwH5ld
 lionagi/protocols/messages/action_response.py,sha256=v5ykwspVEBXYxgFCIgmhdvk8lR3ycxxQdbjwdVLK_w8,5401
 lionagi/protocols/messages/assistant_response.py,sha256=rWQsvElcUdF9KHkDLTNg_o-uzNl5DTcK066lszvCI0k,6412
 lionagi/protocols/messages/base.py,sha256=YcOtj-okN5ldbviPsulIDBAFaGRvqpmt7DTfgLYjpwA,2482
-lionagi/protocols/messages/instruction.py,sha256=tiznMvDAkuGzsUrh5hkCGKPmxOjZLfr8_zC4Rwzj7J4,19603
+lionagi/protocols/messages/instruction.py,sha256=VvOJ4hpZIngWy1lZM4wrzfYPn8YpXw9wkFTvRfa_XZk,21234
 lionagi/protocols/messages/manager.py,sha256=QpOIbQ6zVaBzhh7KkMvXdujB05vJK4waSaos14WwG_Q,17370
 lionagi/protocols/messages/message.py,sha256=ulTjKK4FSWTLFRbMbg8V8cEveKu_fhFnOaAyuVWCXbo,7864
 lionagi/protocols/messages/system.py,sha256=MX9bR76TbrY3a-d3akocDX3cFSYb9OD6lwP4HqvbFLg,4799
@@ -187,9 +187,9 @@ lionagi/service/providers/openrouter_/chat_completions.py,sha256=MRf4ZbMCgzNIL4g
 lionagi/service/providers/perplexity_/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/service/providers/perplexity_/chat_completions.py,sha256=SsDbrtXwQsR4Yu2VMU43KfeS86QWI8UTNhDth5lNWNs,1055
 lionagi/session/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
-lionagi/session/branch.py,sha256=JvErd9YUuGdWyLj37rtKOteSqV0ltn9lg0R2G8GO40c,62539
+lionagi/session/branch.py,sha256=ZLItnjW66__85s0brxhq1LZisjDoVS4HBJSHUpFaxsg,63834
 lionagi/session/session.py,sha256=po6C7PnM0iu_ISHUo4PBzzQ61HFOgcsAUfPoO--eLak,8987
-lionagi-0.8.2.dist-info/METADATA,sha256=lJ_wNrsDsRFOVYHqCc9GbFWe5bFyusOF7-nZW_cMIL4,22819
-lionagi-0.8.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lionagi-0.8.2.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
-lionagi-0.8.2.dist-info/RECORD,,
+lionagi-0.8.4.dist-info/METADATA,sha256=MHgfVuup5176MSrXlfX2B_tQUaiKNdGAjtITrJVLCWY,22819
+lionagi-0.8.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lionagi-0.8.4.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
+lionagi-0.8.4.dist-info/RECORD,,