webagents 0.2.2__py3-none-any.whl → 0.2.3__py3-none-any.whl

webagents/agents/core/handoffs.py

@@ -114,14 +114,18 @@ class LocalAgentHandoff:
             if preserve_context:
                 target_context = create_context(
                     request_id=current_context.request_id,
-                    peer_user_id=current_context.peer_user_id,
-                    payment_user_id=current_context.payment_user_id,
-                    origin_user_id=current_context.origin_user_id,
-                    agent_owner_user_id=current_context.agent_owner_user_id,
                     messages=current_context.messages.copy(),
-                    stream=current_context.stream
+                    stream=current_context.stream,
+                    agent=target
                 )
                 
+                # Copy auth context if present
+                if current_context.auth:
+                    target_context.auth = current_context.auth
+                
+                # Copy usage data
+                target_context.usage = current_context.usage.copy()
+                
                 # Add handoff context data
                 target_context.set("handoff_context", context_data)
                 target_context.update_agent_context(target, target_agent)
@@ -135,6 +139,10 @@ class LocalAgentHandoff:
                 current_context.set("handoff_data", context_data)
                 await source._execute_hooks("before_handoff", current_context)
             
+            # Ensure handoff_data is a dict
+            if handoff_data is None:
+                handoff_data = {}
+            
             # Create handoff message for target agent
             handoff_message = {
                 "role": "system",
@@ -143,7 +151,7 @@ class LocalAgentHandoff:
             
             # Add handoff context to messages
             messages_with_handoff = current_context.messages + [handoff_message]
-            if handoff_data and handoff_data.get('user_message'):
+            if handoff_data.get('user_message'):
                 messages_with_handoff.append({
                     "role": "user", 
                     "content": handoff_data['user_message']

webagents/agents/skills/base.py

@@ -106,14 +106,45 @@ class Skill(ABC):
     def get_dependencies(self) -> List[str]:
         """Return list of skill dependencies"""
         return self.dependencies.copy()
+    
+    def request_handoff(self, target_name: str) -> str:
+        """Return a handoff request marker for the given target
+        
+        This is a helper for skills that want to trigger dynamic handoff.
+        Return this value from a tool, and the framework will execute the handoff.
+        
+        Args:
+            target_name: Target name of the handoff to switch to
+        
+        Returns:
+            Handoff request marker string
+        
+        Example:
+            @tool
+            async def use_specialist(self):
+                return self.request_handoff("specialist_agent")
+        """
+        return f"__HANDOFF_REQUEST__:{target_name}"
 
 
 # Base dataclasses for handoffs and other components
 @dataclass
 class Handoff:
-    """Configuration for handoff operations"""
+    """Configuration for handoff operations
+    
+    Handoffs are chat completion handlers that can be local LLMs or remote agents.
+    The first registered handoff (lowest priority) becomes the default completion handler.
+    
+    Attributes:
+        target: Unique identifier for this handoff
+        description: Human-readable description (from @handoff prompt parameter)
+        scope: Access control scope(s)
+        metadata: Additional data including:
+            - function: The actual handoff function
+            - priority: Execution priority (lower = higher priority)
+            - is_generator: Whether function is async generator (streaming)
+    """
     target: str
-    handoff_type: str
     description: str = ""
     scope: Union[str, List[str]] = "all"
     metadata: Dict[str, Any] = None