autobyteus 1.1.9__py3-none-any.whl → 1.2.1__py3-none-any.whl

autobyteus/tools/tool_meta.py

@@ -35,26 +35,8 @@ class ToolMeta(ABCMeta):
                  logger.error(f"Tool class {name} ({tool_name}) must return a valid string from get_description(). Skipping registration.")
                  return
 
-            argument_schema: ParameterSchema = None 
-            try:
-                argument_schema = cls.get_argument_schema() 
-                if argument_schema is not None and not isinstance(argument_schema, ParameterSchema): 
-                    logger.error(f"Tool class {name} ({tool_name}) get_argument_schema() must return a ParameterSchema or None. Got {type(argument_schema)}. Skipping registration.")
-                    return
-            except Exception as e:
-                logger.error(f"Tool class {name} ({tool_name}) failed to provide argument_schema via get_argument_schema(): {e}. Skipping registration.", exc_info=True)
-                return
+            # Note: We do not call the schema methods here. We pass them as providers.
             
-            instantiation_config_schema: ParameterSchema = None 
-            if hasattr(cls, 'get_config_schema'):
-                try:
-                    instantiation_config_schema = cls.get_config_schema()
-                    if instantiation_config_schema is not None and not isinstance(instantiation_config_schema, ParameterSchema): 
-                        logger.warning(f"Tool class {name} ({tool_name}) get_config_schema() returned non-ParameterSchema type: {type(instantiation_config_schema)}. Treating as no config schema.")
-                        instantiation_config_schema = None
-                except Exception as e:
-                    logger.warning(f"Tool class {name} ({tool_name}) has get_config_schema() but it failed: {e}. Assuming no instantiation config.")
-
             # Get category from class attribute, defaulting to "General"
             category_str = getattr(cls, 'CATEGORY', ToolCategory.GENERAL)
             
@@ -64,16 +46,14 @@ class ToolMeta(ABCMeta):
                 description=general_description, 
                 tool_class=cls,
                 custom_factory=None,
-                argument_schema=argument_schema,
-                config_schema=instantiation_config_schema,
+                argument_schema_provider=cls.get_argument_schema,
+                config_schema_provider=cls.get_config_schema,
                 origin=ToolOrigin.LOCAL,
                 category=category_str
             )
             default_tool_registry.register_tool(definition)
             
-            arg_schema_info = f"args: {len(argument_schema) if argument_schema else '0'}"
-            config_info = f"inst_config: {len(instantiation_config_schema) if instantiation_config_schema else '0'}"
-            logger.info(f"Auto-registered tool: '{tool_name}' from class {name} ({arg_schema_info}, {config_info})")
+            logger.info(f"Auto-registered tool: '{tool_name}' from class {name}")
 
         except AttributeError as e:
              logger.error(f"Tool class {name} is missing a required method ({e}). Skipping registration.")

autobyteus/tools/usage/parsers/_string_decoders.py

@@ -0,0 +1,18 @@
+# file: autobyteus/autobyteus/tools/usage/parsers/_string_decoders.py
+"""Utility helpers for normalizing string content inside parsed tool payloads."""
+
+from __future__ import annotations
+
+import html
+from typing import Any
+
+
+def decode_html_entities(data: Any) -> Any:
+    """Recursively decode HTML/XML entities in strings within a data structure."""
+    if isinstance(data, dict):
+        return {key: decode_html_entities(value) for key, value in data.items()}
+    if isinstance(data, list):
+        return [decode_html_entities(item) for item in data]
+    if isinstance(data, str):
+        return html.unescape(data)
+    return data

autobyteus/tools/usage/parsers/default_json_tool_usage_parser.py

@@ -3,6 +3,8 @@ import json
 import logging
 from typing import Dict, Any, TYPE_CHECKING, List
 
+from ._string_decoders import decode_html_entities
+
 from autobyteus.agent.tool_invocation import ToolInvocation
 from .base_parser import BaseToolUsageParser
 from .exceptions import ToolUsageParseException
@@ -53,9 +55,15 @@ class DefaultJsonToolUsageParser(BaseToolUsageParser):
                         logger.debug(f"Skipping tool block with invalid 'parameters' type ({type(arguments)}): {tool_block}")
                         continue
                     
+                    decoded_arguments = decode_html_entities(arguments)
+                    decoded_tool_name = decode_html_entities(tool_name)
                     try:
                         # Pass id=None to trigger deterministic ID generation.
-                        tool_invocation = ToolInvocation(name=tool_name, arguments=arguments, id=None)
+                        tool_invocation = ToolInvocation(
+                            name=decoded_tool_name,
+                            arguments=decoded_arguments,
+                            id=None,
+                        )
                         invocations.append(tool_invocation)
                         logger.info(f"Successfully parsed default JSON tool invocation for '{tool_name}'.")
                     except Exception as e:

autobyteus/tools/usage/parsers/default_xml_tool_usage_parser.py

@@ -1,5 +1,6 @@
 import logging
 import re
+import html
 from typing import TYPE_CHECKING, Dict, Any, List
 from dataclasses import dataclass, field
 
@@ -124,6 +125,7 @@ class _XmlArgumentsParser:
         
         final_args = context.stack[0]
         self._cleanup_internal_keys(final_args)
+        final_args = self._decode_entities_inplace(final_args)
         return final_args
 
     def process_tag(self, tag_content: str, context: '_ParsingContext'):
@@ -213,6 +215,19 @@ class _XmlArgumentsParser:
             for item in data:
                 self._cleanup_internal_keys(item)
 
+    def _decode_entities_inplace(self, data: Any):
+        if isinstance(data, dict):
+            for key, value in list(data.items()):
+                data[key] = self._decode_entities_inplace(value)
+            return data
+        if isinstance(data, list):
+            for index, item in enumerate(data):
+                data[index] = self._decode_entities_inplace(item)
+            return data
+        if isinstance(data, str):
+            return html.unescape(data)
+        return data
+
 
 # --- Main Parser Class ---
 
@@ -299,4 +314,3 @@ class DefaultXmlToolUsageParser(BaseToolUsageParser):
         """
         parser = _XmlArgumentsParser(xml_string)
         return parser.parse()
-

autobyteus/tools/usage/parsers/gemini_json_tool_usage_parser.py

@@ -7,6 +7,7 @@ from autobyteus.agent.tool_invocation import ToolInvocation
 from .base_parser import BaseToolUsageParser
 from .exceptions import ToolUsageParseException
 from ._json_extractor import _find_json_blobs
+from ._string_decoders import decode_html_entities
 
 if TYPE_CHECKING:
     from autobyteus.llm.utils.response_types import CompleteResponse
@@ -55,8 +56,10 @@ class GeminiJsonToolUsageParser(BaseToolUsageParser):
                     arguments = call_data.get("args")
 
                     if tool_name and isinstance(tool_name, str) and isinstance(arguments, dict):
+                        decoded_tool_name = decode_html_entities(tool_name)
+                        decoded_arguments = decode_html_entities(arguments)
                         # Pass id=None to trigger deterministic ID generation in ToolInvocation
-                        tool_invocation = ToolInvocation(name=tool_name, arguments=arguments)
+                        tool_invocation = ToolInvocation(name=decoded_tool_name, arguments=decoded_arguments)
                         invocations.append(tool_invocation)
                         logger.info(f"Successfully parsed Gemini JSON tool invocation for '{tool_name}'.")
                     else:

autobyteus/tools/usage/parsers/openai_json_tool_usage_parser.py

@@ -7,6 +7,7 @@ from autobyteus.agent.tool_invocation import ToolInvocation
 from .base_parser import BaseToolUsageParser
 from .exceptions import ToolUsageParseException
 from ._json_extractor import _find_json_blobs
+from ._string_decoders import decode_html_entities
 
 if TYPE_CHECKING:
     from autobyteus.llm.utils.response_types import CompleteResponse
@@ -138,7 +139,9 @@ class OpenAiJsonToolUsageParser(BaseToolUsageParser):
                 
         try:
             # The ToolInvocation constructor will generate a deterministic ID if 'id' is None.
-            tool_invocation = ToolInvocation(name=tool_name, arguments=arguments, id=None)
+            decoded_tool_name = decode_html_entities(tool_name)
+            decoded_arguments = decode_html_entities(arguments)
+            tool_invocation = ToolInvocation(name=decoded_tool_name, arguments=decoded_arguments, id=None)
             logger.info(f"Successfully parsed OpenAI-style JSON tool invocation for '{tool_name}'.")
             return tool_invocation
         except Exception as e:

autobyteus/workflow/bootstrap_steps/coordinator_prompt_preparation_step.py

@@ -91,7 +91,7 @@ class CoordinatorPromptPreparationStep(BaseWorkflowBootstrapStep):
 
             prompt_parts.append(tools_section)
                 
-            final_instruction = "### Your Task\nAnalyze the user's request, formulate a plan, and use the `SendMessageTo` tool to delegate tasks to your team. Address team members by their unique ID as listed under 'Your Team'."
+            final_instruction = "### Your Task\nAnalyze the user's request, formulate a plan, and use the `send_message_to` tool to delegate tasks to your team. Address team members by their unique ID as listed under 'Your Team'."
             prompt_parts.append(final_instruction)
         else:
             role_and_goal = (
@@ -105,4 +105,3 @@ class CoordinatorPromptPreparationStep(BaseWorkflowBootstrapStep):
             prompt_parts.append(final_instruction)
 
         return "\n\n".join(prompt_parts)
-

{autobyteus-1.1.9.dist-info → autobyteus-1.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: autobyteus
-Version: 1.1.9
+Version: 1.2.1
 Summary: Multi-Agent framework
 Home-page: https://github.com/AutoByteus/autobyteus
 Author: Ryan Zheng
@@ -16,14 +16,15 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: aiohttp
 Requires-Dist: anthropic
-Requires-Dist: autobyteus-llm-client==1.1.3
 Requires-Dist: beautifulsoup4
 Requires-Dist: boto3
 Requires-Dist: botocore
 Requires-Dist: brui-core==1.0.9
 Requires-Dist: certifi==2025.4.26
+Requires-Dist: cryptography
 Requires-Dist: google-api-python-client
 Requires-Dist: google-genai==1.38.0
+Requires-Dist: httpx
 Requires-Dist: Jinja2
 Requires-Dist: mcp[cli]
 Requires-Dist: mistral_common
@@ -98,11 +99,11 @@ Launch and monitor your agent teams with our built-in Textual-based TUI.
 -   **Hierarchical View**: See the structure of your team, including sub-teams and their agents.
 -   **Real-Time Status**: Agent and team statuses are updated live, showing you who is idle, thinking, or executing a tool.
 -   **Detailed Logs**: Select any agent to view a detailed, streaming log of their thoughts, actions, and tool interactions.
--   **Live Task Board**: Watch your team's `TaskBoard` update in real-time as the coordinator publishes a plan and agents complete their tasks.
+-   **Live Task Plan**: Watch your team's `TaskPlan` update in real-time as the coordinator publishes a plan and agents complete their tasks.
 
-| TUI - Detailed Agent Log | TUI - Task Board with Completed Task |
+| TUI - Detailed Agent Log | TUI - Task Plan with Completed Task |
 | :---: | :---: |
-| ![Autobyteus Agent Log](docs/images/image_4.png) | ![Autobyteus Task Board](docs/images/image_3.png) |
+| ![Autobyteus Agent Log](docs/images/image_4.png) | ![Autobyteus Task Plan](docs/images/image_3.png) |
 
 #### 🏗️ Fluent Team Building
 Define complex agent and team structures with an intuitive, fluent API. The `AgentTeamBuilder` makes composing your team simple and readable.
@@ -129,7 +130,7 @@ Autobyteus intelligently handles tool communication with LLMs while giving you f
 #### 📈 Flexible Communication Protocols
 Choose the collaboration pattern that best fits your use case with configurable `TaskNotificationMode`s.
 -   **`AGENT_MANUAL_NOTIFICATION` (Default)**: A traditional approach where a coordinator agent is responsible for creating a plan and then explicitly notifying other agents to begin their work via messages.
--   **`SYSTEM_EVENT_DRIVEN`**: A more automated approach where the coordinator's only job is to publish a plan to the `TaskBoard`. The framework then monitors the board and automatically notifies agents when their tasks become unblocked, enabling parallel execution and reducing coordinator overhead.
+-   **`SYSTEM_EVENT_DRIVEN`**: A more automated approach where the coordinator's only job is to publish a plan to the `TaskPlan`. The framework then monitors the board and automatically notifies agents when their tasks become unblocked, enabling parallel execution and reducing coordinator overhead.
 
 ## Requirements