原始代码

_qwen_xinference_demo/opro/session_state.py

@@ -0,0 +1,56 @@
+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