Spaces:
Running
Running
File size: 1,172 Bytes
c09dbef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
import streamlit as st
def set_session_state(mode: str, sys: str, tokens: int, temp: float):
if f"{mode}_sys" not in st.session_state:
st.session_state[f"{mode}_sys"] = sys
if f"{mode}_msg" not in st.session_state:
st.session_state[f"{mode}_msg"] = []
if f"{mode}_cache" not in st.session_state:
st.session_state[f"{mode}_cache"] = []
if f"{mode}_tokens" not in st.session_state:
st.session_state[f"{mode}_tokens"] = tokens
if f"{mode}_temp" not in st.session_state:
st.session_state[f"{mode}_temp"] = temp
if f"{mode}_topp" not in st.session_state:
st.session_state[f"{mode}_topp"] = 0.70
if f"{mode}_freq" not in st.session_state:
st.session_state[f"{mode}_freq"] = 0.00
if f"{mode}_pres" not in st.session_state:
st.session_state[f"{mode}_pres"] = 0.00
if f"{mode}_stop" not in st.session_state:
st.session_state[f"{mode}_stop"] = None
if f"{mode}_stop_str" not in st.session_state:
st.session_state[f"{mode}_stop_str"] = ""
if f"{mode}_retry" not in st.session_state:
st.session_state[f"{mode}_retry"] = False
|