quantalogic 0.61.0__py3-none-any.whl → 0.61.2__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/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/utils/generate_database_report.py

@@ -1,4 +1,4 @@
-from datetime import UTC, datetime
+from datetime import timezone, datetime  # noqa: I001
 from typing import Dict, List
 
 import networkx as nx
@@ -127,7 +127,7 @@ def generate_markdown_report(
     md.append(f"**Database Type**: {db_metadata['dialect'].capitalize()}\n")
     md.append(f"**Database Name**: {db_metadata['name']}\n")
     md.append(f"**Total Tables**: {len(db_metadata['tables'])}\n")
-    md.append(f"**Generated At**: {datetime.now(UTC).strftime('%Y-%m-%d %H:%M:%S UTC')}\n\n")
+    md.append(f"**Generated At**: {datetime.now(timezone.utc).strftime('%Y-%m-%d %H:%M:%S UTC')}\n\n")
 
     # ERD Section
     md.append("## Entity Relationship Diagram\n")

quantalogic/tools/write_file_tool.py

@@ -4,6 +4,8 @@ import os
 from pathlib import Path
 
 from loguru import logger
+from pydantic import Field
+
 from quantalogic.tools.tool import Tool, ToolArgument
 
 
@@ -13,41 +15,45 @@ class WriteFileTool(Tool):
     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."
     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,7 +105,8 @@ 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)
+            if not self.disable_ensure_tmp_path:
+                file_path = self._ensure_tmp_path(file_path)
 
             # Ensure parent directory exists (only within /tmp)
             parent_dir = os.path.dirname(file_path)

{quantalogic-0.61.0.dist-info → quantalogic-0.61.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: quantalogic
-Version: 0.61.0
+Version: 0.61.2
 Summary: QuantaLogic ReAct Agents
 Author: Raphaël MANSUY
 Author-email: raphael.mansuy@gmail.com
@@ -10,6 +10,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: aiofiles (>=24.1.0,<25.0.0)
 Requires-Dist: beautifulsoup4 (>=4.12.3,<5.0.0)
 Requires-Dist: boto3 (>=1.34.0,<2.0.0)
 Requires-Dist: botocore (>=1.29.123,<2.0.0)
@@ -21,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)
@@ -387,7 +388,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.
 

{quantalogic-0.61.0.dist-info → quantalogic-0.61.2.dist-info}/RECORD

@@ -1,6 +1,6 @@
 quantalogic/__init__.py,sha256=qFbvfHOd_chAu536pH816E3uo6CdyAgXCpQOwMXXVnY,1076
 quantalogic/agent.py,sha256=VChZXFLEsIIrBtXVQZ-FGZ72GCUXDL3g9br8Vo1t5V8,75072
-quantalogic/agent_config.py,sha256=65dvgJ1rwC0eEysMtTv_Dsjln4WS0IvAIcn6r39Zq4k,8222
+quantalogic/agent_config.py,sha256=HNXHmu3BiT4q9i-L2wJKUh6PuiZ_ryTNaB7ga24kaJo,8334
 quantalogic/agent_factory.py,sha256=soActorasOqs5g6NmlyeEjRYbJceIGGg7wuUS-vA898,6878
 quantalogic/codeact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 quantalogic/codeact/agent.py,sha256=oajib7OsfBp230sg_IjmknqYsFz_lSURC39fj5brsP4,22454
@@ -14,7 +14,7 @@ quantalogic/codeact/prompts/generate_program.j2,sha256=05_1bAwIJLLBKt4o9c7sRWLHI
 quantalogic/codeact/prompts/response_format.j2,sha256=TwO43dEG-3justNigpX4yyzGR1TGS3YDlpJQq8Z5Vf4,355
 quantalogic/codeact/tools_manager.py,sha256=l2cpHLuQP-MW0uP8d2e6IdMToeveaIOdilwDGPjPciY,4464
 quantalogic/codeact/utils.py,sha256=B7Xowk4id6Sgoyl0eBJsi6JiKa-BAwnj0a5z70vrV1M,5308
-quantalogic/coding_agent.py,sha256=i4QYY1Q7AXfJFCVfrowxukH33nuqByUOX6bt-hAcGDE,5504
+quantalogic/coding_agent.py,sha256=WFfabRwwPZFV3Pw3seLKpSrFE9Li4pz8Z8mCdsUIDi8,5532
 quantalogic/config.py,sha256=bmPI2rvQ9M8Zdl1H7K588JgJdnkxSE0L5_i5aBqsagA,564
 quantalogic/console_print_events.py,sha256=yDtfOr7s5r_gLTgwkl_XoKSkUqNRZhqqq4hwR_oJsUw,2050
 quantalogic/console_print_token.py,sha256=5IRVoPhwWZtSc4LpNoAsCQhCB_RnAW9chycGgyD3_5U,437
@@ -184,11 +184,11 @@ quantalogic/tools/utilities/selenium_tool.py,sha256=UrfQSF62UoADPuRPqP8x9_jIL6Yy
 quantalogic/tools/utilities/vscode_tool.py,sha256=M248_464I3FWHuvrTVPbrF6czXe0n45_o315TFRD-6U,4363
 quantalogic/tools/utils/__init__.py,sha256=-NtMSwxRt_G79Oo_DcDaCdLU2vLvuXIoCd34TOYEUmI,334
 quantalogic/tools/utils/create_sample_database.py,sha256=h5c_uxv3eztQvHlloTZxzWt5gEzai8zfnR8-_QrUqxU,3724
-quantalogic/tools/utils/generate_database_report.py,sha256=3PT34ClMvZ2O62-24cp_5lOyZHY_pBjVObMHpfyVi-s,10140
+quantalogic/tools/utils/generate_database_report.py,sha256=IU_XGTDNXfJXxzpHR1F4cAV5ggmiTs2hj-WyfWRe9Tk,10168
 quantalogic/tools/web_navigation/__init__.py,sha256=O7SkVqbGwN4zt8Sm3H8AHF9451FSgI5h0J3fDj1rFS4,142
 quantalogic/tools/web_navigation/web_tool.py,sha256=AxAxQLUNwCElqxP2ceOOjHVY80ck-Md-uNsjHdR9ErA,4721
 quantalogic/tools/wikipedia_search_tool.py,sha256=LXQSPH8961Efw2QNxKe-cD5ZiIYD3ufEgrxH4y5uB74,5180
-quantalogic/tools/write_file_tool.py,sha256=qyRhDwE77rakRb_CuAC00V0E2vqIRNxwKD4PHLGgyW8,5230
+quantalogic/tools/write_file_tool.py,sha256=aevO9koB1Gz96PJCu-hQuW014qTzrnoSpcMc6eke5mI,5583
 quantalogic/utils/__init__.py,sha256=E442CJQuTohKzgI0Wrd4NZEpKascFjz6F4Vy8Y1c_0Y,634
 quantalogic/utils/ask_user_validation.py,sha256=kSr7TXPTpsLR9zgwpGWgvffx8-cKAC_rdFRdLqwC22A,1176
 quantalogic/utils/async_utils.py,sha256=FOizWRbHdsZwoD36dNErzunfwPlE7zDprS6RXcWuWSo,963
@@ -209,8 +209,8 @@ quantalogic/version_check.py,sha256=JyQFTNMDWtpHCLnN-BiakzB2cyXf6kUFsTjvmSruZi4,
 quantalogic/welcome_message.py,sha256=o4tHdgabNuIV9kbIDPgS3_2yzJhayK30oKad2UouYDc,3020
 quantalogic/xml_parser.py,sha256=bLLwIwO-VEHWF3heNS7nuPC8wgdYw9F_fVZZNW1figY,11728
 quantalogic/xml_tool_parser.py,sha256=hGHA1q20JUoTNTbZYmi4FTdA5I25-AGEIP8DwZgQCNA,3897
-quantalogic-0.61.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-quantalogic-0.61.0.dist-info/METADATA,sha256=htzdPnOadswC70Yh7pfSyOVhsM364yrtwOZufiMJSCc,32955
-quantalogic-0.61.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-quantalogic-0.61.0.dist-info/entry_points.txt,sha256=h74O_Q3qBRCrDR99qvwB4BpBGzASPUIjCfxHq6Qnups,183
-quantalogic-0.61.0.dist-info/RECORD,,
+quantalogic-0.61.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+quantalogic-0.61.2.dist-info/METADATA,sha256=38TzrZH-h5sMLtIgRGcMlOdTBXtGh-tSndq7AeFmJxA,32975
+quantalogic-0.61.2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+quantalogic-0.61.2.dist-info/entry_points.txt,sha256=h74O_Q3qBRCrDR99qvwB4BpBGzASPUIjCfxHq6Qnups,183
+quantalogic-0.61.2.dist-info/RECORD,,