Create markdown_writer.py

tools/markdown_writer.py

@@ -0,0 +1,33 @@
+import datetime
+import os
+from typing import Any, Dict, Union
+
+from smolagents.tools import Tool
+
+
+class MarkdownWriterTool(Tool):
+    name = "markdown_writer"
+    description = """Writes the given text content, which should be a summary in Markdown format, to a Markdown file with a timestamped filename.
+    The assistant should provide a summary as text in Markdown format.
+    Args:
+        markdown_content: The text content in Markdown format to be written to the file.
+    """
+    inputs = {"markdown_content": {"type": "string", "description": "The Markdown content to write to the file."}}
+    output_type = "string"
+
+    def forward(self, markdown_content: str) -> str:
+        """Writes the given Markdown content to a file with a timestamped name."""
+        timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+        filename = f"summary_{timestamp}.md"
+        filepath = os.path.join(os.getcwd(), filename)  # Save in the current working directory
+
+        try:
+            with open(filepath, "w", encoding="utf-8") as f:
+                f.write(markdown_content)
+            return f"Successfully wrote content to {filepath}"
+        except Exception as e:
+            return f"Error writing to file: {str(e)}"
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False
+