quantalogic 0.2.26__py3-none-any.whl → 0.2.29__py3-none-any.whl

quantalogic/coding_agent.py

@@ -4,6 +4,7 @@ from quantalogic.tools import (
     DuckDuckGoSearchTool,
     EditWholeContentTool,
     ExecuteBashCommandTool,
+    GrepAppTool,
     InputQuestionTool,
     JinjaTool,
     ListDirectoryTool,
@@ -75,7 +76,8 @@ def create_coding_agent(
         InputQuestionTool(),
         DuckDuckGoSearchTool(),
         JinjaTool(),
-        ReadHTMLTool()
+        ReadHTMLTool(),
+        GrepAppTool()
     ]
 
     if vision_model_name:

quantalogic/tools/__init__.py

@@ -7,6 +7,8 @@ from .duckduckgo_search_tool import DuckDuckGoSearchTool
 from .edit_whole_content_tool import EditWholeContentTool
 from .elixir_tool import ElixirTool
 from .execute_bash_command_tool import ExecuteBashCommandTool
+from .generate_database_report_tool import GenerateDatabaseReportTool
+from .grep_app_tool import GrepAppTool
 from .input_question_tool import InputQuestionTool
 from .jinja_tool import JinjaTool
 from .list_directory_tool import ListDirectoryTool
@@ -22,6 +24,7 @@ from .replace_in_file_tool import ReplaceInFileTool
 from .ripgrep_tool import RipgrepTool
 from .search_definition_names import SearchDefinitionNames
 from .serpapi_search_tool import SerpApiSearchTool
+from .sql_query_tool import SQLQueryTool
 from .task_complete_tool import TaskCompleteTool
 from .tool import Tool, ToolArgument
 from .unified_diff_tool import UnifiedDiffTool
@@ -56,5 +59,8 @@ __all__ = [
     "EditWholeContentTool",
     "JinjaTool",
     "LLMImageGenerationTool",
-    "ReadHTMLTool"
+    "ReadHTMLTool",
+    "GrepAppTool",
+    "GenerateDatabaseReportTool",
+    'SQLQueryTool'
 ]

quantalogic/tools/execute_bash_command_tool.py

@@ -1,14 +1,18 @@
-"""Tool for executing bash commands and capturing their output."""
+"""Tool for executing bash commands with interactive input support."""
 
 import os
+import pty
+import select
+import signal
 import subprocess
+import sys
 from typing import Dict, Optional, Union
 
 from quantalogic.tools.tool import Tool, ToolArgument
 
 
 class ExecuteBashCommandTool(Tool):
-    """Tool for executing bash commands and capturing their output."""
+    """Tool for executing bash commands with real-time I/O handling."""
 
     name: str = "execute_bash_tool"
     description: str = "Executes a bash command and returns its output."
@@ -35,7 +39,6 @@ class ExecuteBashCommandTool(Tool):
             required=False,
             example="60",
         ),
-        # Removed the `env` argument from ToolArgument since it doesn't support `dict` type
     ]
 
     def execute(
@@ -45,72 +48,86 @@ class ExecuteBashCommandTool(Tool):
         timeout: Union[int, str, None] = 60,
         env: Optional[Dict[str, str]] = None,
     ) -> str:
-        """Executes a bash command and returns its output.
-
-        Args:
-            command (str): The bash command to execute.
-            working_dir (str, optional): Working directory for command execution. Defaults to the current directory.
-            timeout (int or str, optional): Maximum execution time in seconds. Defaults to 60 seconds.
-            env (dict, optional): Environment variables to set for the command execution. Defaults to the current environment.
-
-        Returns:
-            str: The command output or error message.
-
-        Raises:
-            subprocess.TimeoutExpired: If the command execution exceeds the timeout.
-            subprocess.CalledProcessError: If the command returns a non-zero exit status.
-            ValueError: If the timeout cannot be converted to an integer.
-        """
-        # Convert timeout to integer, defaulting to 60 if None or invalid
-        try:
-            timeout_seconds = int(timeout) if timeout else 60
-        except (ValueError, TypeError):
-            timeout_seconds = 60
-
-        # Use the current working directory if no working directory is specified
-        cwd = working_dir if working_dir else os.getcwd()
-
-        # Use the current environment if no custom environment is specified
+        """Executes a bash command with interactive input handling."""
+        timeout_seconds = int(timeout) if timeout else 60
+        cwd = working_dir or os.getcwd()
         env_vars = os.environ.copy()
         if env:
             env_vars.update(env)
 
         try:
-            # Execute the command with specified timeout, working directory, and environment
-            result = subprocess.run(
+            master, slave = pty.openpty()
+            proc = subprocess.Popen(
                 command,
                 shell=True,
+                stdin=slave,
+                stdout=slave,
+                stderr=subprocess.STDOUT,
                 cwd=cwd,
-                capture_output=True,
-                text=True,
-                timeout=timeout_seconds,
                 env=env_vars,
+                preexec_fn=os.setsid,
+                close_fds=True,
             )
-
-            formated_result = (
+            os.close(slave)
+
+            stdout_buffer = []
+            break_loop = False
+
+            try:
+                while True:
+                    rlist, _, _ = select.select([master, sys.stdin], [], [], timeout_seconds)
+                    if not rlist:
+                        if proc.poll() is not None:
+                            break  # Process completed but select timed out
+                        raise subprocess.TimeoutExpired(command, timeout_seconds)
+                    
+                    for fd in rlist:
+                        if fd == master:
+                            data = os.read(master, 1024).decode()
+                            if not data:
+                                break_loop = True
+                                break
+                            stdout_buffer.append(data)
+                            sys.stdout.write(data)
+                            sys.stdout.flush()
+                        elif fd == sys.stdin:
+                            user_input = os.read(sys.stdin.fileno(), 1024)
+                            os.write(master, user_input)
+                    
+                    # Check if process completed or EOF received
+                    if break_loop or proc.poll() is not None:
+                        # Read any remaining output
+                        while True:
+                            data = os.read(master, 1024).decode()
+                            if not data:
+                                break
+                            stdout_buffer.append(data)
+                            sys.stdout.write(data)
+                            sys.stdout.flush()
+                        break
+
+            except subprocess.TimeoutExpired:
+                os.killpg(os.getpgid(proc.pid), signal.SIGTERM)
+                return f"Command timed out after {timeout_seconds} seconds."
+            except EOFError:
+                pass  # Process exited normally
+            finally:
+                os.close(master)
+                proc.wait()
+
+            stdout_content = ''.join(stdout_buffer)
+            return_code = proc.returncode
+            formatted_result = (
                 "<command_output>"
-                f" <stdout>"
-                f"{result.stdout.strip()}"
-                f" </stdout>"
-                f" <stderr>"
-                f"{result.stderr.strip()}"
-                f" </stderr>"
-                f" <returncode>"
-                f" {result.returncode}"
-                f" </returncode>"
+                f" <stdout>{stdout_content.strip()}</stdout>"
+                f" <returncode>{return_code}</returncode>"
                 f"</command_output>"
             )
-
-            return formated_result
-
-        except subprocess.TimeoutExpired:
-            return f"Command timed out after {timeout_seconds} seconds."
-        except subprocess.CalledProcessError as e:
-            return f"Command failed with error: {e.stderr.strip()}"
+            return formatted_result
         except Exception as e:
             return f"Unexpected error executing command: {str(e)}"
 
 
 if __name__ == "__main__":
     tool = ExecuteBashCommandTool()
-    print(tool.to_markdown())
+    print(tool.to_markdown())

quantalogic/tools/generate_database_report_tool.py

@@ -0,0 +1,52 @@
+"""Tool for generating comprehensive database documentation reports."""
+
+from pydantic import Field, ValidationError
+
+from quantalogic.tools.tool import Tool
+from quantalogic.tools.utils.generate_database_report import generate_database_report
+
+
+class GenerateDatabaseReportTool(Tool):
+    """Tool for generating database documentation reports from a connection string."""
+
+    name: str = "generate_database_report_tool"
+    description: str = (
+        "Generates a comprehensive Markdown database documentation report with ER diagram. "
+    )
+    arguments: list = []  # No execution arguments - connection string is configured during tool setup
+    connection_string: str = Field(
+        ...,
+        description="SQLAlchemy-compatible database connection string (e.g., 'sqlite:///database.db')",
+        example="postgresql://user:password@localhost/mydatabase"
+    )
+
+    def execute(self) -> str:
+        """Generates a database documentation report using the configured connection string.
+
+        Returns:
+            str: Markdown-formatted database report
+
+        Raises:
+            ValueError: For invalid connection strings or database connection errors
+            RuntimeError: For errors during report generation
+        """
+        try:
+            return generate_database_report(self.connection_string)
+        except ValidationError as e:
+            raise ValueError(f"Invalid connection configuration: {e}") from e
+        except Exception as e:
+            raise RuntimeError(f"Database report generation failed: {e}") from e
+
+
+if __name__ == "__main__":
+
+    from quantalogic.tools.utils.create_sample_database import create_sample_database
+
+    # Create and document sample database
+    create_sample_database("sample.db")
+
+    # Example usage
+    tool = GenerateDatabaseReportTool(
+        connection_string="sqlite:///sample.db"
+    )
+    print(tool.execute())