Spaces:
Sleeping
Sleeping
Add metadata on llm chain
Browse files
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
|