llm_hub/app/utils/sessions.py

14 lines
437 B
Python

# app/utils.py
import uuid
from flask import session
# 初始化用户的会话信息,包括对话历史
def init_session():
if 'user_id' not in session:
session['user_id'] = str(uuid.uuid4()) # 生成用户唯一ID
if 'history' not in session:
session['history'] = [] # 初始化会话历史
session['session_info'] = {}
if 'llm_service' not in session:
session['llm_service'] = 'zhipu'