lionagi 0.0.306__py3-none-any.whl → 0.0.307__py3-none-any.whl

lionagi/integrations/bridge/llamaindex_/textnode.py

@@ -10,17 +10,17 @@ def to_llama_index_node(lion_node, node_type: Any = None, **kwargs: Any) -> Any:
     the expected Llama Index node schema, and then creates a Llama Index node object of the specified type.
 
     Args:
-        lion_node: The Lion node to convert. Must have a `to_dict` method.
-        node_type (Any, optional): The type of Llama Index node to create. Can be a string name of a node class
-            within the Llama Index schema or a class that inherits from `BaseNode`. Defaults to 'TextNode'.
-        **kwargs: Additional keyword arguments to be included in the Llama Index node's initialization.
+            lion_node: The Lion node to convert. Must have a `to_dict` method.
+            node_type (Any, optional): The type of Llama Index node to create. Can be a string name of a node class
+                    within the Llama Index schema or a class that inherits from `BaseNode`. Defaults to 'TextNode'.
+            **kwargs: Additional keyword arguments to be included in the Llama Index node's initialization.
 
     Returns:
-        Any: A new instance of the specified Llama Index node type populated with data from the Lion node.
+            Any: A new instance of the specified Llama Index node type populated with data from the Lion node.
 
     Raises:
-        TypeError: If `node_type` is neither a string nor a subclass of `BaseNode`.
-        AttributeError: If an error occurs due to an invalid node type or during the creation of the node object.
+            TypeError: If `node_type` is neither a string nor a subclass of `BaseNode`.
+            AttributeError: If an error occurs due to an invalid node type or during the creation of the node object.
     """
 
     SysUtil.check_import("llama_index", pip_name="llama-index")

lionagi/integrations/config/openrouter_configs.py

@@ -56,7 +56,6 @@ openrouter_finetune_schema = {
     "config": openrouter_finetune_llmconfig,
 }
 
-
 openrouter_schema = {
     "chat/completions": openrouter_chat_schema,
     "finetune": openrouter_finetune_schema,

lionagi/integrations/provider/oai.py

@@ -8,19 +8,19 @@ class OpenAIService(BaseService):
     A service to interact with OpenAI's API endpoints.
 
     Attributes:
-        base_url (str): The base URL for the OpenAI API.
-        available_endpoints (list): A list of available API endpoints.
-        schema (dict): The schema configuration for the API.
-        key_scheme (str): The environment variable name for OpenAI API key.
-        token_encoding_name (str): The default token encoding scheme.
+            base_url (str): The base URL for the OpenAI API.
+            available_endpoints (list): A list of available API endpoints.
+            schema (dict): The schema configuration for the API.
+            key_scheme (str): The environment variable name for OpenAI API key.
+            token_encoding_name (str): The default token encoding scheme.
 
     Examples:
-        >>> service = OpenAIService(api_key="your_api_key")
-        >>> asyncio.run(service.serve("Hello, world!","chat/completions"))
-        (payload, completion)
+            >>> service = OpenAIService(api_key="your_api_key")
+            >>> asyncio.run(service.serve("Hello, world!","chat/completions"))
+            (payload, completion)
 
-        >>> service = OpenAIService()
-        >>> asyncio.run(service.serve("Convert this text to speech.","audio_speech"))
+            >>> service = OpenAIService()
+            >>> asyncio.run(service.serve("Convert this text to speech.","audio_speech"))
     """
 
     base_url = "https://api.openai.com/v1/"
@@ -57,25 +57,25 @@ class OpenAIService(BaseService):
         Serves the input using the specified endpoint and method.
 
         Args:
-            input_: The input text to be processed.
-            endpoint: The API endpoint to use for processing.
-            method: The HTTP method to use for the request.
-            **kwargs: Additional keyword arguments to pass to the payload creation.
+                input_: The input text to be processed.
+                endpoint: The API endpoint to use for processing.
+                method: The HTTP method to use for the request.
+                **kwargs: Additional keyword arguments to pass to the payload creation.
 
         Returns:
-            A tuple containing the payload and the completion assistant_response from the API.
+                A tuple containing the payload and the completion assistant_response from the API.
 
         Raises:
-            ValueError: If the specified endpoint is not supported.
+                ValueError: If the specified endpoint is not supported.
 
         Examples:
-            >>> service = OpenAIService(api_key="your_api_key")
-            >>> asyncio.run(service.serve("Hello, world!","chat/completions"))
-            (payload, completion)
+                >>> service = OpenAIService(api_key="your_api_key")
+                >>> asyncio.run(service.serve("Hello, world!","chat/completions"))
+                (payload, completion)
 
-            >>> service = OpenAIService()
-            >>> asyncio.run(service.serve("Convert this text to speech.","audio_speech"))
-            ValueError: 'audio_speech' is currently not supported
+                >>> service = OpenAIService()
+                >>> asyncio.run(service.serve("Convert this text to speech.","audio_speech"))
+                ValueError: 'audio_speech' is currently not supported
         """
         if endpoint not in self.active_endpoint:
             await self.init_endpoint(endpoint)
@@ -89,14 +89,14 @@ class OpenAIService(BaseService):
         Serves the chat completion request with the given messages.
 
         Args:
-            messages: The messages to be included in the chat completion.
-            **kwargs: Additional keyword arguments for payload creation.
+                messages: The messages to be included in the chat completion.
+                **kwargs: Additional keyword arguments for payload creation.
 
         Returns:
-            A tuple containing the payload and the completion assistant_response from the API.
+                A tuple containing the payload and the completion assistant_response from the API.
 
         Raises:
-            Exception: If the API call fails.
+                Exception: If the API call fails.
         """
         if "chat/completions" not in self.active_endpoint:
             await self.init_endpoint("chat/completions")

lionagi/integrations/provider/services.py

@@ -6,15 +6,15 @@ class Services:
         A provider to interact with OpenAI's API endpoints.
 
         Attributes:
-            api_key (Optional[str]): The API key used for authentication.
-            schema (Dict[str, Any]): The schema defining the provider's endpoints.
-            status_tracker (StatusTracker): The object tracking the status of API calls.
-            endpoints (Dict[str, EndPoint]): A dictionary of endpoint objects.
-            base_url (str): The base URL for the OpenAI API.
-            available_endpoints (list): A list of available API endpoints, including
-                'chat/completions'
-            key_scheme (str): The environment variable name for API key.
-            token_encoding_name (str): The default token encoding scheme.
+                api_key (Optional[str]): The API key used for authentication.
+                schema (Dict[str, Any]): The schema defining the provider's endpoints.
+                status_tracker (StatusTracker): The object tracking the status of API calls.
+                endpoints (Dict[str, EndPoint]): A dictionary of endpoint objects.
+                base_url (str): The base URL for the OpenAI API.
+                available_endpoints (list): A list of available API endpoints, including
+                        'chat/completions'
+                key_scheme (str): The environment variable name for API key.
+                token_encoding_name (str): The default token encoding scheme.
         """
 
         from lionagi.integrations.provider.oai import OpenAIService
@@ -27,15 +27,15 @@ class Services:
         A provider to interact with OpenRouter's API endpoints.
 
         Attributes:
-            api_key (Optional[str]): The API key used for authentication.
-            schema (Dict[str, Any]): The schema defining the provider's endpoints.
-            status_tracker (StatusTracker): The object tracking the status of API calls.
-            endpoints (Dict[str, EndPoint]): A dictionary of endpoint objects.
-            base_url (str): The base URL for the OpenAI API.
-            available_endpoints (list): A list of available API endpoints, including
-                'chat/completions'
-            key_scheme (str): The environment variable name for API key.
-            token_encoding_name (str): The default token encoding scheme.
+                api_key (Optional[str]): The API key used for authentication.
+                schema (Dict[str, Any]): The schema defining the provider's endpoints.
+                status_tracker (StatusTracker): The object tracking the status of API calls.
+                endpoints (Dict[str, EndPoint]): A dictionary of endpoint objects.
+                base_url (str): The base URL for the OpenAI API.
+                available_endpoints (list): A list of available API endpoints, including
+                        'chat/completions'
+                key_scheme (str): The environment variable name for API key.
+                token_encoding_name (str): The default token encoding scheme.
         """
 
         from lionagi.integrations.provider.openrouter import OpenRouterService
@@ -48,24 +48,24 @@ class Services:
         A provider to interact with Transformers' pipeline
 
         Attributes:
-            task (str): The specific task to be performed by the transformer model.
-                Currently, only 'conversational' tasks are supported.
-            model (Union[str, Any]): Identifier for the transformer model to be used. This
-                can be a model name or a path to a model.
-            config (Union[str, Dict, Any]): Configuration for the transformer model. Can
-                include tokenizer information among others.
-            pipe (pipeline): The loaded transformer pipeline for the specified task, model,
-                and configuration.
+                task (str): The specific task to be performed by the transformer model.
+                        Currently, only 'conversational' tasks are supported.
+                model (Union[str, Any]): Identifier for the transformer model to be used. This
+                        can be a model name or a path to a model.
+                config (Union[str, Dict, Any]): Configuration for the transformer model. Can
+                        include tokenizer information among others.
+                pipe (pipeline): The loaded transformer pipeline for the specified task, model,
+                        and configuration.
 
         Warnings:
-            - Ensure the selected model is suitable for conversational tasks to avoid
-            unexpected behavior.
-            - As this provider heavily relies on external libraries (Hugging Face's
-            Transformers), ensure they are installed and updated to compatible versions.
+                - Ensure the selected model is suitable for conversational tasks to avoid
+                unexpected behavior.
+                - As this provider heavily relies on external libraries (Hugging Face's
+                Transformers), ensure they are installed and updated to compatible versions.
 
         Dependencies:
-            - Requires the `transformers` library by Hugging Face and `asyncio` for
-            asynchronous operations.
+                - Requires the `transformers` library by Hugging Face and `asyncio` for
+                asynchronous operations.
         """
 
         from lionagi.integrations.provider.transformers import TransformersService
@@ -99,8 +99,8 @@ class Services:
         A provider to interact with Ollama
 
         Attributes:
-            model (str): name of the model to use
-            kwargs (Optional[Any]): additional kwargs for calling the model
+                model (str): name of the model to use
+                kwargs (Optional[Any]): additional kwargs for calling the model
         """
 
         from lionagi.integrations.provider.ollama import OllamaService
@@ -113,8 +113,8 @@ class Services:
         A provider to interact with Litellm
 
         Attributes:
-            model (str): name of the model to use
-            kwargs (Optional[Any]): additional kwargs for calling the model
+                model (str): name of the model to use
+                kwargs (Optional[Any]): additional kwargs for calling the model
         """
 
         from .litellm import LiteLLMService
@@ -127,8 +127,8 @@ class Services:
         A provider to interact with MlX
 
         Attributes:
-            model (str): name of the model to use
-            kwargs (Optional[Any]): additional kwargs for calling the model
+                model (str): name of the model to use
+                kwargs (Optional[Any]): additional kwargs for calling the model
         """
 
         from lionagi.integrations.provider.mlx_service import MlXService

lionagi/libs/__init__.py

@@ -1 +1,34 @@
-from . import *
+from lionagi.libs.sys_util import SysUtil
+from lionagi.libs.ln_async import AsyncUtil
+
+import lionagi.libs.ln_convert as convert
+import lionagi.libs.ln_dataframe as dataframe
+import lionagi.libs.ln_func_call as func_call
+from lionagi.libs.ln_func_call import CallDecorator
+import lionagi.libs.ln_nested as nested
+from lionagi.libs.ln_parse import ParseUtil, StringMatch
+
+from lionagi.libs.ln_api import (
+    APIUtil,
+    SimpleRateLimiter,
+    StatusTracker,
+    BaseService,
+    PayloadPackage,
+)
+
+__all__ = [
+    "SysUtil",
+    "convert",
+    "func_call",
+    "dataframe",
+    "nested",
+    "AsyncUtil",
+    "ParseUtil",
+    "StringMatch",
+    "APIUtil",
+    "BaseService",
+    "PayloadPackage",
+    "StatusTracker",
+    "SimpleRateLimiter",
+    "CallDecorator",
+]