Merge pull request #34 from dbravender/fix-cookbook-rag
Remove get_simple_embedding from node.py for pocketflow-rag example
This commit is contained in:
commit
92fa92685c
|
|
@ -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):
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue