smarta2a 0.4.17__py3-none-any.whl → 0.4.19__py3-none-any.whl

smarta2a/client/a2a_client.py

@@ -54,15 +54,6 @@ 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,
@@ -93,15 +84,6 @@ 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,
@@ -207,9 +189,6 @@ class A2AClient:
                     webhook_url, json=request.model_dump(), timeout=30
                 )
                 response.raise_for_status()
-                print("--- response from webhook ---")
-                print(response.json())
-                print("--- end of response from webhook ---")
                 return response.json()
             except httpx.HTTPStatusError as e:
                 raise A2AClientHTTPError(e.response.status_code, str(e)) from e

smarta2a/model_providers/openai_provider.py

@@ -193,10 +193,12 @@ class OpenAIProvider(BaseLLMProvider):
 
             # Call the actual tool
             try:
+                
                 override_args = {
                     'id': state.task_id,
                     'sessionId': state.task.sessionId,
-                    'metadata': state.task.metadata
+                    'metadata': state.task.metadata,
+                    'push_notification': state.push_notification_config
                 }
                 
                 tool_result = await self.tools_manager.call_tool(fn_name, fn_args, override_args)

smarta2a/server/nats_client.py

@@ -19,10 +19,8 @@ class NATSClient:
                 # Use the current running loop by default
                 await self.nats.connect(self.server_url)
                 self._connected = True
-                print(f"Connected to NATS at {self.server_url}")
             except Exception as e:
-                print(f"Failed to connect to NATS: {e}")
-                raise
+                pass
 
     async def publish(self, subject: str, payload: Dict[str, Any]) -> None:
         """Publishes a JSON-encoded message to a NATS subject, auto-connecting if needed."""
@@ -42,7 +40,6 @@ class NATSClient:
         if self._connected:
             await self.nats.close()
             self._connected = False
-            print("NATS connection closed")
 
     @property
     def is_connected(self) -> bool:

smarta2a/server/request_handler.py

@@ -157,13 +157,9 @@ class RequestHandler:
             # If push_notification_config is set send the task to the push notification url
             if push_notification_config and forward_to_webhook:
                 try:
-                    print("call to send_to_webhook")
-                    print(push_notification_config.url)
-                    print(task.model_dump())
-                    print("--- end ---")
                     self.a2a_aclient.send_to_webhook(webhook_url=push_notification_config.url,id=task_id,task=task.model_dump())
                 except Exception as e:
-                    print(f"Error sending task to webhook: {e}")
+                    pass
             
 
             # Send the task back to the client
@@ -296,13 +292,12 @@ class RequestHandler:
                                 context_stream_history = new_context_history
                             stream_metadata = new_metadata
 
-                            # TODO: If push_notification_config is set, send the task to the push notification url 
                             # If push_notification_config is set send the task to the push notification url
                             if push_notification_config and forward_to_webhook:
                                 try:
                                     self.a2a_aclient.send_to_webhook(webhook_url=push_notification_config.url,id=task_id,task=task)
                                 except Exception as e:
-                                    print(f"Error sending task to webhook: {e}")
+                                    pass
 
                             
                         elif isinstance(item, TaskStatusUpdateEvent):

smarta2a/server/server.py

@@ -37,7 +37,6 @@ class SmartA2A:
                 name: str,
                 agent_card: Optional[AgentCard] = None,
                 state_manager: Optional[StateManager] = None,
-                has_frontend: bool = False,
                 **fastapi_kwargs
                 ):
         self.name = name
@@ -46,7 +45,6 @@ class SmartA2A:
         self.state_mgr = state_manager
         self.app = FastAPI(title=name, **fastapi_kwargs)
         self.router = APIRouter()
-        self.has_frontend = has_frontend
         self._setup_cors()
         self._setup_routes()
         self.server_config = {

smarta2a/server/state_manager.py

@@ -10,11 +10,12 @@ from smarta2a.utils.types import Message, StateData, Task, TaskStatus, TaskState
 from smarta2a.server.nats_client import NATSClient
 
 class StateManager:
-    def __init__(self, state_store: BaseStateStore, file_store: BaseFileStore, 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", push_notification_config: Optional[PushNotificationConfig] = None):
         self.state_store = state_store
         self.file_store = file_store
         self.strategy = history_strategy
         self.nats_client = NATSClient(server_url=nats_server_url)
+        self.push_notification_config = push_notification_config
     
     async def load(self):
         await self.nats_client.connect()
@@ -40,11 +41,15 @@ class StateManager:
             history=[],
             metadata={}
         )
+
+        # Use self.push_notification_config if push_notification_config is None
+        notification_config = push_notification_config if push_notification_config is not None else self.push_notification_config
+
         state = StateData(
             task_id=task_id,
             task=initial_task,
             context_history=[],
-            push_notification_config=push_notification_config
+            push_notification_config=notification_config
         )
         self.state_store.initialize_state(state)
         return state
@@ -82,7 +87,7 @@ class StateManager:
 
         return latest_state
     
-    def get_and_update_state_from_webhook(self, task_id: str, result: Task) -> StateData:
+    async def get_and_update_state_from_webhook(self, task_id: str, result: Task) -> StateData:
         """
         Update existing state with webhook result data, including:
         - Merges task history from result
@@ -141,6 +146,11 @@ class StateManager:
         # Update task status if provided
         if result.status:
             updated_state.task.status = result.status
+        
+         # Process files before persistence
+        await self._process_file_parts(updated_state)
+
+        await self.update_state(updated_state)
 
         return updated_state
     
@@ -152,9 +162,6 @@ class StateManager:
 
         # Publish update through NATS client
         payload = self._prepare_update_payload(state_data)
-        print("--- NATS payload ---")
-        print(payload)
-        print("--- NATS payload end ---")
         await self.nats_client.publish("state.updates", payload)
     
     def get_store(self) -> Optional[BaseStateStore]:

smarta2a/server/webhook_request_processor.py

@@ -15,9 +15,6 @@ class WebhookRequestProcessor:
     async def process_request(self, request: WebhookRequest) -> WebhookResponse:
         if self.state_manager:
             state_data = self.state_manager.get_and_update_state_from_webhook(request.id, request.result)
-            print("--- state_data in webhook_request_processor.py ---")
-            print(state_data)
-            print("--- end of state_data in webhook_request_processor.py ---")
             return await self._handle_webhook(request, state_data)
         else:
             return await self._handle_webhook(request)
@@ -37,29 +34,6 @@ class WebhookRequestProcessor:
                         context_history=[],
                         push_notification_config=None
                     )
-                else:
-                    existing_task = state_data.task
-                    
-                    # Overwrite artifacts
-                    existing_task.artifacts = incoming_task.artifacts.copy() if incoming_task.artifacts else []
-                    
-                    # Merge metadata
-                    existing_task.metadata = {**(existing_task.metadata or {}), **(incoming_task.metadata or {})}
-                    
-                    # Build messages from artifact parts (role="agent" as in handle_send_task)
-                    all_parts = [part for artifact in incoming_task.artifacts for part in artifact.parts] if incoming_task.artifacts else []
-                    new_messages = [Message(role="agent", parts=all_parts, metadata=incoming_task.metadata)]
-                    
-                    # Update context history using strategy
-                    history_strategy = self.state_manager.get_history_strategy()
-                    state_data.context_history = history_strategy.update_history(
-                        existing_history=state_data.context_history,
-                        new_messages=new_messages
-                    )
-                    
-                    # Persist state
-                    print("call to update state - before webhook function")
-                    await self.state_manager.update_state(state_data)
 
             # --- Step 2: Call Webhook Function ---
             webhook_response = await self.webhook_fn(request, state_data) if state_data else await self.webhook_fn(request)
@@ -80,20 +54,17 @@ class WebhookRequestProcessor:
                 updated_messages = [Message(role="agent", parts=updated_parts, metadata=updated_task.metadata)]
                 
                 # Update context history again
-                state_data.context_history = history_strategy.update_history(
-                    existing_history=state_data.context_history,
-                    new_messages=updated_messages
-                )
-                
-                print("call to update state - after webhook function")
-                await self.state_manager.update_state(state_data)
+                if self.state_manager:
+                    history_strategy = self.state_manager.get_history_strategy()
+                    state_data.context_history = history_strategy.update_history(
+                        existing_history=state_data.context_history,
+                        new_messages=updated_messages
+                    )
+                    await self.state_manager.update_state(state_data)
 
             # --- Step 4: Push Notification ---
-            push_url = (
-                state_data.push_notification_config.url 
-                if state_data and state_data.push_notification_config 
-                else None
-            )
+            push_url = state_data.push_notification_config.url if state_data and state_data.push_notification_config else None
+            
             if push_url:
                 try:
                     self.a2a_aclient.send_to_webhook(

smarta2a/utils/tools_manager.py

@@ -91,25 +91,12 @@ class ToolsManager:
             client = self.get_client(tool_key)
             tool_name = self._get_tool_name(tool_key)
             new_args = self._replace_with_override_args(args, override_args)
-            print("--- new_args ---")
-            print(new_args)
-            print("--- end of new_args ---")
-            print("--- tool_name ---")
-            print(tool_name)
-            print("--- end of tool_name ---")
             result = await client.call_tool(tool_name, new_args)
-            print("--- result ---")
-            print(result)
-            print("--- end of result ---")
+            
             return result
 
         except Exception as e:
             # This will catch ANY error in the body above
-            print("--- EXCEPTION CAUGHT ---")
-            print(f"Error Type: {type(e).__name__}")
-            print(f"Error Message: {str(e)}")
-            print("--- END OF EXCEPTION DETAILS ---")
-        
             raise
     
     def _get_tool_name(self, tool_key: str) -> str:

{smarta2a-0.4.17.dist-info → smarta2a-0.4.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: smarta2a
-Version: 0.4.17
+Version: 0.4.19
 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.17.dist-info → smarta2a-0.4.19.dist-info}/RECORD

@@ -6,7 +6,7 @@ smarta2a/archive/smart_mcp_client.py,sha256=0s2OWFKWSv-_UF7rb9fOrsh1OIYsYOsGukkX
 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=yxxsIcInVwU-DMCcWGsrLzaBeEicbpIPUOoKAEo6PqQ,12874
+smarta2a/client/a2a_client.py,sha256=apDkKFtq61T79LpkbkzVTKWA0mSjR_eTNdGPUYozyvk,12100
 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
@@ -16,16 +16,16 @@ smarta2a/history_update_strategies/history_update_strategy.py,sha256=n2sfIGu8ztK
 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=dGVnQ94H6-pohUYwT_IzmUf0Nc7BDMJtI_dpjHLL2ak,12131
+smarta2a/model_providers/openai_provider.py,sha256=cqAAhoImHJtBVOGeIERphXqiNlq6MSe_p3-3gImhAeM,12221
 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
-smarta2a/server/nats_client.py,sha256=akyNg1hLd9XYoLSH_qQVs8uoiTQerztgvqu_3TifSgE,1617
-smarta2a/server/request_handler.py,sha256=QPVWWEKAzMAf4TiwXSEuPBiKGn8_f4ahTRs0mo3qjEE,26569
+smarta2a/server/nats_client.py,sha256=K97e8awvAxsqgs3BxsTZU_mB3JPPZ0yjLX1LnZZjfXE,1450
+smarta2a/server/request_handler.py,sha256=IuKY9L3WYt-G5u55Qz2eq2OORb2bSb140HGIGGvHxQI,26178
 smarta2a/server/send_task_handler.py,sha256=fiBeCCHCu9c2H4EJOUc0t3EZgpHVFJy4B_6qZOC140s,6336
-smarta2a/server/server.py,sha256=Xdj3WN-Zgxf9RQDoN8_qGMXJgkADb8Dzk3QwgjDDKb4,6882
-smarta2a/server/state_manager.py,sha256=mFh0r002sUAzgASkQKEv3ffAAiMtI8kxHxtfGWwVxwA,7666
-smarta2a/server/webhook_request_processor.py,sha256=LAVA4FiICci-6R2htxRf4bSnKFYoNoaQ3zKI83380ZE,5634
+smarta2a/server/server.py,sha256=E58tvOjAvHf_gYjST76sPEC5piaNyrHePa9k-fSadZA,6797
+smarta2a/server/state_manager.py,sha256=jFxf9cDxLqjIo87tv_Ynh2dH1vCTFLrVSLPuVg6YWnA,8063
+smarta2a/server/webhook_request_processor.py,sha256=eD8APJ6tgb72TMxXCviQG4p77mfSPK6Iob7wiV7JpQU,4077
 smarta2a/state_stores/__init__.py,sha256=vafxAqpwvag_cYFH2XKGk3DPmJIWJr4Ioey30yLFkVQ,220
 smarta2a/state_stores/base_state_store.py,sha256=_3LInM-qepKwwdypJTDNs9-DozBNrKVycwPwUm7bYdU,512
 smarta2a/state_stores/inmemory_state_store.py,sha256=nEBBUiiqhEluP2MYJjFUImcjIwLJEvL8BWwMbLCb8Fw,1268
@@ -34,9 +34,9 @@ smarta2a/utils/agent_discovery_manager.py,sha256=6KpRSQH_EDUOZbF4wFRsZneZGIPLXFP
 smarta2a/utils/prompt_helpers.py,sha256=M3UUjFQEspEAnNm54Dip0-D7mMFFZLrP_s_89ZPe6fs,1438
 smarta2a/utils/task_builder.py,sha256=wqSyfVHNTaXuGESu09dhlaDi7D007gcN3-8tH-nPQ40,5159
 smarta2a/utils/task_request_builder.py,sha256=6cOGOqj2Rg43xWM03GRJQzlIZHBptsMCJRp7oD-TDAQ,3362
-smarta2a/utils/tools_manager.py,sha256=igKYeSi0SaYzd36jUqOMPvnYd5kK55EPQ0X_pdTo5e4,4857
+smarta2a/utils/tools_manager.py,sha256=oR5cbwzPZ36hQAsWAgb-c6wFv5BthmCPraD7DSv-Bv8,4332
 smarta2a/utils/types.py,sha256=kzA6Vv5xXfu1sJuxhEXrglI9e9S6eZVIljMnsrQVyN0,13650
-smarta2a-0.4.17.dist-info/METADATA,sha256=eUfquvHQDEf_zqNn-Yw7vQqCUtoDWrs50P7mwHYscGA,13051
-smarta2a-0.4.17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-smarta2a-0.4.17.dist-info/licenses/LICENSE,sha256=lDbqrxVnzDMY5KJ8JS1WhvkWE8TJaw-O-CHDy-ecsJA,2095
-smarta2a-0.4.17.dist-info/RECORD,,
+smarta2a-0.4.19.dist-info/METADATA,sha256=JZr1tVu5pJsSzm6nUoQvTTU6IwN6ydTSZlVov-Mz38o,13051
+smarta2a-0.4.19.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+smarta2a-0.4.19.dist-info/licenses/LICENSE,sha256=lDbqrxVnzDMY5KJ8JS1WhvkWE8TJaw-O-CHDy-ecsJA,2095
+smarta2a-0.4.19.dist-info/RECORD,,