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({}), {}) vectorstore = FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {})
# ---------- Worker Node ---------- # ---------- Worker Node ----------
@tool("search", return_direct=True) @tool
class WorkerNode: class WorkerNode:
def __init__(self, llm, tools, vectorstore): def __init__(self, llm, tools, vectorstore):
self.llm = llm self.llm = llm

Loading…
Cancel
Save