57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
|
import uuid
|
||
|
|
|
||
|
|
SESSIONS = {}
|
||
|
|
USER_FEEDBACK_LOG = []
|
||
|
|
|
||
|
|
def create_session(query: str) -> str:
|
||
|
|
sid = uuid.uuid4().hex
|
||
|
|
SESSIONS[sid] = {
|
||
|
|
"original_query": query,
|
||
|
|
"round": 0,
|
||
|
|
"history_candidates": [],
|
||
|
|
"user_feedback": [],
|
||
|
|
"rejected": [],
|
||
|
|
"selected_prompt": None,
|
||
|
|
"chat_history": [],
|
||
|
|
"model_name": None
|
||
|
|
}
|
||
|
|
return sid
|
||
|
|
|
||
|
|
def get_session(sid: str):
|
||
|
|
return SESSIONS.get(sid)
|
||
|
|
|
||
|
|
def update_session_add_candidates(sid: str, candidates: list):
|
||
|
|
s = SESSIONS[sid]
|
||
|
|
s["round"] += 1
|
||
|
|
s["history_candidates"].extend(candidates)
|
||
|
|
|
||
|
|
def log_user_choice(sid: str, choice: str):
|
||
|
|
SESSIONS[sid]["user_feedback"].append(
|
||
|
|
{"round": SESSIONS[sid]["round"], "choice": choice}
|
||
|
|
)
|
||
|
|
USER_FEEDBACK_LOG.append({
|
||
|
|
"session_id": sid,
|
||
|
|
"round": SESSIONS[sid]["round"],
|
||
|
|
"choice": choice
|
||
|
|
})
|
||
|
|
|
||
|
|
def log_user_reject(sid: str, candidate: str, reason: str | None = None):
|
||
|
|
SESSIONS[sid]["rejected"].append(candidate)
|
||
|
|
USER_FEEDBACK_LOG.append({
|
||
|
|
"session_id": sid,
|
||
|
|
"round": SESSIONS[sid]["round"],
|
||
|
|
"reject": candidate,
|
||
|
|
"reason": reason or ""
|
||
|
|
})
|
||
|
|
|
||
|
|
def set_selected_prompt(sid: str, prompt: str):
|
||
|
|
SESSIONS[sid]["selected_prompt"] = prompt
|
||
|
|
|
||
|
|
def log_chat_message(sid: str, role: str, content: str):
|
||
|
|
SESSIONS[sid]["chat_history"].append({"role": role, "content": content})
|
||
|
|
|
||
|
|
def set_session_model(sid: str, model_name: str | None):
|
||
|
|
s = SESSIONS.get(sid)
|
||
|
|
if s is not None:
|
||
|
|
s["model_name"] = model_name
|