Update main.py

main.py

@@ -55,13 +55,7 @@ async def on_chat_start():
     llm = HuggingFaceEndpoint(
         repo_id=repo_id, max_new_tokens=8000, temperature=1.0, task="text2text-generation", streaming=True
     )
-    os.environ['PINECONE_API_KEY'] = os.environ['PINECONE_API_KEY']
-    embeddings = HuggingFaceEmbeddings()
-    index_name = "all-venus"
-    pc = Pinecone(
-        api_key=os.environ['PINECONE_API_KEY']
-    )
-    index = pc.Index(index_name) 
+    
     await cl.Message(f"> REVIEWSTREAM").send()
     await cl.Message(f"Nous avons le plaisir de vous accueillir dans l'application de recherche et d'analyse des publications.").send()
     listPrompts_name = f"Liste des revues de recherche"
@@ -98,12 +92,19 @@ async def on_chat_start():
         RunnablePassthrough.assign(
             history=RunnableLambda(memory.load_memory_variables) | itemgetter("history")
         )
-        | prompt | llm | StrOutputParser()
+        | prompt | llm
     )
     cl.user_session.set("runnable", chain)
     
 @cl.on_message
 async def on_message(message: cl.Message):
+    os.environ['PINECONE_API_KEY'] = os.environ['PINECONE_API_KEY']
+    embeddings = HuggingFaceEmbeddings()
+    index_name = "all-venus"
+    pc = Pinecone(
+        api_key=os.environ['PINECONE_API_KEY']
+    )
+    index = pc.Index(index_name)
     memory = cl.user_session.get("memory")
     runnable = cl.user_session.get("runnable")