QA Using Gemini Langchain ChromaDB PDF
QA Using Gemini Langchain ChromaDB PDF
import os
#os.environ["GOOGLE_GENAI_CLIENT_INFO_KEY"] = ""
os.environ["GOOGLE_API_KEY"] = "AIzaSyBc1M6tEFkimB8AuQcSHNqluPW5E3Wc6Ys"
!wget -q https://www.dropbox.com/s/zoj9rnm7oyeaivb/new_papers.zip
!unzip -q new_papers.zip -d new_papers
retriever = vectordb.as_retriever()
docs = retriever.get_relevant_documents("What is Flash attention?")
len(docs)
retriever = vectordb.as_retriever(search_kwargs={"k": 3})
retriever.search_type
retriever.search_kwargs
## Cite sources
import textwrap
return wrapped_text
def process_llm_response(llm_response):
print(wrap_text_preserve_newlines(llm_response['result']))
print('\n\nSources:')
for source in llm_response["source_documents"]:
print(source.metadata['source'])