datacipen commited on
Commit
8160932
1 Parent(s): 6b80cf7

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -3
main.py CHANGED
@@ -60,8 +60,6 @@ vectorstore = PineconeVectorStore(
60
  )
61
  retriever = vectorstore.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": .7, "k": 150,"filter": {'categorie': {'$eq': 'bibliographie-OPP-DGDIN'}}})
62
  #search = vectorstore.similarity_search(query,k=50, filter={"categorie": {"$eq": "bibliographie-OPP-DGDIN"}, 'Source': {'$eq': 'Source : Persée'}})
63
- search = vectorstore.similarity_search(query,k=50, filter={"categorie": {"$eq": "bibliographie-OPP-DGDIN"}})
64
- cl.user_session.set("search", search)
65
 
66
  @cl.on_chat_start
67
  async def on_chat_start():
@@ -184,7 +182,8 @@ async def on_message(message: cl.Message):
184
  answer = results["answer"]
185
 
186
  await cl.Message(content=answer).send()
187
- if cl.user_session.get("search"):
 
188
  test = []
189
  sources_text = ""
190
  count = 0
 
60
  )
61
  retriever = vectorstore.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": .7, "k": 150,"filter": {'categorie': {'$eq': 'bibliographie-OPP-DGDIN'}}})
62
  #search = vectorstore.similarity_search(query,k=50, filter={"categorie": {"$eq": "bibliographie-OPP-DGDIN"}, 'Source': {'$eq': 'Source : Persée'}})
 
 
63
 
64
  @cl.on_chat_start
65
  async def on_chat_start():
 
182
  answer = results["answer"]
183
 
184
  await cl.Message(content=answer).send()
185
+ search = vectorstore.similarity_search(message.content,k=50, filter={"categorie": {"$eq": "bibliographie-OPP-DGDIN"}})
186
+ if search:
187
  test = []
188
  sources_text = ""
189
  count = 0