.

bill.py

@@ -159,7 +159,7 @@ def main():
         # If the context prompt hasn't been added yet, build & inject it once;
         # otherwise, just add the user's raw question.
         if not st.session_state.context_prompt_added:
-            final_prompt = process_query(prompt, st.session_state["user_id"])
+            final_prompt = process_query(prompt, st.session_state["user_id"], model_name)
             if final_prompt is None:
                 st.stop()
             st.session_state["messages"].append({"role": "user", "content": final_prompt})