o e_@sPddlZddlmZddZddZddZd d Zd d Zd dZddZ dS)N) LangChainModecCs|od|v}|tjj}|durt|tksJ|dr|r)|||dd|d<|dr/|r?d}d|vr9|d}||d<dSdS)Nusername)id0)rMY_DATAvaluelen length_db1)db1srequests_state1get_userid_authforcedb1 username1r/h2ogpt/src/db_utils.py set_userids     rcCs |tjj}||d<||d<dS)Nrrrrr)r useridrrrrrset_userid_directs  rcC|dur |tjjdSdS)Nrrr rrrget_userid_directrcCr)Nrrrrrrrget_username_directrrcCs|dSNrrrrrrget_dbid!srcCs>|dur t|tks J|ddurtt|d<dSdSr)r r struuiduuid4rrrrset_dbid%s r#cCsdS)Nrrrrrr -s r ) r!enumsrrrrrrr#r rrrrs