smarta2a 0.4.11__py3-none-any.whl → 0.4.12__py3-none-any.whl

smarta2a/agent/a2a_human.py

@@ -7,7 +7,6 @@ from typing import Optional
 
 # Local imports
 from smarta2a.server import SmartA2A
-from smarta2a.model_providers.base_llm_provider import BaseLLMProvider
 from smarta2a.server.state_manager import StateManager
 from smarta2a.utils.types import StateData, SendTaskRequest, AgentCard, WebhookRequest, WebhookResponse, TextPart, DataPart, FilePart
 from smarta2a.client.a2a_client import A2AClient
@@ -16,11 +15,9 @@ class A2AHuman:
     def __init__(
             self,
             name: str,
-            model_provider: BaseLLMProvider,
             agent_card: AgentCard = None,
             state_manager: StateManager = None,
         ):
-        self.model_provider = model_provider
         self.state_manager = state_manager
         self.app = SmartA2A(
             name=name,
@@ -32,7 +29,7 @@ class A2AHuman:
     def __register_handlers(self):
         @self.app.on_event("startup")
         async def on_startup():
-            await self.model_provider.load()
+            pass
         
         @self.app.app.get("/tasks")
         async def get_tasks(fields: Optional[str] = Query(None)):
@@ -42,7 +39,7 @@ class A2AHuman:
         
         @self.app.on_send_task(forward_to_webhook=True)
         async def on_send_task(request: SendTaskRequest, state: StateData):
-            return "Give me some time to respond"
+            return "I am human and take some time to respond, but I will definitely respond to your request"
         
         @self.app.webhook()
         async def on_webhook(request: WebhookRequest, state: StateData):

smarta2a/client/a2a_client.py

@@ -54,6 +54,15 @@ class A2AClient:
         metadata: dict[str, Any] | None = None,
     ):
         """Send a task to another Agent."""
+
+        # Auto-create PushNotificationConfig if not provided and we have a URL
+        if push_notification is None and self.url:
+            push_notification = PushNotificationConfig(
+                url=f"{self.url}/webhook",
+                token=None,
+                authentication=None
+            )
+
         params = TaskRequestBuilder.build_send_task_request(
             id=id,
             role=role,
@@ -84,6 +93,15 @@ class A2AClient:
         metadata: dict[str, Any] | None = None,
     ):
         """Send to another Agent and receive a stream of responses."""
+        
+        # Auto-create PushNotificationConfig if not provided and we have a URL
+        if push_notification is None and self.url:
+            push_notification = PushNotificationConfig(
+                url=f"{self.url}/webhook",
+                token=None,
+                authentication=None
+            )
+
         params = TaskRequestBuilder.build_send_task_request(
             id=id,
             role=role,

smarta2a/file_stores/base_file_store.py

@@ -0,0 +1,27 @@
+# Library imports
+from abc import ABC, abstractmethod
+from typing import Optional
+
+class BaseFileStore(ABC):
+    """Separate interface for file operations"""
+    
+    @abstractmethod
+    async def upload(
+        self,
+        content: bytes,
+        task_id: str,
+        filename: Optional[str] = None
+    ) -> str:
+        pass
+
+    @abstractmethod
+    async def download(self, uri: str) -> bytes:
+        pass
+
+    @abstractmethod
+    async def delete_for_task(self, task_id: str):
+        pass
+
+    @abstractmethod
+    async def list_files(self, task_id: str) -> list[str]:
+        pass

smarta2a/file_stores/local_file_store.py

@@ -0,0 +1,75 @@
+# Library imports
+from pathlib import Path
+from typing import Optional
+import hashlib
+import aiofiles
+import aiofiles.os
+import shutil
+
+# Local imports
+from smarta2a.file_stores.base_file_store import BaseFileStore
+
+class LocalFileStore(BaseFileStore):
+    """
+    Local filesystem implementation that mimics cloud storage patterns
+    - Stores files in task-specific directories
+    - Uses content-addressable storage for deduplication
+    - Generates file:// URIs for compatibility
+    """
+    
+    def __init__(self, base_path: str = "./filestore"):
+        self.base_path = Path(base_path).resolve()
+        self.base_path.mkdir(parents=True, exist_ok=True)
+    
+    async def upload(self, content: bytes, task_id: str, filename: Optional[str] = None) -> str:
+        # Create task directory if not exists
+        task_dir = self.base_path / task_id
+        await aiofiles.os.makedirs(task_dir, exist_ok=True)
+        
+        # Ensure content is bytes for hashing
+        if isinstance(content, str):
+            content = content.encode('utf-8')
+        
+        # Generate content hash for deduplication
+        content_hash = hashlib.sha256(content).hexdigest()
+        file_ext = Path(filename).suffix if filename else ""
+        unique_name = f"{content_hash}{file_ext}"
+        
+        # Write file
+        file_path = task_dir / unique_name
+        async with aiofiles.open(file_path, "wb") as f:
+            await f.write(content)
+            
+        return f"file://{file_path}"
+
+
+    async def download(self, uri: str) -> bytes:
+        # Parse file:// URI
+        path = Path(uri.replace("file://", ""))
+        if not path.exists():
+            raise FileNotFoundError(f"File not found at {uri}")
+            
+        async with aiofiles.open(path, "rb") as f:
+            return await f.read()
+
+    async def delete_for_task(self, task_id: str) -> None:
+        task_dir = self.base_path / task_id
+        if await aiofiles.os.path.exists(task_dir):
+            await aiofiles.os.rmdir(task_dir)
+
+    async def list_files(self, task_id: str) -> list[str]:
+        task_dir = self.base_path / task_id
+        if not await aiofiles.os.path.exists(task_dir):
+            return []
+            
+        return [
+            f"file://{file_path}"
+            for file_path in task_dir.iterdir()
+            if file_path.is_file()
+        ]
+
+    async def clear_all(self) -> None:
+        """Clear entire storage (useful for testing)"""
+        if await aiofiles.os.path.exists(self.base_path):
+            shutil.rmtree(self.base_path)
+        await aiofiles.os.makedirs(self.base_path, exist_ok=True)

smarta2a/model_providers/openai_provider.py

@@ -69,35 +69,33 @@ class OpenAIProvider(BaseLLMProvider):
             self.agent_cards
         )
     
+    
     def _convert_part(self, part: Union[TextPart, FilePart, DataPart]) -> dict:
         """Convert a single part to OpenAI-compatible format"""
         if isinstance(part, TextPart):
             return {"type": "text", "text": part.text}
             
         elif isinstance(part, FilePart):
-            if part.file.mimeType not in self.supported_media_types:
-                raise ValueError(f"Unsupported media type: {part.file.mimeType}")
-                
-            if part.file.uri:
-                return {
-                    "type": "image_url",
-                    "image_url": {"url": part.file.uri}
-                }
-            elif part.file.bytes:
-                return {
-                    "type": "image_url",
-                    "image_url": {
-                        "url": f"data:{part.file.mimeType};base64,{part.file.bytes}"
-                    }
-                }
-                
-        elif isinstance(part, DataPart):
+            # Treat all files as attachments with placeholder metadata
+            fc = part.file
+            # Determine URI (either direct or data URL)
+            if fc.uri:
+                uri = fc.uri
+            else:
+                uri = f"data:{fc.mimeType};base64,{fc.bytes}"
+
+            placeholder = (
+                f'[FileAttachment name="{fc.name or ""}" '
+                f'mimeType="{fc.mimeType or ""}" '
+                f'uri="{uri}"]'
+            )
+            return {"type": "text", "text": placeholder}
+
+        else:
             return {
                 "type": "text",
                 "text": f"[Structured Data]\n{json.dumps(part.data, indent=2)}"
             }
-            
-        raise ValueError(f"Unsupported part type: {type(part)}")
 
 
     def _convert_messages(self, messages: List[Message]) -> List[dict]:

smarta2a/server/state_manager.py

@@ -1,15 +1,18 @@
 # Library imports
 from typing import Optional, Dict, Any, List
+import base64
 
 # Local imports
 from smarta2a.state_stores.base_state_store import BaseStateStore
+from smarta2a.file_stores.base_file_store import BaseFileStore
 from smarta2a.history_update_strategies.history_update_strategy import HistoryUpdateStrategy
-from smarta2a.utils.types import Message, StateData, Task, TaskStatus, TaskState, PushNotificationConfig, Part
+from smarta2a.utils.types import Message, StateData, Task, TaskStatus, TaskState, PushNotificationConfig, Part, FilePart
 from smarta2a.server.nats_client import NATSClient
 
 class StateManager:
-    def __init__(self, state_store: BaseStateStore, history_strategy: HistoryUpdateStrategy, nats_server_url: Optional[str] = "nats://localhost:4222"):
+    def __init__(self, state_store: BaseStateStore, file_store: BaseFileStore, history_strategy: HistoryUpdateStrategy, nats_server_url: Optional[str] = "nats://localhost:4222"):
         self.state_store = state_store
+        self.file_store = file_store
         self.strategy = history_strategy
         self.nats_client = NATSClient(server_url=nats_server_url)
     
@@ -72,6 +75,9 @@ class StateManager:
         )
         latest_state.task.metadata = metadata
 
+        # Process files before persistence
+        await self._process_file_parts(latest_state)
+
         await self.update_state(latest_state)
 
         return latest_state
@@ -187,3 +193,16 @@ class StateManager:
                     except ValueError as e:
                         print(f"Invalid part in artifact: {e}")
         return parts
+    
+    async def _process_file_parts(self, state: StateData):
+        """Replace file bytes with URIs and persist files"""
+        for msg in state.context_history:
+            for part in msg.parts:
+                if isinstance(part, FilePart) and part.file.bytes:
+                    uri = await self.file_store.upload(
+                        content=base64.b64decode(part.file.bytes),
+                        task_id=state.task_id,
+                        filename=part.file.name
+                    )
+                    part.file.uri = uri
+                    part.file.bytes = None  # Remove from state

{smarta2a-0.4.11.dist-info → smarta2a-0.4.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: smarta2a
-Version: 0.4.11
+Version: 0.4.12
 Summary: a Python framework that helps you build servers and AI agents that communicate using the A2A protocol
 Project-URL: Homepage, https://github.com/siddharthsma/smarta2a
 Project-URL: Bug Tracker, https://github.com/siddharthsma/smarta2a/issues

{smarta2a-0.4.11.dist-info → smarta2a-0.4.12.dist-info}/RECORD

@@ -1,20 +1,22 @@
 smarta2a/__init__.py,sha256=T_EECYqWrxshix0FbgUv22zlKRX22HFU-HKXcYTOb3w,175
 smarta2a/agent/a2a_agent.py,sha256=EurcxpV14e3OPWCMutYL0EXMHb5ZKQqAHEGZZF6pNgg,1892
-smarta2a/agent/a2a_human.py,sha256=tXrEtl0OqZEXZfmPTLnbbAiREMTmlqA42XIf7KocqMU,1850
+smarta2a/agent/a2a_human.py,sha256=EZP9TD54Gum509OMOGmfFeCeoMk2Lyi1Y24RoRWh4hY,1720
 smarta2a/agent/a2a_mcp_server.py,sha256=X_mxkgYgCA_dSNtCvs0rSlOoWYc-8d3Qyxv0e-a7NKY,1015
 smarta2a/archive/smart_mcp_client.py,sha256=0s2OWFKWSv-_UF7rb9fOrsh1OIYsYOsGukkXXp_E1cU,4158
 smarta2a/archive/subscription_service.py,sha256=vftmZD94HbdjPFa_1UBvsBm-WkW-s3ZCVq60fF7OCgA,4109
 smarta2a/archive/task_service.py,sha256=ptf-oMHy98Rw4XSxyK1-lpqc1JtkCkEEHTmwAaunet4,8199
 smarta2a/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-smarta2a/client/a2a_client.py,sha256=apDkKFtq61T79LpkbkzVTKWA0mSjR_eTNdGPUYozyvk,12100
+smarta2a/client/a2a_client.py,sha256=Tl9FLADlK5n_y_0C3qcFB25T6Uu02iXDSxopm40LbKs,12726
 smarta2a/client/mcp_client.py,sha256=JeXhBqxM9TYAArpExLRtEr3lZeQZMcnTmGFl6XKsdu8,3797
+smarta2a/file_stores/base_file_store.py,sha256=fcwFIOoFjLQiIKb8lIRVujnV6udyuI9Dq8cEc2ldmIQ,591
+smarta2a/file_stores/local_file_store.py,sha256=4GLDrsKxSoLWn2Oha4OD-P2r5vBpfV-8ePvZ5bhP1e8,2616
 smarta2a/history_update_strategies/__init__.py,sha256=x5WtiE9rG5ze8d8hA6E6wJOciBhWHa_ZgGgoIAZcXEQ,213
 smarta2a/history_update_strategies/append_strategy.py,sha256=j7Qbhs69Wwr-HBLB8GJ3-mEPaBSHiBV2xz9ZZi86k2w,312
 smarta2a/history_update_strategies/history_update_strategy.py,sha256=n2sfIGu8ztKI7gJTwRX26m4tZr28B8Xdhrk6RlBFlI8,373
 smarta2a/history_update_strategies/rolling_window_strategy.py,sha256=7Ch042JWt4TM_r1-sFKlSIxHj8VX1P3ZoqjCvIdeSqA,540
 smarta2a/model_providers/__init__.py,sha256=hJj0w00JjqTiBgJmHmOWwL6MU_hwmro9xTiX3XYf6ts,148
 smarta2a/model_providers/base_llm_provider.py,sha256=iQUqjnypl0f2M929iU0WhHoxAE4ek-NUFJPbEnNQ8-4,412
-smarta2a/model_providers/openai_provider.py,sha256=SJfEOktgbjyZ3FtE-Q_K7dYJJI0khrltj2kJoYwz0WQ,12317
+smarta2a/model_providers/openai_provider.py,sha256=dGVnQ94H6-pohUYwT_IzmUf0Nc7BDMJtI_dpjHLL2ak,12131
 smarta2a/server/__init__.py,sha256=f2X454Ll4vJc02V4JLJHTN-h8u0TBm4d_FkiO4t686U,53
 smarta2a/server/handler_registry.py,sha256=OVRG5dTvxB7qUNXgsqWxVNxIyRljUShSYxb1gtbi5XM,820
 smarta2a/server/json_rpc_request_processor.py,sha256=qRB3sfj_n9ImkIOCdaUKMsDmKcO7CiMhaZ4VdQS7Mb4,6993
@@ -22,7 +24,7 @@ smarta2a/server/nats_client.py,sha256=akyNg1hLd9XYoLSH_qQVs8uoiTQerztgvqu_3TifSg
 smarta2a/server/request_handler.py,sha256=5KMtfpHQX6bOgk1DJbhs1fUCQ5tSvMYXWzheT3IW2Bo,26374
 smarta2a/server/send_task_handler.py,sha256=fiBeCCHCu9c2H4EJOUc0t3EZgpHVFJy4B_6qZOC140s,6336
 smarta2a/server/server.py,sha256=RKkQM8jpSndt_nOuUB0kswOqLdm7JfvjZA1O424sYdY,6722
-smarta2a/server/state_manager.py,sha256=omjN3C9Sl4hqw5FepnX2qBm1do6QTSE5P9uip78dvEI,6807
+smarta2a/server/state_manager.py,sha256=JhF6jma8t2YtBzb3sADGvxtmUMOlDafqxllPzLb3DU0,7668
 smarta2a/server/webhook_request_processor.py,sha256=_0XoUDmueSl9CvFQE-1zgKRSts-EW8QxbmolPTfFER8,5306
 smarta2a/state_stores/__init__.py,sha256=vafxAqpwvag_cYFH2XKGk3DPmJIWJr4Ioey30yLFkVQ,220
 smarta2a/state_stores/base_state_store.py,sha256=_3LInM-qepKwwdypJTDNs9-DozBNrKVycwPwUm7bYdU,512
@@ -34,7 +36,7 @@ smarta2a/utils/task_builder.py,sha256=wqSyfVHNTaXuGESu09dhlaDi7D007gcN3-8tH-nPQ4
 smarta2a/utils/task_request_builder.py,sha256=6cOGOqj2Rg43xWM03GRJQzlIZHBptsMCJRp7oD-TDAQ,3362
 smarta2a/utils/tools_manager.py,sha256=igKYeSi0SaYzd36jUqOMPvnYd5kK55EPQ0X_pdTo5e4,4857
 smarta2a/utils/types.py,sha256=kzA6Vv5xXfu1sJuxhEXrglI9e9S6eZVIljMnsrQVyN0,13650
-smarta2a-0.4.11.dist-info/METADATA,sha256=qiNnMJUcRkkKVHJwXD9yT01MB9pVdvxtQeP_3K2i5Es,12988
-smarta2a-0.4.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-smarta2a-0.4.11.dist-info/licenses/LICENSE,sha256=lDbqrxVnzDMY5KJ8JS1WhvkWE8TJaw-O-CHDy-ecsJA,2095
-smarta2a-0.4.11.dist-info/RECORD,,
+smarta2a-0.4.12.dist-info/METADATA,sha256=_dkL7gOJcF84lsAYt8vnal-hetsh3gWC8J-CDCGLQww,12988
+smarta2a-0.4.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+smarta2a-0.4.12.dist-info/licenses/LICENSE,sha256=lDbqrxVnzDMY5KJ8JS1WhvkWE8TJaw-O-CHDy-ecsJA,2095
+smarta2a-0.4.12.dist-info/RECORD,,