Spaces:
Sleeping
Sleeping
Commit
·
b94a242
1
Parent(s):
85b88ff
Update main.py
Browse files
main.py
CHANGED
@@ -86,7 +86,6 @@ def to_cache(file):
|
|
86 |
|
87 |
@cl.cache
|
88 |
def retriever_to_cache():
|
89 |
-
from langchain.cache import InMemoryCache
|
90 |
index_name = os.environ['PINECONE_INDEX_NAME']
|
91 |
embeddings = HuggingFaceEmbeddings()
|
92 |
pinecone.init(
|
@@ -97,7 +96,6 @@ def retriever_to_cache():
|
|
97 |
index_name=index_name, embedding=embeddings
|
98 |
)
|
99 |
retriever = vectorstore.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": .7, "k": 30,"filter": {'categorie': {'$eq': 'OF'}}})
|
100 |
-
retriever = InMemoryCache()
|
101 |
return retriever
|
102 |
|
103 |
@cl.set_chat_profiles
|
|
|
86 |
|
87 |
@cl.cache
|
88 |
def retriever_to_cache():
|
|
|
89 |
index_name = os.environ['PINECONE_INDEX_NAME']
|
90 |
embeddings = HuggingFaceEmbeddings()
|
91 |
pinecone.init(
|
|
|
96 |
index_name=index_name, embedding=embeddings
|
97 |
)
|
98 |
retriever = vectorstore.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": .7, "k": 30,"filter": {'categorie': {'$eq': 'OF'}}})
|
|
|
99 |
return retriever
|
100 |
|
101 |
@cl.set_chat_profiles
|