Spaces:
Running
Running
[email protected]
commited on
Commit
·
8c27c79
1
Parent(s):
72cdb72
Add prompt system in YAML config
Browse files- app.py +3 -2
- config.yaml +2 -0
app.py
CHANGED
@@ -15,7 +15,7 @@ LOGO = "assets/logo.png"
|
|
15 |
|
16 |
def init_app():
|
17 |
|
18 |
-
|
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"] =
|
|
|
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}
|