agent-framework-lib 0.8.2__tar.gz → 0.8.2.post2__tar.gz

{agent_framework_lib-0.8.2/agent_framework_lib.egg-info → agent_framework_lib-0.8.2.post2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-framework-lib
-Version: 0.8.2
+Version: 0.8.2.post2
 Summary: A comprehensive Python framework for building and serving conversational AI agents with FastAPI
 Author-email: Sebastian Pavel <sebastian@cinco.ai>, Elliott Girard <elliott.girard@icloud.com>
 Maintainer-email: Sebastian Pavel <sebastian@cinco.ai>

{agent_framework_lib-0.8.2 → agent_framework_lib-0.8.2.post2}/agent_framework/__init__.py

@@ -23,11 +23,11 @@ Key Components:
 Example Usage:
     ```python
     from agent_framework import AgentInterface, create_basic_agent_server
-    
+
     class MyAgent(AgentInterface):
         async def handle_message(self, session_id: str, agent_input):
             return StructuredAgentOutput(response_text="Hello!")
-    
+
     # Start server
     create_basic_agent_server(MyAgent, port=8000)
     ```
@@ -48,29 +48,29 @@ logger = logging.getLogger(__name__)
 def _auto_setup_dependencies() -> None:
     """
     Automatically check and install required dependencies at import time.
-    
+
     This runs synchronously when the package is first imported, ensuring
     Playwright browsers and Deno are available before any async tools try to use them.
-    
+
     Set AGENT_FRAMEWORK_SKIP_AUTO_SETUP=1 to disable this behavior.
     """
     if os.environ.get("AGENT_FRAMEWORK_SKIP_AUTO_SETUP", "").lower() in ("1", "true", "yes"):
         return
-    
+
     # Only run once per process
     global _AUTO_SETUP_DONE
     if "_AUTO_SETUP_DONE" in globals() and _AUTO_SETUP_DONE:
         return
     _AUTO_SETUP_DONE = True
-    
+
     try:
         from .utils.post_install import ensure_playwright_browsers, ensure_deno
-        
+
         # Check/install Playwright (silent if already installed)
         success, error = ensure_playwright_browsers()
         if not success and error:
             logger.warning(f"Playwright setup: {error}")
-        
+
         # Check/install Deno (silent if already installed)
         success, message, deno_path = ensure_deno()
         if not success and message:
@@ -79,7 +79,7 @@ def _auto_setup_dependencies() -> None:
             # Success but with PATH instructions
             logger.info(f"Deno: {message}")
             logger.info(f"Deno will be available via: {deno_path}")
-            
+
     except Exception as e:
         # Don't fail the import if setup fails - tools will show errors when used
         logger.debug(f"Auto-setup skipped: {e}")
@@ -157,19 +157,19 @@ from .session.session_storage import (
 
 # File system management (consolidated)
 from .storage.file_system_management import (
-    FileStorageManager, 
-    FileStorageFactory, 
+    FileStorageManager,
+    FileStorageFactory,
     process_file_inputs,
     process_response_file_links,
     get_download_url,
-    get_file_processing_summary, 
-    FileInputMixin
+    get_file_processing_summary,
+    FileInputMixin,
 )
 from .storage.file_storages import (
     FileStorageInterface,
     MetadataStorageInterface,
-    FileMetadata, 
-    LocalFileStorage
+    FileMetadata,
+    LocalFileStorage,
 )
 
 # Optional file storage backends (only available if dependencies are installed)
@@ -196,13 +196,11 @@ __all__ = [
     # Version info
     "__version__",
     "__author__",
-    
     # Core interfaces
     "AgentInterface",
-    "StructuredAgentInput", 
+    "StructuredAgentInput",
     "StructuredAgentOutput",
     "AgentConfig",
-    
     # Base implementations
     "BaseAgent",
     "LlamaIndexAgent",
@@ -210,16 +208,15 @@ __all__ = [
     "StateManager",
     "AgentIdentity",
     "AgentManager",
-    
     # Input/Output types
     "TextInputPart",
-    "ImageUrlInputPart", 
+    "ImageUrlInputPart",
     "FileDataInputPart",
     "AgentInputPartUnion",
     "TextOutputPart",
     "TextOutputStreamPart",
     "JsonOutputPart",
-    "YamlOutputPart", 
+    "YamlOutputPart",
     "FileContentOutputPart",
     "FileReferenceInputPart",
     "FileReferenceOutputPart",
@@ -231,87 +228,77 @@ __all__ = [
     "FileDownloadLinkOutputPart",
     "ImageOutputPart",
     "AgentOutputPartUnion",
-    
     # Utility functions
     "consolidate_text_parts",
     "strip_technical_details",
-    
     # Model configuration
     "ModelConfigManager",
-    "ModelProvider", 
+    "ModelProvider",
     "model_config",
     "ModelClientFactory",
     "client_factory",
-    
     # Session storage
     "SessionStorageInterface",
     "SessionStorageFactory",
     "SessionData",
     "MessageData",
-    "MessageInsight", 
+    "MessageInsight",
     "MessageMetadata",
     "AgentLifecycleData",
     "MemorySessionStorage",
     "MongoDBSessionStorage",
     "history_message_to_message_data",
     "message_data_to_history_message",
-    
     # File storage implementations (consolidated)
     "FileStorageInterface",
     "MetadataStorageInterface",
-    "FileMetadata", 
+    "FileMetadata",
     "LocalFileStorage",
     "S3FileStorage",
     "MinIOFileStorage",
     "S3_AVAILABLE",
     "MINIO_AVAILABLE",
-    
     # File system management (consolidated)
     "FileStorageManager",
-    "FileStorageFactory", 
+    "FileStorageFactory",
     "process_file_inputs",
     "get_file_processing_summary",
     "FileInputMixin",
-    
     # Server
     "app",
     "start_server",
-
     # Utilities
     "get_deno_command",
-
     # Convenience functions
     "create_basic_agent_server",
 ]
 
+
 # Quick start function for convenience
 def create_basic_agent_server(
-    agent_class: type[AgentInterface], 
-    host: str = "0.0.0.0", 
-    port: int = 8000, 
-    reload: bool = False
+    agent_class: type[AgentInterface], host: str = "0.0.0.0", port: int = 8000, reload: bool = False
 ) -> None:
     """
     Quick start function to create and run an agent server.
-    
+
     This function allows external projects to quickly start an agent server
     without needing to create their own server.py file or set environment variables.
-    
+
     Args:
         agent_class: The agent class that implements AgentInterface
         host: Host to bind the server to (default: "0.0.0.0")
         port: Port to run the server on (default: 8000)
         reload: Whether to enable auto-reload for development (default: False)
-                Note: When reload=True, the agent class is temporarily stored in an 
+                Note: When reload=True, the agent class is temporarily stored in an
                 environment variable to survive module reloads.
-    
+
     Returns:
         None (starts the server and blocks)
-    
+
     Raises:
         ImportError: If uvicorn is not available
         ValueError: If agent_class does not implement AgentInterface
-    
+
     Example:
         >>> from agent_framework import create_basic_agent_server, AgentInterface
         >>> from my_agent import MyAgent
@@ -323,17 +310,16 @@ def create_basic_agent_server(
         raise ImportError(
             "uvicorn is required to run the server. Install it with: uv add uvicorn"
         ) from e
-    
+
     # Validate that agent_class implements AgentInterface
     if not issubclass(agent_class, AgentInterface):
-        raise ValueError(
-            f"agent_class must implement AgentInterface, got {agent_class.__name__}"
-        )
-    
+        raise ValueError(f"agent_class must implement AgentInterface, got {agent_class.__name__}")
+
     # Store the agent class globally for immediate use
     from .web import server
+
     server._GLOBAL_AGENT_CLASS = agent_class
-    
+
     # If reload is enabled, also store in environment variable to survive reloads
     # We use the class's module and name to recreate the import path
     if reload:
@@ -341,37 +327,32 @@ def create_basic_agent_server(
         class_name = agent_class.__name__
         agent_class_path = f"{module_name}:{class_name}"
         os.environ["AGENT_CLASS_PATH"] = agent_class_path
-        logger.info(f"[create_basic_agent_server] Reload enabled. Set AGENT_CLASS_PATH={agent_class_path}")
-    
-    logger.info(f"[create_basic_agent_server] Starting server for {agent_class.__name__} on {host}:{port}")
+        logger.info(
+            f"[create_basic_agent_server] Reload enabled. Set AGENT_CLASS_PATH={agent_class_path}"
+        )
+
+    logger.info(
+        f"[create_basic_agent_server] Starting server for {agent_class.__name__} on {host}:{port}"
+    )
     logger.info(f"[create_basic_agent_server] Reload: {reload}")
 
     # Ensure AGENT_HOST and AGENT_PORT are in the environment so that
     # subprocesses (e.g. skill scripts via ShellTool) can reach this server.
     os.environ["AGENT_HOST"] = host
     os.environ["AGENT_PORT"] = str(port)
-    
+
     # When reload=True, uvicorn requires an import string, not the app object directly
     if reload:
         # Use the agent_framework.web.server:app import string for reload mode
-        uvicorn.run(
-            "agent_framework.web.server:app",
-            host=host,
-            port=port,
-            reload=reload
-        )
+        uvicorn.run("agent_framework.web.server:app", host=host, port=port, reload=reload)
     else:
         # Import the app after setting the global variable for non-reload mode
         from .web.server import app
+
         # For non-reload mode, we can pass the app object directly
-        uvicorn.run(
-            app,
-            host=host,
-            port=port,
-            reload=reload
-        )
+        uvicorn.run(app, host=host, port=port, reload=reload)
 
 
 # Run auto-setup when package is imported
 # This ensures Playwright browsers and Deno are installed before any async tools need them
-_auto_setup_dependencies()
+_auto_setup_dependencies()

{agent_framework_lib-0.8.2 → agent_framework_lib-0.8.2.post2}/agent_framework/a2a/base.py

@@ -11,6 +11,7 @@ logger = logging.getLogger(__name__)
 
 class A2ATaskProviderError(Exception):
     """Base exception for task provider errors."""
+
     def __init__(self, message: str, code: int = -32603):
         super().__init__(message)
         self.code = code
@@ -18,6 +19,7 @@ class A2ATaskProviderError(Exception):
 
 class TaskNotFoundError(A2ATaskProviderError):
     """Raised when a task ID is not found."""
+
     def __init__(self, task_id: str):
         super().__init__(f"Task not found: {task_id}", code=-32001)
         self.task_id = task_id
@@ -25,6 +27,7 @@ class TaskNotFoundError(A2ATaskProviderError):
 
 class TaskNotCancelableError(A2ATaskProviderError):
     """Raised when trying to cancel a task in a terminal state."""
+
     def __init__(self, task_id: str, current_state: str):
         super().__init__(
             f"Task {task_id} is in terminal state '{current_state}' and cannot be canceled",
@@ -36,6 +39,7 @@ class TaskNotCancelableError(A2ATaskProviderError):
 
 class TaskTerminalStateError(A2ATaskProviderError):
     """Raised when trying to update a task already in a terminal state."""
+
     def __init__(self, task_id: str, current_state: str):
         super().__init__(
             f"Task {task_id} is in terminal state '{current_state}' and cannot be modified",

{agent_framework_lib-0.8.2 → agent_framework_lib-0.8.2.post2}/agent_framework/a2a/endpoints/agent_card_builder.py

@@ -48,7 +48,6 @@ class AgentCardBuilder:
         }
         return card
 
-
     def _resolve_url(
         self,
         injected_url: str | None = None,

{agent_framework_lib-0.8.2 → agent_framework_lib-0.8.2.post2}/agent_framework/a2a/endpoints/agent_card_skill_builder.py

@@ -9,12 +9,42 @@ from typing import Any
 
 logger = logging.getLogger(__name__)
 
-_STOP_WORDS = frozenset({
-    "the", "and", "for", "with", "from", "that", "this", "are", "was",
-    "will", "can", "has", "have", "been", "not", "but", "all", "any",
-    "each", "more", "also", "into", "over", "such", "than", "its",
-    "about", "between", "through", "during", "before", "after",
-})
+_STOP_WORDS = frozenset(
+    {
+        "the",
+        "and",
+        "for",
+        "with",
+        "from",
+        "that",
+        "this",
+        "are",
+        "was",
+        "will",
+        "can",
+        "has",
+        "have",
+        "been",
+        "not",
+        "but",
+        "all",
+        "any",
+        "each",
+        "more",
+        "also",
+        "into",
+        "over",
+        "such",
+        "than",
+        "its",
+        "about",
+        "between",
+        "through",
+        "during",
+        "before",
+        "after",
+    }
+)
 
 
 class AgentCardSkillBuilder:
@@ -85,12 +115,14 @@ class AgentCardSkillBuilder:
             joined = "; ".join(descriptions)
             description = self._truncate_description(joined, len(framework_skills))
 
-            return [{
-                "id": "framework-capabilities",
-                "name": "Framework Capabilities",
-                "description": description,
-                "tags": names,
-            }]
+            return [
+                {
+                    "id": "framework-capabilities",
+                    "name": "Framework Capabilities",
+                    "description": description,
+                    "tags": names,
+                }
+            ]
         except Exception:
             logger.warning("Failed to build framework skill from registry")
             return [self._make_default_framework_skill()]
@@ -108,12 +140,14 @@ class AgentCardSkillBuilder:
 
         result = []
         for server_key in mcp_params:
-            result.append({
-                "id": f"mcp-{server_key}",
-                "name": f"MCP: {server_key}",
-                "description": f"MCP server: {server_key}",
-                "tags": ["mcp", server_key],
-            })
+            result.append(
+                {
+                    "id": f"mcp-{server_key}",
+                    "name": f"MCP: {server_key}",
+                    "description": f"MCP server: {server_key}",
+                    "tags": ["mcp", server_key],
+                }
+            )
         return result
 
     def _build_tool_skills(self, agent_instance: Any) -> list[dict]:
@@ -128,16 +162,16 @@ class AgentCardSkillBuilder:
         for tool in agent_tools:
             try:
                 tool_name = (
-                    getattr(tool, "__name__", None)
-                    or getattr(tool, "name", None)
-                    or str(tool)
+                    getattr(tool, "__name__", None) or getattr(tool, "name", None) or str(tool)
+                )
+                result.append(
+                    {
+                        "id": f"tool-{tool_name}",
+                        "name": tool_name,
+                        "description": f"Tool: {tool_name}",
+                        "tags": ["tool", tool_name],
+                    }
                 )
-                result.append({
-                    "id": f"tool-{tool_name}",
-                    "name": tool_name,
-                    "description": f"Tool: {tool_name}",
-                    "tags": ["tool", tool_name],
-                })
             except Exception:
                 logger.warning("Failed to build skill for tool %s", tool)
         return result
@@ -192,9 +226,7 @@ class AgentCardSkillBuilder:
                 break
         return tags
 
-    def _truncate_description(
-        self, description: str, total_skills: int = 0
-    ) -> str:
+    def _truncate_description(self, description: str, total_skills: int = 0) -> str:
         """Truncates description to MAX_FRAMEWORK_DESC_LENGTH with suffix.
 
         Args:

{agent_framework_lib-0.8.2 → agent_framework_lib-0.8.2.post2}/agent_framework/a2a/endpoints/jsonrpc_dispatcher.py

@@ -79,7 +79,9 @@ class JSONRPCDispatcher:
                 try:
                     send_params = MessageSendParams(**params)
                 except ValidationError as e:
-                    return self._error_response(request_id, INVALID_PARAMS, "Invalid params", {"details": str(e)})
+                    return self._error_response(
+                        request_id, INVALID_PARAMS, "Invalid params", {"details": str(e)}
+                    )
                 result = await self._translation_layer.handle_message_send(send_params, request_id)
                 return self._success_response(request_id, result)
 
@@ -87,15 +89,21 @@ class JSONRPCDispatcher:
                 try:
                     stream_params = MessageSendParams(**params)
                 except ValidationError as e:
-                    return self._error_response(request_id, INVALID_PARAMS, "Invalid params", {"details": str(e)})
-                generator = await self._translation_layer.handle_message_stream(stream_params, request_id)
+                    return self._error_response(
+                        request_id, INVALID_PARAMS, "Invalid params", {"details": str(e)}
+                    )
+                generator = await self._translation_layer.handle_message_stream(
+                    stream_params, request_id
+                )
                 return StreamingResponse(generator, media_type="text/event-stream")
 
             elif method == "tasks/get":
                 try:
                     get_params = TaskGetParams(**params)
                 except ValidationError as e:
-                    return self._error_response(request_id, INVALID_PARAMS, "Invalid params", {"details": str(e)})
+                    return self._error_response(
+                        request_id, INVALID_PARAMS, "Invalid params", {"details": str(e)}
+                    )
                 result = await self._translation_layer.handle_tasks_get(get_params, request_id)
                 return self._success_response(request_id, result)
 
@@ -103,8 +111,12 @@ class JSONRPCDispatcher:
                 try:
                     cancel_params = TaskCancelParams(**params)
                 except ValidationError as e:
-                    return self._error_response(request_id, INVALID_PARAMS, "Invalid params", {"details": str(e)})
-                result = await self._translation_layer.handle_tasks_cancel(cancel_params, request_id)
+                    return self._error_response(
+                        request_id, INVALID_PARAMS, "Invalid params", {"details": str(e)}
+                    )
+                result = await self._translation_layer.handle_tasks_cancel(
+                    cancel_params, request_id
+                )
                 return self._success_response(request_id, result)
 
             else:

{agent_framework_lib-0.8.2 → agent_framework_lib-0.8.2.post2}/agent_framework/a2a/endpoints/sse_wrapper.py

@@ -42,9 +42,7 @@ class SSEWrapperA2A:
         """
         is_first_chunk = True
 
-        yield self._format_sse(
-            request_id, self._status_event(task_id, context_id, "working")
-        )
+        yield self._format_sse(request_id, self._status_event(task_id, context_id, "working"))
 
         try:
             async for output in internal_stream:
@@ -57,7 +55,7 @@ class SSEWrapperA2A:
 
                             # Skip activity markers and emit status event
                             if text.startswith("__STREAM_ACTIVITY__"):
-                                activity_text = text[len("__STREAM_ACTIVITY__"):]
+                                activity_text = text[len("__STREAM_ACTIVITY__") :]
                                 yield self._format_sse(
                                     request_id,
                                     self._status_event(
@@ -68,15 +66,13 @@ class SSEWrapperA2A:
 
                             # Strip chunk marker prefix
                             if text.startswith("__STREAM_CHUNK__"):
-                                text = text[len("__STREAM_CHUNK__"):]
+                                text = text[len("__STREAM_CHUNK__") :]
 
                             if text:
                                 append = not is_first_chunk
                                 yield self._format_sse(
                                     request_id,
-                                    self._artifact_event(
-                                        task_id, context_id, text, append
-                                    ),
+                                    self._artifact_event(task_id, context_id, text, append),
                                 )
                                 is_first_chunk = False
 
@@ -85,14 +81,10 @@ class SSEWrapperA2A:
                             content = getattr(part, "content", "") or activity_type
                             yield self._format_sse(
                                 request_id,
-                                self._status_event(
-                                    task_id, context_id, "working", content
-                                ),
+                                self._status_event(task_id, context_id, "working", content),
                             )
 
-            yield self._format_sse(
-                request_id, self._status_event(task_id, context_id, "completed")
-            )
+            yield self._format_sse(request_id, self._status_event(task_id, context_id, "completed"))
 
         except Exception as e:
             logger.exception("Error during SSE streaming")

{agent_framework_lib-0.8.2 → agent_framework_lib-0.8.2.post2}/agent_framework/a2a/endpoints/translation_layer.py

@@ -104,7 +104,9 @@ class A2ATranslationLayer:
         from ...core.agent_interface import StructuredAgentInput, TextInputPart
 
         text = self.extract_text(params.message.parts)
-        context_id_param = getattr(params.configuration, "contextId", None) if params.configuration else None
+        context_id_param = (
+            getattr(params.configuration, "contextId", None) if params.configuration else None
+        )
         session_id, _is_new = self._resolve_context(context_id_param)
         context_id = session_id
 
@@ -183,7 +185,7 @@ class A2ATranslationLayer:
                         text = getattr(part, "text", "") or ""
                         # Strip the streaming marker prefix
                         if text.startswith("__STREAM_CHUNK__"):
-                            text = text[len("__STREAM_CHUNK__"):]
+                            text = text[len("__STREAM_CHUNK__") :]
                         if text:
                             chunks.append(text)
             if output.response_text:
@@ -208,7 +210,9 @@ class A2ATranslationLayer:
         from ...core.agent_interface import StructuredAgentInput, TextInputPart
 
         text = self.extract_text(params.message.parts)
-        context_id_param = getattr(params.configuration, "contextId", None) if params.configuration else None
+        context_id_param = (
+            getattr(params.configuration, "contextId", None) if params.configuration else None
+        )
         session_id, _is_new = self._resolve_context(context_id_param)
         context_id = session_id
 
@@ -256,6 +260,7 @@ class A2ATranslationLayer:
                 ],
             )
             import json
+
             yield f"data: {json.dumps(result.model_dump())}\n\n"
 
     async def _consume_stream_from_generator(self, stream: AsyncGenerator) -> str:
@@ -275,7 +280,7 @@ class A2ATranslationLayer:
                     if part_type in ("text_output_stream", "text_output"):
                         text = getattr(part, "text", "") or ""
                         if text.startswith("__STREAM_CHUNK__"):
-                            text = text[len("__STREAM_CHUNK__"):]
+                            text = text[len("__STREAM_CHUNK__") :]
                         if text:
                             chunks.append(text)
             if output.response_text:
@@ -311,10 +316,14 @@ class A2ATranslationLayer:
             status=A2ATaskStatus(
                 state=task.status.value,
                 timestamp=task.updated_at.isoformat() if task.updated_at else None,
-                message=A2AMessage(
-                    role="agent",
-                    parts=[A2APart(kind="text", text=task.status_message.get("text", ""))],
-                ) if task.status_message and task.status_message.get("text") else None,
+                message=(
+                    A2AMessage(
+                        role="agent",
+                        parts=[A2APart(kind="text", text=task.status_message.get("text", ""))],
+                    )
+                    if task.status_message and task.status_message.get("text")
+                    else None
+                ),
             ),
             artifacts=artifacts if artifacts else None,
             history=history,

{agent_framework_lib-0.8.2 → agent_framework_lib-0.8.2.post2}/agent_framework/a2a/models.py

@@ -12,6 +12,7 @@ logger = logging.getLogger(__name__)
 
 class TaskStatus(str, Enum):
     """A2A task lifecycle states."""
+
     SUBMITTED = "submitted"
     WORKING = "working"
     INPUT_REQUIRED = "input-required"
@@ -25,12 +26,14 @@ class TaskStatus(str, Enum):
         return self in _TERMINAL_STATES
 
 
-_TERMINAL_STATES = frozenset({
-    TaskStatus.COMPLETED,
-    TaskStatus.FAILED,
-    TaskStatus.CANCELED,
-    TaskStatus.REJECTED,
-})
+_TERMINAL_STATES = frozenset(
+    {
+        TaskStatus.COMPLETED,
+        TaskStatus.FAILED,
+        TaskStatus.CANCELED,
+        TaskStatus.REJECTED,
+    }
+)
 
 
 @dataclass
@@ -40,6 +43,7 @@ class A2ATask:
     Links an A2A task_id to a framework session_id + message_id.
     The actual conversational content lives in the session store.
     """
+
     id: str
     context_id: str
     session_id: str