cdupland commited on
Commit
4515a36
·
verified ·
1 Parent(s): 71794e2

Add metadata on llm chain

Browse files
Files changed (1) hide show
  1. rag.py +6 -1
rag.py CHANGED
@@ -120,7 +120,12 @@ class Rag:
120
  chain_input.update(extra_vars)
121
 
122
 
123
- return self.chain.stream(chain_input)
 
 
 
 
 
124
 
125
  def clear(self):
126
  self.document_vector_store = None
 
120
  chain_input.update(extra_vars)
121
 
122
 
123
+ return self.chain.stream(chain_input, config={
124
+ "metadata" : {
125
+ "id_user": st.session_state.get("id_user", None),
126
+ "use_case": st.session_state.get("use_case", None)
127
+ }
128
+ })
129
 
130
  def clear(self):
131
  self.document_vector_store = None