swarmauri-standard 0.7.5.dev1__tar.gz → 0.9.0.dev2__tar.gz

{swarmauri_standard-0.7.5.dev1 → swarmauri_standard-0.9.0.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: swarmauri-standard
-Version: 0.7.5.dev1
+Version: 0.9.0.dev2
 Summary: This repository includes standard components within the Swarmauri framework.
 License: Apache-2.0
 Author: Jacob Stewart
@@ -11,6 +11,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: aiofiles
+Requires-Dist: h11 (>=0.16.0)
 Requires-Dist: httpx (>=0.27.0)
 Requires-Dist: joblib (>=1.4.0)
 Requires-Dist: numpy

{swarmauri_standard-0.7.5.dev1 → swarmauri_standard-0.9.0.dev2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "swarmauri-standard"
-version = "0.7.5.dev1"
+version = "0.9.0.dev2"
 description = "This repository includes standard components within the Swarmauri framework."
 authors = [
     { name = "Jacob Stewart", email = "jacob@swarmauri.com" },
@@ -18,6 +18,7 @@ requires-python = ">=3.10,<3.13"
 dependencies = [
 "toml>=0.10.2",
 "pydantic>=2.9.2",
+"h11>=0.16.0",
 "httpx>=0.27.0",
 "numpy",
 "pandas",

{swarmauri_standard-0.7.5.dev1 → swarmauri_standard-0.9.0.dev2}/swarmauri_standard/chunkers/SentenceChunker.py

@@ -1,4 +1,4 @@
-from typing import Literal
+from typing import Any, List, Literal
 import re
 from swarmauri_base.chunkers.ChunkerBase import ChunkerBase
 from swarmauri_base.ComponentBase import ComponentBase
@@ -14,7 +14,7 @@ class SentenceChunker(ChunkerBase):
 
     type: Literal["SentenceChunker"] = "SentenceChunker"
 
-    def chunk_text(self, text, *args, **kwargs):
+    def chunk_text(self, text: str, *args: Any, **kwargs: Any) -> List[str]:
         """
         Chunks the given text into sentences using basic punctuation.
 

{swarmauri_standard-0.7.5.dev1 → swarmauri_standard-0.9.0.dev2}/swarmauri_standard/embeddings/CohereEmbedding.py

@@ -1,4 +1,4 @@
-from typing import List, Literal, Optional, Union
+from typing import Any, Dict, List, Literal, Optional, Union
 
 import httpx
 from pydantic import PrivateAttr, SecretStr
@@ -65,17 +65,12 @@ class CohereEmbedding(EmbeddingBase):
 
     def __init__(
         self,
-        **kwargs,
+        **kwargs: Dict[str, Any],
     ):
         """
         Initialize the CohereEmbedding instance.
 
         Args:
-            api_key (str, optional): The API key for accessing the Cohere API.
-            model (str, optional): The Cohere embedding model to use.
-            task_type (str, optional): The type of task for which embeddings are generated.
-            embedding_types (str, optional): The type of embedding to generate.
-            truncate (str, optional): The truncation strategy to use.
             **kwargs: Additional keyword arguments.
 
         Raises:

{swarmauri_standard-0.7.5.dev1 → swarmauri_standard-0.9.0.dev2}/swarmauri_standard/image_gens/BlackForestImgGenModel.py

@@ -1,7 +1,7 @@
 import asyncio
 import contextlib
 import time
-from typing import Dict, List, Literal, Optional
+from typing import Any, Dict, List, Literal, Optional
 
 import httpx
 from pydantic import PrivateAttr, SecretStr
@@ -30,9 +30,11 @@ class BlackForestImgGenModel(ImageGenBase):
     name: str = "flux-pro-1.1"
     type: Literal["BlackForestImgGenModel"] = "BlackForestImgGenModel"
 
-    def __init__(self, **kwargs):
+    def __init__(self, **kwargs: Dict[str, Any]):
         """
         Initializes the BlackForestImgGenModel instance with HTTP clients.
+        Args:
+            **kwarg (Dict[str, Any]): Additional arguments including api_key and allowed_models.
         """
         super().__init__(**kwargs)
         self._headers = {
@@ -161,7 +163,7 @@ class BlackForestImgGenModel(ImageGenBase):
 
         raise TimeoutError(f"Image generation timed out after {max_wait_time} seconds")
 
-    async def agenerate_image(self, prompt: str, **kwargs) -> Dict:
+    async def agenerate_image(self, prompt: str, **kwargs: Dict[str, Any]) -> Dict:
         """
         Asynchronously generates an image based on the prompt and waits for the result.
 
@@ -219,13 +221,15 @@ class BlackForestImgGenModel(ImageGenBase):
         finally:
             await self._close_async_client()
 
-    def batch_generate(self, prompts: List[str], **kwargs) -> List[Dict]:
+    def batch_generate(
+        self, prompts: List[str], **kwargs: Dict[str, Any]
+    ) -> List[Dict]:
         """
         Generates images for a batch of prompts synchronously.
 
         Args:
             prompts (List[str]): List of text prompts
-            **kwargs: Additional arguments passed to generate_image
+            **kwargs (Dict[str, Any]): Additional arguments passed to generate_image
 
         Returns:
             List[Dict]: List of result dictionaries
@@ -233,7 +237,7 @@ class BlackForestImgGenModel(ImageGenBase):
         return [self.generate_image(prompt=prompt, **kwargs) for prompt in prompts]
 
     async def abatch_generate(
-        self, prompts: List[str], max_concurrent: int = 5, **kwargs
+        self, prompts: List[str], max_concurrent: int = 5, **kwargs: Dict[str, Any]
     ) -> List[Dict]:
         """
         Asynchronously generates images for a batch of prompts.
@@ -241,7 +245,7 @@ class BlackForestImgGenModel(ImageGenBase):
         Args:
             prompts (List[str]): List of text prompts
             max_concurrent (int): Maximum number of concurrent tasks
-            **kwargs: Additional arguments passed to agenerate_image
+            **kwargs (Dict[str, Any]): Additional arguments passed to agenerate_image
 
         Returns:
             List[Dict]: List of result dictionaries

{swarmauri_standard-0.7.5.dev1 → swarmauri_standard-0.9.0.dev2}/swarmauri_standard/image_gens/DeepInfraImgGenModel.py

@@ -1,6 +1,6 @@
 import asyncio
 import contextlib
-from typing import List, Literal
+from typing import Any, Dict, List, Literal
 
 import httpx
 from pydantic import PrivateAttr, SecretStr
@@ -40,7 +40,7 @@ class DeepInfraImgGenModel(ImageGenBase):
     name: str = "black-forest-labs/FLUX-1-dev"
     type: Literal["DeepInfraImgGenModel"] = "DeepInfraImgGenModel"
 
-    def __init__(self, **kwargs):
+    def __init__(self, **kwargs: Dict[str, Any]):
         """
         Initializes the DeepInfraImgGenModel instance.
 
@@ -48,7 +48,7 @@ class DeepInfraImgGenModel(ImageGenBase):
         operations and configures request headers with the provided API key.
 
         Args:
-            **data: Keyword arguments for model initialization.
+            **kwargs (Dict[str, Any]): Additional keyword arguments, which may includes api_key and allowed_models.
         """
         super().__init__(**kwargs)
         self._headers = {

{swarmauri_standard-0.7.5.dev1 → swarmauri_standard-0.9.0.dev2}/swarmauri_standard/image_gens/FalAIImgGenModel.py

@@ -1,6 +1,6 @@
 import asyncio
 import time
-from typing import Dict, List, Literal, Optional
+from typing import Any, Dict, List, Literal, Optional
 
 import httpx
 from pydantic import Field, PrivateAttr, SecretStr
@@ -103,12 +103,12 @@ class FalAIImgGenModel(ImageGenBase):
     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, **kwargs):
+    def __init__(self, **kwargs: Dict[str, Any]):
         """
         Initializes the model with the specified API key and model name.
 
         Args:
-            **data: Configuration parameters for the model.
+            **kwargs (Dict[str, Any]): Additional keyword arguments, which may includes api_key and allowed_models.
 
         Raises:
             ValueError: If an invalid model name is provided.
@@ -141,13 +141,13 @@ class FalAIImgGenModel(ImageGenBase):
             await self._async_client.aclose()
             self._async_client = None
 
-    def _create_request_payload(self, prompt: str, **kwargs) -> dict:
+    def _create_request_payload(self, prompt: str, **kwargs: Dict[str, Any]) -> dict:
         """
         Creates a payload for the image generation request.
 
         Args:
             prompt (str): The text prompt for image generation.
-            **kwargs: Additional parameters for the request.
+            **kwargs (Dict[str, Any]): Additional parameters for the request.
 
         Returns:
             dict: The request payload.
@@ -155,13 +155,13 @@ class FalAIImgGenModel(ImageGenBase):
         return {"prompt": prompt, **kwargs}
 
     @retry_on_status_codes((429, 529), max_retries=1)
-    def _send_request(self, prompt: str, **kwargs) -> Dict:
+    def _send_request(self, prompt: str, **kwargs: Dict[str, Any]) -> Dict:
         """
         Sends an image generation request to the queue and returns the request ID.
 
         Args:
             prompt (str): The text prompt for image generation.
-            **kwargs: Additional parameters for the request.
+            **kwargs (Dict[str, Any]): Additional parameters for the request.
 
         Returns:
             Dict: The response containing the request ID.
@@ -206,13 +206,13 @@ class FalAIImgGenModel(ImageGenBase):
         return response.json()
 
     @retry_on_status_codes((429, 529), max_retries=1)
-    async def _async_send_request(self, prompt: str, **kwargs) -> Dict:
+    async def _async_send_request(self, prompt: str, **kwargs: Dict[str, Any]) -> Dict:
         """
         Asynchronously sends an image generation request to the queue.
 
         Args:
             prompt (str): The text prompt for image generation.
-            **kwargs: Additional parameters for the request.
+            **kwargs (Dict[str, Any]): Additional parameters for the request.
 
         Returns:
             Dict: The response containing the request ID.
@@ -311,13 +311,13 @@ class FalAIImgGenModel(ImageGenBase):
             f"Request {request_id} did not complete within the timeout period"
         )
 
-    def generate_image(self, prompt: str, **kwargs) -> str:
+    def generate_image(self, prompt: str, **kwargs: Dict[str, Any]) -> str:
         """
         Generates an image based on the prompt and returns the image URL.
 
         Args:
             prompt (str): The text prompt for image generation.
-            **kwargs: Additional parameters for the request.
+            **kwargs (Dict[str, Any]): Additional parameters for the request.
 
         Returns:
             str: The URL of the generated image.
@@ -327,13 +327,13 @@ class FalAIImgGenModel(ImageGenBase):
         final_response = self._wait_for_completion(request_id)
         return final_response["images"][0]["url"]
 
-    async def agenerate_image(self, prompt: str, **kwargs) -> str:
+    async def agenerate_image(self, prompt: str, **kwargs: Dict[str, Any]) -> str:
         """
         Asynchronously generates an image based on the prompt and returns the image URL.
 
         Args:
             prompt (str): The text prompt for image generation
-            **kwargs: Additional parameters to pass to the API
+            **kwargs (Dict[str, Any]): Additional parameters to pass to the API
 
         Returns:
             str: The URL of the generated image
@@ -346,13 +346,13 @@ class FalAIImgGenModel(ImageGenBase):
         finally:
             await self._close_async_client()
 
-    def batch_generate(self, prompts: List[str], **kwargs) -> List[str]:
+    def batch_generate(self, prompts: List[str], **kwargs: Dict[str, Any]) -> List[str]:
         """
         Generates images for a batch of prompts.
 
         Args:
             prompts (List[str]): List of text prompts
-            **kwargs: Additional parameters to pass to the API
+            **kwargs (Dict[str, Any]): Additional parameters to pass to the API
 
         Returns:
             List[str]: List of image URLs
@@ -360,7 +360,7 @@ class FalAIImgGenModel(ImageGenBase):
         return [self.generate_image(prompt, **kwargs) for prompt in prompts]
 
     async def abatch_generate(
-        self, prompts: List[str], max_concurrent: int = 5, **kwargs
+        self, prompts: List[str], max_concurrent: int = 5, **kwargs: Dict[str, Any]
     ) -> List[str]:
         """
         Asynchronously generates images for a batch of prompts.
@@ -368,7 +368,7 @@ class FalAIImgGenModel(ImageGenBase):
         Args:
             prompts (List[str]): List of text prompts
             max_concurrent (int): Maximum number of concurrent requests
-            **kwargs: Additional parameters to pass to the API
+            **kwargs (Dict[str, Any]): Additional parameters to pass to the API
 
         Returns:
             List[str]: List of image URLs

{swarmauri_standard-0.7.5.dev1 → swarmauri_standard-0.9.0.dev2}/swarmauri_standard/image_gens/HyperbolicImgGenModel.py

@@ -1,6 +1,6 @@
 import asyncio
 import contextlib
-from typing import List, Literal
+from typing import Any, Dict, List, Literal
 
 import httpx
 from pydantic import PrivateAttr, SecretStr
@@ -57,7 +57,7 @@ class HyperbolicImgGenModel(ImageGenBase):
     enable_refiner: bool = False
     backend: str = "auto"
 
-    def __init__(self, **kwargs):
+    def __init__(self, **kwargs: Dict[str, Any]):
         """
         Initializes the HyperbolicImgGenModel instance.
 
@@ -65,7 +65,7 @@ class HyperbolicImgGenModel(ImageGenBase):
         operations and configures request headers with the provided API key.
 
         Args:
-            **data: Keyword arguments for model initialization.
+            **kwargs (Dict[str, Any]): Additional keyword arguments, which may include api_key and allowed_models.
         """
         super().__init__(**kwargs)
         self._headers = {

{swarmauri_standard-0.7.5.dev1 → swarmauri_standard-0.9.0.dev2}/swarmauri_standard/image_gens/OpenAIImgGenModel.py

@@ -1,5 +1,5 @@
 import asyncio
-from typing import Dict, List, Literal, Optional
+from typing import Any, Dict, List, Literal, Optional
 
 import httpx
 from pydantic import PrivateAttr, SecretStr
@@ -31,12 +31,12 @@ class OpenAIImgGenModel(ImageGenBase):
     _BASE_URL: str = PrivateAttr(default="https://api.openai.com/v1/images/generations")
     _headers: Dict[str, str] = PrivateAttr(default=None)
 
-    def __init__(self, **kwargs) -> None:
+    def __init__(self, **kwargs: Dict[str, Any]) -> None:
         """
         Initialize the GroqAIAudio class with the provided data.
 
         Args:
-            **data: Arbitrary keyword arguments containing initialization data.
+            **kwargs (Dict[str, Any]): Additional keyword arguments, which may include api_key and allowed_models.
         """
         super().__init__(**kwargs)
         self._headers = {

swarmauri_standard-0.9.0.dev2/swarmauri_standard/key_providers/InMemoryKeyProvider.py

@@ -0,0 +1,147 @@
+from __future__ import annotations
+
+from typing import Dict, Iterable, Mapping, Optional, Tuple, Literal
+import secrets
+import hashlib
+import hmac
+
+from swarmauri_base.keys.KeyProviderBase import KeyProviderBase
+from swarmauri_core.crypto.types import KeyRef, KeyType, ExportPolicy
+from swarmauri_core.keys.types import KeySpec, KeyClass
+
+
+class InMemoryKeyProvider(KeyProviderBase):
+    """Simple in-memory key provider for testing or ephemeral usage."""
+
+    type: Literal["InMemoryKeyProvider"] = "InMemoryKeyProvider"
+
+    def __init__(self, **kwargs) -> None:
+        super().__init__(**kwargs)
+        self._store: Dict[str, Dict[int, KeyRef]] = {}
+
+    def supports(self) -> Mapping[str, Iterable[str]]:
+        return {
+            "class": ("sym", "asym"),
+            "algs": (),
+            "features": ("create", "import", "rotate"),
+        }
+
+    async def create_key(self, spec: KeySpec) -> KeyRef:
+        kid = secrets.token_hex(8)
+        version = 1
+        material = secrets.token_bytes(32)
+        ref = KeyRef(
+            kid=kid,
+            version=version,
+            type=(
+                KeyType.SYMMETRIC
+                if spec.klass == KeyClass.symmetric
+                else KeyType.OPAQUE
+            ),
+            uses=tuple(spec.uses),
+            export_policy=spec.export_policy,
+            material=(material if spec.export_policy != ExportPolicy.NONE else None),
+            tags={"label": spec.label or "", **(spec.tags or {})},
+        )
+        self._store.setdefault(kid, {})[version] = ref
+        return ref
+
+    async def import_key(
+        self,
+        spec: KeySpec,
+        material: bytes,
+        *,
+        public: Optional[bytes] = None,
+    ) -> KeyRef:
+        kid = secrets.token_hex(8)
+        version = 1
+        ref = KeyRef(
+            kid=kid,
+            version=version,
+            type=(
+                KeyType.SYMMETRIC
+                if spec.klass == KeyClass.symmetric
+                else KeyType.OPAQUE
+            ),
+            uses=tuple(spec.uses),
+            export_policy=spec.export_policy,
+            material=(material if spec.export_policy != ExportPolicy.NONE else None),
+            public=public,
+            tags={"label": spec.label or "", **(spec.tags or {})},
+        )
+        self._store.setdefault(kid, {})[version] = ref
+        return ref
+
+    async def rotate_key(
+        self, kid: str, *, spec_overrides: Optional[dict] = None
+    ) -> KeyRef:
+        bucket = self._store.get(kid)
+        if not bucket:
+            raise KeyError(f"Unknown kid: {kid}")
+        latest = max(bucket)
+        base = bucket[latest]
+        material = secrets.token_bytes(len(base.material or b"\x00" * 32))
+        version = latest + 1
+        ref = KeyRef(
+            kid=kid,
+            version=version,
+            type=base.type,
+            uses=base.uses,
+            export_policy=base.export_policy,
+            material=(material if base.export_policy != ExportPolicy.NONE else None),
+            public=base.public,
+            tags=base.tags,
+            uri=base.uri,
+        )
+        bucket[version] = ref
+        return ref
+
+    async def destroy_key(self, kid: str, version: Optional[int] = None) -> bool:
+        bucket = self._store.get(kid)
+        if not bucket:
+            return False
+        if version is None:
+            del self._store[kid]
+            return True
+        bucket.pop(version, None)
+        if not bucket:
+            del self._store[kid]
+        return True
+
+    async def get_key(
+        self,
+        kid: str,
+        version: Optional[int] = None,
+        *,
+        include_secret: bool = False,
+    ) -> KeyRef:
+        bucket = self._store[kid]
+        v = version or max(bucket)
+        return bucket[v]
+
+    async def list_versions(self, kid: str) -> Tuple[int, ...]:
+        bucket = self._store.get(kid)
+        if not bucket:
+            raise KeyError(f"Unknown kid: {kid}")
+        return tuple(sorted(bucket.keys()))
+
+    async def get_public_jwk(self, kid: str, version: Optional[int] = None) -> dict:
+        raise NotImplementedError("JWK export not supported")
+
+    async def jwks(self, *, prefix_kids: Optional[str] = None) -> dict:
+        raise NotImplementedError("JWKS export not supported")
+
+    async def random_bytes(self, n: int) -> bytes:
+        return secrets.token_bytes(n)
+
+    async def hkdf(self, ikm: bytes, *, salt: bytes, info: bytes, length: int) -> bytes:
+        """Derive key material using HKDF-SHA256."""
+        prk = hmac.new(salt, ikm, hashlib.sha256).digest()
+        t = b""
+        okm = b""
+        counter = 1
+        while len(okm) < length:
+            t = hmac.new(prk, t + info + bytes([counter]), hashlib.sha256).digest()
+            okm += t
+            counter += 1
+        return okm[:length]

swarmauri_standard-0.9.0.dev2/swarmauri_standard/key_providers/__init__.py

@@ -0,0 +1,3 @@
+from .InMemoryKeyProvider import InMemoryKeyProvider
+
+__all__ = ["InMemoryKeyProvider"]

{swarmauri_standard-0.7.5.dev1 → swarmauri_standard-0.9.0.dev2}/swarmauri_standard/llms/AI21StudioModel.py

@@ -1,6 +1,6 @@
 import asyncio
 import json
-from typing import AsyncIterator, Iterator, List, Literal, Type
+from typing import Any, AsyncIterator, Dict, Iterator, List, Literal, Type
 
 import httpx
 from pydantic import PrivateAttr, SecretStr
@@ -33,8 +33,13 @@ class AI21StudioModel(LLMBase):
     """
 
     api_key: SecretStr
-    allowed_models: List[str] = ["jamba-1.5-large", "jamba-1.5-mini"]
-    name: str = "jamba-1.5-large"
+    allowed_models: List[str] = [
+        "jamba-large",
+        "jamba-mini",
+        "jamba-large-1.7",
+        "jamba-mini-1.7",
+    ]
+    name: str = "jamba-large"
     type: Literal["AI21StudioModel"] = "AI21StudioModel"
     _client: httpx.Client = PrivateAttr(default=None)
     _async_client: httpx.AsyncClient = PrivateAttr(default=None)
@@ -42,7 +47,7 @@ class AI21StudioModel(LLMBase):
         default="https://api.ai21.com/studio/v1/chat/completions"
     )
 
-    def __init__(self, **data) -> None:
+    def __init__(self, **data: Dict[str, Any]) -> None:
         """
         Initializes the GroqToolModel instance, setting up headers for API requests.
 
@@ -425,4 +430,4 @@ class AI21StudioModel(LLMBase):
         Returns:
             List[str]: List of allowed model names.
         """
-        return ["jamba-1.5-large", "jamba-1.5-mini"]
+        return ["jamba-large", "jamba-mini", "jamba-large-1.7", "jamba-mini-1.7"]

{swarmauri_standard-0.7.5.dev1 → swarmauri_standard-0.9.0.dev2}/swarmauri_standard/llms/AnthropicModel.py

@@ -34,19 +34,15 @@ class AnthropicModel(LLMBase):
     _async_client: httpx.AsyncClient = PrivateAttr()
     api_key: SecretStr
     allowed_models: List[str] = [
+        "claude-opus-4-1",
+        "claude-opus-4-0",
+        "claude-sonnet-4-0",
         "claude-3-7-sonnet-latest",
         "claude-3-5-haiku-latest",
-        "claude-3-5-sonnet-latest",
-        "claude-3-opus-latest",
-        "claude-3-5-sonnet-20241022",
         "claude-3-5-haiku-20241022",
         "claude-3-7-sonnet-20250219",
-        "claude-3-5-sonnet-20240620",
-        "claude-3-opus-20240229",
-        "claude-3-sonnet-20240229",
-        "claude-3-haiku-20240307",
     ]
-    name: str = "claude-3-7-sonnet-latest"
+    name: str = "claude-opus-4-1"
     type: Literal["AnthropicModel"] = "AnthropicModel"
 
     timeout: float = 600.0