signalwire-agents 0.1.23__py3-none-any.whl → 0.1.24__py3-none-any.whl

signalwire_agents/cli/types.py

@@ -0,0 +1,72 @@
+#!/usr/bin/env python3
+"""
+Type definitions for the CLI tools
+"""
+
+from typing import TypedDict, Dict, Any, Optional, List, Union
+
+
+class CallData(TypedDict, total=False):
+    """Call data structure for SWML post_data"""
+    id: str
+    node_id: str
+    state: str
+    type: str
+    direction: str
+    project_id: str
+    space_id: str
+    from_number: str
+    to_number: str
+    from_: str
+    to: str
+    from_name: str
+    headers: Dict[str, str]
+    timeout: int
+    tag: str
+
+
+class VarsData(TypedDict, total=False):
+    """Variables data structure for SWML post_data"""
+    userVariables: Dict[str, Any]
+    environment: str
+    call_data: Dict[str, Any]
+
+
+class PostData(TypedDict, total=False):
+    """Complete post_data structure for SWML requests"""
+    call_id: str
+    call: CallData
+    vars: VarsData
+    params: Dict[str, Any]
+    project_id: str
+    space_id: str
+    meta_data: Dict[str, Any]
+    post_prompt_data: Dict[str, Any]
+    error: Optional[str]
+    protocol_error: Optional[bool]
+    parse_error: Optional[bool]
+
+
+class DataMapConfig(TypedDict, total=False):
+    """DataMap function configuration"""
+    function: str
+    data_map: Dict[str, Any]
+    description: str
+    parameters: Dict[str, Any]
+
+
+class AgentInfo(TypedDict):
+    """Information about a discovered agent"""
+    class_name: str
+    file_path: str
+    is_instance: bool
+    instance_name: Optional[str]
+
+
+class FunctionInfo(TypedDict):
+    """Information about a SWAIG function"""
+    name: str
+    description: str
+    parameters: Dict[str, Any]
+    type: str  # 'local', 'external', 'datamap'
+    webhook_url: Optional[str]

signalwire_agents/core/agent/__init__.py

@@ -9,6 +9,4 @@ See LICENSE file in the project root for full license information.
 
 """Agent refactored components."""
 
-from .config.ephemeral import EphemeralAgentConfig
-
-__all__ = ['EphemeralAgentConfig']
+__all__ = []

signalwire_agents/core/agent/config/__init__.py

@@ -9,6 +9,4 @@ See LICENSE file in the project root for full license information.
 
 """Configuration management modules."""
 
-from .ephemeral import EphemeralAgentConfig
-
-__all__ = ['EphemeralAgentConfig']
+__all__ = []

signalwire_agents/core/agent/prompt/manager.py

@@ -192,12 +192,29 @@ class PromptManager:
             bullets: Optional list of bullet points to add
         """
         if self.agent._use_pom and self.agent.pom:
-            self.agent.pom.add_to_section(
-                title=title,
-                body=body,
-                bullet=bullet,
-                bullets=bullets
-            )
+            # Find the section first
+            section = self.agent.pom.find_section(title)
+            
+            if section is None:
+                # Section doesn't exist, create it
+                section = self.agent.pom.add_section(title=title)
+            
+            # Add content to the section
+            if body:
+                if section.body:
+                    section.body = f"{section.body}\n\n{body}"
+                else:
+                    section.body = body
+            
+            # Process bullets
+            bullets_to_add = []
+            if bullet:
+                bullets_to_add.append(bullet)
+            if bullets:
+                bullets_to_add.extend(bullets)
+            
+            if bullets_to_add:
+                section.add_bullets(bullets_to_add)
     
     def prompt_add_subsection(
         self,
@@ -248,7 +265,8 @@ class PromptManager:
             True if section exists, False otherwise
         """
         if self.agent._use_pom and self.agent.pom:
-            return self.agent.pom.has_section(title)
+            # Use find_section method from POM
+            return self.agent.pom.find_section(title) is not None
         return False
     
     def get_prompt(self) -> Optional[Union[str, List[Dict[str, Any]]]]:

signalwire_agents/core/agent/tools/decorator.py

@@ -50,6 +50,7 @@ class ToolDecorator:
                 secure = kwargs.pop("secure", True)
                 fillers = kwargs.pop("fillers", None)
                 webhook_url = kwargs.pop("webhook_url", None)
+                required = kwargs.pop("required", None)
                 
                 registry.define_tool(
                     name=name,
@@ -59,6 +60,7 @@ class ToolDecorator:
                     secure=secure,
                     fillers=fillers,
                     webhook_url=webhook_url,
+                    required=required,
                     **kwargs  # Pass through any additional swaig_fields
                 )
                 return func

signalwire_agents/core/agent/tools/registry.py

@@ -41,6 +41,7 @@ class ToolRegistry:
         secure: bool = True,
         fillers: Optional[Dict[str, List[str]]] = None,
         webhook_url: Optional[str] = None,
+        required: Optional[List[str]] = None,
         **swaig_fields
     ) -> None:
         """
@@ -54,6 +55,7 @@ class ToolRegistry:
             secure: Whether to require token validation
             fillers: Optional dict mapping language codes to arrays of filler phrases
             webhook_url: Optional external webhook URL to use instead of local handling
+            required: Optional list of required parameter names
             **swaig_fields: Additional SWAIG fields to include in function definition
             
         Raises:
@@ -70,6 +72,7 @@ class ToolRegistry:
             secure=secure,
             fillers=fillers,
             webhook_url=webhook_url,
+            required=required,
             **swaig_fields
         )
         
@@ -96,6 +99,9 @@ class ToolRegistry:
         # These don't have handlers since they execute on SignalWire's server
         self._swaig_functions[function_name] = function_dict
         
+        # Debug logging using the module logger with proper format
+        logger.debug(f"Registered SWAIG function in registry: {function_name} (registry_id={id(self)}, agent_id={id(self.agent) if hasattr(self, 'agent') else None}, total_functions={len(self._swaig_functions)})")
+        
         logger.debug(f"Registered SWAIG function: {function_name}")
     
     def register_class_decorated_tools(self) -> None:
@@ -127,6 +133,7 @@ class ToolRegistry:
                     secure = tool_params_copy.pop("secure", True)
                     fillers = tool_params_copy.pop("fillers", None)
                     webhook_url = tool_params_copy.pop("webhook_url", None)
+                    required = tool_params_copy.pop("required", None)
                     
                     # Register the tool with any remaining params as swaig_fields
                     self.define_tool(
@@ -137,6 +144,7 @@ class ToolRegistry:
                         secure=secure,
                         fillers=fillers,
                         webhook_url=webhook_url,
+                        required=required,
                         **tool_params_copy  # Pass through any additional swaig_fields
                     )