feat: add user ID handling from query parameters and include in metadata for chain input

app.py

@@ -22,6 +22,10 @@ def init_app():
         pinecone_connector = PineconeConnector()
         vs_manager = VectoreStoreManager(pinecone_connector)
 
+        query_params = st.query_params
+        if "user" in query_params:
+            st.session_state["id_user"] = query_params["user"]
+            
         st.session_state["messages"] = []
         st.session_state["assistant"] = Rag(vectore_store=vs_manager)
         st.session_state["data_dict"] = config['variables']

rag.py

@@ -11,6 +11,7 @@ from langchain.schema.runnable import RunnablePassthrough
 from langchain.prompts import PromptTemplate
 from langchain_community.vectorstores.utils import filter_complex_metadata
 from langchain_community.document_loaders.csv_loader import CSVLoader
+import streamlit as st
 
 from util import getYamlConfig
 
@@ -110,7 +111,12 @@ class Rag:
             chain_input.update(extra_vars)
         
 
-        return self.chain.stream(chain_input)
+        return self.chain.stream(chain_input, config={
+            "metadata" : {
+                "id_user": st.session_state.get("id_user", None),
+                "use_case": st.session_state.get("use_case", None)
+            }
+        })
 
     def clear(self):
         self.document_vector_store = None