[email protected] commited on
Commit
8c27c79
·
1 Parent(s): 72cdb72

Add prompt system in YAML config

Browse files
Files changed (2) hide show
  1. app.py +3 -2
  2. config.yaml +2 -0
app.py CHANGED
@@ -15,7 +15,7 @@ LOGO = "assets/logo.png"
15
 
16
  def init_app():
17
 
18
- data_dict = getYamlConfig()['variables']
19
 
20
  if len(st.session_state) == 0:
21
  # Define Vectore store strategy
@@ -24,7 +24,8 @@ def init_app():
24
 
25
  st.session_state["messages"] = []
26
  st.session_state["assistant"] = Rag(vectore_store=vs_manager)
27
- st.session_state["data_dict"] = data_dict
 
28
 
29
 
30
  def main():
 
15
 
16
  def init_app():
17
 
18
+ config = getYamlConfig()
19
 
20
  if len(st.session_state) == 0:
21
  # Define Vectore store strategy
 
24
 
25
  st.session_state["messages"] = []
26
  st.session_state["assistant"] = Rag(vectore_store=vs_manager)
27
+ st.session_state["data_dict"] = config['variables']
28
+ st.session_state["prompt_system"] = config['prompt_system']
29
 
30
 
31
  def main():
config.yaml CHANGED
@@ -22,6 +22,8 @@ variables:
22
  key : param3
23
  value :
24
 
 
 
25
  prompt_template: "
26
  Paramètre 1 : {param1}
27
  Paramètre 2 : {param2}
 
22
  key : param3
23
  value :
24
 
25
+ prompt_system: "Prompt system par défaut"
26
+
27
  prompt_template: "
28
  Paramètre 1 : {param1}
29
  Paramètre 2 : {param2}