taaha3244 commited on
Commit
11a1fa9
1 Parent(s): 4e5ac28

Update retrieve.py

Browse files
Files changed (1) hide show
  1. retrieve.py +3 -0
retrieve.py CHANGED
@@ -5,6 +5,9 @@ from langchain_core.runnables import RunnablePassthrough
5
  from langchain.prompts import PromptTemplate
6
  from utils import setup_openai_embeddings,setup_qdrant_client,openai_llm,format_document_metadata
7
  from dotenv import load_dotenv
 
 
 
8
  from langfuse.callback import CallbackHandler
9
  load_dotenv()
10
 
 
5
  from langchain.prompts import PromptTemplate
6
  from utils import setup_openai_embeddings,setup_qdrant_client,openai_llm,format_document_metadata
7
  from dotenv import load_dotenv
8
+ from langchain.retrievers.contextual_compression import ContextualCompressionRetriever
9
+ from langchain_cohere import CohereRerank
10
+ from langchain_community.llms import Cohere
11
  from langfuse.callback import CallbackHandler
12
  load_dotenv()
13