insurance_advisor_wb / test_this.py
Asaad Almutareb
added ReAct agent with its tools
fb95c43
raw
history blame
1.02 kB
from rag_app.loading_data.load_urls_recurisvely import load_docs_from_urls
from rag_app.create_embedding import create_embeddings
from rag_app.generate_summary import generate_description, generate_keywords
from rag_app.build_vector_store import build_vector_store
# 1. load the urls
# 2. build the vectorstore -> the function will create the chunking and embeddings
# 3. initialize the db retriever
# 4.
# docs = load_docs_from_urls(["https://www.wuerttembergische.de/"],6)
# # for doc in docs:
# # keywords=generate_keywords(doc)
# # description=generate_description(doc)
# # doc.metadata['keywords']=keywords
# # doc.metadata['description']=description
# # print(doc.metadata)
# build_vector_store(docs, './vectorstore/faiss-insurance-agent-1500','sentence-transformers/multi-qa-mpnet-base-dot-v1',True,1500,150)
# print(create_embeddings(docs))
from rag_app.loading_data.scrap_website import scrap_website
print(scrap_website(target_url='https://www.wuerttembergische.de/',depth=1))