[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}
|