main
Kye 2 years ago
parent 44590f9857
commit 20d9488936

@ -58,7 +58,7 @@ index = faiss.IndexFlatL2(embedding_size)
vectorstore = FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {})
# ---------- Worker Node ----------
@tool("search", return_direct=True)
@tool
class WorkerNode:
def __init__(self, llm, tools, vectorstore):
self.llm = llm

Loading…
Cancel
Save