Ludovicollin commited on
Commit
b94a242
·
1 Parent(s): 85b88ff

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +0 -2
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