lionagi 0.9.4__py3-none-any.whl → 0.9.5__py3-none-any.whl

lionagi/operations/ReAct/ReAct.py

@@ -47,6 +47,8 @@ async def ReAct(
     analysis_model: iModel | None = None,
     verbose_analysis: bool = False,
     verbose_length: int = None,
+    include_token_usage_to_model: bool = True,
+    continue_after_failed_response: bool = False,
     **kwargs,
 ):
     outs = []
@@ -73,6 +75,8 @@ async def ReAct(
             verbose_analysis=verbose_analysis,
             display_as=display_as,
             verbose_length=verbose_length,
+            include_token_usage_to_model=include_token_usage_to_model,
+            continue_after_failed_response=continue_after_failed_response,
             **kwargs,
         ):
             analysis, str_ = i
@@ -101,6 +105,8 @@ async def ReAct(
             analysis_model=analysis_model,
             display_as=display_as,
             verbose_length=verbose_length,
+            include_token_usage_to_model=include_token_usage_to_model,
+            continue_after_failed_response=continue_after_failed_response,
             **kwargs,
         ):
             outs.append(i)
@@ -131,6 +137,8 @@ async def ReActStream(
     verbose_analysis: bool = False,
     display_as: Literal["json", "yaml"] = "yaml",
     verbose_length: int = None,
+    include_token_usage_to_model: bool = True,
+    continue_after_failed_response: bool = False,
     **kwargs,
 ) -> AsyncGenerator:
     irfm: FieldModel | None = None
@@ -213,6 +221,9 @@ async def ReActStream(
     kwargs_for_operate = copy(kwargs)
     kwargs_for_operate["actions"] = True
     kwargs_for_operate["reason"] = True
+    kwargs_for_operate["include_token_usage_to_model"] = (
+        include_token_usage_to_model
+    )
 
     # Step 1: Generate initial ReAct analysis
     analysis: ReActAnalysis = await branch.operate(
@@ -255,7 +266,7 @@ async def ReActStream(
             if isinstance(analysis, dict)
             else False
         )
-        and (extensions if max_extensions else 0) > 0
+        and (extensions - 1 if max_extensions else 0) > 0
     ):
         new_instruction = None
         if extensions == max_extensions:
@@ -272,6 +283,9 @@ async def ReActStream(
         operate_kwargs["reason"] = True
         operate_kwargs["response_format"] = ReActAnalysis
         operate_kwargs["action_strategy"] = analysis.action_strategy
+        operate_kwargs["include_token_usage_to_model"] = (
+            include_token_usage_to_model
+        )
         if analysis.action_batch_size:
             operate_kwargs["action_batch_size"] = analysis.action_batch_size
         if irfm:
@@ -289,6 +303,7 @@ async def ReActStream(
             operate_kwargs["guidance"] = guide + operate_kwargs.get(
                 "guidance", ""
             )
+            operate_kwargs["reasoning_effort"] = reasoning_effort
 
         analysis = await branch.operate(
             instruction=new_instruction,
@@ -298,6 +313,16 @@ async def ReActStream(
         )
         round_count += 1
 
+        if isinstance(analysis, dict) and all(
+            i is None for i in analysis.values()
+        ):
+            if not continue_after_failed_response:
+                raise ValueError(
+                    "All values in the response are None. "
+                    "This might be due to a failed response. "
+                    "Set `continue_after_failed_response=True` to ignore this error."
+                )
+
         # If verbose, show round analysis
         if verbose_analysis:
             str_ = f"\n### ReAct Round No.{round_count} Analysis:\n"
@@ -329,6 +354,15 @@ async def ReActStream(
             response_format=response_format,
             **(response_kwargs or {}),
         )
+        if isinstance(analysis, dict) and all(
+            i is None for i in analysis.values()
+        ):
+            if not continue_after_failed_response:
+                raise ValueError(
+                    "All values in the response are None. "
+                    "This might be due to a failed response. "
+                    "Set `continue_after_failed_response=True` to ignore this error."
+                )
     except Exception:
         out = branch.msgs.last_response.response
 

lionagi/operations/ReAct/utils.py

@@ -30,6 +30,8 @@ class ReActAnalysis(BaseModel):
     2) A list of planned actions to perform before finalizing,
     3) Indication whether more expansions/rounds are needed,
     4) Additional tuning knobs: how to handle validation, how to execute actions, etc.
+    Remember do not repeat yourself, and aim to use the most efficient way to achieve
+    the goal to user's satisfaction.
     """
 
     # Standard ReAct strings for controlling expansions:
@@ -38,11 +40,12 @@ class ReActAnalysis(BaseModel):
         "If you are not ready to finalize, set extension_needed to True. "
         "hint: you should set extension_needed to True if the overall goal"
         "is not yet achieved. Do not set it to False, if you are just providing"
-        "an interim answer. You have up to {extensions} expansions. Please continue."
+        "an interim answer. You have up to {extensions} expansions. Please "
+        "strategize accordingly and continue."
     )
     CONTINUE_EXT_PROMPT: ClassVar[str] = (
         "Another round is available. You may do multiple actions if needed. "
-        "You have up to {extensions} expansions. Please continue."
+        "You have up to {extensions} expansions. Please strategize accordingly and continue."
     )
     ANSWER_PROMPT: ClassVar[str] = (
         "Given your reasoning and actions, please now provide the final answer "

lionagi/operations/chat/chat.py

@@ -36,6 +36,7 @@ async def chat(
     image_detail: Literal["low", "high", "auto"] = None,
     plain_content: str = None,
     return_ins_res_message: bool = False,
+    include_token_usage_to_model: bool = False,
     **kwargs,
 ) -> tuple[Instruction, AssistantResponse]:
     ins: Instruction = branch.msgs.create_instruction(
@@ -151,11 +152,12 @@ async def chat(
     kwargs["messages"] = [i.chat_msg for i in messages]
     imodel = imodel or branch.chat_model
 
-    meth = (
-        imodel.invoke
-        if ("stream" not in kwargs or not kwargs["stream"])
-        else imodel.stream
-    )
+    meth = imodel.invoke
+    if "stream" not in kwargs or not kwargs["stream"]:
+        kwargs["include_token_usage_to_model"] = include_token_usage_to_model
+    else:
+        meth = imodel.stream
+
     api_call = await meth(**kwargs)
     branch._log_manager.log(Log.create(api_call))
 

lionagi/operations/communicate/communicate.py

@@ -35,6 +35,7 @@ async def communicate(
     fuzzy_match_kwargs=None,
     clear_messages=False,
     operative_model=None,
+    include_token_usage_to_model: bool = False,
     **kwargs,
 ):
     if operative_model:
@@ -80,6 +81,7 @@ async def communicate(
         image_detail=image_detail,
         plain_content=plain_content,
         return_ins_res_message=True,
+        include_token_usage_to_model=include_token_usage_to_model,
         **kwargs,
     )
     branch.msgs.add_message(instruction=ins)

lionagi/operations/interpret/interpret.py

@@ -20,7 +20,8 @@ async def interpret(
     instruction = (
         "You are given a user's raw instruction or question. Your task is to rewrite it into a clearer,"
         "more structured prompt for an LLM or system, making any implicit or missing details explicit. "
-        "Return only the re-written prompt."
+        "Return only the re-written prompt. Do not assume any details not mentioned in the input, nor "
+        "give additional instruction than what is explicitly stated."
     )
     guidance = (
         f"Domain hint: {domain or 'general'}. "

lionagi/operations/operate/operate.py

@@ -63,6 +63,7 @@ async def operate(
     ] = "return_value",
     operative_model: type[BaseModel] = None,
     request_model: type[BaseModel] = None,
+    include_token_usage_to_model: bool = False,
     **kwargs,
 ) -> list | BaseModel | None | dict | str:
     if operative_model:
@@ -138,6 +139,7 @@ async def operate(
         image_detail=image_detail,
         tool_schemas=tool_schemas,
         return_ins_res_message=True,
+        include_token_usage_to_model=include_token_usage_to_model,
         **kwargs,
     )
     branch.msgs.add_message(instruction=ins)

lionagi/service/endpoints/base.py

@@ -349,11 +349,37 @@ class APICalling(Event):
     endpoint: EndPoint = Field(exclude=True)
     is_cached: bool = Field(default=False, exclude=True)
     should_invoke_endpoint: bool = Field(default=True, exclude=True)
+    include_token_usage_to_model: bool = Field(
+        default=False,
+        exclude=True,
+        description="Whether to include token usage information into instruction messages",
+    )
 
     @model_validator(mode="after")
     def _validate_streaming(self) -> Self:
         if self.payload.get("stream") is True:
             self.streaming = True
+
+        if self.include_token_usage_to_model:
+            if isinstance(self.payload["messages"][-1], dict):
+                required_tokens = self.required_tokens
+                self.payload["messages"][-1][
+                    "content"
+                ] += f"\n\nEstimated Current Token Usage: {required_tokens}"
+                if "model" in self.payload:
+                    if (
+                        self.payload["model"].startswith("gpt-4")
+                        or "o1mini" in self.payload["model"]
+                        or "o1-preview" in self.payload["model"]
+                    ):
+                        self.payload["messages"][-1]["content"] += "/128_000"
+                    elif "o1" in self.payload["model"]:
+                        self.payload["messages"][-1]["content"] += "/200_000"
+                    elif "sonnet" in self.payload["model"]:
+                        self.payload["messages"][-1]["content"] += "/200_000"
+                    elif "haiku" in self.payload["model"]:
+                        self.payload["messages"][-1]["content"] += "/200_000"
+
         return self
 
     @property

lionagi/service/imodel.py

@@ -162,7 +162,9 @@ class iModel:
         else:
             self.streaming_process_func = streaming_process_func
 
-    def create_api_calling(self, **kwargs) -> APICalling:
+    def create_api_calling(
+        self, include_token_usage_to_model: bool = False, **kwargs
+    ) -> APICalling:
         """Constructs an `APICalling` object from endpoint-specific payload.
 
         Args:
@@ -183,6 +185,7 @@ class iModel:
             endpoint=self.endpoint,
             is_cached=payload.get("is_cached", False),
             should_invoke_endpoint=self.should_invoke_endpoint,
+            include_token_usage_to_model=include_token_usage_to_model,
         )
 
     async def process_chunk(self, chunk) -> None:
@@ -200,7 +203,12 @@ class iModel:
                 return await self.streaming_process_func(chunk)
             return self.streaming_process_func(chunk)
 
-    async def stream(self, api_call=None, **kwargs) -> AsyncGenerator:
+    async def stream(
+        self,
+        api_call=None,
+        include_token_usage_to_model: bool = False,
+        **kwargs,
+    ) -> AsyncGenerator:
         """Performs a streaming API call with the given arguments.
 
         Args:
@@ -214,7 +222,10 @@ class iModel:
         """
         if api_call is None:
             kwargs["stream"] = True
-            api_call = self.create_api_calling(**kwargs)
+            api_call = self.create_api_calling(
+                include_token_usage_to_model=include_token_usage_to_model,
+                **kwargs,
+            )
             await self.executor.append(api_call)
 
         if (

lionagi/service/providers/openai_/chat_completions.py

@@ -89,6 +89,8 @@ class OpenAIChatCompletionEndPoint(ChatCompletionEndPoint):
             payload.pop("top_p", None)
             if payload["messages"][0].get("role") == "system":
                 payload["messages"][0]["role"] = "developer"
+        else:
+            payload.pop("reasoning_effort", None)
 
         return {
             "payload": payload,

lionagi/session/branch.py

@@ -941,6 +941,7 @@ class Branch(Element, Communicatable, Relational):
         ] = "return_value",
         operative_model: type[BaseModel] = None,
         request_model: type[BaseModel] = None,
+        include_token_usage_to_model: bool = False,
         **kwargs,
     ) -> list | BaseModel | None | dict | str:
         """
@@ -1028,6 +1029,8 @@ class Branch(Element, Communicatable, Relational):
                 Alias for `response_format`.
             request_model (type[BaseModel], optional):
                 Another alias for `response_format`.
+            include_token_usage_to_model:
+                If `True`, includes token usage in the model messages.
             **kwargs:
                 Additional keyword arguments passed to the LLM via `branch.chat()`.
 
@@ -1080,6 +1083,7 @@ class Branch(Element, Communicatable, Relational):
             operative_model=operative_model,
             request_model=request_model,
             imodel=imodel,
+            include_token_usage_to_model=include_token_usage_to_model,
             **kwargs,
         )
 
@@ -1106,6 +1110,7 @@ class Branch(Element, Communicatable, Relational):
         fuzzy_match_kwargs: dict = None,
         clear_messages: bool = False,
         operative_model: type[BaseModel] = None,
+        include_token_usage_to_model: bool = False,
         **kwargs,
     ):
         """
@@ -1190,6 +1195,7 @@ class Branch(Element, Communicatable, Relational):
             fuzzy_match_kwargs=fuzzy_match_kwargs,
             clear_messages=clear_messages,
             operative_model=operative_model,
+            include_token_usage_to_model=include_token_usage_to_model,
             **kwargs,
         )
 
@@ -1639,6 +1645,7 @@ class Branch(Element, Communicatable, Relational):
         analysis_model: iModel | None = None,
         verbose: bool = False,
         verbose_length: int = None,
+        include_token_usage_to_model: bool = True,
         **kwargs,
     ):
         """
@@ -1688,6 +1695,12 @@ class Branch(Element, Communicatable, Relational):
             analysis_model (iModel | None, optional):
                 A custom LLM model for generating the ReAct analysis steps. If `None`,
                 uses the branch's default `chat_model`.
+            include_token_usage_to_model:
+                If `True`, includes token usage in the model messages.
+            verbose (bool):
+                If `True`, logs detailed information about the process.
+            verbose_length (int):
+                If `verbose=True`, limits the length of logged strings to this value.
             **kwargs:
                 Additional keyword arguments passed into the initial `branch.operate()` call.
 
@@ -1733,6 +1746,7 @@ class Branch(Element, Communicatable, Relational):
             intermediate_listable=intermediate_listable,
             reasoning_effort=reasoning_effort,
             display_as=display_as,
+            include_token_usage_to_model=include_token_usage_to_model,
             **kwargs,
         )
 
@@ -1758,6 +1772,7 @@ class Branch(Element, Communicatable, Relational):
         verbose: bool = False,
         display_as: Literal["json", "yaml"] = "yaml",
         verbose_length: int = None,
+        include_token_usage_to_model: bool = True,
         **kwargs,
     ) -> AsyncGenerator:
         from lionagi.operations.ReAct.ReAct import ReActStream
@@ -1784,6 +1799,7 @@ class Branch(Element, Communicatable, Relational):
             verbose_analysis=True,
             display_as=display_as,
             verbose_length=verbose_length,
+            include_token_usage_to_model=include_token_usage_to_model,
             **kwargs,
         ):
             analysis, str_ = result

lionagi/version.py

@@ -1 +1 @@
-__version__ = "0.9.4"
+__version__ = "0.9.5"

{lionagi-0.9.4.dist-info → lionagi-0.9.5.dist-info}/METADATA

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

{lionagi-0.9.4.dist-info → lionagi-0.9.5.dist-info}/RECORD

@@ -4,7 +4,7 @@ lionagi/_errors.py,sha256=JlBTFJnRWtVYcRxKb7fWFiJHLbykl1E19mSJ8sXYVxg,455
 lionagi/_types.py,sha256=9g7iytvSj3UjZxD-jL06_fxuNfgZyWT3Qnp0XYp1wQU,63
 lionagi/settings.py,sha256=W52mM34E6jXF3GyqCFzVREKZrmnUqtZm_BVDsUiDI_s,1627
 lionagi/utils.py,sha256=K36D9AAGiMPR4eM9tYoiVgvH-NdPPSeMQPls09s7keQ,73223
-lionagi/version.py,sha256=e56AvHfJCtG2ZwwINqsxINVbehWdKxMYgIDbjd7P-II,22
+lionagi/version.py,sha256=ORAtCCI2THBDcdzIbh6oBsoshDvkkmXUWpmO4Q5McAk,22
 lionagi/libs/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/libs/parse.py,sha256=JRS3bql0InHJqATnAatl-hQv4N--XXw4P77JHhTFnrc,1011
 lionagi/libs/file/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
@@ -48,24 +48,24 @@ lionagi/operations/__init__.py,sha256=2HExKTx2J3iKWsvS9YaF6i5SZgqcpAJkVmWbi2H5A5
 lionagi/operations/manager.py,sha256=H7UY86PIxvxKdzJY9YVsWyJcqlwLWhVyvm4sYePH_uY,565
 lionagi/operations/types.py,sha256=fM8HphnbBifMzhoKKvdl3JxGCBHlEGPJEYkLWj9b7vE,704
 lionagi/operations/utils.py,sha256=LrWr_JEyJmSw5RL03KZhWgDKYsjFk0-OS8SoaGU7Jhs,1220
-lionagi/operations/ReAct/ReAct.py,sha256=eBCQOzVHv85uEL4YH1qv8NhbzM0t-qyKZ4JI7tQRXQ8,11971
+lionagi/operations/ReAct/ReAct.py,sha256=f3_Q3Sc9U7i_4IBzFN3KxohUnMTMa_yFJ2I9rSlERvE,13578
 lionagi/operations/ReAct/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
-lionagi/operations/ReAct/utils.py,sha256=jXf9LafAg0GtZajXqT4jOLJHW_Y4eL-hbz5_OlFCBh8,3612
+lionagi/operations/ReAct/utils.py,sha256=84Giel5ToqfbN5F6Tm0uw8yZTTnxiM_jWuFEhnKOxM8,3800
 lionagi/operations/_act/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/operations/_act/act.py,sha256=CunHTTZcS6xNUe0xKSDgtMJ7-ucSvHeW4BtmVjXnaxk,2958
 lionagi/operations/brainstorm/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/operations/brainstorm/brainstorm.py,sha256=iRZUW_V-0Ncw7Av0_WBk3oNRWb4LqQU5i2gQDxulYWY,17222
 lionagi/operations/brainstorm/prompt.py,sha256=Dqi4NNeztdI4iutggRqjnOrG4a4E2JtwIAtRnjZ_ghQ,610
 lionagi/operations/chat/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
-lionagi/operations/chat/chat.py,sha256=GHhudJlQk-okoP-S8ceUnoWF-ccxbdQW_p8gqsE7t6w,5348
+lionagi/operations/chat/chat.py,sha256=xJAH2H0zyVvxiL3XtW3MC6YrwCCB1uCkwcQIJ1YsIOk,5466
 lionagi/operations/communicate/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lionagi/operations/communicate/communicate.py,sha256=f8dQGzFSCrBHcoe_z75GcD8zxgcWQ5umK_Y8BUVaQ3c,2962
+lionagi/operations/communicate/communicate.py,sha256=dPaPqg898biY6j_FlgH4HEJxTK6T_87ixXWhD6kbk40,3077
 lionagi/operations/instruct/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/operations/instruct/instruct.py,sha256=-HDdCgvRGCNB5vMCV0xM2KEa9VrjosApXSKtQQ9d0xQ,795
 lionagi/operations/interpret/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
-lionagi/operations/interpret/interpret.py,sha256=zdsPx8qylFOTH3QYfpQeqNj3wddaqJORKy0_1flDWnk,1399
+lionagi/operations/interpret/interpret.py,sha256=8_F3oYaoYK8MDcK4iCwksBP7sI0UlgBiZSrUusdlKNo,1528
 lionagi/operations/operate/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
-lionagi/operations/operate/operate.py,sha256=NFC1EqykJkz__4oFpzzR8wGmH3_nOBRW4vdVnG2C1tI,7218
+lionagi/operations/operate/operate.py,sha256=j5dGWhHlcWnO-aaLZ4Xe0Hb1M7FGp9BGm356-LOH6rA,7333
 lionagi/operations/parse/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/operations/parse/parse.py,sha256=i2q6YGRwsp2WEu0aySfZ6io7iLNAqPAx1wWd2hUjpgM,3296
 lionagi/operations/plan/__init__.py,sha256=yGBPll6lOqVjadbTvDLGrTlMx3FfBW-e00z7AMvg7Uo,156
@@ -161,11 +161,11 @@ lionagi/protocols/messages/templates/instruction_message.jinja2,sha256=L-ptw5OHx
 lionagi/protocols/messages/templates/system_message.jinja2,sha256=JRKJ0aFpYfaXSFouKc_N4unZ35C3yZTOWhIrIdCB5qk,215
 lionagi/protocols/messages/templates/tool_schemas.jinja2,sha256=ozIaSDCRjIAhLyA8VM6S-YqS0w2NcctALSwx4LjDwII,126
 lionagi/service/__init__.py,sha256=DMGXIqPsmut9H5GT0ZeSzQIzYzzPwI-2gLXydpbwiV8,21
-lionagi/service/imodel.py,sha256=w3cqrJSz2q7k_Y3BXsuS85ZTpBOfa0bNM7Gr58IdTaA,14589
+lionagi/service/imodel.py,sha256=TcqjpkRkPEEny6uaY383tL_45GcLvjju3-NVZS4EbLU,14907
 lionagi/service/manager.py,sha256=FkuqAtLErqLmXNnDtuAdTUFo4uuE_VL660BBGBhzInU,1435
 lionagi/service/types.py,sha256=CHPi8Bxl_yJ1pl2jYZBOrTHbT8_oO9sK75d4LMB651g,486
 lionagi/service/endpoints/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
-lionagi/service/endpoints/base.py,sha256=WuXs2tDrOxbbv9-UkiAgpVcM_6nuCNmvvry7eN1GuoI,23482
+lionagi/service/endpoints/base.py,sha256=nr53A1Z6zGmP5aqkyG-Gc-vbKG22CqvOj1x4nPwnBJ8,24763
 lionagi/service/endpoints/chat_completion.py,sha256=nihV7kCYm7ixdm8dH0JW7vKjqH9yIom4QDXGeDwuO6E,2964
 lionagi/service/endpoints/match_endpoint.py,sha256=hPCqFwVirj5g9Husec980OCUynjRmr0zQzrs7O4yP74,1874
 lionagi/service/endpoints/rate_limited_processor.py,sha256=P0CsMyhuG8OHCPYe2qez92Bm7v2ZRq4L5I6LOiAoGYs,5199
@@ -181,14 +181,14 @@ lionagi/service/providers/exa_/types.py,sha256=8ODjXpFajBE9-DGqBJNS--GObwmLSDi66
 lionagi/service/providers/groq_/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/service/providers/groq_/chat_completions.py,sha256=q1p_1qus4vduWWBzs9V_KbNrqU2Tu2o8TZm6Fh09fw4,1343
 lionagi/service/providers/openai_/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
-lionagi/service/providers/openai_/chat_completions.py,sha256=d-tRlGBmNCcKS_06Ps6dZ6zUa18l0rHHXKo3xPKYKj8,2803
+lionagi/service/providers/openai_/chat_completions.py,sha256=y3RAgI5WQH5EwT1wZxp5ttnkCxUJEcOM3Cta6u9cpQo,2867
 lionagi/service/providers/openrouter_/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/service/providers/openrouter_/chat_completions.py,sha256=0pdXjJCXmCPPbKKVubrnqofaodTOxWTJam8fd3NgrNk,1525
 lionagi/service/providers/perplexity_/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/service/providers/perplexity_/chat_completions.py,sha256=O4MIS_3xIINGjkAZdlw0Bu_jAfBDR4VZA1F8JW2EU1M,1197
 lionagi/service/providers/perplexity_/models.py,sha256=gXH4XGkhZ4aFxvMSDTlHq9Rz1mhu3aTENXAtE-BIr6U,4866
 lionagi/session/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
-lionagi/session/branch.py,sha256=XL0P507Jfqk9LhC8rDvajkjVE8FkPx7hnLltb-LdqHw,71503
+lionagi/session/branch.py,sha256=dKlaM6hh_q7OoXkz4E5S3aS4ksqC2yzdhjzI7xe6pzU,72439
 lionagi/session/prompts.py,sha256=AhuHL19s0TijVZX3tMKUKMi6l88xeVdpkuEn2vJSRyU,3236
 lionagi/session/session.py,sha256=8SuNMiJX6IAW6Ou8aDK0LsVG7zcD5yd22sakMyrd3pw,8987
 lionagi/tools/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
@@ -211,7 +211,7 @@ lionagi/tools/file/writer.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,
 lionagi/tools/file/providers/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/tools/file/providers/docling_.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/tools/query/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
-lionagi-0.9.4.dist-info/METADATA,sha256=46U0oyzqf8A6wuF-KlmsEZWVDiRQCFyZrYfPeG5ZgOQ,18053
-lionagi-0.9.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lionagi-0.9.4.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
-lionagi-0.9.4.dist-info/RECORD,,
+lionagi-0.9.5.dist-info/METADATA,sha256=RgrktdGFQrMKZn0BfHReT-JQ03IVXAkvTvqq6C5g_is,18053
+lionagi-0.9.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lionagi-0.9.5.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
+lionagi-0.9.5.dist-info/RECORD,,