update

pages/1_Injection.py

@@ -48,7 +48,7 @@ else:
     st.session_state.endpoint_url = st.sidebar.text_input("Endpoint URL", value=st.session_state.endpoint_url)
     st.session_state.deployment_name = st.sidebar.text_input("Model Name", value=st.session_state.deployment_name)
     api_version = '2024-02-15-preview' if model_type == 'GPTAgent' else ''
-    st.session_state.temperature = st.sidebar.slider("Temperature", 0.0, 1.0, st.session_state.temperature, 0.01)
+    st.session_state.temperature = st.sidebar.slider("Temperature", 0.0, 1.0, st.session_state.temperature, 0.00)
     st.session_state.max_tokens = st.sidebar.number_input("Max Tokens", 1, 1000, st.session_state.max_tokens)
 
     if st.sidebar.button("Reset Model Info"):

util/injection.py

@@ -104,7 +104,7 @@ def process_scores_multiple(df, num_run, parameters, privilege_label, protect_la
 
             for key, label in zip(['Privilege', 'Protect', 'Neutral'], [privilege_label, protect_label, False]):
                 prompt_temp = create_summary(row, group_name, label, occupation, additional_charateristics)
-                print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{prompt_temp}")
+                print(f"Run {run + 1} - Entry {index + 1} - {key}:\n{additional_charateristics}")
                 print("=============================================================")
                 result = invoke_retry(prompt_temp, agent, parameters)
                 scores[key][index].append(result)