Skip to content

Fix: Model uses the system prompt and query rewriting prompt from the database if it exists. #124

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
gcapuzzi merged 1 commit into from
Jul 6, 2025
Merged

Fix: Model uses the system prompt and query rewriting prompt from the database if it exists. #124

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions src/mvt/pages/chatbot.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from utils import load_yaml_file
from utils import load_yaml_file_with_db_prompts
from main import get_ragchain
import streamlit as st
from menu import menu_with_redirect
Expand All @@ -23,7 +23,7 @@

config_path = "./config.yaml"
logo_path = "https://github.com/hyperledger-labs/aifaq/blob/mvt-streamlit/images/logo.png?raw=true"
config_data = load_yaml_file(config_path)
config_data = load_yaml_file_with_db_prompts(config_path)

# filter public document in case of guest user
filter = None
Expand Down
9 changes: 7 additions & 2 deletions src/mvt/pages/prompt_management.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import streamlit as st
import yaml
from utils import load_yaml_file
from utils import load_yaml_file, load_yaml_file_with_db_prompts
from database import create_connection, create_prompts_table, save_prompt, get_prompt
from menu import menu_with_redirect

Expand Down Expand Up @@ -87,7 +87,12 @@ def load_prompts_from_db():

# Initialize session state for prompts if not already set
if "current_prompts" not in st.session_state:
st.session_state.current_prompts = load_prompts_from_db()
# Use the utility function that loads from database with config fallback
config_data = load_yaml_file_with_db_prompts("config.yaml")
st.session_state.current_prompts = {
"system_prompt": config_data.get("system_prompt", ""),
"query_rewriting_prompt": config_data.get("query_rewriting_prompt", "")
}

# Create columns for better layout
col1, col2 = st.columns([3, 1])
Expand Down