Method get_variables

src/tool/context.py:1113–1159

Get variables from tools, where each tool's code is used as the variable value. Args: tool_name (Optional[str]): Name of a specific tool. If None, returns variables for all tools. Returns: Dict[str, Variable]: Dictionary mapping tool name

(self, tool_name: Optional[str] = None)

            logger.error(f"| ❌ Error during tool context manager cleanup: {e}")
            
    async def get_variables(self, tool_name: Optional[str] = None) -> Dict[str, 'Variable']:
        """Get variables from tools, where each tool's code is used as the variable value.
        
        Args:
            tool_name (Optional[str]): Name of a specific tool. If None, returns variables for all tools.
            
        Returns:
            Dict[str, Variable]: Dictionary mapping tool names to Variable objects. Each Variable has:
                - name: tool name
                - type: "tool_code"
                - description: tool description
                - require_grad: tool's require_grad value
                - variables: tool's code (as string value)
        """
        # Lazy import to avoid circular dependency
        from src.optimizer.types import Variable
        
        variables: Dict[str, Variable] = {}
        
        if tool_name is not None:
            # Get specific tool
            tool_config = await self.get_info(tool_name)
            if tool_config is None:
                logger.warning(f"| ⚠️ Tool {tool_name} not found")
                return variables
            
            tool_configs = {tool_name: tool_config}
        else:
            # Get all tools
            tool_configs = self._tool_configs
        
        for name, tool_config in tool_configs.items():
            # Get tool code
            tool_code = tool_config.code or ""
            
            # Create Variable for this tool
            variable = Variable(
                name=name,
                type="tool_code",
                description=tool_config.description or f"Code for tool {name}",
                require_grad=tool_config.require_grad,
                template=None,
                variables=tool_code  # Store code as the variable value
            )
            variables[name] = variable
        
        return variables
    
    async def get_trainable_variables(self, tool_name: Optional[str] = None) -> Dict[str, 'Variable']:
        """Get trainable variables from tools, filtering out tools with require_grad=False.