Former-commit-id: dfbc6f1942
group-chat
Kye 2 years ago
parent 985a55ea14
commit a8be028ca9

@ -131,7 +131,9 @@ class Worker:
temperature=0.5) temperature=0.5)
``` ```
""" """
query_website_tool = WebpageQATool(qa_chain=load_qa_with_sources_chain(self.llm)) query_website_tool = WebpageQATool(
qa_chain=load_qa_with_sources_chain(self.llm)
)
self.tools = [ self.tools = [
WriteFileTool(root_dir=ROOT_DIR), WriteFileTool(root_dir=ROOT_DIR),
@ -155,7 +157,12 @@ class Worker:
embedding_size = 1536 embedding_size = 1536
index = faiss.IndexFlatL2(embedding_size) index = faiss.IndexFlatL2(embedding_size)
self.vectorstore = FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {}) self.vectorstore = FAISS(
embeddings_model.embed_query,
index,
InMemoryDocstore({}), {}
)
except Exception as error: except Exception as error:
raise RuntimeError(f"Error setting up memory perhaps try try tuning the embedding size: {error}") raise RuntimeError(f"Error setting up memory perhaps try try tuning the embedding size: {error}")

Loading…
Cancel
Save