lionagi 0.2.6__py3-none-any.whl → 0.2.7__py3-none-any.whl

lionagi/core/collections/model.py

@@ -26,6 +26,8 @@ load_dotenv()
 
 _oai_price_map = {
     "gpt-4o": (5, 15),
+    "gpt-4o-2024-08-06": (2.5, 10),
+    "gpt-4o-mini": (0.15, 0.6),
     "gpt-4-turbo": (10, 30),
     "gpt-3.5-turbo": (0.5, 1.5),
 }

lionagi/core/session/branch.py

@@ -58,7 +58,7 @@ class Branch(Node, DirectiveMixin):
     messages: Pile = Field(None)
     progress: Progression = Field(None)
     tool_manager: ToolManager = Field(None)
-    system: System = Field(None)
+    system: System | None = Field(None)
     user: str = Field(None)
     mailbox: Exchange[Mail] = Field(None)
     imodel: iModel = Field(None)
@@ -110,14 +110,14 @@ class Branch(Node, DirectiveMixin):
             system (System): The system message to set.
             sender (str, optional): The sender of the system message.
         """
-        system = system or "You are a helpful assistant."
-        if len(self.progress) == 0:
-            self.add_message(system=system, sender=sender)
-        else:
-            _msg = System(system=system, sender=sender)
-            _msg.recipient = self.ln_id
-            self._remove_system()
-            self.system = _msg
+        if system is not None:
+            if len(self.progress) == 0:
+                self.add_message(system=system, sender=sender)
+            else:
+                _msg = System(system=system, sender=sender)
+                _msg.recipient = self.ln_id
+                self._remove_system()
+                self.system = _msg
 
     def add_message(
         self,

lionagi/version.py

@@ -1 +1 @@
-__version__ = "0.2.6"
+__version__ = "0.2.7"

{lionagi-0.2.6.dist-info → lionagi-0.2.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lionagi
-Version: 0.2.6
+Version: 0.2.7
 Summary: Towards automated general intelligence.
 Author: HaiyangLi
 Author-email: Haiyang Li <ocean@lionagi.ai>
@@ -237,7 +237,7 @@ Requires-Dist: boto3 >=1.34.131
 ### an AGentic Intelligence Operating System
 
 ```
-pip install lionagi==0.2.5
+pip install lionagi==0.2.6
 ```
 
 **Powerful Intelligent Workflow Automation**

{lionagi-0.2.6.dist-info → lionagi-0.2.7.dist-info}/RECORD

@@ -1,5 +1,5 @@
 lionagi/__init__.py,sha256=amQal6CUIv4QDI4Qmb0M5qwTbn3_F430Wl5vaSNG6-U,1952
-lionagi/version.py,sha256=Oz5HbwHMyE87nmwV80AZzpkJPf-wBg7eDuJr_BXZkhU,22
+lionagi/version.py,sha256=XHypfHSPdgXFKmOdoewn7czU670gt8InhHhzlP5j_aA,22
 lionagi/core/__init__.py,sha256=Il5Q9ATdX8yXqVxtP_nYqUhExzxPC_qk_WXQ_4h0exg,16
 lionagi/core/_setting/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/core/_setting/_setting.py,sha256=p23fHtrzIZlQ8s8CDZICn8C6k7mdB_088nIDx19JaqM,1907
@@ -23,7 +23,7 @@ lionagi/core/collections/__init__.py,sha256=5374e0qHH75IUvW-xA3Dte25N_oc_YXjtu7e
 lionagi/core/collections/_logger.py,sha256=_jwZe6BsRaWB2YkHwG01jmiudJkoi5u90Ry3KKa9T-o,12104
 lionagi/core/collections/exchange.py,sha256=Skx2al0PtNkzdaApb8DdbOO0ukC19kSQnlskHpAHJHw,4832
 lionagi/core/collections/flow.py,sha256=0tBeVJhENXjVFjtOlrEHXvI4q-rgTyD-kcE7X29AqDE,13189
-lionagi/core/collections/model.py,sha256=q7Zp4ccHjzkuH1QKFojbkInLgAbaPlOiEw8V5NaExOQ,16011
+lionagi/core/collections/model.py,sha256=U9yFhQ31NH24HwJjnnRxOfZcnUWq8Vuk_iF0r8SZc5A,16079
 lionagi/core/collections/pile.py,sha256=MF2G846TolIiKZkY0mjSOvD7Bc3__Fb_z-WiiLG1lbM,29444
 lionagi/core/collections/progression.py,sha256=eDhSbsQWb9TVsNExaJAN27tjoyb6QEKWPxBZKUbNbs0,7295
 lionagi/core/collections/util.py,sha256=JRu_gCDYopqIUByEtJgnJNkeoWnhge7XbSFejrdbIec,1962
@@ -82,7 +82,7 @@ lionagi/core/rule/rulebook.py,sha256=HlJCIjhsRB0yJs-uF85aK9bJasCDI0K845UGmymzD9Q
 lionagi/core/rule/string.py,sha256=8nSxEhHUFs-gWM1MW5fiNuDP1Yd_IIZ2ipBVphgCmQ0,1487
 lionagi/core/rule/util.py,sha256=qqtUqt4BL_sikW2XZ73NDzb3p-Awu4za_2B5mZvzFdA,1042
 lionagi/core/session/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lionagi/core/session/branch.py,sha256=6ySBqEZHYbQ1AKJmcoBtU9MwF5WT5wAr3z2NB6TF8gg,14761
+lionagi/core/session/branch.py,sha256=338N-jw-AfnjksYFo2WiLIeXdzxOYJ_qZBxMuk-U8bU,14769
 lionagi/core/session/directive_mixin.py,sha256=vCoOIA_AYEH7M1ptDBUaOtSuxC5KQSAnsKn9X-LpjFs,12275
 lionagi/core/session/session.py,sha256=uESwIyGfRrX8Ii03-QIQYUlQKR7lsJWREdSZrHxfNW4,11012
 lionagi/core/structure/__init__.py,sha256=DkeLUlrb7rGx3nZ04aADU9HXXu5mZTf_DBwT0xhzIv4,7
@@ -251,8 +251,8 @@ lionagi/tests/test_core/test_structure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JC
 lionagi/tests/test_core/test_structure/test_base_structure.py,sha256=qsaYP745fFIst5ZNiPFb9-ATScLyUFqQ5UQzyyBgJPM,9388
 lionagi/tests/test_core/test_structure/test_graph.py,sha256=hLsTZmZMs9vCZW-KiOwYY3npk6WxaVU6zZSA9-ltDvQ,2162
 lionagi/tests/test_core/test_structure/test_tree.py,sha256=PvMJXDsNPpJFgEQCan-5Q5JREgMrBOpYIaWcwHd-WDY,1944
-lionagi-0.2.6.dist-info/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
-lionagi-0.2.6.dist-info/METADATA,sha256=jYXPLq2e8RMVKs3rC2hb7f_qv_ET3u0mXuLP8KoM0do,16276
-lionagi-0.2.6.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
-lionagi-0.2.6.dist-info/top_level.txt,sha256=szvch_d2jE1Lu9ZIKsl26Ll6BGfYfbOgt5lm-UpFSo4,8
-lionagi-0.2.6.dist-info/RECORD,,
+lionagi-0.2.7.dist-info/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
+lionagi-0.2.7.dist-info/METADATA,sha256=WI8pyy3AHxkvSBmxFFQKtYGoL-tCXEplXj6xU9rzcWs,16276
+lionagi-0.2.7.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
+lionagi-0.2.7.dist-info/top_level.txt,sha256=szvch_d2jE1Lu9ZIKsl26Ll6BGfYfbOgt5lm-UpFSo4,8
+lionagi-0.2.7.dist-info/RECORD,,