File size: 478 Bytes
06821d7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from langchain.tools import StructuredTool
from pydantic.v1 import BaseModel
def write_report(filename, html):
with open(filename, 'w') as f:
f.write(html)
class WriteReportArgsSchema(BaseModel):
filename: str
html: str
write_report_tool = StructuredTool.from_function(
name="write_report",
description="Write an HTML file to disk. Use this tool whenever someone asks for a report.",
func=write_report,
args_schema=WriteReportArgsSchema
) |