solace-agent-mesh 1.4.12__py3-none-any.whl → 1.5.1__py3-none-any.whl

solace_agent_mesh/common/utils/message_utils.py

@@ -5,10 +5,11 @@ to ensure they don't exceed configured limits. The size calculation matches
 the exact serialization format used by the Solace AI Connector (JSON + UTF-8).
 """
 
+import logging
 import json
 from typing import Any, Dict, Tuple
 
-from solace_ai_connector.common.log import log
+log = logging.getLogger(__name__)
 
 # Maximum bytes per character in UTF-8 encoding (4 bytes for full Unicode range)
 MAX_UTF8_BYTES_PER_CHARACTER = 4

solace_agent_mesh/common/utils/push_notification_auth.py

@@ -1,3 +1,4 @@
+import logging
 from jwcrypto import jwk
 import uuid
 from starlette.responses import JSONResponse
@@ -10,10 +11,10 @@ import json
 import hashlib
 import httpx
 
-from solace_ai_connector.common.log import log
-
 from jwt import PyJWK, PyJWKClient
 
+log = logging.getLogger(__name__)
+
 AUTH_HEADER_PREFIX = "Bearer "
 
 

solace_agent_mesh/common/utils/utils_llm.txt

@@ -13,6 +13,8 @@ The `utils` directory provides essential utility functions and tools for the Sol
 - **`message_utils.py`** - Message size calculation and validation utilities
 - **`mime_helpers.py`** - MIME type classification and file extension utilities
 - **`push_notification_auth.py`** - JWT-based authentication for push notifications
+- **`pydantic_utils.py`** - Pydantic BaseModel with dict-like access for configuration
+- **`type_utils.py`** - Robust type checking utilities for development environments
 
 ### Subdirectories:
 - **`embeds/`** - Dynamic expression evaluation system using `«...»` syntax for mathematical calculations, datetime formatting, UUID generation, and artifact content processing
@@ -105,6 +107,26 @@ The `utils` directory provides essential utility functions and tools for the Sol
   - `load_jwks(jwks_url: str) -> None` - Load public keys from JWKS endpoint
   - `verify_push_notification(request: Request) -> bool` - Verify notification authenticity
 
+#### pydantic_utils.py
+**Purpose:** Provides a Pydantic BaseModel for SAM configuration with dict-like access
+**Import:** `from solace_agent_mesh.common.utils.pydantic_utils import SamConfigBase`
+
+**Classes:**
+- **`SamConfigBase(BaseModel)`** - Pydantic BaseModel with dict-like access
+  - `model_validate_and_clean(cls: Type[T], obj: Any) -> T` - Validates dict after removing None values
+  - `get(key: str, default: Any = None) -> Any` - Dict-like .get() method
+  - `__getitem__(key: str) -> Any` - Dict-like ['key'] access
+  - `__setitem__(key: str, value: Any)` - Dict-like ['key'] = value assignment
+  - `__contains__(key: str) -> bool` - Dict-like 'in' support
+  - `keys()`, `values()`, `items()`, `__iter__()` - Dict-like iteration methods
+
+#### type_utils.py
+**Purpose:** Utilities for robust type checking, especially in development environments
+**Import:** `from solace_agent_mesh.common.utils.type_utils import is_subclass_by_name`
+
+**Functions:**
+- `is_subclass_by_name(cls_to_check: type, base_class_name: str) -> bool` - Checks if a class is a subclass by looking for the base class name in the MRO
+
 ### Subdirectory APIs
 
 #### embeds/
@@ -154,7 +176,55 @@ extension = get_extension_for_mime_type("image/png")  # Returns ".png"
 filename = f"image_{uuid.uuid4()}{extension}"
 ```
 
-### 2. Platform Compatibility and System Initialization
+### 2. Configuration and Type Utilities
+
+```python
+from solace_agent_mesh.common.utils.pydantic_utils import SamConfigBase
+from solace_agent_mesh.common.utils.type_utils import is_subclass_by_name
+from pydantic import Field
+from typing import Optional
+
+# Define configuration with Pydantic validation and dict-like access
+class AgentConfig(SamConfigBase):
+    name: str
+    timeout: int = 30
+    debug: bool = False
+    api_key: Optional[str] = None
+
+# Load config from YAML/dict with None value cleaning
+config_dict = {
+    "name": "my_agent",
+    "timeout": None,  # Will use default value of 30
+    "debug": True,
+    "api_key": None   # Will use default value of None
+}
+
+config = AgentConfig.model_validate_and_clean(config_dict)
+
+# Use both Pydantic and dict-style access
+print(config.name)              # Pydantic style: "my_agent"
+print(config["timeout"])        # Dict style: 30 (default applied)
+print(config.get("debug", False))  # Dict .get(): True
+
+# Check if field was explicitly set
+if "api_key" in config:
+    print("API key was provided")
+else:
+    print("API key not provided, using default")
+
+# Robust type checking for development
+class BaseAgent:
+    pass
+
+class MyAgent(BaseAgent):
+    pass
+
+# This works even if BaseAgent is loaded from different paths
+if is_subclass_by_name(MyAgent, "BaseAgent"):
+    print("MyAgent is a BaseAgent subclass")
+```
+
+### 3. Platform Compatibility and System Initialization
 
 ```python
 # Early in application startup - import for side effects
@@ -184,7 +254,7 @@ async def run_command(cmd: str):
 result = await run_command("echo Hello World")
 ```
 
-### 3. Structured Logging Setup
+### 4. Structured Logging Setup
 
 ```python
 import logging
@@ -212,7 +282,7 @@ logger.info("User action completed", extra={
 # Output will be JSON with timestamp, level, service, code location, etc.
 ```
 
-### 4. Secure Push Notification System
+### 5. Secure Push Notification System
 
 ```python
 from solace_agent_mesh.common.utils.push_notification_auth import (
@@ -261,88 +331,6 @@ async def webhook_handler(request: Request):
         else:
             return Response("Unauthorized", status_code=401)
     except Exception as e:
-        print(f"Verification failed: {e}")
-        return Response("Bad Request", status_code=400)
-
-# JWKS endpoint for sender
-@app.route("/.well-known/jwks.json")
-async def jwks_endpoint(request: Request):
-    return sender_auth.handle_jwks_endpoint(request)
-```
-
-### 5. Advanced Embed Processing
-
-```python
-from solace_agent_mesh.common.utils.embeds import (
-    resolve_embeds_recursively_in_string, 
-    evaluate_embed,
-    EARLY_EMBED_TYPES,
-    LATE_EMBED_TYPES
-)
-
-# Set up context for embed resolution
-context = {
-    "artifact_service": my_artifact_service,
-    "session_context": {
-        "app_name": "sales_analyzer",
-        "user_id": "user123", 
-        "session_id": "sess456"
-    }
-}
-
-# Simple embed resolution for dynamic content
-template = """
-Sales Report Generated: «datetime:%Y-%m-%d %H:%M:%S»
-Total Revenue: $«math:«artifact_content:sales.csv >>> jsonpath:$.total_revenue» * 1.08 | .2f»
-Report ID: «uuid:new»
-Tax Rate Applied: 8%
-"""
-
-# Resolve embeds recursively with safety limits
-resolved_content = await resolve_embeds_recursively_in_string(
-    text=template,
-    context=context,
-    resolver_func=evaluate_embed,
-    types_to_resolve=EARLY_EMBED_TYPES | LATE_EMBED_TYPES,
-    log_identifier="[SalesReport]",
-    config={"max_artifact_size": 1024*1024},  # 1MB limit
-    max_depth=5
-)
-
-# Complex data transformation pipeline
-data_analysis = """
-Product Analysis:
-«artifact_content:products.csv:latest >>> select_cols:name,category,price,sales >>> filter_rows_eq:category:electronics >>> slice_rows:0:10 >>> format:json_pretty»
-
-Summary Statistics:
-«artifact_content:summary.txt >>> head:5»
-
-Top Performers:
-«artifact_content:sales.json >>> jsonpath:$.top_products[*].name >>> format:text»
-"""
-
-analysis_result = await resolve_embeds_recursively_in_string(
-    text=data_analysis,
-    context=context,
-    resolver_func=evaluate_embed,
-    types_to_resolve=LATE_EMBED_TYPES,
-    log_identifier="[Analysis]",
-    config={},
-    max_depth=3
-)
-```
-
-### 6. Integrated Workflow Example
-
-```python
-from solace_agent_mesh.common.utils import is_text_based_mime_type
-from solace_agent_mesh.common.utils.in_memory_cache import InMemoryCache
-from solace_agent_mesh.common.utils.message_utils import validate_message_size
-from solace_agent_mesh.common.utils.embeds import resolve_embeds_recursively_in_string, evaluate_embed
-from solace_agent_mesh.common.utils.push_notification_auth import PushNotificationSenderAuth
-import hashlib
-
-async def process_and_notify_report(user_id: str, template: str, context: dict, notification_urls: list):
-    """Complete workflow integrating
+        print(f"
 
-# content_hash: 0f48341a035bc4fbdb720885186fe04075f8fbca7ddeee6a151b00b4c667c47a
+# content_hash: d80ffe631ca180ea5aacca592313af42838960a949c8664559a957c2492fa86f