add jina embeddings and reranker

yijinaembed.py

@@ -20,8 +20,8 @@ import spaces
 from langchain.retrievers.contextual_compression import ContextualCompressionRetriever
 from langchain_community.document_compressors.jina_rerank import JinaRerank
 from langchain import hub
-from langchain.chains import create_retrieval_chain
-from langchain.chains.retrieval import create_stuff_documents_chain
+from langchain.chains.retrieval import create_retrieval_chain
+from langchain.chains.combine_documents.stuff  import create_stuff_documents_chain
 
 load_dotenv()
 
@@ -140,7 +140,7 @@ def add_documents_to_chroma(documents: list, embedding_function: JinaEmbeddingFu
 @spaces.GPU
 def rerank_documents(query: str, documents: List[str]) -> List[str]:
     compressor = JinaRerank()
-    retriever = chroma_db.as_retriever(search_kwargs={"k": 20})
+    retriever = chroma_db.as_retriever(search_kwargs={"k": 15})
     compression_retriever = ContextualCompressionRetriever(
         base_compressor=compressor, base_retriever=retriever
     )
@@ -153,7 +153,7 @@ def query_chroma(query_text: str, embedding_function: JinaEmbeddingFunction):
     query_embeddings, query_metadata = embedding_function.compute_embeddings(query_text)
     result_docs = chroma_collection.query(
         query_embeddings=[query_embeddings],
-        n_results=3
+        n_results=5
     )
     return result_docs