microbots 0.0.7__tar.gz → 0.0.8__tar.gz

{microbots-0.0.7 → microbots-0.0.8}/.vscode/launch.json

@@ -5,7 +5,7 @@
     "version": "0.2.0",
     "configurations": [
         {
-            "name": "Python Debugger: Module",
+            "name": "PyTest: test_microbot_ro_mount",
             "type": "debugpy",
             "request": "launch",
             "module": "pytest",

{microbots-0.0.7 → microbots-0.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: microbots
-Version: 0.0.7
+Version: 0.0.8
 Summary: container-based autonomous agent framework
 Author-email: xxx <xxx@example.com>
 License: MIT License
@@ -48,8 +48,8 @@ Dynamic: license-file
 
 # 🤖 Microbots
 
-MicroBots is a lightweight, extensible AI agent for code comprehension and controlled file edits. It integrates cleanly 
-into automation pipelines, mounting a target directory with explicit read-only or read/write modes so LLMs can safely 
+MicroBots is a lightweight, extensible AI agent for code comprehension and controlled file edits. It integrates cleanly
+into automation pipelines, mounting a target directory with explicit read-only or read/write modes so LLMs can safely
 inspect, refactor, or generate files with least‑privilege access.
 
 
@@ -61,14 +61,14 @@ myWritingBot = WritingBot(
     folder_to_mount=str("myReactApp"),
 )
 
-data = myWritingBot.run("""when doing npm run build, I get an error. 
+data = myWritingBot.run("""when doing npm run build, I get an error.
 Fix the error and make sure the build is successful.""", timeout_in_seconds=600)
 print(data.results)
 ```
 
-## ⚠️ Project Status: Highly Unstable
-This project is currently **under active development** and is considered **highly unstable**. Features, APIs, and internal structures are subject to change without notice, and unexpected behavior may occur.
-Please **use with caution** in production environments.
+## ⚠️ Project Status: Under Fast Development
+
+This project is currently **under active development**. Features, APIs, and internal structures are subject to change across versions without notice, and unexpected behavior may occur. Please **use with caution** in production environments.
 
 ## 🚀 How to install
 
@@ -85,7 +85,7 @@ pip install microbots
 
 
 ## ✨LLM Support
-    
+
 Azure OpenAI Models - Add the below environment variables in a `.env` file in the root of your application
 
 ```env
@@ -95,7 +95,7 @@ OPEN_AI_KEY=XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
 
 ## 🤖 Bots & Usage Examples
 
-Pre-requisite for the below example code of Bots:   
+Pre-requisite for the below example code of Bots:
 From the root of your application, Create a folder called  `code` inside which clone the repo `https://github.com/swe-agent/test-repo/`. Now run the code
 
 
@@ -120,7 +120,7 @@ The `ReadingBot` will read the files inside `code` folder and will extract infor
 
 ### ✍️ WritingBot
 
-Pre-requisite for the example code: 
+Pre-requisite for the example code:
 From the root the application, Create a folder called  `code` inside which clone the repo `https://github.com/swe-agent/test-repo/`. Now run the code
 
 ```py

{microbots-0.0.7 → microbots-0.0.8}/README.md

@@ -1,7 +1,7 @@
 # 🤖 Microbots
 
-MicroBots is a lightweight, extensible AI agent for code comprehension and controlled file edits. It integrates cleanly 
-into automation pipelines, mounting a target directory with explicit read-only or read/write modes so LLMs can safely 
+MicroBots is a lightweight, extensible AI agent for code comprehension and controlled file edits. It integrates cleanly
+into automation pipelines, mounting a target directory with explicit read-only or read/write modes so LLMs can safely
 inspect, refactor, or generate files with least‑privilege access.
 
 
@@ -13,14 +13,14 @@ myWritingBot = WritingBot(
     folder_to_mount=str("myReactApp"),
 )
 
-data = myWritingBot.run("""when doing npm run build, I get an error. 
+data = myWritingBot.run("""when doing npm run build, I get an error.
 Fix the error and make sure the build is successful.""", timeout_in_seconds=600)
 print(data.results)
 ```
 
-## ⚠️ Project Status: Highly Unstable
-This project is currently **under active development** and is considered **highly unstable**. Features, APIs, and internal structures are subject to change without notice, and unexpected behavior may occur.
-Please **use with caution** in production environments.
+## ⚠️ Project Status: Under Fast Development
+
+This project is currently **under active development**. Features, APIs, and internal structures are subject to change across versions without notice, and unexpected behavior may occur. Please **use with caution** in production environments.
 
 ## 🚀 How to install
 
@@ -37,7 +37,7 @@ pip install microbots
 
 
 ## ✨LLM Support
-    
+
 Azure OpenAI Models - Add the below environment variables in a `.env` file in the root of your application
 
 ```env
@@ -47,7 +47,7 @@ OPEN_AI_KEY=XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
 
 ## 🤖 Bots & Usage Examples
 
-Pre-requisite for the below example code of Bots:   
+Pre-requisite for the below example code of Bots:
 From the root of your application, Create a folder called  `code` inside which clone the repo `https://github.com/swe-agent/test-repo/`. Now run the code
 
 
@@ -72,7 +72,7 @@ The `ReadingBot` will read the files inside `code` folder and will extract infor
 
 ### ✍️ WritingBot
 
-Pre-requisite for the example code: 
+Pre-requisite for the example code:
 From the root the application, Create a folder called  `code` inside which clone the repo `https://github.com/swe-agent/test-repo/`. Now run the code
 
 ```py

{microbots-0.0.7 → microbots-0.0.8}/src/microbots/MicroBot.py

@@ -51,6 +51,33 @@ class BotRunResult:
 
 
 class MicroBot:
+    """
+    The core Microbot class.
+
+    MicroBot class is the core class representing the autonomous agent. Other bots are extensions of this class.
+    If you want to create a custom bot, you can directly use this class or extend it into your own bot class.
+
+    Attributes
+    ----------
+        model : str
+            The model to use for the bot, in the format <provider>/<model_name>.
+        bot_type : BotType
+            The type of bot being created. It's unused. Will be removed soon.
+        system_prompt : Optional[str]
+            The system prompt to guide the bot's behavior.
+        environment : Optional[any]
+            The execution environment for the bot. If not provided, a default
+            LocalDockerEnvironment will be created.
+        additional_tools : Optional[list[Tool]]
+            A list of additional tools to install in the bot's environment.
+        folder_to_mount : Optional[Mount]
+            A folder to mount into the bot's environment. The bot will be given
+            access to this folder based on the specified permissions. This will
+            be the main code folder where the bot will work. Additional folders
+            can be mounted during the run() method. Refer to `Mount` class
+            regarding the directory structure and permission details. Defaults
+            to None.
+    """
 
     def __init__(
         self,
@@ -61,6 +88,33 @@ class MicroBot:
         additional_tools: Optional[list[Tool]] = [],
         folder_to_mount: Optional[Mount] = None,
     ):
+        """
+        Init function for MicroBot class.
+
+        Parameters
+        ----------
+            model :str
+                The model to use for the bot, in the format <provider>/<model_name>.
+            bot_type :BotType
+                The type of bot being created. It's unused. Will be removed soon.
+            system_prompt :Optional[str]
+                The system prompt to guide the bot's behavior. Defaults to None.
+            environment :Optional[any]
+                The execution environment for the bot. If not provided, a default
+                LocalDockerEnvironment will be created.
+            additional_tools :Optional[list[Tool]]
+                A list of additional tools to install in the bot's environment.
+                Defaults to [].
+            folder_to_mount :Optional[Mount]
+                A folder to mount into the bot's environment. The bot will be given
+                access to this folder based on the specified permissions. This will
+                be the main code folder where the bot will work. Additional folders
+                can be mounted using the run() method. Refer to `Mount` class
+                regarding the directory structure and permission details. Defaults
+                to None.
+
+                Note: Supports only mount type MountType.MOUNT for now.
+        """
 
         self.folder_to_mount = folder_to_mount
 
@@ -74,6 +128,7 @@ class MicroBot:
 
         self.mounted = []
         if folder_to_mount is not None:
+            self._validate_folder_to_mount(folder_to_mount)
             self.mounted.append(folder_to_mount)
 
         self.system_prompt = system_prompt
@@ -195,10 +250,7 @@ class MicroBot:
 
         self.environment = LocalDockerEnvironment(
             port=free_port,
-            folder_to_mount=(
-                folder_to_mount.host_path_info.abs_path if folder_to_mount else None
-            ),
-            permission=folder_to_mount.permission if folder_to_mount else None,
+            folder_to_mount=folder_to_mount,
         )
 
     def _create_llm(self):
@@ -215,11 +267,12 @@ class MicroBot:
         if provider not in [e.value for e in ModelProvider]:
             raise ValueError(f"Unsupported model provider: {provider}")
 
-    # def __del__(self):
-    #     if self.environment:
-    #         try:
-    #             self.environment.stop()
-    #         except Exception as e:
-    #             logger.error(
-    #                 "%s Error while stopping environment: %s", LogLevelEmoji.ERROR, e
-    #             )
+    def _validate_folder_to_mount(self, folder_to_mount: Mount):
+        if folder_to_mount.mount_type != MountType.MOUNT:
+            logger.error(
+                "%s Only MOUNT mount type is supported for folder_to_mount",
+                LogLevelEmoji.ERROR,
+            )
+            raise ValueError(
+                "Only MOUNT mount type is supported for folder_to_mount"
+            )

{microbots-0.0.7 → microbots-0.0.8}/src/microbots/bot/LogAnalysisBot.py

@@ -1,4 +1,5 @@
 import logging
+import os
 from typing import Optional
 
 from microbots.constants import DOCKER_WORKING_DIR, LOG_FILE_DIR, PermissionLabels
@@ -22,7 +23,9 @@ class LogAnalysisBot(MicroBot):
         bot_type = BotType.LOG_ANALYSIS_BOT
 
         folder_mount_info = Mount(
-            folder_to_mount, DOCKER_WORKING_DIR, PermissionLabels.READ_ONLY
+            folder_to_mount,
+            f"/{DOCKER_WORKING_DIR}/{os.path.basename(folder_to_mount)}",
+            PermissionLabels.READ_ONLY
         )
 
         system_prompt = f"""

{microbots-0.0.7 → microbots-0.0.8}/src/microbots/bot/ReadingBot.py

@@ -1,3 +1,4 @@
+import os
 from typing import Optional
 
 from microbots.constants import DOCKER_WORKING_DIR, PermissionLabels
@@ -18,8 +19,9 @@ class ReadingBot(MicroBot):
         # validate init values before assigning
         bot_type = BotType.READING_BOT
 
+        base_name = os.path.basename(folder_to_mount)
         folder_mount_info = Mount(
-            folder_to_mount, f"/{DOCKER_WORKING_DIR}", PermissionLabels.READ_ONLY
+            folder_to_mount, f"/{DOCKER_WORKING_DIR}/{base_name}", PermissionLabels.READ_ONLY
         )
 
         system_prompt = f"""

{microbots-0.0.7 → microbots-0.0.8}/src/microbots/bot/WritingBot.py

@@ -1,3 +1,4 @@
+import os
 from typing import Optional
 
 from microbots.constants import DOCKER_WORKING_DIR, PermissionLabels
@@ -19,7 +20,7 @@ class WritingBot(MicroBot):
         bot_type = BotType.WRITING_BOT
 
         folder_mount_info = Mount(
-            folder_to_mount, DOCKER_WORKING_DIR, PermissionLabels.READ_WRITE
+            folder_to_mount, f"/{DOCKER_WORKING_DIR}/{os.path.basename(folder_to_mount)}", PermissionLabels.READ_WRITE
         )
 
         system_prompt = f"""

{microbots-0.0.7 → microbots-0.0.8}/src/microbots/environment/local_docker/LocalDockerEnvironment.py

@@ -10,7 +10,8 @@ import docker
 import requests
 
 from microbots.environment.Environment import CmdReturn, Environment
-from microbots.constants import DOCKER_WORKING_DIR, WORKING_DIR
+from microbots.constants import DOCKER_WORKING_DIR, WORKING_DIR, PermissionLabels
+from microbots.extras.mount import Mount
 
 logger = logging.getLogger(__name__)
 
@@ -19,23 +20,13 @@ class LocalDockerEnvironment(Environment):
     def __init__(
         self,
         port: int,
-        folder_to_mount: Optional[str] = None,
-        permission: Optional[str] = None,
+        folder_to_mount: Optional[Mount] = None,
         image: str = "kavyasree261002/shell_server:latest",
     ):
-        if folder_to_mount is None and permission is not None:
-            raise ValueError("permission provided but folder_to_mount is None")
-        elif permission is None and folder_to_mount is not None:
-            raise ValueError("folder_to_mount provided but permission is None")
-        if permission is not None and permission not in ["READ_ONLY", "READ_WRITE"]:
-            raise ValueError(
-                "permission must be 'READ_ONLY' or 'READ_WRITE' when provided"
-            )
 
         self.image = image
         self.folder_to_mount = folder_to_mount
         self.overlay_mount = False
-        self.permission = permission
         self.container = None
         self.client = docker.from_env()
         self.port = port  # required host port
@@ -58,26 +49,26 @@ class LocalDockerEnvironment(Environment):
     def start(self):
         mode_map = {"READ_ONLY": "ro", "READ_WRITE": "rw"}
         volumes_config = {WORKING_DIR: {"bind": DOCKER_WORKING_DIR, "mode": "rw"}}
-        if self.folder_to_mount and self.permission:
-            if self.permission == "READ_ONLY":
-                volumes_config[self.folder_to_mount] = {
-                    "bind": f"/ro/{os.path.basename(self.folder_to_mount)}",
-                    "mode": mode_map[self.permission],
+        if self.folder_to_mount:
+            if self.folder_to_mount.permission == PermissionLabels.READ_ONLY:
+                volumes_config[self.folder_to_mount.host_path_info.abs_path] = {
+                    "bind": f"/ro/{os.path.basename(self.folder_to_mount.sandbox_path)}",
+                    "mode": mode_map[self.folder_to_mount.permission],
                 }
                 logger.info(
                     "📦 Volume mapping: %s → /ro/%s",
-                    self.folder_to_mount,
-                    os.path.basename(self.folder_to_mount),
+                    self.folder_to_mount.host_path_info.abs_path,
+                    os.path.basename(self.folder_to_mount.sandbox_path),
                 )
             else:
-                volumes_config[self.folder_to_mount] = {
-                    "bind": f"/{DOCKER_WORKING_DIR}/{os.path.basename(self.folder_to_mount)}",
-                    "mode": mode_map[self.permission],
+                volumes_config[self.folder_to_mount.host_path_info.abs_path] = {
+                    "bind": self.folder_to_mount.sandbox_path,
+                    "mode": mode_map[self.folder_to_mount.permission],
                 }
                 logger.debug(
-                    "📦 Volume mapping: %s → /{DOCKER_WORKING_DIR}/%s",
-                    self.folder_to_mount,
-                    os.path.basename(self.folder_to_mount),
+                    "📦 Volume mapping: %s → %s",
+                    self.folder_to_mount.host_path_info.abs_path,
+                    self.folder_to_mount.sandbox_path,
                 )
 
         # Port mapping
@@ -100,20 +91,22 @@ class LocalDockerEnvironment(Environment):
         )
         time.sleep(2)  # Give some time for the server to start
 
-        if self.permission == "READ_ONLY" and self.folder_to_mount:
-            self._setup_overlay_mount(self.folder_to_mount)
+        if self.folder_to_mount and self.folder_to_mount.permission == PermissionLabels.READ_ONLY:
+            self._setup_overlay_mount()
 
         if self.folder_to_mount:
-            self.execute(f"cd /{DOCKER_WORKING_DIR}/{os.path.basename(self.folder_to_mount)}")
+            self.execute(f"cd {self.folder_to_mount.sandbox_path}")
         else:
             self.execute("cd /")
 
-    def _setup_overlay_mount(self, folder_to_mount: str):
-        path_name = os.path.basename(os.path.abspath(folder_to_mount))
+    def _setup_overlay_mount(self):
+        # NOTE: Don't use this for any other read-only mounts except the main code folder.
+
+        path_name = os.path.basename(self.folder_to_mount.sandbox_path)
         # Mount /ro/path_name to /{WORKING_DIR}/path_name using overlayfs
         mount_command = (
-            f"mkdir -p /{DOCKER_WORKING_DIR}/{path_name} /{DOCKER_WORKING_DIR}/overlay/{path_name}/upper /{DOCKER_WORKING_DIR}/overlay/{path_name}/work && "
-            f"mount -t overlay overlay -o lowerdir=/ro/{path_name}/,upperdir={DOCKER_WORKING_DIR}/overlay/{path_name}/upper/,workdir={DOCKER_WORKING_DIR}/overlay/{path_name}/work/ {DOCKER_WORKING_DIR}/{path_name}/"
+            f"mkdir -p {self.folder_to_mount.sandbox_path} /{DOCKER_WORKING_DIR}/overlay/{path_name}/upper /{DOCKER_WORKING_DIR}/overlay/{path_name}/work && sleep 5 && "
+            f"mount -t overlay overlay -o lowerdir=/ro/{path_name}/,upperdir={DOCKER_WORKING_DIR}/overlay/{path_name}/upper/,workdir={DOCKER_WORKING_DIR}/overlay/{path_name}/work/ {self.folder_to_mount.sandbox_path}"
         )
         self.execute(mount_command)
         logger.info(
@@ -121,12 +114,12 @@ class LocalDockerEnvironment(Environment):
         )
         self.overlay_mount = True
 
-    def _teardown_overlay_mount(self, folder_to_mount: str):
-        path_name = os.path.basename(os.path.abspath(folder_to_mount))
+    def _teardown_overlay_mount(self):
+        path_name = os.path.basename(os.path.abspath(self.folder_to_mount.sandbox_path))
 
         try:
             logger.info("🛠️  Tearing down overlay mount for %s", path_name)
-            unmount_command = f"umount -l {DOCKER_WORKING_DIR}/{path_name}/"
+            unmount_command = f"umount -l {self.folder_to_mount.sandbox_path}"
             ret: CmdReturn = self.execute(unmount_command)
             if ret.return_code != 0:
                 logger.error("❌  Failed to unmount overlay: %s", ret.stderr)
@@ -134,10 +127,10 @@ class LocalDockerEnvironment(Environment):
                 logger.info("✅  Unmounted overlay for %s", path_name)
 
             logger.info(
-                f"🛑  Removing overlay dirs at {DOCKER_WORKING_DIR}/{path_name} and {DOCKER_WORKING_DIR}/overlay/"
+                f"🛑  Removing overlay dirs at {self.folder_to_mount.sandbox_path} and {DOCKER_WORKING_DIR}/overlay/"
             )
             remove_dir_command = (
-                f"rm -rf {DOCKER_WORKING_DIR}/{path_name} && "
+                f"rm -rf {self.folder_to_mount.sandbox_path} && "
                 f"rm -rf {DOCKER_WORKING_DIR}/overlay/"
             )
             ret: CmdReturn = self.execute(remove_dir_command)
@@ -156,7 +149,7 @@ class LocalDockerEnvironment(Environment):
         """Stop and remove the container"""
         if self.container:
             if self.overlay_mount:
-                self._teardown_overlay_mount(self.folder_to_mount)
+                self._teardown_overlay_mount()
 
             self.container.stop()
             self.container.remove()
@@ -236,7 +229,7 @@ class LocalDockerEnvironment(Environment):
                 e,
             )
             return CmdReturn(stdout="", stderr="Unexpected error", return_code=1)
-        
+
     def copy_to_container(self, src_path: str, dest_path: str) -> bool:
         """
         Copy a file or folder from the host machine to the Docker container.

microbots-0.0.8/src/microbots/extras/mount.py

@@ -0,0 +1,73 @@
+from dataclasses import dataclass, field
+from enum import StrEnum
+from pathlib import Path
+
+from microbots.constants import PermissionLabels, PermissionMapping
+from microbots.utils.path import PathInfo, get_path_info, ends_with_separator
+
+
+class MountType(StrEnum):
+    """
+    Enum representing the type of mount operation.
+
+    MOUNT : Mount the folder from host to sandbox environment.
+    COPY : Copy the folder from host to sandbox environment.
+    """
+    MOUNT = "mount"
+    COPY = "copy"
+
+
+@dataclass
+class Mount:
+    """
+    Folder mount configuration for a microbot environment.
+
+    All the folders and files to be presented for the Bot should be
+    either mounted or copied to the Bot's sandbox environment using
+    this class.
+
+    Attributes
+    ----------
+    host_path : str
+        The absolute path on the host machine to be mounted or copied.
+    sandbox_path : str
+        The absolute path inside the Bot's sandbox environment where the
+        host_path will be mounted or copied. If the host_path is a file
+        and the sandbox_path ends with a path separator ("/"), then
+        the file will be placed inside the sandbox_path directory with
+        the same base name as the host file.
+    permission : PermissionLabels
+        The permission level for the mounted/copied folder. It should
+        be one of the values from PermissionLabels enum.
+    mount_type : MountType, optional
+        The type of mount operation. Mounting and copying have are the
+        options. Possible values are MountType enum values.
+        Default is MountType.MOUNT.
+    """
+    host_path: str
+    sandbox_path: str
+    permission: PermissionLabels
+    mount_type: MountType = MountType.MOUNT
+
+    # These will be set in __post_init__
+    permission_key: str = field(init=False)
+    host_path_info: PathInfo = field(init=False)
+
+    def __post_init__(self):
+        self.permission_key = PermissionMapping.MAPPING.get(self.permission)
+        self.host_path_info = get_path_info(self.host_path)
+
+        sandbox_path = Path(self.sandbox_path)
+
+        # Validate that sandbox_path is absolute
+        if not sandbox_path.is_absolute():
+            raise ValueError(
+                f"sandbox_path must be an absolute path. Given: {self.sandbox_path}"
+            )
+
+        if not ends_with_separator(self.host_path) and ends_with_separator(self.sandbox_path):
+            # If host_path is a file and sandbox_path ends with separator,
+            # place the file inside the sandbox_path directory with same base name
+            sandbox_path = str(sandbox_path / self.host_path_info.base_name)
+
+        self.sandbox_path = str(sandbox_path)

{microbots-0.0.7 → microbots-0.0.8}/src/microbots/utils/path.py

@@ -37,3 +37,12 @@ def get_path_info(file_or_folder: str) -> PathInfo:
             abs_path=os.path.abspath(file_or_folder),
         )
     return PathInfo(path_valid=False, base_name="", abs_path="")
+
+
+def ends_with_separator(path_str: str) -> bool:
+    # Accept both separators on Windows; only '/' on POSIX
+    seps = {os.sep}
+    if os.altsep:
+        seps.add(os.altsep)
+    return len(path_str) > 0 and path_str[-1] in seps
+

{microbots-0.0.7 → microbots-0.0.8}/src/microbots.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: microbots
-Version: 0.0.7
+Version: 0.0.8
 Summary: container-based autonomous agent framework
 Author-email: xxx <xxx@example.com>
 License: MIT License
@@ -48,8 +48,8 @@ Dynamic: license-file
 
 # 🤖 Microbots
 
-MicroBots is a lightweight, extensible AI agent for code comprehension and controlled file edits. It integrates cleanly 
-into automation pipelines, mounting a target directory with explicit read-only or read/write modes so LLMs can safely 
+MicroBots is a lightweight, extensible AI agent for code comprehension and controlled file edits. It integrates cleanly
+into automation pipelines, mounting a target directory with explicit read-only or read/write modes so LLMs can safely
 inspect, refactor, or generate files with least‑privilege access.
 
 
@@ -61,14 +61,14 @@ myWritingBot = WritingBot(
     folder_to_mount=str("myReactApp"),
 )
 
-data = myWritingBot.run("""when doing npm run build, I get an error. 
+data = myWritingBot.run("""when doing npm run build, I get an error.
 Fix the error and make sure the build is successful.""", timeout_in_seconds=600)
 print(data.results)
 ```
 
-## ⚠️ Project Status: Highly Unstable
-This project is currently **under active development** and is considered **highly unstable**. Features, APIs, and internal structures are subject to change without notice, and unexpected behavior may occur.
-Please **use with caution** in production environments.
+## ⚠️ Project Status: Under Fast Development
+
+This project is currently **under active development**. Features, APIs, and internal structures are subject to change across versions without notice, and unexpected behavior may occur. Please **use with caution** in production environments.
 
 ## 🚀 How to install
 
@@ -85,7 +85,7 @@ pip install microbots
 
 
 ## ✨LLM Support
-    
+
 Azure OpenAI Models - Add the below environment variables in a `.env` file in the root of your application
 
 ```env
@@ -95,7 +95,7 @@ OPEN_AI_KEY=XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
 
 ## 🤖 Bots & Usage Examples
 
-Pre-requisite for the below example code of Bots:   
+Pre-requisite for the below example code of Bots:
 From the root of your application, Create a folder called  `code` inside which clone the repo `https://github.com/swe-agent/test-repo/`. Now run the code
 
 
@@ -120,7 +120,7 @@ The `ReadingBot` will read the files inside `code` folder and will extract infor
 
 ### ✍️ WritingBot
 
-Pre-requisite for the example code: 
+Pre-requisite for the example code:
 From the root the application, Create a folder called  `code` inside which clone the repo `https://github.com/swe-agent/test-repo/`. Now run the code
 
 ```py

{microbots-0.0.7 → microbots-0.0.8}/src/microbots.egg-info/SOURCES.txt

@@ -54,6 +54,7 @@ test/bot/calculator/calculator.log
 test/bot/calculator/code/calculator.py
 test/environment/local_docker/test_local_docker_environment.py
 test/environment/swe-rex/test_local_docker.py.disabled
+test/extras/test_mount.py
 test/fixtures/fixture_issue_1.py
 test/fixtures/fixture_issue_22.py
 test/fixtures/fixture_test_repo.py

{microbots-0.0.7 → microbots-0.0.8}/test/bot/test_browsing_bot.py

@@ -19,7 +19,7 @@ from microbots import BrowsingBot, BotRunResult
 @pytest.mark.slow
 class TestBrowsingBot:
     """Integration tests for BrowsingBot functionality."""
-    
+
     @pytest.fixture(scope="function")
     def browsing_bot(self):
         """Create a BrowsingBot instance for testing."""
@@ -31,42 +31,43 @@ class TestBrowsingBot:
                 bot.environment.stop()
             except Exception as e:
                 logger.warning(f"Error stopping environment: {e}")
-    
+
+    # Google search may fail due to captcha, so this test may be flaky in CI environments.
     def test_simple_question_response(self, browsing_bot):
         """Test that the bot can answer a simple factual question."""
         response: BotRunResult = browsing_bot.run(
-            "What is the capital of France?",
+            "What is the capital of France? Get this info from https://en.wikipedia.org/wiki/France",
             timeout_in_seconds=300,
         )
-        
+
         # Assert the response was successful
         assert response.status == True, f"Bot failed with error: {response.error}"
         assert response.result is not None, "Bot returned no result"
         assert isinstance(response.result, str), "Result should be a string"
-        
+
         # Check that the result contains the expected answer
         result_lower = response.result.lower()
         assert "paris" in result_lower, f"Expected 'Paris' in result, got: {response.result}"
-        
+
         logger.info(f"Test passed. Bot response: {response.result}")
-    
-    
+
+
+    # Google search may fail due to captcha, so this test may be flaky in CI environments.
     @pytest.mark.parametrize("query,expected_keywords", [
-        ("What is the capital of Germany?", ["berlin"]),
+        ("Get capital of Germany from https://en.wikipedia.org/wiki/Germany", ["berlin"]),
         ("What is 2+2?", ["4", "four"]),
-        ("What time is it in Tokyo?", ["am", "pm", "time"]),
-        ("Who is the current President of the United States?", ["Trump"]),
+        ("Get the description of this CVE-2024-11738 from nvd.nist.gov website", ["Rustls"]),
     ])
     def test_multiple_queries(self, browsing_bot, query, expected_keywords):
         """Test the bot with multiple different queries."""
         response: BotRunResult = browsing_bot.run(query, timeout_in_seconds=300)
-        
+
         assert response.status == True, f"Query '{query}' failed: {response.error}"
         assert response.result is not None, f"No result for query: {query}"
-        
+
         result_lower = response.result.lower()
         # At least one expected keyword should be in the result
         keyword_found = any(keyword.lower() in result_lower for keyword in expected_keywords)
         assert keyword_found, f"None of {expected_keywords} found in result: {response.result}"
-        
+
         logger.info(f"Query '{query}' passed with result: {response.result[:100]}...")

{microbots-0.0.7 → microbots-0.0.8}/test/bot/test_log_analysis_bot.py

@@ -65,7 +65,7 @@ class TestLogAnalysisBot:
             log_file.write(result.stderr)
 
         response: BotRunResult = log_analysis_bot.run(
-            log_file_path, timeout_in_seconds=300
+            str(log_file_path), timeout_in_seconds=300
         )
 
         print(f"Status: {response.status}, Result: {response.result}, Error: {response.error}")

{microbots-0.0.7 → microbots-0.0.8}/test/bot/test_microbot.py

@@ -47,17 +47,17 @@ You must send `task_done` as true only when you have completed the task. It mean
 class TestMicroBot:
 
     @pytest.fixture(scope="function")
-    def log_file_path(self, tmpdir):
+    def log_file_path(self, tmpdir: Path):
         assert tmpdir.exists()
         yield tmpdir / "error.log"
         if tmpdir.exists():
             subprocess.run(["sudo", "rm", "-rf", str(tmpdir)])
 
     @pytest.fixture(scope="function")
-    def ro_mount(self, test_repo):
+    def ro_mount(self, test_repo: Path):
         assert test_repo is not None
         return Mount(
-            str(test_repo), DOCKER_WORKING_DIR, PermissionLabels.READ_ONLY
+            str(test_repo), f"{DOCKER_WORKING_DIR}/{test_repo.name}", PermissionLabels.READ_ONLY
         )
 
     @pytest.fixture(scope="function")
@@ -98,7 +98,9 @@ class TestMicroBot:
         verify_function = issue_1[1]
 
         test_repo_mount_ro = Mount(
-            str(test_repo), DOCKER_WORKING_DIR, PermissionLabels.READ_ONLY
+            str(test_repo),
+            f"{DOCKER_WORKING_DIR}/{test_repo.name}",
+            PermissionLabels.READ_ONLY
         )
         testing_bot = MicroBot(
             model="azure-openai/mini-swe-agent-gpt5",
@@ -121,7 +123,7 @@ class TestMicroBot:
             log_file.write(response.result)
 
         test_repo_mount_rw = Mount(
-            str(test_repo), DOCKER_WORKING_DIR, PermissionLabels.READ_WRITE
+            str(test_repo), f"{DOCKER_WORKING_DIR}/{test_repo.name}", PermissionLabels.READ_WRITE
         )
         coding_bot = MicroBot(
             model="azure-openai/mini-swe-agent-gpt5",
@@ -130,7 +132,7 @@ class TestMicroBot:
         )
 
         additional_mounts = Mount(
-            log_file_path,
+            str(log_file_path),
             "/var/log",
             PermissionLabels.READ_ONLY,
             MountType.COPY,
@@ -148,12 +150,30 @@ class TestMicroBot:
 
         verify_function(test_repo)
 
-    def test_incorrect_mount_type(self, log_file_path, test_repo):
+    def test_incorrect_code_mount_type(self, log_file_path, test_repo):
         assert test_repo is not None
         assert log_file_path is not None
 
         test_repo_mount_ro = Mount(
-            str(test_repo), DOCKER_WORKING_DIR, PermissionLabels.READ_ONLY
+            str(test_repo),
+            f"{DOCKER_WORKING_DIR}/{test_repo.name}",
+            PermissionLabels.READ_ONLY,
+            MountType.COPY,
+        )
+
+        with pytest.raises(ValueError, match="Only MOUNT mount type is supported for folder_to_mount"):
+            testing_bot = MicroBot(
+                model="azure-openai/mini-swe-agent-gpt5",
+                system_prompt=SYSTEM_PROMPT,
+                folder_to_mount=test_repo_mount_ro,
+            )
+
+    def test_incorrect_copy_mount_type(self, log_file_path, test_repo):
+        assert test_repo is not None
+        assert log_file_path is not None
+
+        test_repo_mount_ro = Mount(
+            str(test_repo), f"{DOCKER_WORKING_DIR}/{test_repo.name}", PermissionLabels.READ_ONLY
         )
         testing_bot = MicroBot(
             model="azure-openai/mini-swe-agent-gpt5",
@@ -162,7 +182,7 @@ class TestMicroBot:
         )
 
         additional_mounts = Mount(
-            log_file_path,
+            str(log_file_path),
             "/var/log",
             PermissionLabels.READ_ONLY,
             MountType.MOUNT, # MOUNT is not supported yet
@@ -178,7 +198,7 @@ class TestMicroBot:
         assert test_repo is not None
 
         test_repo_mount_ro = Mount(
-            str(test_repo), DOCKER_WORKING_DIR, PermissionLabels.READ_ONLY
+            str(test_repo), f"{DOCKER_WORKING_DIR}/{test_repo.name}", PermissionLabels.READ_ONLY
         )
         with pytest.raises(ValueError, match="Unsupported model provider: provider"):
             MicroBot(
@@ -191,7 +211,7 @@ class TestMicroBot:
         assert test_repo is not None
 
         test_repo_mount_ro = Mount(
-            str(test_repo), DOCKER_WORKING_DIR, PermissionLabels.READ_ONLY
+            str(test_repo), f"{DOCKER_WORKING_DIR}/{test_repo.name}", PermissionLabels.READ_ONLY
         )
         with pytest.raises(ValueError, match="Model should be in the format <provider>/<model_name>"):
             MicroBot(

{microbots-0.0.7 → microbots-0.0.8}/test/environment/local_docker/test_local_docker_environment.py

@@ -2,19 +2,17 @@
 Integration tests for LocalDockerEnvironment
 """
 import pytest
-import shutil
 import os
-import time
 import socket
-from pathlib import Path
 import re
 
 # Add src to path for imports
 import sys
+
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../../../src")))
 
 from microbots.environment.local_docker.LocalDockerEnvironment import LocalDockerEnvironment
-from microbots.environment.Environment import CmdReturn
+from microbots.extras.mount import Mount
 
 # Use the correct working directory path
 DOCKER_WORKING_DIR = "workdir"
@@ -82,22 +80,6 @@ class TestLocalDockerEnvironmentIntegration:
         assert "Hello World" in result.stdout
 
 
-    @pytest.mark.integration
-    @pytest.mark.docker
-    def test_initialization_validation_errors(self):
-        """Test that initialization properly validates parameters"""
-        # Test permission without folder
-        with pytest.raises(ValueError, match="permission provided but folder_to_mount is None"):
-            LocalDockerEnvironment(port=8999, permission="READ_ONLY")
-
-        # Test folder without permission
-        with pytest.raises(ValueError, match="folder_to_mount provided but permission is None"):
-            LocalDockerEnvironment(port=8999, folder_to_mount="/some/path")
-
-        # Test invalid permission
-        with pytest.raises(ValueError, match="permission must be 'READ_ONLY' or 'READ_WRITE' when provided"):
-            LocalDockerEnvironment(port=8999, folder_to_mount="/some/path", permission="INVALID")
-
     @pytest.mark.integration
     @pytest.mark.docker
     def test_command_execution_basic(self, shared_env):
@@ -191,11 +173,15 @@ EOF"""
             mount_port = s.getsockname()[1]
 
         env = None
+        test_dir_mount = Mount(
+            host_path=test_dir,
+            sandbox_path=f"/{DOCKER_WORKING_DIR}/{os.path.basename(test_dir)}",
+            permission="READ_WRITE"
+        )
         try:
             env = LocalDockerEnvironment(
                 port=mount_port,
-                folder_to_mount=test_dir,
-                permission="READ_WRITE"
+                folder_to_mount=test_dir_mount,
             )
 
             folder_name = os.path.basename(test_dir)
@@ -246,11 +232,16 @@ EOF"""
             mount_port = s.getsockname()[1]
 
         env = None
+        test_dir_mount = Mount(
+            host_path=test_dir,
+            sandbox_path=f"/{DOCKER_WORKING_DIR}/{os.path.basename(test_dir)}",
+            permission="READ_ONLY"
+        )
+
         try:
             env = LocalDockerEnvironment(
                 port=mount_port,
-                folder_to_mount=test_dir,
-                permission="READ_ONLY"
+                folder_to_mount=test_dir_mount,
             )
 
             folder_name = os.path.basename(test_dir)

microbots-0.0.8/test/extras/test_mount.py

@@ -0,0 +1,37 @@
+import sys
+from pathlib import Path
+
+import pytest
+
+sys.path.insert(
+    0, f"{Path(__file__).resolve().parents[2].as_posix()}/src"
+)
+
+import logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+from microbots.extras.mount import Mount
+
+@pytest.mark.integration
+def test_non_abs_sandbox_path(tmpdir: Path):
+    with pytest.raises(ValueError, match="sandbox_path must be an absolute path"):
+        Mount(
+            host_path=str(tmpdir),
+            sandbox_path="relative/path/in/sandbox",
+            permission="READ_ONLY",
+        )
+
+@pytest.mark.integration
+def test_host_file_sandbox_dir(tmpdir: Path):
+    host_file = tmpdir / "testfile.txt"
+    host_file.write_text("Sample content", encoding="utf-8")
+
+    mount = Mount(
+        host_path=str(host_file),
+        sandbox_path="/sandbox/dir/",
+        permission="READ_ONLY",
+        mount_type="copy",
+    )
+
+    assert mount.sandbox_path == "/sandbox/dir/testfile.txt"

{microbots-0.0.7 → microbots-0.0.8}/test/fixtures/fixture_test_repo.py

@@ -17,12 +17,19 @@ def test_repo(tmpdir):
             capture_output=True,
             text=True
         )
+
+        if result.returncode != 0:
+            pytest.fail(
+                f"Failed to clone repository: {TEST_REPO}\n"
+                f"Return code: {result.returncode}\n"
+                f"Stdout: {result.stdout}\n"
+                f"Stderr: {result.stderr}"
+            )
+
     except subprocess.CalledProcessError as e:
         pytest.fail(
-            f"Failed to clone repository: {TEST_REPO}\n"
-            f"Return code: {e.returncode}\n"
-            f"Stdout: {e.stdout}\n"
-            f"Stderr: {e.stderr}"
+            f"Exception while cloning repository: {TEST_REPO}\n"
+            f"Exception: {str(e)}\n"
         )
 
     repo_path = Path(tmpdir / os.listdir(tmpdir)[0])

microbots-0.0.7/src/microbots/extras/mount.py

@@ -1,27 +0,0 @@
-from dataclasses import dataclass, field
-from enum import StrEnum
-
-from microbots.constants import PermissionLabels, PermissionMapping
-from microbots.utils.path import PathInfo, get_path_info
-
-
-class MountType(StrEnum):
-    MOUNT = "mount"
-    COPY = "copy"
-
-
-@dataclass
-class Mount:
-    host_path: str
-    sandbox_path: str
-    permission: PermissionLabels
-    mount_type: MountType = MountType.MOUNT
-
-    # These will be set in __post_init__
-    permission_key: str = field(init=False)
-    host_path_info: PathInfo = field(init=False)
-
-    def __post_init__(self):
-        self.permission_key = PermissionMapping.MAPPING.get(self.permission)
-        self.host_path_info = get_path_info(self.host_path)
-        self.sandbox_path = f"{self.sandbox_path}/{self.host_path_info.base_name}"