File size: 3,843 Bytes
dc893fb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 |
"""Example 1: Basic Tools Usage
This example demonstrates how to use the basic tools:
- ReadTool: Read file contents
- WriteTool: Create new files
- EditTool: Edit existing files
- BashTool: Execute bash commands
Based on: tests/test_tools.py
"""
import asyncio
import tempfile
from pathlib import Path
from mini_agent.tools import BashTool, EditTool, ReadTool, WriteTool
async def demo_write_tool():
"""Demo: Write a new file."""
print("\n" + "=" * 60)
print("Demo 1: WriteTool - Create a new file")
print("=" * 60)
with tempfile.TemporaryDirectory() as tmpdir:
file_path = Path(tmpdir) / "hello.txt"
tool = WriteTool()
result = await tool.execute(
path=str(file_path), content="Hello, Mini Agent!\nThis is a test file."
)
if result.success:
print(f"✅ File created: {file_path}")
print(f"Content:\n{file_path.read_text()}")
else:
print(f"❌ Failed: {result.error}")
async def demo_read_tool():
"""Demo: Read a file."""
print("\n" + "=" * 60)
print("Demo 2: ReadTool - Read file contents")
print("=" * 60)
with tempfile.NamedTemporaryFile(mode="w", delete=False, suffix=".txt") as f:
f.write("Line 1: Hello\nLine 2: World\nLine 3: Mini Agent")
temp_path = f.name
try:
tool = ReadTool()
result = await tool.execute(path=temp_path)
if result.success:
print(f"✅ File read successfully")
print(f"Content:\n{result.content}")
else:
print(f"❌ Failed: {result.error}")
finally:
Path(temp_path).unlink()
async def demo_edit_tool():
"""Demo: Edit an existing file."""
print("\n" + "=" * 60)
print("Demo 3: EditTool - Edit file content")
print("=" * 60)
with tempfile.NamedTemporaryFile(mode="w", delete=False, suffix=".txt") as f:
f.write("Python is great!\nI love Python programming.")
temp_path = f.name
try:
print(f"Original content:\n{Path(temp_path).read_text()}\n")
tool = EditTool()
result = await tool.execute(
path=temp_path, old_str="Python", new_str="Agent"
)
if result.success:
print(f"✅ File edited successfully")
print(f"New content:\n{Path(temp_path).read_text()}")
else:
print(f"❌ Failed: {result.error}")
finally:
Path(temp_path).unlink()
async def demo_bash_tool():
"""Demo: Execute bash commands."""
print("\n" + "=" * 60)
print("Demo 4: BashTool - Execute bash commands")
print("=" * 60)
tool = BashTool()
# Example 1: List files
print("\nCommand: ls -la")
result = await tool.execute(command="ls -la")
if result.success:
print(f"✅ Command executed successfully")
print(f"Output:\n{result.content[:200]}...")
# Example 2: Get current directory
print("\nCommand: pwd")
result = await tool.execute(command="pwd")
if result.success:
print(f"✅ Current directory: {result.content.strip()}")
# Example 3: Echo
print("\nCommand: echo 'Hello from BashTool!'")
result = await tool.execute(command="echo 'Hello from BashTool!'")
if result.success:
print(f"✅ Output: {result.content.strip()}")
async def main():
"""Run all demos."""
print("=" * 60)
print("Basic Tools Usage Examples")
print("=" * 60)
print("\nThese examples show how to use the core tools directly.")
print("In a real agent scenario, the LLM decides which tools to use.\n")
await demo_write_tool()
await demo_read_tool()
await demo_edit_tool()
await demo_bash_tool()
print("\n" + "=" * 60)
print("All demos completed! ✅")
print("=" * 60)
if __name__ == "__main__":
asyncio.run(main())
|