swarmauri 0.5.3.dev1__tar.gz → 0.5.3.dev4__tar.gz

{swarmauri-0.5.3.dev1 → swarmauri-0.5.3.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: swarmauri
-Version: 0.5.3.dev1
+Version: 0.5.3.dev4
 Summary: This repository includes base classes, concrete generics, and concrete standard components within the Swarmauri framework.
 Home-page: http://github.com/swarmauri/swarmauri-sdk
 License: Apache-2.0
@@ -21,7 +21,7 @@ Requires-Dist: Pillow (>=8.0,<11.0)
 Requires-Dist: aiofiles (==24.1.0) ; extra == "io" or extra == "full"
 Requires-Dist: aiohttp (>=3.10.10,<4.0.0) ; extra == "io" or extra == "full"
 Requires-Dist: beautifulsoup4 (==04.12.3) ; extra == "nlp-tools" or extra == "full"
-Requires-Dist: httpx (>=0.27.2,<0.28.0)
+Requires-Dist: httpx (>=0.25.0,<0.26.0)
 Requires-Dist: joblib (>=1.4.0,<2.0.0)
 Requires-Dist: matplotlib (>=3.9.2) ; extra == "visualization" or extra == "full"
 Requires-Dist: numpy
@@ -29,7 +29,7 @@ Requires-Dist: pandas
 Requires-Dist: pydantic (>=2.9.2,<3.0.0)
 Requires-Dist: requests
 Requires-Dist: scipy (>=1.7.0,<1.14.0)
-Requires-Dist: swarmauri_core (==0.5.2)
+Requires-Dist: swarmauri_core (==0.5.3.dev3)
 Requires-Dist: toml (>=0.10.2,<0.11.0)
 Requires-Dist: typing_extensions
 Requires-Dist: yake (==0.4.8) ; extra == "nlp" or extra == "full"

{swarmauri-0.5.3.dev1 → swarmauri-0.5.3.dev4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "swarmauri"
-version = "0.5.3.dev1"
+version = "0.5.3.dev4"
 description = "This repository includes base classes, concrete generics, and concrete standard components within the Swarmauri framework."
 authors = ["Jacob Stewart <jacob@swarmauri.com>"]
 license = "Apache-2.0"
@@ -15,9 +15,9 @@ classifiers = [
 
 [tool.poetry.dependencies]
 python = ">=3.10,<3.13"
-swarmauri_core = "==0.5.2"
+swarmauri_core = "==0.5.3.dev3"
 toml = "^0.10.2"
-httpx = "^0.27.2"
+httpx = "^0.25.0"
 joblib = "^1.4.0"
 numpy = "*"
 pandas = "*"
@@ -52,7 +52,11 @@ matplotlib = { version = ">=3.9.2", optional = true }
 # Extras without versioning, grouped for specific use cases
 io = ["aiofiles", "aiohttp"]
 #llms = ["cohere", "mistralai", "fal-client", "google-generativeai", "openai"]
-nlp = ["nltk", "textblob", "yake"]
+nlp = [
+    #"nltk", 
+    #"textblob", 
+    "yake"
+]
 nlp_tools = ["beautifulsoup4"]
 #ml_toolkits = ["gensim", "scipy", "scikit-learn"]
 #spacy = ["spacy"]

swarmauri-0.5.3.dev4/swarmauri/agent_factories/concrete/__init__.py

@@ -0,0 +1,22 @@
+from swarmauri.utils._lazy_import import _lazy_import
+
+# List of agent factory names (file names without the ".py" extension) and corresponding class names
+agent_factory_files = [
+    ("swarmauri.agent_factories.concrete.agent_factory", "AgentFactory"),
+    (
+        "swarmauri.agent_factories.concrete.conf_driven_agent_factory",
+        "ConfDrivenAgentFactory",
+    ),
+    ("swarmauri.agent_factories.concrete.JsonAgentFactory", "JsonAgentFactory"),
+    (
+        "swarmauri.agent_factories.concrete.ReflectionAgentFactory",
+        "ReflectionAgentFactory",
+    ),
+]
+
+# Lazy loading of agent factories storing them in variables
+for module_name, class_name in agent_factory_files:
+    globals()[class_name] = _lazy_import(module_name, class_name)
+
+# Adding the lazy-loaded agent factories to __all__
+__all__ = [class_name for _, class_name in agent_factory_files]

swarmauri-0.5.3.dev4/swarmauri/agents/concrete/__init__.py

@@ -0,0 +1,16 @@
+from swarmauri.utils._lazy_import import _lazy_import
+
+# List of agent names (file names without the ".py" extension) and corresponding class names
+agent_files = [
+    ("swarmauri.agents.concrete.QAAgent", "QAAgent"),
+    ("swarmauri.agents.concrete.RagAgent", "RagAgent"),
+    ("swarmauri.agents.concrete.SimpleConversationAgent", "SimpleConversationAgent"),
+    ("swarmauri.agents.concrete.ToolAgent", "ToolAgent"),
+]
+
+# Lazy loading of agent classes, storing them in variables
+for module_name, class_name in agent_files:
+    globals()[class_name] = _lazy_import(module_name, class_name)
+
+# Adding the lazy-loaded agent classes to __all__
+__all__ = [class_name for _, class_name in agent_files]

swarmauri-0.5.3.dev4/swarmauri/chains/concrete/__init__.py

@@ -0,0 +1,15 @@
+from swarmauri.utils._lazy_import import _lazy_import
+
+chains_files = [
+    ("swarmauri.chains.concrete.CallableChain import", "CallableChain"),
+    ("swarmauri.chains.concrete.ChainStep", "ChainStep"),
+    ("swarmauri.chains.concrete.PromptContextChain", "PromptContextChain"),
+    ("swarmauri.chains.concrete.ContextChain", "ContextChain"),
+]
+
+# Lazy loading of chain classes, storing them in variables
+for module_name, class_name in chains_files:
+    globals()[class_name] = _lazy_import(module_name, class_name)
+
+# Adding the lazy-loaded chain classes to __all__
+__all__ = [class_name for _, class_name in chains_files]

swarmauri-0.5.3.dev4/swarmauri/chunkers/concrete/__init__.py

@@ -0,0 +1,17 @@
+from swarmauri.utils._lazy_import import _lazy_import
+
+# List of chunker names (file names without the ".py" extension) and corresponding class names
+chunkers_files = [
+    ("swarmauri.chunkers.concrete.DelimiterBasedChunker", "DelimiterBasedChunker"),
+    ("swarmauri.chunkers.concrete.FixedLengthChunker", "FixedLengthChunker"),
+    ("swarmauri.chunkers.concrete.MdSnippetChunker", "MdSnippetChunker"),
+    ("swarmauri.chunkers.concrete.SentenceChunker", "SentenceChunker"),
+    ("swarmauri.chunkers.concrete.SlidingWindowChunker", "SlidingWindowChunker"),
+]
+
+# Lazy loading of chunker classes, storing them in variables
+for module_name, class_name in chunkers_files:
+    globals()[class_name] = _lazy_import(module_name, class_name)
+
+# Adding the lazy-loaded chunker classes to __all__
+__all__ = [class_name for _, class_name in chunkers_files]

swarmauri-0.5.3.dev4/swarmauri/conversations/concrete/__init__.py

@@ -0,0 +1,22 @@
+from swarmauri.utils._lazy_import import _lazy_import
+
+# List of conversations names (file names without the ".py" extension) and corresponding class names
+conversations_files = [
+    ("swarmauri.conversations.concrete.Conversation", "Conversation"),
+    (
+        "swarmauri.conversations.concrete.MaxSystemContextConversation",
+        "MaxSystemContextConversation",
+    ),
+    ("swarmauri.conversations.concrete.MaxSizeConversation", "MaxSizeConversation"),
+    (
+        "swarmauri.conversations.concrete.SessionCacheConversation",
+        "SessionCacheConversation",
+    ),
+]
+
+# Lazy loading of conversations classes, storing them in variables
+for module_name, class_name in conversations_files:
+    globals()[class_name] = _lazy_import(module_name, class_name)
+
+# Adding the lazy-loaded conversations classes to __all__
+__all__ = [class_name for _, class_name in conversations_files]

swarmauri-0.5.3.dev4/swarmauri/distances/concrete/__init__.py

@@ -0,0 +1,27 @@
+from swarmauri.utils._lazy_import import _lazy_import
+
+# List of distances names (file names without the ".py" extension) and corresponding class names
+distances_files = [
+    ("swarmauri.distances.concrete.CanberraDistance", "CanberraDistance"),
+    ("swarmauri.distances.concrete.ChebyshevDistance", "ChebyshevDistance"),
+    ("swarmauri.distances.concrete.ChiSquaredDistance", "ChiSquaredDistance"),
+    ("swarmauri.distances.concrete.CosineDistance", "CosineDistance"),
+    ("swarmauri.distances.concrete.EuclideanDistance", "EuclideanDistance"),
+    ("swarmauri.distances.concrete.HaversineDistance", "HaversineDistance"),
+    ("swarmauri.distances.concrete.JaccardIndexDistance", "JaccardIndexDistance"),
+    ("swarmauri.distances.concrete.LevenshteinDistance", "LevenshteinDistance"),
+    ("swarmauri.distances.concrete.ManhattanDistance", "ManhattanDistance"),
+    ("swarmauri.distances.concrete.MinkowskiDistance", "MinkowskiDistance"),
+    ("swarmauri.distances.concrete.SorensenDiceDistance", "SorensenDiceDistance"),
+    (
+        "swarmauri.distances.concrete.SquaredEuclideanDistance",
+        "SquaredEuclideanDistance",
+    ),
+]
+
+# Lazy loading of distances classes, storing them in variables
+for module_name, class_name in distances_files:
+    globals()[class_name] = _lazy_import(module_name, class_name)
+
+# Adding the lazy-loaded distances classes to __all__
+__all__ = [class_name for _, class_name in distances_files]

swarmauri-0.5.3.dev4/swarmauri/documents/concrete/__init__.py

@@ -0,0 +1,11 @@
+from swarmauri.utils._lazy_import import _lazy_import
+
+# List of documents names (file names without the ".py" extension) and corresponding class names
+documents_files = [("swarmauri.documents.concrete.Document", "Document")]
+
+# Lazy loading of documents classes, storing them in variables
+for module_name, class_name in documents_files:
+    globals()[class_name] = _lazy_import(module_name, class_name)
+
+# Adding the lazy-loaded documents classes to __all__
+__all__ = [class_name for _, class_name in documents_files]

swarmauri-0.5.3.dev4/swarmauri/embeddings/concrete/__init__.py

@@ -0,0 +1,19 @@
+from swarmauri.utils._lazy_import import _lazy_import
+
+# List of embeddings names (file names without the ".py" extension) and corresponding class names
+embeddings_files = [
+    ("swarmauri.embeddings.concrete.CohereEmbedding", "CohereEmbedding"),
+    ("swarmauri.embeddings.concrete.GeminiEmbedding", "GeminiEmbedding"),
+    ("swarmauri.embeddings.concrete.MistralEmbedding", "MistralEmbedding"),
+    ("swarmauri.embeddings.concrete.NmfEmbedding", "NmfEmbedding"),
+    ("swarmauri.embeddings.concrete.OpenAIEmbedding", "OpenAIEmbedding"),
+    ("swarmauri.embeddings.concrete.TfidfEmbedding", "TfidfEmbedding"),
+    ("swarmauri.embeddings.concrete.VoyageEmbedding", "VoyageEmbedding"),
+]
+
+# Lazy loading of embeddings classes, storing them in variables
+for module_name, class_name in embeddings_files:
+    globals()[class_name] = _lazy_import(module_name, class_name)
+
+# Adding the lazy-loaded embeddings classes to __all__
+__all__ = [class_name for _, class_name in embeddings_files]

swarmauri-0.5.3.dev4/swarmauri/exceptions/concrete/__init__.py

@@ -0,0 +1,13 @@
+from swarmauri.utils._lazy_import import _lazy_import
+
+# List of exceptions names (file names without the ".py" extension) and corresponding class names
+exceptions_files = [
+    ("swarmauri.exceptions.concrete.IndexErrorWithContext", "IndexErrorWithContext"),
+]
+
+# Lazy loading of exceptions classes, storing them in variables
+for module_name, class_name in exceptions_files:
+    globals()[class_name] = _lazy_import(module_name, class_name)
+
+# Adding the lazy-loaded exceptions classes to __all__
+__all__ = [class_name for _, class_name in exceptions_files]

swarmauri-0.5.3.dev4/swarmauri/image_gens/base/ImageGenBase.py

@@ -0,0 +1,51 @@
+from abc import abstractmethod
+from typing import Optional, List, Literal
+from pydantic import ConfigDict, model_validator, Field
+from swarmauri_core.image_gens.IGenImage import IGenImage
+from swarmauri_core.ComponentBase import ComponentBase, ResourceTypes
+
+
+class ImageGenBase(IGenImage, ComponentBase):
+    allowed_models: List[str] = []
+    resource: Optional[str] = Field(default=ResourceTypes.IMAGE_GEN.value, frozen=True)
+    model_config = ConfigDict(extra="forbid", arbitrary_types_allowed=True)
+    type: Literal["ImageGenBase"] = "ImageGenBase"
+
+    @model_validator(mode="after")
+    @classmethod
+    def _validate_name_in_allowed_models(cls, values):
+        name = values.name
+        allowed_models = values.allowed_models
+        if name and name not in allowed_models:
+            raise ValueError(
+                f"Model name {name} is not allowed. Choose from {allowed_models}"
+            )
+        return values
+
+    @abstractmethod
+    def generate_image(self, *args, **kwargs) -> any:
+        """
+        Generate images based on the input data provided to the model.
+        """
+        raise NotImplementedError("generate_image() not implemented in subclass yet.")
+
+    @abstractmethod
+    async def agenerate_image(self, *args, **kwargs) -> any:
+        """
+        Generate images based on the input data provided to the model.
+        """
+        raise NotImplementedError("agenerate_image() not implemented in subclass yet.")
+
+    @abstractmethod
+    def batch_generate(self, *args, **kwargs) -> any:
+        """
+        Generate images based on the input data provided to the model.
+        """
+        raise NotImplementedError("batch_generate() not implemented in subclass yet.")
+
+    @abstractmethod
+    async def abatch_generate(self, *args, **kwargs) -> any:
+        """
+        Generate images based on the input data provided to the model.
+        """
+        raise NotImplementedError("abatch_generate() not implemented in subclass yet.")

{swarmauri-0.5.3.dev1/swarmauri/llms → swarmauri-0.5.3.dev4/swarmauri/image_gens}/concrete/BlackForestImgGenModel.py

@@ -1,35 +1,35 @@
 import httpx
 import time
-from typing import List, Literal, Optional, Dict, ClassVar
+from typing import List, Literal, Optional, Dict
 from pydantic import PrivateAttr
 from swarmauri.utils.retry_decorator import retry_on_status_codes
-from swarmauri.llms.base.LLMBase import LLMBase
+from swarmauri.image_gens.base.ImageGenBase import ImageGenBase
 import asyncio
 import contextlib
 
 
-class BlackForestImgGenModel(LLMBase):
+class BlackForestImgGenModel(ImageGenBase):
     """
     A model for generating images using FluxPro's image generation models through the Black Forest API.
     Link to API key: https://api.bfl.ml/auth/profile
     """
 
     _BASE_URL: str = PrivateAttr("https://api.bfl.ml")
-    _client: httpx.Client = PrivateAttr()
+    _client: httpx.Client = PrivateAttr(default=None)
     _async_client: httpx.AsyncClient = PrivateAttr(default=None)
+    _headers: Dict[str, str] = PrivateAttr(default=None)
 
     api_key: str
     allowed_models: List[str] = ["flux-pro-1.1", "flux-pro", "flux-dev"]
 
-    asyncio: ClassVar = asyncio
     name: str = "flux-pro"  # Default model
     type: Literal["BlackForestImgGenModel"] = "BlackForestImgGenModel"
 
-    def __init__(self, **data):
+    def __init__(self, **kwargs):
         """
         Initializes the BlackForestImgGenModel instance with HTTP clients.
         """
-        super().__init__(**data)
+        super().__init__(**kwargs)
         self._headers = {
             "Content-Type": "application/json",
             "X-Key": self.api_key,

{swarmauri-0.5.3.dev1/swarmauri/llms → swarmauri-0.5.3.dev4/swarmauri/image_gens}/concrete/DeepInfraImgGenModel.py

@@ -2,12 +2,12 @@ import httpx
 from typing import List, Literal
 from pydantic import PrivateAttr
 from swarmauri.utils.retry_decorator import retry_on_status_codes
-from swarmauri.llms.base.LLMBase import LLMBase
+from swarmauri.image_gens.base.ImageGenBase import ImageGenBase
 import asyncio
 import contextlib
 
 
-class DeepInfraImgGenModel(LLMBase):
+class DeepInfraImgGenModel(ImageGenBase):
     """
     A model class for generating images from text prompts using DeepInfra's image generation API.
 
@@ -37,7 +37,7 @@ class DeepInfraImgGenModel(LLMBase):
     name: str = "stabilityai/stable-diffusion-2-1"  # Default model
     type: Literal["DeepInfraImgGenModel"] = "DeepInfraImgGenModel"
 
-    def __init__(self, **data):
+    def __init__(self, **kwargs):
         """
         Initializes the DeepInfraImgGenModel instance.
 
@@ -47,7 +47,7 @@ class DeepInfraImgGenModel(LLMBase):
         Args:
             **data: Keyword arguments for model initialization.
         """
-        super().__init__(**data)
+        super().__init__(**kwargs)
         self._headers = {
             "Content-Type": "application/json",
             "Authorization": f"Bearer {self.api_key}",

{swarmauri-0.5.3.dev1/swarmauri/llms → swarmauri-0.5.3.dev4/swarmauri/image_gens}/concrete/FalAIImgGenModel.py

@@ -3,11 +3,11 @@ import asyncio
 from typing import List, Literal, Optional, Dict
 from pydantic import Field, PrivateAttr
 from swarmauri.utils.retry_decorator import retry_on_status_codes
-from swarmauri.llms.base.LLMBase import LLMBase
+from swarmauri.image_gens.base.ImageGenBase import ImageGenBase
 import time
 
 
-class FalAIImgGenModel(LLMBase):
+class FalAIImgGenModel(ImageGenBase):
     """
     A model class for generating images from text using FluxPro's image generation model,
     provided by FalAI. This class uses a queue-based API to handle image generation requests.
@@ -34,7 +34,7 @@ class FalAIImgGenModel(LLMBase):
     max_retries: int = Field(default=60)  # Maximum number of status check retries
     retry_delay: float = Field(default=1.0)  # Delay between status checks in seconds
 
-    def __init__(self, **data):
+    def __init__(self, **kwargs):
         """
         Initializes the model with the specified API key and model name.
 
@@ -44,7 +44,7 @@ class FalAIImgGenModel(LLMBase):
         Raises:
             ValueError: If an invalid model name is provided.
         """
-        super().__init__(**data)
+        super().__init__(**kwargs)
         self._headers = {
             "Content-Type": "application/json",
             "Authorization": f"Key {self.api_key}",

{swarmauri-0.5.3.dev1/swarmauri/llms → swarmauri-0.5.3.dev4/swarmauri/image_gens}/concrete/HyperbolicImgGenModel.py

@@ -1,13 +1,13 @@
 import httpx
-from typing import List, Literal, Optional
+from typing import List, Literal
 from pydantic import PrivateAttr
 from swarmauri.utils.retry_decorator import retry_on_status_codes
-from swarmauri.llms.base.LLMBase import LLMBase
+from swarmauri.image_gens.base.ImageGenBase import ImageGenBase
 import asyncio
 import contextlib
 
 
-class HyperbolicImgGenModel(LLMBase):
+class HyperbolicImgGenModel(ImageGenBase):
     """
     A model class for generating images from text prompts using Hyperbolic's image generation API.
 
@@ -29,7 +29,7 @@ class HyperbolicImgGenModel(LLMBase):
     """
 
     _BASE_URL: str = PrivateAttr("https://api.hyperbolic.xyz/v1/image/generation")
-    _client: httpx.Client = PrivateAttr()
+    _client: httpx.Client = PrivateAttr(default=None)
     _async_client: httpx.AsyncClient = PrivateAttr(default=None)
 
     api_key: str
@@ -52,7 +52,7 @@ class HyperbolicImgGenModel(LLMBase):
     enable_refiner: bool = False
     backend: str = "auto"
 
-    def __init__(self, **data):
+    def __init__(self, **kwargs):
         """
         Initializes the HyperbolicImgGenModel instance.
 
@@ -62,7 +62,7 @@ class HyperbolicImgGenModel(LLMBase):
         Args:
             **data: Keyword arguments for model initialization.
         """
-        super().__init__(**data)
+        super().__init__(**kwargs)
         self._headers = {
             "Content-Type": "application/json",
             "Authorization": f"Bearer {self.api_key}",

{swarmauri-0.5.3.dev1/swarmauri/llms → swarmauri-0.5.3.dev4/swarmauri/image_gens}/concrete/OpenAIImgGenModel.py

@@ -2,11 +2,11 @@ from pydantic import PrivateAttr
 import asyncio
 import httpx
 from typing import Dict, List, Literal, Optional
+from swarmauri.image_gens.base.ImageGenBase import ImageGenBase
 from swarmauri.utils.retry_decorator import retry_on_status_codes
-from swarmauri.llms.base.LLMBase import LLMBase
 
 
-class OpenAIImgGenModel(LLMBase):
+class OpenAIImgGenModel(ImageGenBase):
     """
     OpenAIImgGenModel is a class for generating images using OpenAI's DALL-E models.
 
@@ -26,14 +26,14 @@ class OpenAIImgGenModel(LLMBase):
     _BASE_URL: str = PrivateAttr(default="https://api.openai.com/v1/images/generations")
     _headers: Dict[str, str] = PrivateAttr(default=None)
 
-    def __init__(self, **data) -> None:
+    def __init__(self, **kwargs) -> None:
         """
         Initialize the GroqAIAudio class with the provided data.
 
         Args:
             **data: Arbitrary keyword arguments containing initialization data.
         """
-        super().__init__(**data)
+        super().__init__(**kwargs)
         self._headers = {
             "Authorization": f"Bearer {self.api_key}",
             "Content-Type": "application/json",

swarmauri-0.5.3.dev4/swarmauri/llms/concrete/__init__.py

@@ -0,0 +1,43 @@
+from swarmauri.utils._lazy_import import _lazy_import
+
+# List of llms names (file names without the ".py" extension) and corresponding class names
+llms_files = [
+    ("swarmauri.llms.concrete.AI21StudioModel", "AI21StudioModel"),
+    ("swarmauri.llms.concrete.AnthropicModel", "AnthropicModel"),
+    ("swarmauri.llms.concrete.AnthropicToolModel", "AnthropicToolModel"),
+    ("swarmauri.llms.concrete.BlackForestImgGenModel", "BlackForestImgGenModel"),
+    ("swarmauri.llms.concrete.CohereModel", "CohereModel"),
+    ("swarmauri.llms.concrete.CohereToolModel", "CohereToolModel"),
+    ("swarmauri.llms.concrete.DeepInfraImgGenModel", "DeepInfraImgGenModel"),
+    ("swarmauri.llms.concrete.DeepInfraModel", "DeepInfraModel"),
+    ("swarmauri.llms.concrete.DeepSeekModel", "DeepSeekModel"),
+    ("swarmauri.llms.concrete.FalAIImgGenModel", "FalaiImgGenModel"),
+    ("swarmauri.llms.concrete.FalAIVisionModel", "FalAIVisionModel"),
+    ("swarmauri.llms.concrete.GeminiProModel", "GeminiProModel"),
+    ("swarmauri.llms.concrete.GeminiToolModel", "GeminiToolModel"),
+    ("swarmauri.llms.concrete.GroqAIAudio", "GroqAIAudio"),
+    ("swarmauri.llms.concrete.GroqModel", "GroqModel"),
+    ("swarmauri.llms.concrete.GroqToolModel", "GroqToolModel"),
+    ("swarmauri.llms.concrete.GroqVisionModel", "GroqVisionModel"),
+    ("swarmauri.llms.concrete.HyperbolicAudioTTS", "HyperbolicAudioTTS"),
+    ("swarmauri.llms.concrete.HyperbolicImgGenModel", "HyperbolicImgGenModel"),
+    ("swarmauri.llms.concrete.HyperbolicModel", "HyperbolicModel"),
+    ("swarmauri.llms.concrete.HyperbolicVisionModel", "HyperbolicVisionModel"),
+    ("swarmauri.llms.concrete.MistralModel", "MistralModel"),
+    ("swarmauri.llms.concrete.MistralToolModel", "MistralToolModel"),
+    ("swarmauri.llms.concrete.OpenAIAudio", "OpenAIAudio"),
+    ("swarmauri.llms.concrete.OpenAIAudioTTS", "OpenAIAudioTTS"),
+    ("swarmauri.llms.concrete.OpenAIImgGenModel", "OpenAIImgGenModel"),
+    ("swarmauri.llms.concrete.OpenAIModel", "OpenAIModel"),
+    ("swarmauri.llms.concrete.OpenAIToolModel", "OpenAIToolModel"),
+    ("swarmauri.llms.concrete.PerplexityModel", "PerplexityModel"),
+    ("swarmauri.llms.concrete.PlayHTModel", "PlayHTModel"),
+    ("swarmauri.llms.concrete.WhisperLargeModel", "WhisperLargeModel"),
+]
+
+# Lazy loading of llms classes, storing them in variables
+for module_name, class_name in llms_files:
+    globals()[class_name] = _lazy_import(module_name, class_name)
+
+# Adding the lazy-loaded llms classes to __all__
+__all__ = [class_name for _, class_name in llms_files]

swarmauri-0.5.3.dev4/swarmauri/measurements/concrete/__init__.py

@@ -0,0 +1,41 @@
+from swarmauri.utils._lazy_import import _lazy_import
+
+# List of measurements names (file names without the ".py" extension) and corresponding class names
+measurements_files = [
+    (
+        "swarmauri.measurements.concrete.CompletenessMeasurement",
+        "CompletenessMeasurement",
+    ),
+    (
+        "swarmauri.measurements.concrete.DistinctivenessMeasurement",
+        "DistinctivenessMeasurement",
+    ),
+    (
+        "swarmauri.measurements.concrete.FirstImpressionMeasurement",
+        "FirstImpressionMeasurement",
+    ),
+    ("swarmauri.measurements.concrete.MeanMeasurement", "MeanMeasurement"),
+    ("swarmauri.measurements.concrete.MiscMeasurement", "MiscMeasurement"),
+    (
+        "swarmauri.measurements.concrete.MissingnessMeasurement",
+        "MissingnessMeasurement",
+    ),
+    (
+        "swarmauri.measurements.concrete.PatternMatchingMeasurement",
+        "PatternMatchingMeasurement",
+    ),
+    (
+        "swarmauri.measurements.concrete.RatioOfSumsMeasurement",
+        "RatioOfSumsMeasurement",
+    ),
+    ("swarmauri.measurements.concrete.StaticMeasurement", "StaticMeasurement"),
+    ("swarmauri.measurements.concrete.UniquenessMeasurement", "UniquenessMeasurement"),
+    ("swarmauri.measurements.concrete.ZeroMeasurement", "ZeroMeasurement"),
+]
+
+# Lazy loading of measurements classes, storing them in variables
+for module_name, class_name in measurements_files:
+    globals()[class_name] = _lazy_import(module_name, class_name)
+
+# Adding the lazy-loaded measurements classes to __all__
+__all__ = [class_name for _, class_name in measurements_files]

swarmauri-0.5.3.dev4/swarmauri/messages/concrete/__init__.py

@@ -0,0 +1,16 @@
+from swarmauri.utils._lazy_import import _lazy_import
+
+# List of messages names (file names without the ".py" extension) and corresponding class names
+messages_files = [
+    ("swarmauri.messages.concrete.HumanMessage", "HumanMessage"),
+    ("swarmauri.messages.concrete.AgentMessage", "AgentMessage"),
+    ("from swarmauri.messages.concrete.FunctionMessage", "FunctionMessage"),
+    ("swarmauri.messages.concrete.SystemMessage", "SystemMessage"),
+]
+
+# Lazy loading of messages classes, storing them in variables
+for module_name, class_name in messages_files:
+    globals()[class_name] = _lazy_import(module_name, class_name)
+
+# Adding the lazy-loaded messages classes to __all__
+__all__ = [class_name for _, class_name in messages_files]

swarmauri-0.5.3.dev4/swarmauri/parsers/concrete/__init__.py

@@ -0,0 +1,29 @@
+from swarmauri.utils._lazy_import import _lazy_import
+
+# List of parsers names (file names without the ".py" extension) and corresponding class names
+parsers_files = [
+    (
+        "swarmauri.parsers.concrete.BeautifulSoupElementParser",
+        "BeautifulSoupElementParser",
+    ),
+    ("swarmauri.parsers.concrete.CSVParser", "CSVParser"),
+    ("swarmauri.parsers.concrete.HTMLTagStripParser", "HTMLTagStripParser"),
+    ("swarmauri.parsers.concrete.KeywordExtractorParser", "KeywordExtractorParser"),
+    ("swarmauri.parsers.concrete.Md2HtmlParser", "Md2HtmlParser"),
+    ("swarmauri.parsers.concrete.OpenAPISpecParser", "OpenAPISpecParser"),
+    (
+        "swarmauri.parsers.concrete.PhoneNumberExtractorParser",
+        "PhoneNumberExtractorParser",
+    ),
+    ("swarmauri.parsers.concrete.PythonParser", "PythonParser"),
+    ("swarmauri.parsers.concrete.RegExParser", "RegExParser"),
+    ("swarmauri.parsers.concrete.URLExtractorParser", "URLExtractorParser"),
+    ("swarmauri.parsers.concrete.XMLParser", "XMLParser"),
+]
+
+# Lazy loading of parsers classes, storing them in variables
+for module_name, class_name in parsers_files:
+    globals()[class_name] = _lazy_import(module_name, class_name)
+
+# Adding the lazy-loaded parsers classes to __all__
+__all__ = [class_name for _, class_name in parsers_files]

swarmauri-0.5.3.dev4/swarmauri/prompts/concrete/__init__.py

@@ -0,0 +1,16 @@
+from swarmauri.utils._lazy_import import _lazy_import
+
+# List of prompts names (file names without the ".py" extension) and corresponding class names
+prompts_files = [
+    ("swarmauri.prompts.concrete.Prompt", "Prompt"),
+    ("swarmauri.prompts.concrete.PromptGenerator", "PromptGenerator"),
+    ("swarmauri.prompts.concrete.PromptMatrix", "PromptMatrix"),
+    ("from swarmauri.prompts.concrete.PromptTemplate", "PromptTemplate"),
+]
+
+# Lazy loading of prompts classes, storing them in variables
+for module_name, class_name in prompts_files:
+    globals()[class_name] = _lazy_import(module_name, class_name)
+
+# Adding the lazy-loaded prompts classes to __all__
+__all__ = [class_name for _, class_name in prompts_files]