Update system prompt config

jonfairbanks · Mar 22, 2024 · 7d63c13 · 7d63c13
1 parent 414ddfb
commit 7d63c13
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utils/rag_pipeline.py b/utils/rag_pipeline.py
@@ -59,6 +59,7 @@ def rag_pipeline(uploaded_files: list = None):
         llm = ollama.create_ollama_llm(
             st.session_state["selected_model"],
             st.session_state["ollama_endpoint"],
+            st.session_state["system_prompt"],
         )
         st.session_state["llm"] = llm
         st.caption("✔️ LLM Initialized")