swarmauri-standard 0.7.3.dev2__tar.gz → 0.7.4.dev9__tar.gz

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: swarmauri-standard
-Version: 0.7.3.dev2
+Version: 0.7.4.dev9
 Summary: This repository includes standard components within the Swarmauri framework.
 License: Apache-2.0
 Author: Jacob Stewart
@@ -17,6 +17,7 @@ Requires-Dist: numpy
 Requires-Dist: pandas
 Requires-Dist: pillow (>=11.1.0)
 Requires-Dist: pydantic (>=2.9.2)
+Requires-Dist: rich
 Requires-Dist: swarmauri_base
 Requires-Dist: swarmauri_core
 Requires-Dist: toml (>=0.10.2)
@@ -28,8 +29,8 @@ Description-Content-Type: text/markdown
 <p align="center">
     <a href="https://pypi.org/project/swarmauri-standard/">
         <img src="https://img.shields.io/pypi/dm/swarmauri-standard" alt="PyPI - Downloads"/></a>
-    <a href="https://github.com/swarmauri/swarmauri-sdk/blob/master/pkgs/swarmauri_standard/README.md">
-        <img src="https://hits.seeyoufarm.com/api/count/incr/badge.svg?url=https://github.com/swarmauri/swarmauri-sdk/pkgs/swarmauri_standard/README.md&count_bg=%2379C83D&title_bg=%23555555&icon=&icon_color=%23E7E7E7&title=hits&edge_flat=false" alt="GitHub Hits"/></a>
+    <a href="https://hits.sh/github.com/swarmauri/swarmauri-sdk/tree/master/pkgs/swarmauri_standard/">
+        <img alt="Hits" src="https://hits.sh/github.com/swarmauri/swarmauri-sdk/tree/master/pkgs/swarmauri_standard.svg"/></a>
     <a href="https://pypi.org/project/swarmauri-standard/">
         <img src="https://img.shields.io/pypi/pyversions/swarmauri-standard" alt="PyPI - Python Version"/></a>
     <a href="https://pypi.org/project/swarmauri-standard/">

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/README.md

@@ -3,8 +3,8 @@
 <p align="center">
     <a href="https://pypi.org/project/swarmauri-standard/">
         <img src="https://img.shields.io/pypi/dm/swarmauri-standard" alt="PyPI - Downloads"/></a>
-    <a href="https://github.com/swarmauri/swarmauri-sdk/blob/master/pkgs/swarmauri_standard/README.md">
-        <img src="https://hits.seeyoufarm.com/api/count/incr/badge.svg?url=https://github.com/swarmauri/swarmauri-sdk/pkgs/swarmauri_standard/README.md&count_bg=%2379C83D&title_bg=%23555555&icon=&icon_color=%23E7E7E7&title=hits&edge_flat=false" alt="GitHub Hits"/></a>
+    <a href="https://hits.sh/github.com/swarmauri/swarmauri-sdk/tree/master/pkgs/swarmauri_standard/">
+        <img alt="Hits" src="https://hits.sh/github.com/swarmauri/swarmauri-sdk/tree/master/pkgs/swarmauri_standard.svg"/></a>
     <a href="https://pypi.org/project/swarmauri-standard/">
         <img src="https://img.shields.io/pypi/pyversions/swarmauri-standard" alt="PyPI - Python Version"/></a>
     <a href="https://pypi.org/project/swarmauri-standard/">

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "swarmauri-standard"
-version = "0.7.3.dev2"
+version = "0.7.4.dev9"
 description = "This repository includes standard components within the Swarmauri framework."
 authors = [
     { name = "Jacob Stewart", email = "jacob@swarmauri.com" },
@@ -24,6 +24,7 @@ dependencies = [
 "joblib>=1.4.0",
 "pillow>=11.1.0",
 "aiofiles",
+"rich",
 "typing_extensions",
 "swarmauri_core",
 "swarmauri_base",

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/swarmauri_standard/distances/CanberraDistance.py

@@ -3,6 +3,13 @@ from typing import List, Literal
 from swarmauri_standard.vectors.Vector import Vector
 from swarmauri_base.distances.DistanceBase import DistanceBase
 from swarmauri_base.ComponentBase import ComponentBase
+import warnings
+
+warnings.warn(
+    "This distance class will be deprecated in v0.10.0",
+    DeprecationWarning,
+    stacklevel=2,
+)
 
 
 @ComponentBase.register_type(DistanceBase, "CanberraDistance")

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/swarmauri_standard/distances/ChebyshevDistance.py

@@ -2,6 +2,13 @@ from typing import List, Literal
 from swarmauri_standard.vectors.Vector import Vector
 from swarmauri_base.distances.DistanceBase import DistanceBase
 from swarmauri_base.ComponentBase import ComponentBase
+import warnings
+
+warnings.warn(
+    "This distance class will be deprecated in v0.10.0",
+    DeprecationWarning,
+    stacklevel=2,
+)
 
 
 @ComponentBase.register_type(DistanceBase, "ChebyshevDistance")

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/swarmauri_standard/distances/ChiSquaredDistance.py

@@ -3,6 +3,13 @@ from typing import List, Literal
 from swarmauri_standard.vectors.Vector import Vector
 from swarmauri_base.distances.DistanceBase import DistanceBase
 from swarmauri_base.ComponentBase import ComponentBase
+import warnings
+
+warnings.warn(
+    "This distance class will be deprecated in v0.10.0",
+    DeprecationWarning,
+    stacklevel=2,
+)
 
 
 @ComponentBase.register_type(DistanceBase, "ChiSquaredDistance")

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/swarmauri_standard/distances/CosineDistance.py

@@ -4,6 +4,13 @@ from typing import List, Literal
 from swarmauri_standard.vectors.Vector import Vector
 from swarmauri_base.distances.DistanceBase import DistanceBase
 from swarmauri_base.ComponentBase import ComponentBase
+import warnings
+
+warnings.warn(
+    "This distance class will be deprecated in v0.10.0",
+    DeprecationWarning,
+    stacklevel=2,
+)
 
 
 @ComponentBase.register_type(DistanceBase, "CosineDistance")

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/swarmauri_standard/distances/EuclideanDistance.py

@@ -3,6 +3,13 @@ from typing import List, Literal
 from swarmauri_standard.vectors.Vector import Vector
 from swarmauri_base.distances.DistanceBase import DistanceBase
 from swarmauri_base.ComponentBase import ComponentBase
+import warnings
+
+warnings.warn(
+    "This distance class will be deprecated in v0.10.0",
+    DeprecationWarning,
+    stacklevel=2,
+)
 
 
 @ComponentBase.register_type(DistanceBase, "EuclideanDistance")

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/swarmauri_standard/distances/HaversineDistance.py

@@ -3,6 +3,13 @@ from math import radians, cos, sin, sqrt, atan2
 from swarmauri_standard.vectors.Vector import Vector
 from swarmauri_base.distances.DistanceBase import DistanceBase
 from swarmauri_base.ComponentBase import ComponentBase
+import warnings
+
+warnings.warn(
+    "This distance class will be deprecated in v0.10.0",
+    DeprecationWarning,
+    stacklevel=2,
+)
 
 
 @ComponentBase.register_type(DistanceBase, "HaversineDistance")

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/swarmauri_standard/distances/JaccardIndexDistance.py

@@ -2,6 +2,13 @@ from typing import List, Literal
 from swarmauri_standard.vectors.Vector import Vector
 from swarmauri_base.distances.DistanceBase import DistanceBase
 from swarmauri_base.ComponentBase import ComponentBase
+import warnings
+
+warnings.warn(
+    "This distance class will be deprecated in v0.10.0",
+    DeprecationWarning,
+    stacklevel=2,
+)
 
 
 @ComponentBase.register_type(DistanceBase, "JaccardIndexDistance")

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/swarmauri_standard/distances/LevenshteinDistance.py

@@ -3,6 +3,13 @@ from typing import List, Literal
 from swarmauri_standard.vectors.Vector import Vector
 from swarmauri_base.distances.DistanceBase import DistanceBase
 from swarmauri_base.ComponentBase import ComponentBase
+import warnings
+
+warnings.warn(
+    "This distance class will be deprecated in v0.10.0",
+    DeprecationWarning,
+    stacklevel=2,
+)
 
 
 @ComponentBase.register_type(DistanceBase, "LevenshteinDistance")

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/swarmauri_standard/distances/ManhattanDistance.py

@@ -2,6 +2,13 @@ from typing import List, Literal
 from swarmauri_standard.vectors.Vector import Vector
 from swarmauri_base.distances.DistanceBase import DistanceBase
 from swarmauri_base.ComponentBase import ComponentBase
+import warnings
+
+warnings.warn(
+    "This distance class will be deprecated in v0.10.0",
+    DeprecationWarning,
+    stacklevel=2,
+)
 
 
 @ComponentBase.register_type(DistanceBase, "ManhattanDistance")

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/swarmauri_standard/distances/SorensenDiceDistance.py

@@ -3,6 +3,13 @@ from typing import List, Literal
 from swarmauri_standard.vectors.Vector import Vector
 from swarmauri_base.distances.DistanceBase import DistanceBase
 from swarmauri_base.ComponentBase import ComponentBase
+import warnings
+
+warnings.warn(
+    "This distance class will be deprecated in v0.10.0",
+    DeprecationWarning,
+    stacklevel=2,
+)
 
 
 @ComponentBase.register_type(DistanceBase, "SorensenDiceDistance")

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/swarmauri_standard/distances/SquaredEuclideanDistance.py

@@ -2,6 +2,13 @@ from typing import List, Literal
 from swarmauri_standard.vectors.Vector import Vector
 from swarmauri_base.distances.DistanceBase import DistanceBase
 from swarmauri_base.ComponentBase import ComponentBase
+import warnings
+
+warnings.warn(
+    "This distance class will be deprecated in v0.10.0",
+    DeprecationWarning,
+    stacklevel=2,
+)
 
 
 @ComponentBase.register_type(DistanceBase, "SquaredEuclideanDistance")

swarmauri_standard-0.7.4.dev9/swarmauri_standard/evaluator_results/EvalResult.py

@@ -0,0 +1,169 @@
+import logging
+from typing import Any, ClassVar, Dict, Literal
+
+from swarmauri_base.ComponentBase import ComponentBase
+from swarmauri_base.evaluator_results.EvalResultBase import EvalResultBase
+from swarmauri_core.programs.IProgram import IProgram
+
+logger = logging.getLogger(__name__)
+
+
+@ComponentBase.register_type(EvalResultBase, "EvalResult")
+class EvalResult(EvalResultBase):
+    """
+    Concrete implementation of EvalResultBase.
+
+    This class holds a program reference, scalar score, and metadata.
+    It provides a fully implemented evaluation result object.
+
+    Attributes
+    ----------
+    type : Literal["EvalResult"]
+        Type identifier for this component
+    _score : float
+        The numerical evaluation score
+    _metadata : Dict[str, Any]
+        Dictionary containing metadata about the evaluation
+    _program : IProgram
+        The program that was evaluated
+    _metadata_schema : ClassVar[Dict[str, Any]]
+        Schema for validating metadata structure
+    """
+
+    type: Literal["EvalResult"] = "EvalResult"
+
+    # Extended metadata schema with specific validation requirements
+    _metadata_schema: ClassVar[Dict[str, Any]] = {
+        "type": "object",
+        "properties": {
+            # Define specific metadata properties here if needed
+        },
+        "additionalProperties": True,
+    }
+
+    def __init__(self, score: float, metadata: Dict[str, Any], program: IProgram):
+        """
+        Initialize a new evaluation result with score, metadata, and program.
+
+        Parameters
+        ----------
+        score : float
+            The numerical evaluation score
+        metadata : Dict[str, Any]
+            Dictionary containing metadata about the evaluation
+        program : IProgram
+            The program that was evaluated
+
+        Raises
+        ------
+        TypeError
+            If score is not a float or if metadata keys are not strings
+        ValueError
+            If program is None
+        """
+        # Validate score is a real number
+        if not isinstance(score, (int, float)):
+            logger.error(f"Score must be a number, got {type(score)}")
+            raise TypeError(f"Score must be a number, got {type(score)}")
+
+        # Validate program is not None
+        if program is None:
+            logger.error("Program cannot be None")
+            raise ValueError("Program cannot be None")
+
+        # Validate metadata keys are strings
+        if not all(isinstance(key, str) for key in metadata.keys()):
+            logger.error("All metadata keys must be strings")
+            raise TypeError("All metadata keys must be strings")
+
+        super().__init__(score, metadata, program)
+        logger.debug(f"Created EvalResult with score {score} for program {program}")
+
+    def _validate_metadata(self, metadata: Dict[str, Any]) -> None:
+        """
+        Validate metadata against the defined schema with additional checks.
+
+        Parameters
+        ----------
+        metadata : Dict[str, Any]
+            The metadata to validate
+
+        Raises
+        ------
+        ValueError
+            If metadata doesn't conform to the schema
+        TypeError
+            If metadata keys are not strings
+        """
+        # First perform basic validation from parent class
+        # Check if metadata is a dictionary
+        if not isinstance(metadata, dict):
+            logger.error("Metadata validation failed: Metadata must be a dictionary")
+            raise ValueError("Metadata must be a dictionary")
+
+        # Ensure all keys are strings
+        non_string_keys = [key for key in metadata.keys() if not isinstance(key, str)]
+        if non_string_keys:
+            logger.error(
+                f"Metadata validation failed: All metadata keys must be strings. Found non-string keys: {non_string_keys}"
+            )
+            raise TypeError(
+                f"All metadata keys must be strings. Found non-string keys: {non_string_keys}"
+            )
+
+        # Additional validation logic would go here
+        logger.debug("Metadata validation passed")
+
+    def update_metadata(self, new_metadata: Dict[str, Any]) -> None:
+        """
+        Update the evaluation result metadata with new values.
+
+        Parameters
+        ----------
+        new_metadata : Dict[str, Any]
+            New metadata to add or update
+
+        Raises
+        ------
+        TypeError
+            If new metadata keys are not strings
+        """
+        # Validate new metadata keys
+        if not all(isinstance(key, str) for key in new_metadata.keys()):
+            logger.error("All metadata keys must be strings")
+            raise TypeError("All metadata keys must be strings")
+
+        # Update metadata
+        self._metadata.update(new_metadata)
+        logger.debug(f"Updated metadata with {len(new_metadata)} new entries")
+
+    def compare_to(self, other: "EvalResult") -> int:
+        """
+        Compare this evaluation result to another based on score.
+
+        Parameters
+        ----------
+        other : EvalResult
+            Another evaluation result to compare with
+
+        Returns
+        -------
+        int
+            1 if this result is better, -1 if other is better, 0 if equal
+
+        Raises
+        ------
+        TypeError
+            If other is not an EvalResult
+        """
+        if not isinstance(other, EvalResult):
+            logger.error(f"Cannot compare EvalResult with {type(other)}")
+            raise TypeError(f"Cannot compare EvalResult with {type(other)}")
+
+        # Higher scores are better
+        if self.score > other.score:
+            return 1
+        elif self.score < other.score:
+            return -1
+        else:
+            return 0

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/swarmauri_standard/image_gens/BlackForestImgGenModel.py

@@ -24,10 +24,10 @@ class BlackForestImgGenModel(ImageGenBase):
     _headers: Dict[str, str] = PrivateAttr(default=None)
 
     api_key: SecretStr
-    allowed_models: List[str] = []
+    allowed_models: List[str] = ["flux-pro-1.1", "flux-pro", "flux-dev"]
     timeout: float = 600.0
 
-    name: str = ""  # Default model
+    name: str = "flux-pro-1.1"
     type: Literal["BlackForestImgGenModel"] = "BlackForestImgGenModel"
 
     def __init__(self, **kwargs):
@@ -40,8 +40,6 @@ class BlackForestImgGenModel(ImageGenBase):
             "X-Key": self.api_key.get_secret_value(),
         }
         self._client = httpx.Client(headers=self._headers, timeout=self.timeout)
-        self.allowed_models = self.allowed_models or self.get_allowed_models()
-        self.name = self.allowed_models[0]
 
     async def _get_async_client(self) -> httpx.AsyncClient:
         """Gets or creates an async client instance."""

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/swarmauri_standard/image_gens/DeepInfraImgGenModel.py

@@ -4,8 +4,8 @@ from typing import List, Literal
 
 import httpx
 from pydantic import PrivateAttr, SecretStr
-from swarmauri_base.image_gens.ImageGenBase import ImageGenBase
 from swarmauri_base.ComponentBase import ComponentBase
+from swarmauri_base.image_gens.ImageGenBase import ImageGenBase
 
 from swarmauri_standard.utils.retry_decorator import retry_on_status_codes
 
@@ -31,8 +31,13 @@ class DeepInfraImgGenModel(ImageGenBase):
     _async_client: httpx.AsyncClient = PrivateAttr(default=None)
     timeout: float = 600.0
     api_key: SecretStr
-    allowed_models: List[str] = []
-    name: str = ""  # Default model
+    allowed_models: List[str] = [
+        "black-forest-labs/FLUX-1-dev",
+        "black-forest-labs/FLUX-1-schnell",
+        "stabilityai/sdxl-turbo",
+        "stabilityai/stable-diffusion-2-1",
+    ]
+    name: str = "black-forest-labs/FLUX-1-dev"
     type: Literal["DeepInfraImgGenModel"] = "DeepInfraImgGenModel"
 
     def __init__(self, **kwargs):

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/swarmauri_standard/image_gens/FalAIImgGenModel.py

@@ -4,8 +4,8 @@ from typing import Dict, List, Literal, Optional
 
 import httpx
 from pydantic import Field, PrivateAttr, SecretStr
-from swarmauri_base.image_gens.ImageGenBase import ImageGenBase
 from swarmauri_base.ComponentBase import ComponentBase
+from swarmauri_base.image_gens.ImageGenBase import ImageGenBase
 
 from swarmauri_standard.utils.retry_decorator import retry_on_status_codes
 
@@ -29,9 +29,75 @@ class FalAIImgGenModel(ImageGenBase):
     _client: httpx.Client = PrivateAttr()
     _async_client: Optional[httpx.AsyncClient] = PrivateAttr(default=None)
 
-    allowed_models: List[str] = []
+    allowed_models: List[str] = [
+        "fal-ai/flux-pro/v1.1-ultra-finetuned",
+        "fal-ai/minimax-image",
+        "fal-ai/aura-flow",
+        "fal-ai/flux/dev",
+        "fal-ai/flux-lora",
+        "fal-ai/flux-lora/inpainting",
+        "fal-ai/flux/schnell",
+        "fal-ai/flux-pro/v1.1",
+        "fal-ai/flux-pro/new",
+        "fal-ai/sana",
+        "fal-ai/omnigen-v1",
+        "fal-ai/lumina-image/v2",
+        "fal-ai/stable-diffusion-v35-large",
+        "fal-ai/switti",
+        "fal-ai/switti/512",
+        "fal-ai/recraft-20b",
+        "fal-ai/ideogram/v2/turbo",
+        "fal-ai/ideogram/v2a",
+        "fal-ai/ideogram/v2a/turbo",
+        "fal-ai/bria/text-to-image/base",
+        "fal-ai/bria/text-to-image/fast",
+        "fal-ai/bria/text-to-image/hd",
+        "fal-ai/flux-control-lora-canny",
+        "fal-ai/flux-control-lora-depth",
+        "fal-ai/flux-general",
+        "rundiffusion-fal/juggernaut-flux/base",
+        "rundiffusion-fal/juggernaut-flux/lightning",
+        "rundiffusion-fal/juggernaut-flux/pro",
+        "rundiffusion-fal/juggernaut-flux-lora",
+        "rundiffusion-fal/rundiffusion-photo-flux",
+        "fal-ai/stable-diffusion-v3-medium",
+        "fal-ai/fast-sdxl",
+        "fal-ai/lora",
+        "fal-ai/imagen3/fast",
+        "fal-ai/imagen3",
+        "fal-ai/janus",
+        "fal-ai/sdxl-controlnet-union",
+        "fal-ai/kolors",
+        "fal-ai/stable-cascade/sote-diffusion",
+        "fal-ai/lightning-models",
+        "fal-ai/realistic-vision",
+        "fal-ai/dreamshaper",
+        "fal-ai/pixart-sigma",
+        "fal-ai/stable-diffusion-v15",
+        "fal-ai/layer-diffusion",
+        "fal-ai/fast-fooocus-sdxl/image-to-image",
+        "fal-ai/fast-fooocus-sdxl",
+        "fal-ai/fooocus/inpaint",
+        "fal-ai/fooocus/image-prompt",
+        "fal-ai/fooocus/upscale-or-vary",
+        "fal-ai/diffusion-edge",
+        "fal-ai/fast-sdxl-controlnet-canny",
+        "fal-ai/illusion-diffusion",
+        "fal-ai/fooocus",
+        "fal-ai/lcm",
+        "fal-ai/playground-v25",
+        "fal-ai/hyper-sdxl",
+        "fal-ai/fast-lightning-sdxl",
+        "fal-ai/fast-lcm-diffusion",
+        "fal-ai/flowedit",
+        "fal-ai/stable-cascade",
+        "fal-ai/luma-photon",
+        "fal-ai/luma-photon/flash",
+        "fal-ai/cogview4",
+        "fal-ai/fast-turbo-diffusion",
+    ]
     api_key: SecretStr = Field(default=None)
-    name: str = ""
+    name: str = "fal-ai/flux-pro/v1.1-ultra-finetuned"
     timeout: float = 600.0
     type: Literal["FalAIImgGenModel"] = "FalAIImgGenModel"
     max_retries: int = Field(default=60)  # Maximum number of status check retries
@@ -53,8 +119,6 @@ class FalAIImgGenModel(ImageGenBase):
             "Authorization": f"Key {self.api_key.get_secret_value()}",
         }
         self._client = httpx.Client(headers=self._headers, timeout=self.timeout)
-        self.allowed_models = self.allowed_models or self.get_allowed_models()
-        self.name = self.allowed_models[0]
 
     async def _get_async_client(self) -> httpx.AsyncClient:
         """
@@ -336,5 +400,69 @@ class FalAIImgGenModel(ImageGenBase):
             List[str]: List of allowed model names.
         """
         return [
-            "fal-ai/flux-pro",
+            "fal-ai/flux-pro/v1.1-ultra-finetuned",
+            "fal-ai/minimax-image",
+            "fal-ai/aura-flow",
+            "fal-ai/flux/dev",
+            "fal-ai/flux-lora",
+            "fal-ai/flux-lora/inpainting",
+            "fal-ai/flux/schnell",
+            "fal-ai/flux-pro/v1.1",
+            "fal-ai/flux-pro/new",
+            "fal-ai/sana",
+            "fal-ai/omnigen-v1",
+            "fal-ai/lumina-image/v2",
+            "fal-ai/stable-diffusion-v35-large",
+            "fal-ai/switti",
+            "fal-ai/switti/512",
+            "fal-ai/recraft-20b",
+            "fal-ai/ideogram/v2/turbo",
+            "fal-ai/ideogram/v2a",
+            "fal-ai/ideogram/v2a/turbo",
+            "fal-ai/bria/text-to-image/base",
+            "fal-ai/bria/text-to-image/fast",
+            "fal-ai/bria/text-to-image/hd",
+            "fal-ai/flux-control-lora-canny",
+            "fal-ai/flux-control-lora-depth",
+            "fal-ai/flux-general",
+            "rundiffusion-fal/juggernaut-flux/base",
+            "rundiffusion-fal/juggernaut-flux/lightning",
+            "rundiffusion-fal/juggernaut-flux/pro",
+            "rundiffusion-fal/juggernaut-flux-lora",
+            "rundiffusion-fal/rundiffusion-photo-flux",
+            "fal-ai/stable-diffusion-v3-medium",
+            "fal-ai/fast-sdxl",
+            "fal-ai/lora",
+            "fal-ai/imagen3/fast",
+            "fal-ai/imagen3",
+            "fal-ai/janus",
+            "fal-ai/sdxl-controlnet-union",
+            "fal-ai/kolors",
+            "fal-ai/stable-cascade/sote-diffusion",
+            "fal-ai/lightning-models",
+            "fal-ai/realistic-vision",
+            "fal-ai/dreamshaper",
+            "fal-ai/pixart-sigma",
+            "fal-ai/stable-diffusion-v15",
+            "fal-ai/layer-diffusion",
+            "fal-ai/fast-fooocus-sdxl/image-to-image",
+            "fal-ai/fast-fooocus-sdxl",
+            "fal-ai/fooocus/inpaint",
+            "fal-ai/fooocus/image-prompt",
+            "fal-ai/fooocus/upscale-or-vary",
+            "fal-ai/diffusion-edge",
+            "fal-ai/fast-sdxl-controlnet-canny",
+            "fal-ai/illusion-diffusion",
+            "fal-ai/fooocus",
+            "fal-ai/lcm",
+            "fal-ai/playground-v25",
+            "fal-ai/hyper-sdxl",
+            "fal-ai/fast-lightning-sdxl",
+            "fal-ai/fast-lcm-diffusion",
+            "fal-ai/flowedit",
+            "fal-ai/stable-cascade",
+            "fal-ai/luma-photon",
+            "fal-ai/luma-photon/flash",
+            "fal-ai/cogview4",
+            "fal-ai/fast-turbo-diffusion",
         ]

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/swarmauri_standard/image_gens/HyperbolicImgGenModel.py

@@ -4,8 +4,8 @@ from typing import List, Literal
 
 import httpx
 from pydantic import PrivateAttr, SecretStr
-from swarmauri_base.image_gens.ImageGenBase import ImageGenBase
 from swarmauri_base.ComponentBase import ComponentBase
+from swarmauri_base.image_gens.ImageGenBase import ImageGenBase
 
 from swarmauri_standard.utils.retry_decorator import retry_on_status_codes
 
@@ -37,7 +37,13 @@ class HyperbolicImgGenModel(ImageGenBase):
     _async_client: httpx.AsyncClient = PrivateAttr(default=None)
 
     api_key: SecretStr
-    allowed_models: List[str] = []
+    allowed_models: List[str] = [
+        "SDXL1.0-base",
+        "SD1.5",
+        "SSD",
+        "SD2",
+        "SDXL-turbo",
+    ]
     timeout: float = 600.0
 
     name: str = ""
@@ -67,8 +73,6 @@ class HyperbolicImgGenModel(ImageGenBase):
             "Authorization": f"Bearer {self.api_key.get_secret_value()}",
         }
         self._client = httpx.Client(headers=self._headers, timeout=30)
-        self.allowed_models = self.allowed_models or self.get_allowed_models()
-        self.name = self.allowed_models[0]
 
     async def _get_async_client(self) -> httpx.AsyncClient:
         """

{swarmauri_standard-0.7.3.dev2 → swarmauri_standard-0.7.4.dev9}/swarmauri_standard/image_gens/OpenAIImgGenModel.py

@@ -24,8 +24,8 @@ class OpenAIImgGenModel(ImageGenBase):
     """
 
     api_key: SecretStr
-    allowed_models: List[str] = []
-    name: str = ""
+    allowed_models: List[str] = ["dall-e-2", "dall-e-3"]
+    name: str = "dall-e-2"
     type: Literal["OpenAIImgGenModel"] = "OpenAIImgGenModel"
     timeout: float = 600.0
     _BASE_URL: str = PrivateAttr(default="https://api.openai.com/v1/images/generations")
@@ -43,8 +43,6 @@ class OpenAIImgGenModel(ImageGenBase):
             "Authorization": f"Bearer {self.api_key.get_secret_value()}",
             "Content-Type": "application/json",
         }
-        self.allowed_models = self.allowed_models or self.get_allowed_models()
-        self.name = self.allowed_models[0]
 
     @retry_on_status_codes((429, 529), max_retries=1)
     def generate_image(