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
|