jaf-py 2.5.10__py3-none-any.whl → 2.5.12__py3-none-any.whl

jaf/__init__.py

@@ -16,6 +16,7 @@ from .core.tools import (
 )
 from .core.tracing import ConsoleTraceCollector, TraceCollector
 from .core.types import *
+
 # Import attachment utilities for convenience
 from .utils.attachments import (
     make_image_attachment,
@@ -83,6 +84,7 @@ try:
         monitor_performance,
         get_performance_summary,
     )
+
     _PERFORMANCE_AVAILABLE = True
 except ImportError:
     _PERFORMANCE_AVAILABLE = False
@@ -96,6 +98,7 @@ try:
         create_sse_response,
         stream_to_websocket,
     )
+
     _STREAMING_AVAILABLE = True
 except ImportError:
     _STREAMING_AVAILABLE = False
@@ -110,6 +113,7 @@ try:
         with_timeout,
         compose,
     )
+
     _COMPOSITION_AVAILABLE = True
 except ImportError:
     _COMPOSITION_AVAILABLE = False
@@ -122,6 +126,7 @@ try:
         PluginRegistry,
         get_plugin_registry,
     )
+
     _PLUGINS_AVAILABLE = True
 except ImportError:
     _PLUGINS_AVAILABLE = False
@@ -135,6 +140,7 @@ try:
         get_analytics_report,
         analyze_conversation_quality,
     )
+
     _ANALYTICS_AVAILABLE = True
 except ImportError:
     _ANALYTICS_AVAILABLE = False
@@ -159,6 +165,7 @@ try:
         create_parallel_workflow,
         execute_workflow_stream,
     )
+
     _WORKFLOWS_AVAILABLE = True
 except ImportError:
     _WORKFLOWS_AVAILABLE = False
@@ -174,6 +181,7 @@ try:
         get_graph_dot,
         validate_graph_options,
     )
+
     _VISUALIZATION_AVAILABLE = True
 except ImportError:
     _VISUALIZATION_AVAILABLE = False
@@ -187,47 +195,89 @@ def generate_trace_id() -> TraceId:
     """Generate a new trace ID."""
     return create_trace_id(str(uuid.uuid4()))
 
+
 def generate_run_id() -> RunId:
     """Generate a new run ID."""
     return create_run_id(str(uuid.uuid4()))
 
-__version__ = "2.5.10"
+
+__version__ = "2.5.12"
 __all__ = [
     # Core types and functions
-    "TraceId", "RunId", "ValidationResult", "Message", "ModelConfig",
-    "Tool", "Agent", "Guardrail", "RunState", "JAFError", "RunResult",
-    "TraceEvent", "ModelProvider", "RunConfig",
-    "create_trace_id", "create_run_id", "generate_trace_id", "generate_run_id",
-    
+    "TraceId",
+    "RunId",
+    "ValidationResult",
+    "Message",
+    "ModelConfig",
+    "Tool",
+    "Agent",
+    "Guardrail",
+    "RunState",
+    "JAFError",
+    "RunResult",
+    "TraceEvent",
+    "ModelProvider",
+    "RunConfig",
+    "create_trace_id",
+    "create_run_id",
+    "generate_trace_id",
+    "generate_run_id",
     # Enums for type safety
-    "Model", "ToolParameterType", "ToolSource", "ContentRole", "PartType",
-    
+    "Model",
+    "ToolParameterType",
+    "ToolSource",
+    "ContentRole",
+    "PartType",
     # Tool factory functions
-    "create_function_tool", "create_function_tool_legacy",
-    "create_async_function_tool", "create_async_function_tool_legacy",
-    "FunctionToolConfig", "ToolExecuteFunction", "function_tool",
-
+    "create_function_tool",
+    "create_function_tool_legacy",
+    "create_async_function_tool",
+    "create_async_function_tool_legacy",
+    "FunctionToolConfig",
+    "ToolExecuteFunction",
+    "function_tool",
     # Exception classes
-    "JAFException", "AgentException", "AgentNotFoundError", "HandoffError",
-    "ToolException", "ToolExecutionError", "ToolValidationError",
-    "ModelException", "ModelProviderError", "ModelResponseError",
-    "ValidationException", "GuardrailViolationError", "InputValidationError",
-    "MemoryException", "MemoryConnectionError", "MemoryStorageError",
-    "SessionException", "SessionStateError", "MaxTurnsExceededError",
-    "A2AException", "A2AProtocolError", "A2ATaskError",
-    "ConfigurationException", "InvalidConfigurationError",
-    "create_agent_error", "create_tool_error", "create_session_error", "create_memory_error",
-
+    "JAFException",
+    "AgentException",
+    "AgentNotFoundError",
+    "HandoffError",
+    "ToolException",
+    "ToolExecutionError",
+    "ToolValidationError",
+    "ModelException",
+    "ModelProviderError",
+    "ModelResponseError",
+    "ValidationException",
+    "GuardrailViolationError",
+    "InputValidationError",
+    "MemoryException",
+    "MemoryConnectionError",
+    "MemoryStorageError",
+    "SessionException",
+    "SessionStateError",
+    "MaxTurnsExceededError",
+    "A2AException",
+    "A2AProtocolError",
+    "A2ATaskError",
+    "ConfigurationException",
+    "InvalidConfigurationError",
+    "create_agent_error",
+    "create_tool_error",
+    "create_session_error",
+    "create_memory_error",
     # Engine
     "run",
-
     # Tracing
-    "TraceCollector", "ConsoleTraceCollector",
-
+    "TraceCollector",
+    "ConsoleTraceCollector",
     # Tool results
-    "ToolResult", "ToolResultStatus", "ToolResponse", "ToolErrorCodes",
-    "with_error_handling", "require_permissions", "tool_result_to_string",
-
+    "ToolResult",
+    "ToolResultStatus",
+    "ToolResponse",
+    "ToolErrorCodes",
+    "with_error_handling",
+    "require_permissions",
+    "tool_result_to_string",
     # Providers
     "make_litellm_provider",
     "FastMCPTool",
@@ -235,45 +285,92 @@ __all__ = [
     "create_mcp_stdio_tools",
     "create_mcp_sse_tools",
     "create_mcp_http_tools",
-
     # Memory system
-    "ConversationMemory", "MemoryProvider", "MemoryQuery", "MemoryConfig",
-    "Result", "Success", "Failure",
-    "InMemoryConfig", "RedisConfig", "PostgresConfig", "MemoryProviderConfig",
-    "MemoryError", "MemoryConnectionError", "MemoryNotFoundError", "MemoryStorageError",
-    "create_memory_provider_from_env", "get_memory_provider_info", "test_memory_provider_connection",
-    "create_in_memory_provider", "create_redis_provider", "create_postgres_provider",
-
+    "ConversationMemory",
+    "MemoryProvider",
+    "MemoryQuery",
+    "MemoryConfig",
+    "Result",
+    "Success",
+    "Failure",
+    "InMemoryConfig",
+    "RedisConfig",
+    "PostgresConfig",
+    "MemoryProviderConfig",
+    "MemoryError",
+    "MemoryConnectionError",
+    "MemoryNotFoundError",
+    "MemoryStorageError",
+    "create_memory_provider_from_env",
+    "get_memory_provider_info",
+    "test_memory_provider_connection",
+    "create_in_memory_provider",
+    "create_redis_provider",
+    "create_postgres_provider",
     # Server
     "run_server",
-    
     # Performance monitoring (NEW)
-    "PerformanceMonitor", "PerformanceMetrics", "monitor_performance", "get_performance_summary",
-    
+    "PerformanceMonitor",
+    "PerformanceMetrics",
+    "monitor_performance",
+    "get_performance_summary",
     # Streaming support (NEW)
-    "run_streaming", "StreamingEvent", "StreamingEventType", "StreamingCollector",
-    "create_sse_response", "stream_to_websocket",
-    
+    "run_streaming",
+    "StreamingEvent",
+    "StreamingEventType",
+    "StreamingCollector",
+    "create_sse_response",
+    "stream_to_websocket",
     # Tool composition (NEW)
-    "create_tool_pipeline", "create_parallel_tools", "create_conditional_tool",
-    "with_retry", "with_cache", "with_timeout", "compose",
-    
+    "create_tool_pipeline",
+    "create_parallel_tools",
+    "create_conditional_tool",
+    "with_retry",
+    "with_cache",
+    "with_timeout",
+    "compose",
     # Plugin system (NEW)
-    "JAFPlugin", "PluginMetadata", "PluginStatus", "PluginRegistry", "get_plugin_registry",
-    
+    "JAFPlugin",
+    "PluginMetadata",
+    "PluginStatus",
+    "PluginRegistry",
+    "get_plugin_registry",
     # Analytics system (NEW)
-    "ConversationAnalytics", "AgentAnalytics", "SystemAnalytics", "AnalyticsEngine",
-    "get_analytics_report", "analyze_conversation_quality",
-    
+    "ConversationAnalytics",
+    "AgentAnalytics",
+    "SystemAnalytics",
+    "AnalyticsEngine",
+    "get_analytics_report",
+    "analyze_conversation_quality",
     # Workflow orchestration (NEW)
-    "Workflow", "WorkflowStep", "WorkflowContext", "WorkflowResult", "StepResult",
-    "WorkflowStatus", "StepStatus", "AgentStep", "ToolStep", "ConditionalStep",
-    "ParallelStep", "LoopStep", "WorkflowBuilder", "create_workflow",
-    "create_sequential_workflow", "create_parallel_workflow", "execute_workflow_stream",
+    "Workflow",
+    "WorkflowStep",
+    "WorkflowContext",
+    "WorkflowResult",
+    "StepResult",
+    "WorkflowStatus",
+    "StepStatus",
+    "AgentStep",
+    "ToolStep",
+    "ConditionalStep",
+    "ParallelStep",
+    "LoopStep",
+    "WorkflowBuilder",
+    "create_workflow",
+    "create_sequential_workflow",
+    "create_parallel_workflow",
+    "execute_workflow_stream",
 ] + (
     # Visualization (conditional)
     [
-        "generate_agent_graph", "generate_tool_graph", "generate_runner_graph",
-        "GraphOptions", "GraphResult", "get_graph_dot", "validate_graph_options"
-    ] if _VISUALIZATION_AVAILABLE else []
+        "generate_agent_graph",
+        "generate_tool_graph",
+        "generate_runner_graph",
+        "GraphOptions",
+        "GraphResult",
+        "get_graph_dot",
+        "validate_graph_options",
+    ]
+    if _VISUALIZATION_AVAILABLE
+    else []
 )

jaf/a2a/__init__.py

@@ -195,28 +195,49 @@ from .types import (
 # Main exports for easy access
 __all__ = [
     # Types
-    "A2AMessage", "A2ATask", "A2AAgent", "A2AAgentTool", "A2AArtifact",
-    "A2ATaskStatus", "A2APart", "A2AStreamEvent", "A2AError", "A2AErrorCodes",
-    "JSONRPCRequest", "JSONRPCResponse", "JSONRPCError",
-    "AgentCard", "AgentSkill", "AgentCapabilities", "AgentProvider",
-    "A2AServerConfig", "A2AClientConfig", "A2AClientState",
-
+    "A2AMessage",
+    "A2ATask",
+    "A2AAgent",
+    "A2AAgentTool",
+    "A2AArtifact",
+    "A2ATaskStatus",
+    "A2APart",
+    "A2AStreamEvent",
+    "A2AError",
+    "A2AErrorCodes",
+    "JSONRPCRequest",
+    "JSONRPCResponse",
+    "JSONRPCError",
+    "AgentCard",
+    "AgentSkill",
+    "AgentCapabilities",
+    "AgentProvider",
+    "A2AServerConfig",
+    "A2AClientConfig",
+    "A2AClientState",
     # Agent utilities
-    "create_a2a_agent", "create_a2a_tool", "transform_a2a_agent_to_jaf",
-    "process_agent_query", "execute_a2a_agent", "execute_a2a_agent_with_streaming",
-
+    "create_a2a_agent",
+    "create_a2a_tool",
+    "transform_a2a_agent_to_jaf",
+    "process_agent_query",
+    "execute_a2a_agent",
+    "execute_a2a_agent_with_streaming",
     # Protocol
-    "validate_jsonrpc_request", "route_a2a_request",
-
+    "validate_jsonrpc_request",
+    "route_a2a_request",
     # Agent Card
-    "generate_agent_card", "validate_agent_card",
-
+    "generate_agent_card",
+    "validate_agent_card",
     # Server
-    "create_a2a_server", "start_a2a_server", "create_server_config",
-
+    "create_a2a_server",
+    "start_a2a_server",
+    "create_server_config",
     # Client
-    "create_a2a_client", "connect_to_a2a_agent", "send_message",
-    "stream_message", "discover_agents"
+    "create_a2a_client",
+    "connect_to_a2a_agent",
+    "send_message",
+    "stream_message",
+    "discover_agents",
 ]
 
 
@@ -224,12 +245,12 @@ __all__ = [
 class A2A:
     """
     Convenience class for A2A operations
-    
+
     Usage:
         # Create client
         a2a = A2A.client("http://localhost:3000")
         response = await a2a.ask("Hello")
-        
+
         # Create server
         server = A2A.server(agents, "My Server", "Description", 3000)
         await server.start()
@@ -279,11 +300,11 @@ A2A_SUPPORTED_METHODS = [
     "message/stream",
     "tasks/get",
     "tasks/cancel",
-    "agent/getAuthenticatedExtendedCard"
+    "agent/getAuthenticatedExtendedCard",
 ]
 A2A_SUPPORTED_TRANSPORTS = ["JSONRPC"]
 A2A_DEFAULT_CAPABILITIES = {
     "streaming": True,
     "pushNotifications": False,
-    "stateTransitionHistory": True
+    "stateTransitionHistory": True,
 }