corr bug

agents_page/recommended_agent.py

@@ -73,7 +73,7 @@ def handle_display_models(index, models_names):
 
 def recommended_agent_main():
     st.title("Agents recommandés")
-    models_names = ["GPT-4o", "GPT-4o-mini"]
+    models_names = ["GPT-4o", "GPT-4o-mini","Mistral Nemo (FR)","Mistral Large 2 (FR)"]
 
     if "chat_history" not in st.session_state:
         st.session_state.chat_history = [

audit_page/audit.py

@@ -112,6 +112,10 @@ def handle_audit(uploaded_file,type:str):
             with st.spinner("Analyse du document..."):
                 st.session_state.audit = {}
                 st.session_state.audit = audit_descriptif_pdf(uploaded_file,100)
+            with st.spinner("Préparation de la DB..."):
+                vectorstore = setup_rag(type,st.session_state.audit["content"])
+                st.session_state.vectorstore = vectorstore
+
         audit = st.session_state.audit["audit"]
         #global audit
         audit_simplified = {
@@ -229,7 +233,5 @@ def audit_main():
     if "audit" in st.session_state and st.session_state.audit != {}:
         display_audit(col1)
         handle_display_content(col2)
-        vectorstore = setup_rag(st.session_state.audit_simplified["type de fichier"],st.session_state.audit["content"])
-        st.session_state.vectorstore = vectorstore
-
+        
 audit_main()