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

autobyteus/tools/browser/session_aware/browser_session_aware_webpage_reader.py

@@ -5,7 +5,7 @@ from typing import Optional, TYPE_CHECKING, Any
 from autobyteus.tools.browser.session_aware.browser_session_aware_tool import BrowserSessionAwareTool
 from autobyteus.tools.browser.session_aware.shared_browser_session import SharedBrowserSession
 from autobyteus.tools.tool_config import ToolConfig 
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType 
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType 
 from autobyteus.tools.tool_category import ToolCategory
 from autobyteus.utils.html_cleaner import clean, CleaningMode
 

autobyteus/tools/browser/session_aware/browser_session_aware_webpage_screenshot_taker.py

@@ -6,7 +6,7 @@ from typing import Optional, TYPE_CHECKING, Any
 from autobyteus.tools.browser.session_aware.browser_session_aware_tool import BrowserSessionAwareTool
 from autobyteus.tools.browser.session_aware.shared_browser_session import SharedBrowserSession
 from autobyteus.tools.tool_config import ToolConfig 
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType 
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType 
 from autobyteus.tools.tool_category import ToolCategory
 
 if TYPE_CHECKING:

autobyteus/tools/browser/standalone/navigate_to.py

@@ -6,7 +6,7 @@ from urllib.parse import urlparse
 from typing import Optional, TYPE_CHECKING, Any
 import logging
 
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
 
 if TYPE_CHECKING:
     from autobyteus.agent.context import AgentContext

autobyteus/tools/browser/standalone/web_page_pdf_generator.py

@@ -7,7 +7,7 @@ import logging
 from typing import Optional, TYPE_CHECKING, Any 
 from urllib.parse import urlparse 
 
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
 
 if TYPE_CHECKING:
     from autobyteus.agent.context import AgentContext

autobyteus/tools/browser/standalone/webpage_image_downloader.py

@@ -7,7 +7,7 @@ import logging
 from urllib.parse import urljoin, urlparse 
 from typing import Optional, TYPE_CHECKING, Any, List 
 
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
 
 if TYPE_CHECKING:
     from autobyteus.agent.context import AgentContext

autobyteus/tools/browser/standalone/webpage_reader.py

@@ -7,7 +7,7 @@ import logging
 from typing import Optional, TYPE_CHECKING, Any
 from autobyteus.tools.base_tool import BaseTool
 from autobyteus.tools.tool_config import ToolConfig 
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType 
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType 
 from autobyteus.tools.tool_category import ToolCategory
 from brui_core.ui_integrator import UIIntegrator 
 from autobyteus.utils.html_cleaner import clean, CleaningMode

autobyteus/tools/browser/standalone/webpage_screenshot_taker.py

@@ -1,7 +1,7 @@
 from typing import Optional, TYPE_CHECKING, Any
 from autobyteus.tools.base_tool import BaseTool
 from autobyteus.tools.tool_config import ToolConfig
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType 
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType 
 from autobyteus.tools.tool_category import ToolCategory
 from brui_core.ui_integrator import UIIntegrator 
 import logging 

autobyteus/tools/functional_tool.py

@@ -5,7 +5,7 @@ import asyncio
 from typing import Callable, Optional, Any, Dict, Tuple, Union, get_origin, get_args, List as TypingList, TYPE_CHECKING, Type
 
 from autobyteus.tools.base_tool import BaseTool
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
 from autobyteus.tools.tool_config import ToolConfig
 from autobyteus.tools.registry import default_tool_registry, ToolDefinition
 from autobyteus.tools.tool_origin import ToolOrigin

autobyteus/tools/google_search.py

@@ -6,7 +6,7 @@ from typing import Optional, TYPE_CHECKING, Any, Dict, List
 
 from autobyteus.tools.base_tool import BaseTool
 from autobyteus.tools.tool_config import ToolConfig
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
 from autobyteus.tools.tool_category import ToolCategory
 
 if TYPE_CHECKING:

autobyteus/tools/image_downloader.py

@@ -6,7 +6,7 @@ from typing import Optional, TYPE_CHECKING, Any
 
 from autobyteus.tools.base_tool import BaseTool
 from autobyteus.tools.tool_config import ToolConfig 
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType 
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType 
 from autobyteus.tools.tool_category import ToolCategory
 from PIL import Image
 from io import BytesIO

autobyteus/tools/mcp/factory.py

@@ -6,7 +6,7 @@ from .tool import GenericMcpTool
 from autobyteus.tools.factory.tool_factory import ToolFactory
 
 if TYPE_CHECKING:
-    from autobyteus.tools.parameter_schema import ParameterSchema
+    from autobyteus.utils.parameter_schema import ParameterSchema
     from autobyteus.tools.tool_config import ToolConfig
     from autobyteus.tools.base_tool import BaseTool
 

autobyteus/tools/mcp/schema_mapper.py

@@ -2,7 +2,7 @@
 import logging
 from typing import Dict, Any, List, Optional
 
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
 
 logger = logging.getLogger(__name__)
 

autobyteus/tools/mcp/tool.py

@@ -3,7 +3,7 @@ import logging
 from typing import Any, Optional, TYPE_CHECKING
 
 from autobyteus.tools.base_tool import BaseTool
-from autobyteus.tools.parameter_schema import ParameterSchema
+from autobyteus.utils.parameter_schema import ParameterSchema
 from .server.proxy import McpServerProxy
 
 if TYPE_CHECKING:

autobyteus/tools/multimedia/__init__.py

@@ -1,8 +1,10 @@
 from .image_tools import GenerateImageTool, EditImageTool
 from .audio_tools import GenerateSpeechTool
+from .media_reader_tool import ReadMediaFile
 
 __all__ = [
     "GenerateImageTool",
     "EditImageTool",
     "GenerateSpeechTool",
+    "ReadMediaFile",
 ]

autobyteus/tools/multimedia/audio_tools.py

@@ -3,7 +3,7 @@ import logging
 from typing import Optional, List
 
 from autobyteus.tools.base_tool import BaseTool
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
 from autobyteus.tools.tool_category import ToolCategory
 from autobyteus.multimedia.audio import audio_client_factory, AudioModel, AudioClientFactory
 
@@ -34,24 +34,8 @@ def _build_dynamic_audio_schema(base_params: List[ParameterDefinition], model_en
         logger.error(f"Cannot generate audio tool schema. Check environment and model registry. Error: {e}")
         raise RuntimeError(f"Failed to configure audio tool. Error: {e}")
 
-    config_schema = ParameterSchema()
-    if model.parameter_schema:
-        for name, meta in model.parameter_schema.items():
-            param_type_str = meta.get("type", "string").upper()
-            param_type = getattr(ParameterType, param_type_str, ParameterType.STRING)
-            
-            allowed_values = meta.get("allowed_values")
-            if param_type == ParameterType.STRING and allowed_values:
-                param_type = ParameterType.ENUM
-
-            config_schema.add_parameter(ParameterDefinition(
-                name=name,
-                param_type=param_type,
-                description=meta.get("description", ""),
-                required=False,
-                default_value=meta.get("default"),
-                enum_values=allowed_values
-            ))
+    # The model's parameter schema is now a ParameterSchema object, so we can use it directly.
+    config_schema = model.parameter_schema
 
     schema = ParameterSchema()
     for param in base_params:
@@ -70,6 +54,7 @@ def _build_dynamic_audio_schema(base_params: List[ParameterDefinition], model_en
 
 class GenerateSpeechTool(BaseTool):
     """
+
     An agent tool for generating speech from text using a Text-to-Speech (TTS) model.
     """
     CATEGORY = ToolCategory.MULTIMEDIA
@@ -93,7 +78,12 @@ class GenerateSpeechTool(BaseTool):
             ParameterDefinition(
                 name="prompt",
                 param_type=ParameterType.STRING,
-                description="The text to be converted into spoken audio.",
+                description=(
+                    "The text to be converted into spoken audio. For multi-speaker mode, you must format the prompt "
+                    "with speaker labels that match the speakers defined in 'speaker_mapping'. "
+                    "CRITICAL: Each speaker's dialogue MUST be on a new line. "
+                    "Example: 'Joe: Hello Jane.\\nJane: Hi Joe, how are you?'"
+                ),
                 required=True
             )
         ]

autobyteus/tools/multimedia/image_tools.py

@@ -3,7 +3,7 @@ import logging
 from typing import Optional, List
 
 from autobyteus.tools.base_tool import BaseTool
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
 from autobyteus.tools.tool_category import ToolCategory
 from autobyteus.multimedia.image import image_client_factory, ImageModel, ImageClientFactory
 
@@ -34,24 +34,8 @@ def _build_dynamic_image_schema(base_params: List[ParameterDefinition], model_en
         logger.error(f"Cannot generate image tool schema. Check environment and model registry. Error: {e}")
         raise RuntimeError(f"Failed to configure image tool. Error: {e}")
 
-    config_schema = ParameterSchema()
-    if model.parameter_schema:
-        for name, meta in model.parameter_schema.items():
-            param_type_str = meta.get("type", "string").upper()
-            param_type = getattr(ParameterType, param_type_str, ParameterType.STRING)
-            
-            allowed_values = meta.get("allowed_values")
-            if param_type == ParameterType.STRING and allowed_values:
-                param_type = ParameterType.ENUM
-
-            config_schema.add_parameter(ParameterDefinition(
-                name=name,
-                param_type=param_type,
-                description=meta.get("description", ""),
-                required=False,
-                default_value=meta.get("default"),
-                enum_values=allowed_values
-            ))
+    # The model's parameter schema is now a ParameterSchema object, so we can use it directly.
+    config_schema = model.parameter_schema
 
     schema = ParameterSchema()
     for param in base_params:
@@ -84,6 +68,7 @@ class GenerateImageTool(BaseTool):
     def get_description(cls) -> str:
         return (
             "Generates one or more images based on a textual description (prompt) using the system's default image model. "
+            "Can optionally accept reference images to influence the style or content. "
             "Returns a list of URLs to the generated images upon success."
         )
 
@@ -95,17 +80,31 @@ class GenerateImageTool(BaseTool):
                 param_type=ParameterType.STRING,
                 description="A detailed textual description of the image to generate.",
                 required=True
+            ),
+            ParameterDefinition(
+                name="input_image_urls",
+                param_type=ParameterType.STRING,
+                description="Optional. A comma-separated string of URLs to reference images. The generated image will try to match the style or content of these images.",
+                required=False
             )
         ]
         return _build_dynamic_image_schema(base_params, cls.MODEL_ENV_VAR, cls.DEFAULT_MODEL)
 
-    async def _execute(self, context, prompt: str, generation_config: Optional[dict] = None) -> List[str]:
+    async def _execute(self, context, prompt: str, input_image_urls: Optional[str] = None, generation_config: Optional[dict] = None) -> List[str]:
         model_identifier = _get_configured_model_identifier(self.MODEL_ENV_VAR, self.DEFAULT_MODEL)
         logger.info(f"GenerateImageTool executing with configured model '{model_identifier}'.")
         client = None
         try:
+            urls_list = None
+            if input_image_urls:
+                urls_list = [url.strip() for url in input_image_urls.split(',') if url.strip()]
+
             client = image_client_factory.create_image_client(model_identifier=model_identifier)
-            response = await client.generate_image(prompt=prompt, generation_config=generation_config)
+            response = await client.generate_image(
+                prompt=prompt, 
+                input_image_urls=urls_list,
+                generation_config=generation_config
+            )
             
             if not response.image_urls:
                 raise ValueError("Image generation failed to return any image URLs.")
@@ -121,7 +120,7 @@ class EditImageTool(BaseTool):
     An agent tool for editing an existing image using a text prompt and a pre-configured model.
     """
     CATEGORY = ToolCategory.MULTIMEDIA
-    MODEL_ENV_VAR = "DEFAULT_IMAGE_GENERATION_MODEL"
+    MODEL_ENV_VAR = "DEFAULT_IMAGE_EDIT_MODEL"
     DEFAULT_MODEL = "gpt-image-1"
 
     @classmethod

autobyteus/tools/multimedia/media_reader_tool.py

@@ -0,0 +1,117 @@
+# file: autobyteus/autobyteus/tools/multimedia/media_reader_tool.py
+import logging
+import os
+from typing import TYPE_CHECKING, Optional
+
+from autobyteus.tools.base_tool import BaseTool
+from autobyteus.tools.tool_category import ToolCategory
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
+from autobyteus.agent.message.context_file import ContextFile
+
+if TYPE_CHECKING:
+    from autobyteus.agent.context import AgentContext
+    from autobyteus.agent.workspace.base_workspace import BaseAgentWorkspace
+
+logger = logging.getLogger(__name__)
+
+class ReadMediaFile(BaseTool):
+    """
+    A tool that loads a media file (image, audio, video) into the context for
+    the next LLM turn. This allows a multimodal LLM to directly 'see' or 'hear'
+    the file's content. The tool's result is a structured object that the system
+    uses to construct a multimodal prompt, not plain text.
+    """
+    TOOL_NAME = "ReadMediaFile"
+    CATEGORY = ToolCategory.MULTIMEDIA
+
+    @classmethod
+    def get_name(cls) -> str:
+        return cls.TOOL_NAME
+
+    @classmethod
+    def get_description(cls) -> str:
+        return (
+            "Loads a media file (image, audio, video) into the context for the next turn, "
+            "allowing the LLM to directly analyze its content. Use this when you need to 'see' an image, "
+            "'listen' to audio, or 'watch' a video that you know exists in the workspace or file system."
+        )
+
+    @classmethod
+    def get_argument_schema(cls) -> Optional[ParameterSchema]:
+        schema = ParameterSchema()
+        schema.add_parameter(ParameterDefinition(
+            name="file_path",
+            param_type=ParameterType.STRING,
+            description="The absolute path or workspace-relative path to the media file.",
+            required=True
+        ))
+        return schema
+
+    async def _execute(self,
+                       context: 'AgentContext',
+                       file_path: str) -> 'ContextFile':
+        """
+        Resolves the file path and returns a ContextFile object, which signals
+        the system to include this file in the next multimodal LLM prompt.
+        It handles both absolute paths and paths relative to the agent's workspace.
+        """
+        logger.debug(f"Tool '{self.get_name()}': Received request to read media file at '{file_path}'.")
+        
+        absolute_path: str
+        workspace: Optional['BaseAgentWorkspace'] = context.workspace
+
+        if os.path.isabs(file_path):
+            absolute_path = os.path.normpath(file_path)
+            logger.debug(f"Path '{file_path}' is absolute. Using resolved path: '{absolute_path}'.")
+            
+            # Security Note: This allows reading from outside the workspace.
+            # We log a warning if this occurs.
+            if workspace and hasattr(workspace, 'get_base_path'):
+                try:
+                    workspace_root = os.path.abspath(workspace.get_base_path())
+                    resolved_target = os.path.abspath(absolute_path)
+                    if not os.path.commonpath([workspace_root]) == os.path.commonpath([workspace_root, resolved_target]):
+                        logger.warning(
+                            f"Security Note: Tool '{self.get_name()}' is accessing an absolute path "
+                            f"'{absolute_path}' which is outside the agent's workspace '{workspace_root}'."
+                        )
+                except Exception:
+                    # Failsafe if get_base_path has an issue.
+                    pass
+        else:
+            # Handle relative paths, which MUST be resolved against a workspace that supports file paths.
+            if not (workspace and hasattr(workspace, 'get_base_path') and callable(getattr(workspace, 'get_base_path'))):
+                raise ValueError(
+                    f"A relative path '{file_path}' was provided, but the agent's workspace does not support "
+                    "file system path resolution. Please provide an absolute path or configure a suitable workspace."
+                )
+
+            try:
+                base_path = os.path.abspath(workspace.get_base_path())
+                # Securely join the path and resolve it to a final absolute path
+                absolute_path = os.path.abspath(os.path.join(base_path, file_path))
+                
+                # Security Check: Ensure the resolved path is still within the workspace directory.
+                if os.path.commonpath([base_path]) != os.path.commonpath([base_path, absolute_path]):
+                    raise ValueError(f"Security error: Path '{file_path}' attempts to access files outside the agent's workspace.")
+                
+                logger.debug(f"Path '{file_path}' is relative. Resolved against workspace to '{absolute_path}'.")
+
+            except ValueError as e:
+                # Re-raise security errors with more context.
+                logger.error(f"Tool '{self.get_name()}': Security error resolving relative path '{file_path}': {e}")
+                raise
+
+        try:
+            if not os.path.exists(absolute_path) or not os.path.isfile(absolute_path):
+                raise FileNotFoundError(f"The file '{file_path}' does not exist or is not a regular file at the resolved path '{absolute_path}'.")
+            
+            logger.info(f"Tool '{self.get_name()}': Staging file '{absolute_path}' for next LLM turn.")
+            
+            # The ContextFile constructor will automatically infer file_type from the path.
+            # This is the special object that the ToolResultEventHandler will look for.
+            return ContextFile(uri=absolute_path)
+
+        except (ValueError, FileNotFoundError) as e:
+            logger.error(f"Tool '{self.get_name()}': Error processing path '{file_path}': {e}")
+            raise

autobyteus/tools/pydantic_schema_converter.py

@@ -2,7 +2,7 @@
 import logging
 from typing import Type, get_origin, get_args, Union, List, Dict
 from pydantic import BaseModel
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType
 
 logger = logging.getLogger(__name__)
 

autobyteus/tools/registry/tool_definition.py

@@ -5,7 +5,7 @@ from typing import Dict, Any, List as TypingList, Type, TYPE_CHECKING, Optional,
 
 from autobyteus.llm.providers import LLMProvider
 from autobyteus.tools.tool_config import ToolConfig
-from autobyteus.tools.parameter_schema import ParameterSchema
+from autobyteus.utils.parameter_schema import ParameterSchema
 from autobyteus.tools.tool_origin import ToolOrigin
 # Import default formatters directly to provide convenience methods
 from autobyteus.tools.usage.formatters import (

autobyteus/tools/timer.py

@@ -2,7 +2,7 @@ import asyncio
 from typing import Optional, TYPE_CHECKING, Any
 from autobyteus.tools.base_tool import BaseTool
 from autobyteus.tools.tool_config import ToolConfig 
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType 
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition, ParameterType 
 from autobyteus.tools.tool_category import ToolCategory
 from autobyteus.events.event_emitter import EventEmitter 
 from autobyteus.events.event_types import EventType

autobyteus/tools/tool_meta.py

@@ -4,7 +4,7 @@ from abc import ABCMeta
 from typing import Dict, Any 
 
 from autobyteus.tools.registry import default_tool_registry, ToolDefinition
-from autobyteus.tools.parameter_schema import ParameterSchema
+from autobyteus.utils.parameter_schema import ParameterSchema
 from autobyteus.tools.tool_origin import ToolOrigin
 from autobyteus.tools.tool_category import ToolCategory
 

autobyteus/tools/usage/formatters/default_json_example_formatter.py

@@ -2,7 +2,7 @@
 import json
 from typing import Dict, Any, TYPE_CHECKING, List, Optional, Union
 
-from autobyteus.tools.parameter_schema import ParameterType, ParameterDefinition, ParameterSchema
+from autobyteus.utils.parameter_schema import ParameterType, ParameterDefinition, ParameterSchema
 from .base_formatter import BaseExampleFormatter
 
 if TYPE_CHECKING:

autobyteus/tools/usage/formatters/default_xml_example_formatter.py

@@ -3,7 +3,7 @@ import xml.sax.saxutils
 import re
 from typing import Any, TYPE_CHECKING, List, Optional
 
-from autobyteus.tools.parameter_schema import ParameterType, ParameterDefinition, ParameterSchema
+from autobyteus.utils.parameter_schema import ParameterType, ParameterDefinition, ParameterSchema
 from .base_formatter import BaseExampleFormatter
 
 if TYPE_CHECKING:

autobyteus/tools/usage/formatters/default_xml_schema_formatter.py

@@ -1,8 +1,8 @@
 # file: autobyteus/autobyteus/tools/usage/formatters/default_xml_schema_formatter.py
 import xml.sax.saxutils
-from typing import TYPE_CHECKING, List
+from typing import TYPE_CHECKING, List, Dict
 
-from autobyteus.tools.parameter_schema import ParameterType, ParameterDefinition, ParameterSchema
+from autobyteus.utils.parameter_schema import ParameterType, ParameterDefinition, ParameterSchema
 from .base_formatter import BaseSchemaFormatter
 
 if TYPE_CHECKING:
@@ -30,6 +30,49 @@ class DefaultXmlSchemaFormatter(BaseSchemaFormatter):
         xml_parts.append("</tool>")
         return "\n".join(xml_parts)
 
+    def _json_schema_props_to_param_defs(self, schema_dict: Dict) -> List[ParameterDefinition]:
+        """
+        Converts a JSON schema's 'properties' dictionary into a list of ParameterDefinition objects.
+        This is used to bridge raw JSON schemas with the internal formatting logic.
+        """
+        param_defs = []
+        properties = schema_dict.get("properties", {})
+        required_fields = schema_dict.get("required", [])
+
+        for prop_name, prop_schema in properties.items():
+            if not isinstance(prop_schema, dict):
+                continue
+
+            param_type_str = prop_schema.get("type", "string").upper()
+            param_type = getattr(ParameterType, param_type_str, ParameterType.STRING)
+
+            # JSON Schema uses 'enum' key for enumerations
+            allowed_values = prop_schema.get("enum")
+            if param_type == ParameterType.STRING and allowed_values:
+                param_type = ParameterType.ENUM
+
+            object_schema = None
+            if param_type == ParameterType.OBJECT and "properties" in prop_schema:
+                # Recursively build a ParameterSchema for nested objects
+                nested_param_defs = self._json_schema_props_to_param_defs(prop_schema)
+                object_schema = ParameterSchema(parameters=nested_param_defs)
+            
+            array_item_schema = None
+            if param_type == ParameterType.ARRAY and "items" in prop_schema:
+                # Pass the nested schema down; it will be handled by the next recursive call
+                array_item_schema = prop_schema["items"]
+
+            param_defs.append(ParameterDefinition(
+                name=prop_name,
+                param_type=param_type,
+                description=prop_schema.get("description", ""),
+                required=(prop_name in required_fields),
+                enum_values=allowed_values,
+                object_schema=object_schema,
+                array_item_schema=array_item_schema
+            ))
+        return param_defs
+
     def _format_params_recursively(self, params: List[ParameterDefinition], indent_level: int) -> List[str]:
         """Recursively formats parameter definitions into XML strings."""
         xml_lines = []
@@ -61,13 +104,26 @@ class DefaultXmlSchemaFormatter(BaseSchemaFormatter):
             elif is_array:
                 xml_lines.append(f'{indent}<arg {" ".join(attrs)}>')
                 if isinstance(param.array_item_schema, ParameterSchema):
-                    # Array of objects
+                    # Array of objects defined with our internal ParameterSchema
                     xml_lines.append(f'{indent}    <items type="object">')
                     xml_lines.extend(self._format_params_recursively(param.array_item_schema.parameters, indent_level + 2))
                     xml_lines.append(f'{indent}    </items>')
                 elif isinstance(param.array_item_schema, ParameterType):
                     # Array of primitives
                     xml_lines.append(f'{indent}    <items type="{param.array_item_schema.value}" />')
+                elif isinstance(param.array_item_schema, dict):
+                    # FIX: Handle array of objects defined with a raw JSON schema dict
+                    item_schema_dict = param.array_item_schema
+                    item_type = item_schema_dict.get("type", "string")
+                    
+                    xml_lines.append(f'{indent}    <items type="{item_type}">')
+                    
+                    if item_type == "object":
+                        # Convert the JSON schema properties to our internal ParameterDefinition list
+                        param_defs = self._json_schema_props_to_param_defs(item_schema_dict)
+                        xml_lines.extend(self._format_params_recursively(param_defs, indent_level + 2))
+
+                    xml_lines.append(f'{indent}    </items>')
                 xml_lines.append(f'{indent}</arg>')
             else:
                 # This is a simple/primitive type or a generic array

autobyteus/tools/usage/formatters/gemini_json_example_formatter.py

@@ -2,7 +2,7 @@
 import json
 from typing import Dict, Any, TYPE_CHECKING, Optional
 
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition
 from .base_formatter import BaseExampleFormatter
 from .default_json_example_formatter import DefaultJsonExampleFormatter # Import for reuse
 

autobyteus/tools/usage/formatters/google_json_example_formatter.py

@@ -2,7 +2,7 @@
 import json
 from typing import Dict, Any, TYPE_CHECKING, Optional
 
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition
 from .base_formatter import BaseExampleFormatter
 # Import for reuse of the intelligent example generation logic
 from .default_json_example_formatter import DefaultJsonExampleFormatter

autobyteus/tools/usage/formatters/openai_json_example_formatter.py

@@ -2,7 +2,7 @@
 import json
 from typing import Dict, Any, TYPE_CHECKING, Optional
 
-from autobyteus.tools.parameter_schema import ParameterSchema, ParameterDefinition
+from autobyteus.utils.parameter_schema import ParameterSchema, ParameterDefinition
 from .base_formatter import BaseExampleFormatter
 from .default_json_example_formatter import DefaultJsonExampleFormatter # Import for reuse
 

autobyteus/{tools → utils}/parameter_schema.py

@@ -1,4 +1,4 @@
-# file: autobyteus/autobyteus/tools/parameter_schema.py
+# file: autobyteus/autobyteus/utils/parameter_schema.py
 from __future__ import annotations
 import logging
 from typing import Dict, Any, List, Optional, Union, Type

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: autobyteus
-Version: 1.1.8
+Version: 1.1.9
 Summary: Multi-Agent framework
 Home-page: https://github.com/AutoByteus/autobyteus
 Author: Ryan Zheng
@@ -23,7 +23,7 @@ Requires-Dist: botocore
 Requires-Dist: brui-core==1.0.9
 Requires-Dist: certifi==2025.4.26
 Requires-Dist: google-api-python-client
-Requires-Dist: google-generativeai
+Requires-Dist: google-genai==1.38.0
 Requires-Dist: Jinja2
 Requires-Dist: mcp[cli]
 Requires-Dist: mistral_common