Mehrdad Esmaeili
commited on
Commit
•
ef79e1c
1
Parent(s):
2600c4e
Update app.py
Browse files
app.py
CHANGED
|
@@ -51,10 +51,10 @@ retriever=docsearch.as_retriever()
|
|
| 51 |
cohereLLM=Cohere(model='command')
|
| 52 |
# cohereLLM=OpenAI()
|
| 53 |
# Initialize the CohereRerank compressor and the ContextualCompressionRetriever
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
|
| 57 |
-
|
| 58 |
|
| 59 |
# delete this to return to production state
|
| 60 |
memory=ConversationSummaryMemory(
|
|
|
|
| 51 |
cohereLLM=Cohere(model='command')
|
| 52 |
# cohereLLM=OpenAI()
|
| 53 |
# Initialize the CohereRerank compressor and the ContextualCompressionRetriever
|
| 54 |
+
compressor = CohereRerank(user_agent='MyTool/1.0 (Linux; x86_64)')
|
| 55 |
+
compression_retriever = ContextualCompressionRetriever(
|
| 56 |
+
base_compressor=compressor, base_retriever=retriever
|
| 57 |
+
)
|
| 58 |
|
| 59 |
# delete this to return to production state
|
| 60 |
memory=ConversationSummaryMemory(
|