diff --git a/cookbook/pocketflow-rag/nodes.py b/cookbook/pocketflow-rag/nodes.py index 4fdbcc2..e0b3cdd 100644 --- a/cookbook/pocketflow-rag/nodes.py +++ b/cookbook/pocketflow-rag/nodes.py @@ -1,7 +1,7 @@ from pocketflow import Node, Flow, BatchNode import numpy as np import faiss -from utils import call_llm, get_embedding, get_simple_embedding, fixed_size_chunk +from utils import call_llm, get_embedding, fixed_size_chunk # Nodes for the offline flow class ChunkDocumentsNode(BatchNode): @@ -140,4 +140,4 @@ Answer: shared["generated_answer"] = exec_res print("\n🤖 Generated Answer:") print(exec_res) - return "default" \ No newline at end of file + return "default"