phowell
/
pocketflow
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remove get_simple_embedding from node.py for pocketflow-rag example

This commit is contained in:
Dan Bravender 2025-04-20 16:42:43 -04:00
parent 7b97105438
commit 0e72aa3526
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cookbook/pocketflow-rag/nodes.py
View File

@ -1,7 +1,7 @@
from pocketflow import Node, Flow, BatchNode from pocketflow import Node, Flow, BatchNode
import numpy as np import numpy as np
import faiss 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 # Nodes for the offline flow
class ChunkDocumentsNode(BatchNode): class ChunkDocumentsNode(BatchNode):
@ -140,4 +140,4 @@ Answer:
shared["generated_answer"] = exec_res shared["generated_answer"] = exec_res
print("\n🤖 Generated Answer:") print("\n🤖 Generated Answer:")
print(exec_res) print(exec_res)
return "default" return "default"