quantalogic 0.61.1__py3-none-any.whl → 0.61.3__py3-none-any.whl

quantalogic/agent_config.py

@@ -75,7 +75,7 @@ def create_agent(
         TaskCompleteTool(),
         ReadFileTool(),
         ReadFileBlockTool(),
-        WriteFileTool(),
+        WriteFileTool(disable_ensure_tmp_path=True),
         EditWholeContentTool(),
         InputQuestionTool(),
         ListDirectoryTool(),
@@ -129,7 +129,7 @@ def create_interpreter_agent(
         TaskCompleteTool(),
         ReadFileTool(),
         ReadFileBlockTool(),
-        WriteFileTool(),
+        WriteFileTool(disable_ensure_tmp_path=True),
         EditWholeContentTool(),
         InputQuestionTool(),
         ListDirectoryTool(),
@@ -176,7 +176,7 @@ def create_full_agent(
         TaskCompleteTool(),
         ReadFileTool(),
         ReadFileBlockTool(),
-        WriteFileTool(),
+        WriteFileTool(disable_ensure_tmp_path=True),
         EditWholeContentTool(),
         InputQuestionTool(),
         ListDirectoryTool(),
@@ -237,7 +237,7 @@ def create_basic_agent(
         SearchDefinitionNamesTool(),
         ReadFileTool(),
         ReplaceInFileTool(),
-        WriteFileTool(),
+        WriteFileTool(disable_ensure_tmp_path=True),
         EditWholeContentTool(),
         ReplaceInFileTool(),
         InputQuestionTool(),

quantalogic/codeact/agent.py

@@ -5,8 +5,8 @@ from typing import Callable, Dict, List, Optional, Tuple
 from jinja2 import Environment, FileSystemLoader
 from loguru import logger
 from lxml import etree
+from quantalogic_pythonbox import execute_async
 
-from quantalogic.python_interpreter import execute_async
 from quantalogic.tools import Tool
 
 from .constants import MAX_GENERATE_PROGRAM_TOKENS, MAX_HISTORY_TOKENS, MAX_TOKENS, TEMPLATE_DIR

quantalogic/codeact/cli.py

@@ -4,11 +4,9 @@ from typing import Callable, List, Optional, Union
 import typer
 from loguru import logger
 
-from quantalogic.tools import create_tool
-
-from .agent import Agent  # Import only Agent from agent.py
-from .constants import DEFAULT_MODEL, LOG_FILE
-from .events import (  # Import events from events.py
+from quantalogic.codeact.agent import Agent  # Import only Agent from agent.py
+from quantalogic.codeact.constants import DEFAULT_MODEL, LOG_FILE
+from quantalogic.codeact.events import (  # Import events from events.py
     ActionExecutedEvent,
     ActionGeneratedEvent,
     ErrorOccurredEvent,
@@ -22,6 +20,8 @@ from .events import (  # Import events from events.py
     ToolExecutionErrorEvent,
     ToolExecutionStartedEvent,
 )
+from quantalogic.tools import create_tool
+
 from .tools_manager import Tool, get_default_tools
 from .utils import XMLResultHandler
 

quantalogic/codeact/tools_manager.py

@@ -130,6 +130,6 @@ def get_default_tools(model: str) -> List[Tool]:
         # RipgrepTool(), 
         # SearchDefinitionNamesTool(),
         #TaskCompleteTool(), 
-        WriteFileTool(), 
+        WriteFileTool(disable_ensure_tmp_path=True),
         AgentTool(model=model),
     ]

quantalogic/coding_agent.py

@@ -65,7 +65,7 @@ def create_coding_agent(
         # Core file manipulation tools
         TaskCompleteTool(),  # Marks task completion
         ReadFileBlockTool(),  # Reads specific file sections
-        WriteFileTool(),  # Creates new files
+        WriteFileTool(disable_ensure_tmp_path=True),  # Creates new files
         ReplaceInFileTool(),  # Updates file sections
         EditWholeContentTool(),  # Modifies entire files
         # Code navigation and search tools

quantalogic/tools/action_gen_safe.py

@@ -0,0 +1,340 @@
+import ast
+import asyncio
+from functools import partial
+from typing import Callable, Dict, List
+
+import litellm
+import typer
+from loguru import logger
+
+from quantalogic.tools.tool import Tool, ToolArgument
+from quantalogic.utils.python_interpreter import ASTInterpreter
+
+# Configure loguru to log to a file with rotation
+logger.add("action_gen_safe.log", rotation="10 MB", level="DEBUG")
+
+# Initialize Typer app
+app = typer.Typer()
+
+# Define tool classes with logging in async_execute
+class AddTool(Tool):
+    def __init__(self):
+        super().__init__(
+            name="add_tool",
+            description="Adds two numbers and returns the sum.",
+            arguments=[
+                ToolArgument(name="a", arg_type="int", description="First number", required=True),
+                ToolArgument(name="b", arg_type="int", description="Second number", required=True)
+            ],
+            return_type="int"
+        )
+    
+    async def async_execute(self, **kwargs) -> str:
+        logger.info(f"Adding {kwargs['a']} and {kwargs['b']}")
+        return str(int(kwargs["a"]) + int(kwargs["b"]))
+
+class MultiplyTool(Tool):
+    def __init__(self):
+        super().__init__(
+            name="multiply_tool",
+            description="Multiplies two numbers and returns the product.",
+            arguments=[
+                ToolArgument(name="x", arg_type="int", description="First number", required=True),
+                ToolArgument(name="y", arg_type="int", description="Second number", required=True)
+            ],
+            return_type="int"
+        )
+    
+    async def async_execute(self, **kwargs) -> str:
+        logger.info(f"Multiplying {kwargs['x']} and {kwargs['y']}")
+        return str(int(kwargs["x"]) * int(kwargs["y"]))
+
+class ConcatTool(Tool):
+    def __init__(self):
+        super().__init__(
+            name="concat_tool",
+            description="Concatenates two strings.",
+            arguments=[
+                ToolArgument(name="s1", arg_type="string", description="First string", required=True),
+                ToolArgument(name="s2", arg_type="string", description="Second string", required=True)
+            ],
+            return_type="string"
+        )
+    
+    async def async_execute(self, **kwargs) -> str:
+        logger.info(f"Concatenating '{kwargs['s1']}' and '{kwargs['s2']}'")
+        return kwargs["s1"] + kwargs["s2"]
+
+class AgentTool(Tool):
+    def __init__(self, model: str = "gemini/gemini-2.0-flash"):
+        super().__init__(
+            name="agent_tool",
+            description="Generates text using a language model based on a system prompt and user prompt.",
+            arguments=[
+                ToolArgument(name="system_prompt", arg_type="string", description="System prompt to guide the model's behavior", required=True),
+                ToolArgument(name="prompt", arg_type="string", description="User prompt to generate a response for", required=True),
+                ToolArgument(name="temperature", arg_type="float", description="Temperature for generation (0 to 1)", required=True)
+            ],
+            return_type="string"
+        )
+        self.model = model
+    
+    async def async_execute(self, **kwargs) -> str:
+        system_prompt = kwargs["system_prompt"]
+        prompt = kwargs["prompt"]
+        temperature = float(kwargs["temperature"])
+        
+        # Validate temperature
+        if not 0 <= temperature <= 1:
+            logger.error(f"Temperature {temperature} is out of range (0-1)")
+            raise ValueError("Temperature must be between 0 and 1")
+        
+        logger.info(f"Generating text with model {self.model}, temperature {temperature}")
+        try:
+            response = await litellm.acompletion(
+                model=self.model,
+                messages=[
+                    {"role": "system", "content": system_prompt},
+                    {"role": "user", "content": prompt}
+                ],
+                temperature=temperature,
+                max_tokens=1000  # Reasonable default for text generation
+            )
+            generated_text = response.choices[0].message.content.strip()
+            logger.debug(f"Generated text: {generated_text}")
+            return generated_text
+        except Exception as e:
+            logger.error(f"Failed to generate text with {self.model}: {str(e)}")
+            raise RuntimeError(f"Text generation failed: {str(e)}")
+
+# Asynchronous function to generate the program
+async def generate_program(task_description: str, tools: List[Tool], model: str, max_tokens: int) -> str:
+    """
+    Asynchronously generate a Python program that solves a given task using a list of tools.
+
+    Args:
+        task_description (str): A description of the task to be solved.
+        tools (List[Tool]): A list of Tool objects available for use.
+        model (str): The litellm model to use for code generation.
+        max_tokens (int): Maximum number of tokens for the generated response.
+
+    Returns:
+        str: A string containing a complete Python program.
+    """
+    logger.debug(f"Generating program for task: {task_description}")
+    # Collect tool docstrings
+    tool_docstrings = "\n\n".join([tool.to_docstring() for tool in tools])
+
+    # Construct the prompt for litellm
+    prompt = f"""
+You are a Python code generator. Your task is to create a Python program that solves the following task:
+"{task_description}"
+
+You have access to the following pre-defined async tool functions, as defined with their signatures and descriptions:
+
+{tool_docstrings}
+
+Instructions:
+1. Generate a Python program as a single string.
+2. Include only the import for asyncio (import asyncio).
+3. Define an async function named main() that solves the task.
+4. Use the pre-defined tool functions (e.g., add_tool, multiply_tool, concat_tool) directly by calling them with await and the appropriate arguments as specified in their descriptions.
+5. Do not redefine the tool functions within the program; assume they are already available in the namespace.
+6. Return the program as a string enclosed in triple quotes (\"\"\"program\"\"\")).
+7. Do not include asyncio.run(main()) or any code outside the main() function definition.
+8. Do not include explanatory text outside the program string.
+9. Express all string variables as multiline strings (\"\"\"\nstring\n\"\"\"), always start a string at the beginning of a line.
+10. Always print the result at the end of the program.
+11. Never call the main() function.
+12. Never use not defined variables, modules, or functions.
+
+Example task: "Add 5 and 7 and print the result"
+Example output:
+\"\"\"import asyncio
+async def main():
+    result = await add_tool(a=5, b=7)
+    print(result)
+\"\"\"
+"""
+    
+    logger.debug(f"Prompt sent to litellm:\n{prompt}")
+    
+    try:
+        # Call litellm asynchronously to generate the program
+        logger.debug(f"Calling litellm with model {model}")
+        response = await litellm.acompletion(
+            model=model,
+            messages=[
+                {"role": "system", "content": "You are a Python code generator."},
+                {"role": "user", "content": prompt}
+            ],
+            max_tokens=max_tokens,
+            temperature=0.3
+        )
+        generated_code = response.choices[0].message.content.strip()
+        logger.debug("Code generation successful")
+    except Exception as e:
+        logger.error(f"Failed to generate code: {str(e)}")
+        raise typer.BadParameter(f"Failed to generate code with model '{model}': {str(e)}")
+
+    # Clean up the output
+    if generated_code.startswith('"""') and generated_code.endswith('"""'):
+        generated_code = generated_code[3:-3]
+    elif generated_code.startswith("```python") and generated_code.endswith("```"):
+        generated_code = generated_code[9:-3].strip()
+    
+    return generated_code
+
+# Async core logic for generate
+async def generate_core(task: str, model: str, max_tokens: int):
+    logger.info(f"Starting generate command for task: {task}")
+    # Input validation
+    if not task.strip():
+        logger.error("Task description is empty")
+        raise typer.BadParameter("Task description cannot be empty")
+    if max_tokens <= 0:
+        logger.error("max-tokens must be positive")
+        raise typer.BadParameter("max-tokens must be a positive integer")
+
+    # Initialize tools
+    tools = [
+        AddTool(),
+        MultiplyTool(),
+        ConcatTool(),
+        AgentTool(model=model)  # Include AgentTool with the specified model
+    ]
+
+    # Generate the program
+    try:
+        program = await generate_program(task, tools, model, max_tokens)
+    except Exception as e:
+        logger.error(f"Failed to generate program: {str(e)}")
+        typer.echo(typer.style(f"Error: {str(e)}", fg=typer.colors.RED))
+        raise typer.Exit(code=1)
+
+    logger.debug(f"Generated program:\n{program}")
+    # Output the generated program
+    typer.echo(typer.style("Generated Python Program:", fg=typer.colors.GREEN, bold=True))
+    typer.echo(program)
+    
+    # Attempt to execute the program using the safe interpreter
+    typer.echo("\n" + typer.style("Executing the program safely:", fg=typer.colors.GREEN, bold=True))
+    try:
+        # Create instances of tools
+        add_tool_instance = AddTool()
+        multiply_tool_instance = MultiplyTool()
+        concat_tool_instance = ConcatTool()
+        agent_tool_instance = AgentTool(model=model)
+
+        # Define the allowed modules
+        allowed_modules = ["asyncio"]
+        
+        # Create a namespace with our tools
+        namespace = {
+            "add_tool": add_tool_instance.async_execute,
+            "multiply_tool": multiply_tool_instance.async_execute,
+            "concat_tool": concat_tool_instance.async_execute,
+            "agent_tool": agent_tool_instance.async_execute,
+        }
+        
+        # Parse the program to AST
+        program_ast = ast.parse(program)
+        
+        # Extract imports first to ensure they're allowed
+        for node in program_ast.body:
+            if isinstance(node, ast.Import):
+                for name in node.names:
+                    if name.name not in allowed_modules:
+                        raise ValueError(f"Import not allowed: {name.name}")
+            elif isinstance(node, ast.ImportFrom):
+                if node.module not in allowed_modules:
+                    raise ValueError(f"Import from not allowed: {node.module}")
+        
+        # Find the main function in the AST
+        main_func_node = None
+        for node in program_ast.body:
+            if isinstance(node, ast.AsyncFunctionDef) and node.name == "main":
+                main_func_node = node
+                break
+        
+        if not main_func_node:
+            logger.warning("No async main() function found in generated code")
+            typer.echo(typer.style("Warning: No async main() function found", fg=typer.colors.YELLOW))
+            return
+            
+        # Execute the code safely to define functions in the namespace
+        # Use a custom ASTInterpreter directly instead of calling interpret_code
+        interpreter = ASTInterpreter(allowed_modules=allowed_modules, source=program)
+        
+        # Add tools to the interpreter's environment
+        interpreter.env_stack[0].update({
+            "add_tool": add_tool_instance.async_execute,
+            "multiply_tool": multiply_tool_instance.async_execute,
+            "concat_tool": concat_tool_instance.async_execute,
+            "agent_tool": agent_tool_instance.async_execute,
+        })
+        
+        # First, execute all non-main code (imports and other top-level code)
+        for node in program_ast.body:
+            if not isinstance(node, ast.AsyncFunctionDef) or node.name != "main":
+                await interpreter.visit(node, wrap_exceptions=True)
+        
+        # Then execute the main function definition
+        for node in program_ast.body:
+            if isinstance(node, ast.AsyncFunctionDef) and node.name == "main":
+                await interpreter.visit(node, wrap_exceptions=True)
+        
+        # Get the main function from the environment
+        main_func = interpreter.get_variable("main")
+        
+        # Execute the main function in the current event loop
+        logger.debug("Executing main() function")
+        await main_func()
+        
+        logger.info("Program executed successfully")
+        typer.echo(typer.style("Execution completed successfully", fg=typer.colors.GREEN))
+    
+    except SyntaxError as e:
+        logger.error(f"Syntax error in generated code: {e}")
+        typer.echo(typer.style(f"Syntax error: {e}", fg=typer.colors.RED))
+    except Exception as e:
+        logger.error(f"Execution error: {e}")
+        typer.echo(typer.style(f"Execution failed: {e}", fg=typer.colors.RED))
+
+# Synchronous callback to invoke async generate_core
+@app.callback(invoke_without_command=True)
+def generate(
+    task: str = typer.Argument(
+        ...,
+        help="The task description to generate a program for (e.g., 'Add 5 and 7 and print the result')"
+    ),
+    model: str = typer.Option(
+        "gemini/gemini-2.0-flash",
+        "--model",
+        "-m",
+        help="The litellm model to use for generation (e.g., 'gpt-3.5-turbo', 'gpt-4')"
+    ),
+    max_tokens: int = typer.Option(
+        4000,
+        "--max-tokens",
+        "-t",
+        help="Maximum number of tokens for the generated response (default: 4000)"
+    )
+):
+    """
+    Asynchronously generate a Python program based on a task description using specified tools and model.
+    Executes the program in a controlled, safe environment.
+
+    Examples:
+        $ python action_gen_safe.py "Add 5 and 7 and print the result"
+        $ python action_gen_safe.py "Concatenate 'Hello' and 'World' and print it" --model gpt-4 --max-tokens 5000
+    """
+    asyncio.run(generate_core(task, model, max_tokens))
+
+# Entry point to start the app
+def main():
+    logger.debug("Starting script execution")
+    app()
+
+if __name__ == "__main__":
+    main()

quantalogic/tools/write_file_tool.py

@@ -4,50 +4,56 @@ import os
 from pathlib import Path
 
 from loguru import logger
+from pydantic import Field
+
 from quantalogic.tools.tool import Tool, ToolArgument
 
 
 class WriteFileTool(Tool):
-    """Tool for writing a text file in /tmp directory."""
+    """Tool for writing a text file to a specified path."""
 
     name: str = "write_file_tool"
-    description: str = "Writes a file with the given content in /tmp directory. The tool will fail if the file already exists when not used in append mode."
+    description: str = "Writes a file to the specified path. The tool will fail if the file already exists when not used in append mode."
     need_validation: bool = True
-    arguments: list = [
-        ToolArgument(
-            name="file_path",
-            arg_type="string",
-            description="The name of the file to write in /tmp directory. Can include subdirectories within /tmp.",
-            required=True,
-            example="/tmp/myfile.txt or myfile.txt",
-        ),
-        ToolArgument(
-            name="content",
-            arg_type="string",
-            description="""
-            The content to write to the file. Use CDATA to escape special characters.
-            Don't add newlines at the beginning or end of the content.
-            """,
-            required=True,
-            example="Hello, world!",
-        ),
-        ToolArgument(
-            name="append_mode",
-            arg_type="string",
-            description="""Append mode. If true, the content will be appended to the end of the file.
-            """,
-            required=False,
-            example="False",
-        ),
-        ToolArgument(
-            name="overwrite",
-            arg_type="string",
-            description="Overwrite mode. If true, existing files can be overwritten. Defaults to False.",
-            required=False,
-            example="False",
-            default="False",
-        ),
-    ]
+
+    disable_ensure_tmp_path: bool = Field(default=False)
+
+    arguments: list = Field(
+        default=[
+            ToolArgument(
+                name="file_path",
+                arg_type="string",
+                description="The path to the file to write. By default, paths will be forced to /tmp directory unless disable_ensure_tmp_path is enabled. Can include subdirectories.",
+                required=True,
+                example="/tmp/myfile.txt or ./myfile.txt",
+            ),
+            ToolArgument(
+                name="content",
+                arg_type="string",
+                description="""The content to write to the file. Use CDATA to escape special characters.
+                Don't add newlines at the beginning or end of the content.
+                """,
+                required=True,
+                example="Hello, world!",
+            ),
+            ToolArgument(
+                name="append_mode",
+                arg_type="string",
+                description="""Append mode. If true, the content will be appended to the end of the file.
+                """,
+                required=False,
+                example="False",
+            ),
+            ToolArgument(
+                name="overwrite",
+                arg_type="string",
+                description="Overwrite mode. If true, existing files can be overwritten. Defaults to False.",
+                required=False,
+                example="False",
+                default="False",
+            ),
+        ],
+    )
 
     def _ensure_tmp_path(self, file_path: str) -> str:
         """Ensures the file path is within /tmp directory.
@@ -99,12 +105,10 @@ class WriteFileTool(Tool):
             overwrite_bool = overwrite.lower() in ["true", "1", "yes"]
 
             # Ensure path is in /tmp and normalize it
-            file_path = self._ensure_tmp_path(file_path)
-
-            # Ensure parent directory exists (only within /tmp)
-            parent_dir = os.path.dirname(file_path)
-            if parent_dir.startswith("/tmp/"):
-                os.makedirs(parent_dir, exist_ok=True)
+            if not self.disable_ensure_tmp_path:
+                file_path = self._ensure_tmp_path(file_path)
+                if not file_path.startswith('/tmp/'):
+                    raise ValueError('File path must be under /tmp when disable_ensure_tmp_path is False')
 
             # Determine file write mode based on append_mode
             mode = "a" if append_mode_bool else "w"
@@ -115,6 +119,8 @@ class WriteFileTool(Tool):
                     f"File {file_path} already exists. Set append_mode=True to append or overwrite=True to overwrite."
                 )
 
+            os.makedirs(os.path.dirname(file_path), exist_ok=True)
+
             with open(file_path, mode, encoding="utf-8") as f:
                 f.write(content)
             

{quantalogic-0.61.1.dist-info → quantalogic-0.61.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: quantalogic
-Version: 0.61.1
+Version: 0.61.3
 Summary: QuantaLogic ReAct Agents
 Author: Raphaël MANSUY
 Author-email: raphael.mansuy@gmail.com
@@ -22,7 +22,7 @@ Requires-Dist: google-search-results (>=2.4.2,<3.0.0)
 Requires-Dist: html2text (>=2024.2.26,<2025.0.0)
 Requires-Dist: instructor (>=1.7.2,<2.0.0)
 Requires-Dist: jinja2 (>=3.1.5,<4.0.0)
-Requires-Dist: litellm (>=1.59.0,<2.0.0)
+Requires-Dist: litellm (>=1.63.14,<2.0.0)
 Requires-Dist: loguru (>=0.7.3,<0.8.0)
 Requires-Dist: markdownify (>=0.14.1,<0.15.0)
 Requires-Dist: markitdown (>=0.0.1a3,<0.0.2)
@@ -34,6 +34,7 @@ Requires-Dist: psutil (>=7.0.0,<8.0.0)
 Requires-Dist: pydantic (>=2.10.4,<3.0.0)
 Requires-Dist: pytest-asyncio (>=0.25.3,<0.26.0)
 Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
+Requires-Dist: quantalogic-pythonbox (>=0.8.3,<0.9.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
 Requires-Dist: rich (>=13.9.4,<14.0.0)
 Requires-Dist: serpapi (>=0.1.5,<0.2.0)
@@ -388,7 +389,7 @@ The **Flow module** is your architect—building workflows that hum with precisi
 
 🔍 **Want to dive deeper?** Check out our comprehensive [Workflow YAML DSL Specification](./quantalogic/flow/flow_yaml.md), a detailed guide that walks you through defining powerful, structured workflows. From basic node configurations to complex transition logic, this documentation is your roadmap to mastering workflow design with QuantaLogic.
 
-📚 **For a deeper understanding of Flow YAML and its applications, please refer to the official [Flow YAML Documentation](https://quantalogic.github.io/quantalogic/flow/flow_yaml).**
+📚 **For a deeper understanding of Flow YAML and its applications, please refer to the official [Flow YAMAL Documentation](./quantalogic/flow/flow_yaml.md).**
 
 The Flow YAML documentation provides a comprehensive overview of the Flow YAML language, including its syntax, features, and best practices. It's a valuable resource for anyone looking to create complex workflows with QuantaLogic.