Python_Data_Driven_ChatBot / note_engine.py
shukdevdatta123's picture
Upload 2 files
738b789 verified
raw
history blame contribute delete
478 Bytes
from llama_index.core.tools import FunctionTool
import os
note_file = os.path.join("data", "notes.txt")
def save_note(note):
if not os.path.exists(note_file):
open(note_file, "w")
with open(note_file, "a") as f:
f.writelines([note + "\n"])
return "note saved"
note_engine = FunctionTool.from_defaults(
fn=save_note,
name="note_saver",
description="this tool can save a text based note to a file for the user",
)