Update main.py
Browse files
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 |
-
|
|
|
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
|