lionagi 0.0.115__py3-none-any.whl → 0.0.204__py3-none-any.whl

{lionagi-0.0.115.dist-info → lionagi-0.0.204.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lionagi
-Version: 0.0.115
+Version: 0.0.204
 Summary: Towards automated general intelligence.
 Author: HaiyangLi
 Author-email: Haiyang Li <ocean@lionagi.ai>
@@ -217,8 +217,11 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: aiohttp >=3.9.0
 Requires-Dist: python-dotenv ==1.0.0
-Requires-Dist: tiktoken ==0.5.1
-Requires-Dist: httpx ==0.25.1
+Requires-Dist: tiktoken >=0.5.1
+Requires-Dist: pydantic >=2.6.0
+Requires-Dist: cryptography >=42.0.0
+Requires-Dist: aiocache ==0.12.2
+Requires-Dist: pandas >=2.1.0
 
 ![PyPI - Version](https://img.shields.io/pypi/v/lionagi?labelColor=233476aa&color=231fc935)  ![Read the Docs](https://img.shields.io/readthedocs/lionagi)  ![PyPI - License](https://img.shields.io/pypi/l/lionagi?color=231fc935) ![PyPI - Downloads](https://img.shields.io/pypi/dm/lionagi?color=blue)
 
@@ -247,12 +250,12 @@ by default we use `OPENAI_API_KEY`.
 
 
 ### Features
-- Create a production ready LLM application **in hours**, with more than 100 models to choose from
-- written in pure python, minimum dependency `aiohttp`, `python-dotenv`, `tiktoken`, `pydantic`
-- Efficient and verstile data operations for reading, chunking, binning, writing, storing data with built-in support for `langchain` and `llamaindex`
+- Robust and scalable. Create a production ready LLM application **in hours**, with more than 100 models
+- Efficient and verstile data operations for reading, chunking, binning, writing, storing data with support for `langchain` and `llamaindex`
+- Built-in support for **chain/graph-of-thoughts, ReAct, Concurrent parallel function calling** 
 - Unified interface with any LLM provider, API or local
   - Fast and **concurrent** API call with **configurable rate limit**
-  - (Work In Progress) support for hundreds of models both API and local
+  - (Work In Progress) support for models both API and local
 ---
 LionAGI is designed to be `asynchronous` only, please check python official documentation on how `async` work: [here](https://docs.python.org/3/library/asyncio.html)
 
@@ -269,7 +272,6 @@ LionAGI is designed to be `asynchronous` only, please check python official docu
 The following example shows how to use LionAGI's `Session` object to interact with `gpt-4` model:
 
 ```python
-import lionagi as li
 
 # define system messages, context and user instruction
 system = "You are a helpful assistant designed to perform calculations."
@@ -279,10 +281,12 @@ context = {"x": 10, "y": 5}
 
 ```python
 # in interactive environment (.ipynb for example)
+import lionagi as li
+
 calculator = li.Session(system=system)
-result = await calculator.initiate(instruction=instruction,
-                                   context=context,
-                                   model="gpt-4-1106-preview")
+result = await calculator.chat(
+  instruction=instruction, context=context, model="gpt-4-1106-preview"
+)
 
 print(f"Calculation Result: {result}")
 ```
@@ -290,15 +294,17 @@ print(f"Calculation Result: {result}")
 ```python
 # or otherwise, you can use
 import asyncio
-from dotenv import loadenv
+from dotenv import load_dotenv
 
 load_dotenv()
 
+import lionagi as li
+
 async def main():
     calculator = li.Session(system=system)
-    result = await calculator.initiate(instruction=instruction,
-                                       context=context, 
-                                       model="gpt-4-1106-preview")
+    result = await calculator.chat(
+      instruction=instruction, context=context, model="gpt-4-1106-preview"
+    )
     print(f"Calculation Result: {result}")
 
 if __name__ == "__main__":
@@ -325,8 +331,6 @@ When referencing LionAGI in your projects or research, please cite:
 }
 ```
 
-## Star History
-![Star History Chart](https://api.star-history.com/svg?repos=lion-agi/lionagi&type=Date)
 
 ### Requirements
 Python 3.9 or higher. 

lionagi-0.0.204.dist-info/RECORD

@@ -0,0 +1,106 @@
+lionagi/__init__.py,sha256=pRm9tnfYQnj_ZCan3xr3HQO7tbUeqwRhSBtwOvXJ-o8,886
+lionagi/version.py,sha256=DFD_QxPb5C0pITp5fF5NUF2YLCrCu9FEfCgCY7YVy4I,24
+lionagi/_services/__init__.py,sha256=zU5sxmSI9-Jtp_WsI-Zsb6hmT8y5zF9YtJ7XikAjnbs,60
+lionagi/_services/anthropic.py,sha256=pLebbnr2H1A41bzXrJrU7yQbZY35swKrSi4mktXoIyk,3195
+lionagi/_services/anyscale.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/_services/azure.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/_services/base_service.py,sha256=bbPQ9xTaY5jxrHe6vW_PeyRkoxVaxWmpWlCqQxSfRI8,17314
+lionagi/_services/bedrock.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/_services/everlyai.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/_services/gemini.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/_services/gpt4all.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/_services/huggingface.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/_services/litellm.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/_services/localai.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/_services/mistralai.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/_services/oai.py,sha256=Jiuw_TibpYhN6x1PmHYUMi_0Nugrxmi5G1eimAuw21I,3958
+lionagi/_services/ollama.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/_services/openllm.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/_services/openrouter.py,sha256=MuuwoT2ro9FmY7O1jzCenRrL2YfiYUMM8-0kKoGZHAY,1799
+lionagi/_services/perplexity.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/_services/predibase.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/_services/rungpt.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/_services/services.py,sha256=kY2TpT98pDS_qCrCO4H1YBlDYNDqekx5S9rWQdWm5Ck,4007
+lionagi/_services/transformers.py,sha256=tHlo9QVV5ycB2xUEKNRf-b665o_21fPC6c4ycH3Hjk4,1444
+lionagi/_services/vllm.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/_services/xinference.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/agents/planner.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/agents/prompter.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/agents/scorer.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/agents/summarizer.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/agents/validator.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/bridge/__init__.py,sha256=YDKi-pniFEknEXTSEvX6yEe4Y69f0kLvbjvEQ0TdrTU,575
+lionagi/bridge/langchain.py,sha256=NuuUOJ5oV0l6ae4EeXxaeB82nJ6FDJHVC4AFBoIqiDs,5726
+lionagi/bridge/llama_index.py,sha256=oggpASmfVA6IhfgBOJS8CWJrQb3zrGaDApUQ7MWT8OM,6652
+lionagi/configs/__init__.py,sha256=QOd4Rs7vjIpNWvIocxWQeU-q-MPRC-AOxh-gM-eBJ2o,142
+lionagi/configs/oai_configs.py,sha256=Q2ESc5QiMprnRc_w7SeMlaTYUWl_Y4SEzZSE4iOkz4Q,2646
+lionagi/configs/openrouter_configs.py,sha256=IBQHqb8mo4Jb3kYAm_7NOHSKRPwSdGbPpDJoiwHxLYw,1269
+lionagi/core/__init__.py,sha256=6uDjq1WCWBahNiCpzGaUCJe7GDg6lwUD_cGNT86GhwM,200
+lionagi/core/core_util.py,sha256=80fmVywc0UC9FklXpQr5_dHdtdHdosv5Zxwy9tC-Ufg,2339
+lionagi/core/branch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/core/branch/branch.py,sha256=w3GZAws8_qgY17XOn6svmO4ECQqhNQ_rWMzXiwBQVO8,23403
+lionagi/core/branch/branch_manager.py,sha256=kLVYUYVUmFQ2CPV34vfVRrkR6fhlizoAr3Dw2ilMX6M,4904
+lionagi/core/branch/cluster.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/core/branch/conversation.py,sha256=q05EwDouTAH5eMDCp2wwRHK0ipbm0MYFngk8R6vOTB0,18664
+lionagi/core/flow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/core/flow/flow.py,sha256=Fd6xVZKgjjpr-rcVAjvuMGP04OxNK4LVkbB9VnEP21k,813
+lionagi/core/flow/flow_util.py,sha256=OoQ2-ktkpQs9f1m1VI1pucUeq75Mx1aKqz8KdINMt8M,2083
+lionagi/core/instruction_set/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/core/instruction_set/instruction_set.py,sha256=bxrxPxLJdaenvZ2CMaiucNB4fZ_5AWVz49MYs3mG2G8,13682
+lionagi/core/messages/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/core/messages/messages.py,sha256=bokGS8hu4wQxa-4Gsf2Eg7Q99ANccjjixZpE-DC0P7g,6533
+lionagi/core/sessions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/core/sessions/session.py,sha256=-dSlIIzHmD2dGd3zu4m_U-UE0BKraKYyt1H_DAzpngk,17694
+lionagi/datastores/__init__.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/datastores/chroma.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/datastores/deeplake.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/datastores/elasticsearch.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/datastores/lantern.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/datastores/pinecone.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/datastores/postgres.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/datastores/qdrant.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/loaders/__init__.py,sha256=vOOwHkdI0yIA_jV_pNyOrdkS5Ghs0k4od75S1U60jJE,451
+lionagi/loaders/chunker.py,sha256=UY6GrC8qC0MLRaHiSfgG5HMnrtWSTuIvaRPhYfdm9ak,6438
+lionagi/loaders/load_util.py,sha256=4fzhMk3H6OvcQcViUwlLPSTMpcY4alfdel16lJgXz8Y,8358
+lionagi/loaders/reader.py,sha256=xI1uxw9qGJ_rWev_s3vtW8Ep9YaK-15q7ts-Jy61PGg,4625
+lionagi/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/models/base_model.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/models/imodel.py,sha256=VultpAAZ5PBT6_Cps-LjgZqHEyuF-km6eVvT-sEPEgo,1491
+lionagi/schema/__init__.py,sha256=XRWXK9sztycoIMCTzwLEVMxxc1wgWKNUDRbWTpn5Ie0,208
+lionagi/schema/async_queue.py,sha256=e_wFyDvCeKTxW6MAIU6Q3jxw24uEQuahaZwNDzZMB4k,5674
+lionagi/schema/base_condition.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/schema/base_node.py,sha256=Ol4Y6gn81Dm8FJb3fdKVkJlq4R-OAGIf891exnVv45I,13211
+lionagi/schema/base_tool.py,sha256=8LR-MYOGkv9zCHd8tWry296OZNYlbTpYeO1xd-oQcOM,1254
+lionagi/schema/data_logger.py,sha256=2rNPBYF29_2xgv2m226BzKdTfFIVU1YlzQ0PK8zLAvY,4313
+lionagi/schema/data_node.py,sha256=k80mv5DCqAHrGNlmaIHAkCA7JF4dkaRaRDh_oku4kUY,2272
+lionagi/schema/status_tracker.py,sha256=6otnTSMrH5jM0PUDiDeK3zG0VOSKfNBDzNN6Bts2tmA,1236
+lionagi/structures/__init__.py,sha256=wMPekT2vbWwUkJ5aW5o-lzJC9Fzhta6RHDiFPTNUm_0,120
+lionagi/structures/graph.py,sha256=dqhRq5j3SOQvelUOanC3_DC4TQ1wRv4qK0QmleoLUZs,7788
+lionagi/structures/relationship.py,sha256=IgiYHD87bc57hzSAYSo27WqLgtk8pzgGbIxfPYJCI-g,6059
+lionagi/structures/structure.py,sha256=YyL3LxgeQWgkWDhACRxo8GoDn0IjE7idh7B0r5U3mZ4,3485
+lionagi/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/tests/test_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/tests/test_utils/test_api_util.py,sha256=7Zyc0J1glZrIWI1HrTRSRhzw8jaUW1L2vVLFAlUhI4g,9721
+lionagi/tests/test_utils/test_call_util.py,sha256=7xmfFaWvniMQfaOyfwasA2enJQVuSlcAwc8gUyAR_7k,26277
+lionagi/tests/test_utils/test_encrypt_util.py,sha256=hlkbFjQs2jodr8fgtPli6f1MO3doQbTcsZfzGKVrG5k,12652
+lionagi/tests/test_utils/test_io_util.py,sha256=cFZCT6EikVeuXB13w-UbtO3YceCHBO5RlNXxGICqg_U,11002
+lionagi/tests/test_utils/test_nested_util.py,sha256=Z1boHufhjZryw51qW2lABOnnyJ1snAFp26KKzzzD8Bs,12612
+lionagi/tests/test_utils/test_sys_util.py,sha256=TDCkzll-JLa6NuBbN_-ay5Rw9KTa_HcSHHAq62RVwGI,13545
+lionagi/tools/__init__.py,sha256=ZEck-ReP5Co05nAA2gUXTpKoDN2QZqrL7DvU9Z09gqg,69
+lionagi/tools/tool_manager.py,sha256=4yCjMVOrKo4PoEMv0xF6wMirpj20qhLx4J7op0v5h2w,6543
+lionagi/tools/tool_util.py,sha256=5ln7lnqC_rjhKDDwef10xrBrjP1yLzsQvphllD5crec,9252
+lionagi/utils/__init__.py,sha256=FJAqYyb19Mp6kiietyjtp0WLlW2857-HAHl7OxqCc5Y,954
+lionagi/utils/api_util.py,sha256=YV-DKPgR4lCKO_riroSKyq6mO-8vz-SilRo_gWtg4Zg,15282
+lionagi/utils/call_util.py,sha256=G3K8dkEZ9AehrBdbKHK6Xtr1pZZWxrtqhQEu5xkItuo,32919
+lionagi/utils/encrypt_util.py,sha256=iZjZdXVvl0lw4Yw_YNzIWriM3F2qKtzai7PgSQ1TExc,9316
+lionagi/utils/io_util.py,sha256=xoVsq8sP5JGsosuC80Kad3GkGjm8Qm0OLYyTw-U5ru8,6455
+lionagi/utils/nested_util.py,sha256=67j-ySQtuMGxtjnC-Ty2mwQgqp2g1gZhXRy1MulUu1U,26656
+lionagi/utils/pd_util.py,sha256=ShLdNRJI-U2nN9TmZEGPdRXHzFMfrmw-sTpUbxNWr1w,2274
+lionagi/utils/sys_util.py,sha256=iZQu3HvYLl-12mmZ0kk4lX-3FnkRh-EAMAefPt_6P7k,10893
+lionagi/utils/url_util.py,sha256=fu1uRFwSR9D3dO1nfSYVNRD1b1BZVClcbpgF7tA_U4s,1864
+lionagi-0.0.204.dist-info/LICENSE,sha256=TBnSyG8fs_tMRtK805GzA1cIyExleKyzoN_kuVxT9IY,11358
+lionagi-0.0.204.dist-info/METADATA,sha256=kxTcrwAWtkK8cRY0CAxJaOZ8lBIVQgFcy-8HT92mTuc,17894
+lionagi-0.0.204.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+lionagi-0.0.204.dist-info/top_level.txt,sha256=szvch_d2jE1Lu9ZIKsl26Ll6BGfYfbOgt5lm-UpFSo4,8
+lionagi-0.0.204.dist-info/RECORD,,

lionagi/core/conversations/__init__.py

@@ -1,5 +0,0 @@
-from .conversation import Conversation
-
-__all__ = [
-    "Conversation"
-]

lionagi/core/conversations/conversation.py

@@ -1,107 +0,0 @@
-from typing import List, Any
-
-from lionagi.schema.base_node import BaseNode
-from lionagi.messages import Message, Response
-from lionagi.objs.messenger import Messenger
-
-
-class Conversation(BaseNode):
-    """
-    A conversation that handles messages and responses.
-
-    Attributes:
-        response_counts (int): A counter for the number of responses in the conversation.
-        messages (List[Message]): A list of message objects in the conversation.
-        msgr (Messenger): An instance of Messenger to create message objects.
-        responses (List[Response]): A list of response objects in the conversation.
-    """    
-
-    response_counts : int = 0
-    messages: List[Message] = []
-    msgr : Any = Messenger()
-    responses: List[Response] = []
-
-    def initiate_conversation(
-        self, system=None, instruction=None, 
-        context=None, name=None
-    ):
-        """
-        Initiates a new conversation, erase previous messages and responses.
-
-        Parameters:
-            system (Any, optional): System information to include in the initial message. Defaults to None.
-            instruction (Any, optional): Instruction details to include in the conversation. Defaults to None.
-            context (Any, optional): Contextual information relevant to the conversation. Defaults to None.
-            name (str, optional): The name associated with the conversation. Defaults to None.
-
-        Returns:
-            None
-        """
-        self.messages, self.responses = [], []
-        self.add_messages(system=system)
-        self.add_messages(instruction=instruction, context=context, name=name)
-
-    # modify the message adding to accomodate tools
-    def add_messages(
-        self, system=None, instruction=None, 
-        context=None, response=None, name=None
-    ):
-        """
-        Adds a new message object to the conversation messages list based on the provided parameters.
-
-        Parameters:
-            system (Any, optional): System information to include in the message. Defaults to None.
-            instruction (Any, optional): Instruction details to include in the message. Defaults to None.
-            context (Any, optional): Contextual information relevant to the message. Defaults to None.
-            response (Any, optional): Response details to include in the message. Defaults to None.
-            name (str, optional): The name associated with the message. Defaults to None.
-
-        Returns:
-            None
-        """
-        msg = self.msgr.create_message(
-            system=system, instruction=instruction, 
-            context=context, response=response, name=name
-        )
-        self.messages.append(msg)
-
-    def change_system(self, system):
-        """
-        Changes the system information of the first message in the conversation.
-
-        Parameters:
-            system (Any): The new system information to be set.
-
-        Returns:
-            None
-        """
-        self.messages[0] = self.msgr.create_message(system=system)
-
-
-    def keep_last_n_exchanges(self, n: int):
-        """
-        Keeps only the last n exchanges in the conversation, where an exchange starts with a user message. This function trims the conversation to retain only the specified number of the most recent exchanges. 
-        An exchange is defined as a sequence of messages starting with a user message. 
-        The first message in the conversation, typically a system message, is always retained.
-
-        Parameters:
-            n (int): The number of exchanges to keep in the conversation.
-
-        Returns:
-            None: The method modifies the conversation in place and does not return a value.
-
-        Raises:
-            ValueError: If n is not a positive integer.
-
-        Note:
-            This function assumes the first message in the conversation is a system message and each user message 
-            marks the beginning of a new exchange.
-        """
-        response_indices = [
-            index for index, message in enumerate(self.messages[1:]) 
-            if message.role == "user"
-        ]
-        if len(response_indices) >= n:
-            first_index_to_keep = response_indices[-n] + 1
-            self.messages = [self.system] + self.messages[first_index_to_keep:]
-            

lionagi/core/flows/__init__.py

@@ -1,8 +0,0 @@
-from .flow_util import run_session
-from .flow import Flow
-
-
-__all__ = [
-    "run_session",
-    "Flow"
-]

lionagi/core/flows/flow.py

@@ -1,8 +0,0 @@
-from lionagi.structures import Relationship, Structure
-from ..sessions import Session
-from .flow_util import run_session
-
-# represents structured sessions
-
-class Flow(Structure):
-    ...

lionagi/core/flows/flow_util.py

@@ -1,62 +0,0 @@
-from ..sessions import Session
-
-def get_config(temperature, max_tokens, key_scheme, n):
-    f = lambda i:{
-        "temperature": temperature[i], 
-        "max_tokens": max_tokens[i],
-    }
-    return {
-        "key": f"{key_scheme}{n+1}", 
-        "config": f(n)
-        }
-
-async def run_workflow(
-    session, prompts, temperature, max_tokens, 
-    key_scheme, num_prompts, context
-):
-    for i in range(num_prompts): 
-        key_, config_ = get_config(temperature, max_tokens, key_scheme, i)
-        if i == 0:
-            await session.initiate(instruction=prompts[key_], context=context, **config_)
-        else: 
-            await session.followup(instruction=prompts[key_], **config_)
-        
-        return session
-
-async def run_auto_workflow(
-    session, prompts, temperature, max_tokens, 
-    key_scheme, num_prompts, context
-):
-    for i in range(num_prompts): 
-        key_, config_ = get_config(temperature, max_tokens, key_scheme, i)
-        if i == 0:
-            await session.initiate(instruction=prompts[key_], context=context, **config_)
-        else: 
-            await session.auto_followup(instruction=prompts[key_], **config_)
-        
-        return session
-    
-async def run_session(
-    prompts, dir, llmconfig, key_scheme, num_prompts, 
-    temperature, max_tokens, type_=None, tools=None
-):
-    prompts_ = prompts.copy()
-    session = Session(
-        system=prompts_.pop('system', 'You are a helpful assistant'), 
-        dir = dir,
-        llmconfig = llmconfig
-    )
-    if tools:
-        session.register_tools(tools)
-    if type_ is None: 
-        session = await run_workflow(
-            session, prompts_, temperature, max_tokens, 
-            key_scheme=key_scheme, num_prompts=num_prompts
-            )
-    elif type_ == 'auto': 
-        session = await run_auto_workflow(
-            session, prompts_, temperature, max_tokens, 
-            key_scheme=key_scheme, num_prompts=num_prompts
-            )
-        
-    return session

lionagi/core/instruction_set/instruction_sets.py

@@ -1,7 +0,0 @@
-from lionagi.structures import Relationship, Structure
-from ..messages import Message, Instruction
-
-# dynamically structured preconfigured instructions
-
-class InstructionSet(Structure):
-    ...

lionagi/core/sessions/sessions.py

@@ -1,185 +0,0 @@
-import json
-from typing import Any
-from dotenv import load_dotenv
-
-from lionagi.schema import DataLogger, Tool
-from lionagi.utils import lcall, alcall
-from lionagi.services import OpenAIService
-from lionagi.endpoints import ChatCompletion
-from lionagi.objs.tool_manager import ToolManager
-from lionagi.configs.oai_configs import oai_schema
-from lionagi.core.conversations.conversation import Conversation
-
-load_dotenv()
-OAIService = OpenAIService()
-
-
-class Session:
-
-    def __init__(
-        self, system, dir=None, llmconfig=oai_schema['chat']['config'], 
-        service=OAIService
-    ):
-        self.conversation = Conversation()
-        self.system = system
-        self.llmconfig = llmconfig
-        self.logger_ = DataLogger(dir=dir)
-        self.service = service
-        self.tool_manager = ToolManager()
-    
-    def set_dir(self, dir):
-        self.logger_.dir = dir
-    
-    def set_system(self, system):
-        self.conversation.change_system(system)
-    
-    def set_llmconfig(self, llmconfig):
-        self.llmconfig = llmconfig
-    
-    def set_service(self, service):
-        self.service = service
-    
-    async def _output(self, invoke=True, out=True):
-        if invoke:
-            try: 
-                # func, args = self.tool_manager._get_function_call(self.conversation.responses[-1]['content'])
-                # outs = await self.tool_manager.invoke(func, args)
-                # self.conversation.add_messages(response=outs)
-
-                tool_uses = json.loads(self.conversation.responses[-1].message_content)
-                if 'function_list' in tool_uses.keys():
-                    func_calls = lcall(tool_uses['function_list'], self.tool_manager.get_function_call)
-                else:
-                    func_calls = lcall(tool_uses['tool_uses'], self.tool_manager.get_function_call)
-
-                outs = await alcall(func_calls, self.tool_manager.invoke)
-                for out, f in zip(outs, func_calls):
-                    response = {"function": f[0], "arguments": f[1], "output": out}
-                    self.conversation.add_messages(response=response)
-            except:
-                pass
-        if out:
-            return self.conversation.responses[-1].message_content
-    
-    def _is_invoked(self):
-        content = self.conversation.messages[-1].message_content
-        try:
-            if json.loads(content).keys() >= {'function', 'arguments', 'output'}:
-                return True
-        except: 
-            return False    
-
-    def register_tools(self, tools): #, update=False, new=False, prefix=None, postfix=None):
-        if not isinstance(tools, list):
-            tools=[tools]
-        self.tool_manager.register_tools(tools=tools) #, update=update, new=new, prefix=prefix, postfix=postfix)
-        # tools_schema = lcall(tools, lambda tool: tool.to_dict()['schema_'])
-        # if self.llmconfig['tools'] is None:
-        #     self.llmconfig['tools'] = tools_schema
-        # else:
-        #     self.llmconfig['tools'] += tools_schema
-
-    def _tool_parser(self, **kwargs):
-        # 1. single schema: dict
-        # 2. tool: Tool
-        # 3. name: str
-        # 4. list: 3 types of lists
-        def tool_check(tool):
-            if isinstance(tool, dict):
-                return tool
-            elif isinstance(tool, Tool):
-                return tool.schema_
-            elif isinstance(tool, str):
-                if self.tool_manager.name_existed(tool):
-                    tool = self.tool_manager.registry[tool]
-                    return tool.schema_
-                else:
-                    raise ValueError(f'Function {tool} is not registered.')
-
-        if 'tools' in kwargs:
-            if not isinstance(kwargs['tools'], list):
-                kwargs['tools']=[kwargs['tools']]
-            kwargs['tools'] = lcall(kwargs['tools'], tool_check)
-
-        else:
-            tool_kwarg = {"tools": self.tool_manager.to_tool_schema_list()}
-            kwargs = {**tool_kwarg, **kwargs}
-
-        return kwargs
-
-    async def initiate(self, instruction, system=None, context=None, 
-                       name=None, invoke=True, out=True, **kwargs) -> Any:
-        # if self.tool_manager.registry != {}:
-        #     if 'tools' not in kwargs:
-        #         tool_kwarg = {"tools": self.tool_manager.to_tool_schema_list()}
-        #         kwargs = {**tool_kwarg, **kwargs}
-        if self.tool_manager.registry != {}:
-            kwargs = self._tool_parser(**kwargs)
-        config = {**self.llmconfig, **kwargs}
-        system = system or self.system
-        self.conversation.initiate_conversation(system=system, instruction=instruction, context=context, name=name)
-        await self.call_chatcompletion(**config)
-        
-        return await self._output(invoke, out)
-
-    async def followup(self, instruction, system=None, context=None, 
-                       out=True, name=None, invoke=True, **kwargs) -> Any:
-        if system:
-            self.conversation.change_system(system)
-        self.conversation.add_messages(instruction=instruction, context=context, name=name)
-
-        if 'tool_parsed' in kwargs:
-            kwargs.pop('tool_parsed')
-        else:
-            if self.tool_manager.registry != {}:
-                kwargs = self._tool_parser(**kwargs)
-        # if self.tool_manager.registry != {}:
-        #     if 'tools' not in kwargs:
-        #         tool_kwarg = {"tools": self.tool_manager.to_tool_schema_list()}
-        #         kwargs = {**tool_kwarg, **kwargs}
-        config = {**self.llmconfig, **kwargs}
-        await self.call_chatcompletion(**config)
-
-        return await self._output(invoke, out)
-
-    async def auto_followup(self, instruct, num=3, **kwargs):
-        # if self.tool_manager.registry != {}:
-        #     if 'tools' not in kwargs:
-        #         tool_kwarg = {"tools": self.tool_manager.to_tool_schema_list()}
-        #         kwargs = {**tool_kwarg, **kwargs}
-        if self.tool_manager.registry != {}:
-            kwargs = self._tool_parser(**kwargs)
-
-        cont_ = True
-        while num > 0 and cont_ is True:
-            await self.followup(instruct, tool_choice="auto", tool_parsed=True, **kwargs)
-            num -= 1
-            cont_ = True if self._is_invoked() else False
-        if num == 0:
-            await self.followup(instruct, **kwargs, tool_parsed=True)
-
-    # def messages_to_csv(self, dir=None, filename="messages.csv", **kwargs):
-    #     dir = dir or self.logger_.dir
-    #     if dir is None:
-    #         raise ValueError("No directory specified.")
-    #     self.conversation.msg.to_csv(dir=dir, filename=filename, **kwargs)
-        
-    # def log_to_csv(self, dir=None, filename="llmlog.csv", **kwargs):
-    #     dir = dir or self.logger_.dir
-    #     if dir is None:
-    #         raise ValueError("No directory specified.")
-    #     self.logger_.to_csv(dir=dir, filename=filename, **kwargs)
-    
-    async def call_chatcompletion(self, schema=oai_schema['chat'], **kwargs):
-        messages = [message.message for message in self.conversation.messages]
-        payload = ChatCompletion.create_payload(messages=messages, schema=schema, llmconfig=self.llmconfig,**kwargs)
-        completion = await self.service.serve(payload=payload)
-        if "choices" in completion:
-            self.logger_({"input":payload, "output": completion})
-            self.conversation.add_messages(response=completion['choices'][0])
-            self.conversation.responses.append(self.conversation.messages[-1])
-            self.conversation.response_counts += 1
-            self.service.status_tracker.num_tasks_succeeded += 1
-        else:
-            self.service.status_tracker.num_tasks_failed += 1
-            

lionagi/endpoints/__init__.py

@@ -1,5 +0,0 @@
-from .chatcompletion import ChatCompletion
-
-__all__ = [
-    "ChatCompletion",
-]

lionagi/endpoints/audio.py

@@ -1,17 +0,0 @@
-# from .base_endpoint import BaseEndpoint
-
-
-# class Audio(BaseEndpoint):
-#     endpoint: str = "chat/completions"
-
-#     @classmethod
-#     def create_payload(scls, messages, llmconfig, schema, **kwargs):
-#         config = {**llmconfig, **kwargs}
-#         payload = {"messages": messages}
-#         for key in schema['required']:
-#             payload.update({key: config[key]})
-
-#         for key in schema['optional']:
-#             if bool(config[key]) is True and str(config[key]).lower() != "none":
-#                 payload.update({key: config[key]})
-#         return payload

lionagi/endpoints/chatcompletion.py

@@ -1,54 +0,0 @@
-from lionagi.objs.abc_objs import BaseEndpoint
-
-
-class ChatCompletion(BaseEndpoint):
-    """
-    Represents an endpoint for chat completions in an API.
-
-    This class is designed to handle the creation of payloads for chat completion requests. The 'endpoint' attribute specifies the API endpoint for chat completions.
-
-    Attributes:
-        endpoint (str): The API endpoint for chat completions.
-    """
-    endpoint: str = "chat/completions"
-
-    @classmethod
-    def create_payload(scls, messages, llmconfig, schema, **kwargs):
-        """
-        Creates a payload for a chat completion request using provided messages, configuration, and schema.
-
-        This method constructs a payload dictionary that includes required and optional parameters 
-        as specified in the schema. Required parameters are extracted from 'llmconfig' and 'kwargs', 
-        while optional parameters are included only if they are truthy and not equal to the string "none".
-
-        Parameters:
-            messages (list): A list of message objects to include in the payload.
-            llmconfig (dict): A dictionary containing configuration settings for the large language model.
-            schema (dict): A dictionary defining required and optional keys for the payload.
-                The 'required' key should map to a list of required parameter names.
-                The 'optional' key should map to a list of optional parameter names.
-            **kwargs: Additional keyword arguments that can override or supplement 'llmconfig'.
-
-        Returns:
-            dict: A dictionary representing the payload for the chat completion request.
-
-        Example:
-            payload = ChatCompletion.create_payload(
-                messages=[{"text": "Hello, world!"}],
-                llmconfig={"max_tokens": 100},
-                schema={"required": ["max_tokens"], "optional": ["temperature"]}
-            )
-        """
-        config = {**llmconfig, **kwargs}
-        payload = {"messages": messages}
-        for key in schema['required']:
-            payload.update({key: config[key]})
-
-        for key in schema['optional']:
-            if bool(config[key]) is True and str(config[key]).lower() != "none":
-                payload.update({key: config[key]})
-        return payload
-    
-    # def process_response(self, session, payload, completion):
-    #     ...
-