stls-a2a 0.1.6__py3-none-any.whl

stls_a2a/__init__.py

@@ -0,0 +1,53 @@
+from .stls_a2a_transport import SecureA2ATransport, A2AMessage
+from .stls_a2a_adapters_model import AgentResponse
+from .stls_a2a_integration import (
+    SecureA2ATransportAdapter,
+    create_secure_a2a_endpoint,
+    create_pqc_a2a_transport,
+    AuthConfig)
+
+# Providers — now live in stls_core (shared with stls-mcp).
+# Only "base" and "arqit_py" are supported (JVM/simple removed).
+from stls_core import (
+    KeyProviderFactory,
+    KeyProvider,
+    KeyProviderClient,
+    KeyProviderServer,
+    KeySessionProvider,
+    KeySessionProviderFactory,
+)
+
+from .stls_a2a_card_model import AgentCard
+from .stls_a2a_utils import (
+    to_a2a,
+    start_task_store_cleanup,
+    extract_a2a_text,
+    extract_a2a_result,
+    extract_a2a_artifacts)
+from .stls_a2a_client import A2AClient
+
+# Auth — now lives in stls_core (shared with stls-mcp).
+from stls_core import (
+    ApiKeyManager,
+    ApiKeyManagerGroup,
+    default_audit_hook,
+    AuditEntry,
+    generate_api_key,
+    hash_key,
+    RedisKeyCache,
+)
+from stls_core.auth.stls_core_auth_loaders import (
+    ApiKeyLoader,
+    StaticLoader,
+    FileLoader,
+    PostgresLoader,
+    create_loader,
+)
+
+from .stls_a2a_agent_registry import AgentRegistry, make_agent_id
+from .stls_a2a_remote import ContextualRemoteA2aAgent
+from .stls_a2a_secure_remote import (
+    SecureSessionManager,
+    STLSHttpxTransport,
+    ContextualSecureRemoteA2aAgent,
+)

stls_a2a/stls_a2a_adapters.py

@@ -0,0 +1,143 @@
+"""
+Agent Adapters for A2A Secure Transport
+========================================
+
+Common interface and adapters for different AI/Agent frameworks,
+enabling secure A2A communication independent of the framework.
+
+Supported frameworks:
+- Google ADK
+- LangChain
+- Semantic Kernel
+- Custom agents
+
+"""
+
+from typing import Dict, Any, Optional
+
+from .stls_a2a_adapters_model import AgentAdapter,AgentRequest,AgentResponse
+
+from .stls_a2a_google_adapter import GoogleADKAdapter
+from .stls_a2a_lc_adapter import LangChainAdapter
+from .stls_a2a_sk_adapter import SemanticKernelAdapter
+from .stls_a2a_custom_adapter import CustomFunctionAdapter
+
+import logging
+logger = logging.getLogger(__name__)
+
+
+# ============================================================================
+# FACTORY TO CREATE ADAPTERS
+# ============================================================================
+
+def create_adapter(
+    agent,
+    framework: str = 'auto',
+    **kwargs
+) -> AgentAdapter:
+    """
+    Factory to create the appropriate adapter
+
+    Args:
+        agent: Agent instance
+        framework: 'google-adk', 'langchain', 'semantic-kernel', 'custom', or 'auto'
+        **kwargs: Additional parameters for the adapter
+
+    Returns:
+        The appropriate AgentAdapter
+
+    """
+    if framework == 'auto':
+        # Framework auto-detection
+        agent_class = agent.__class__.__name__
+        module = agent.__class__.__module__
+
+        if 'google.adk' in module or 'genai.adk' in module:
+            framework = 'google-adk'
+        elif 'langchain' in module:
+            framework = 'langchain'
+        elif 'semantic_kernel' in module:
+            framework = 'semantic-kernel'
+        else:
+            raise ValueError(
+                f"Cannot auto-detect framework for {agent_class}. "
+                "Please specify framework explicitly."
+            )
+    
+    # Create the appropriate adapter
+    if framework == 'google-adk':
+        return GoogleADKAdapter(agent, **kwargs)
+    elif framework == 'langchain':
+        return LangChainAdapter(agent, **kwargs)
+    elif framework == 'semantic-kernel':
+        return SemanticKernelAdapter(agent, **kwargs)
+    elif framework == 'custom':
+        return CustomFunctionAdapter(agent, **kwargs)
+    else:
+        raise ValueError(f"Unknown framework: {framework}")
+
+
+# ============================================================================
+# UTILITIES FOR A2A <-> AgentRequest/Response CONVERSION
+# ============================================================================
+
+def a2a_to_agent_request(a2a_message: Dict[str, Any]) -> AgentRequest:
+    """
+    Converts an A2A (JSON-RPC) message into an AgentRequest
+
+    Args:
+        a2a_message: A2A JSON-RPC message
+
+    Returns:
+        Standardized AgentRequest
+    """
+    return AgentRequest(
+        method=a2a_message.get('method', 'invoke'),
+        params=a2a_message.get('params', {}),
+        id=a2a_message.get('id'),
+        context=a2a_message.get('context')
+    )
+
+
+def agent_response_to_a2a(response: Optional[AgentResponse]) -> Dict[str, Any]:
+    """
+    Converts an AgentResponse into an A2A (JSON-RPC) message.
+
+    Handles the response=None case (adapter that doesn't explicitly return
+    for unsupported methods) by returning a -32603 error instead of
+    raising an AttributeError.
+    """
+    if response is None:
+        logger.error(
+            "agent_response_to_a2a: received None — adapter invoke() returned no value. "
+            "Check that all code paths in the adapter return an AgentResponse."
+        )
+        return {
+            'jsonrpc': '2.0',
+            'error': {
+                'code': -32603,
+                'message': 'Internal error: adapter returned no response. '
+                           'Possible missing else-branch for unsupported method.',
+            },
+            'id': None,
+        }
+
+    a2a_response: Dict[str, Any] = {
+        'jsonrpc': '2.0',
+        'id': response.id,
+    }
+
+    if response.error:
+        a2a_response['error'] = response.error
+    else:
+        # The result is passed as-is: normalization into A2A format
+        # (parts[]) already happens in CustomFunctionAdapter.invoke() for custom
+        # agents, and in _handle_messages_send() for all the others.
+        # DO NOT normalize here: Task dicts (tasks/get, tasks/cancel, messages/send)
+        # don't have "parts" by design and must not be transformed.
+        a2a_response['result'] = response.result
+
+    if response.metadata:
+        a2a_response['metadata'] = response.metadata
+
+    return a2a_response