Poi ddlZddlZddlmZddlmZmZmZmZddl m Z ddl m Z e e Ze jdz ZeddGd d ZdS) N)Path)ListDictAnyOptional) setup_logger)config slack_contextT)parentsexist_okc eZdZdZdZdefdZdeeee ffdZ deeee fddfd Z dd ed ed e eddfd Z deeee ffdZddZdS)ContextStoragez;Manages persistent conversation history for Slack channels.( channel_idc8||_t|dz |_dS)Nz.json)r CONTEXT_DIR file_path)selfrs 2/root/projects/butler/slack_bot/context/storage.py__init__zContextStorage.__init__s"$$*';';';;returnc@|jsgS t|jdd5}tj|cdddS#1swxYwYdS#t $r1}t d|jd|gcYd}~Sd}~wwxYw)Nrutf-8encodingzFailed to load context for : ) rexistsopenjsonload Exceptionloggererrorr)rfes r_loadzContextStorage._loads~$$&& I dncG<<< $y|| $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $    LLMtMM!MM N N NIIIIII s@A"A A"AA"AA"" B,&BBBcontextNc t|jdd5}tj||ddddddS#1swxYwYdS#t$r0}t d|jd|Yd}~dSd}~wwxYw) NwrrF) ensure_asciiindentzFailed to save context for r)r rr!dumpr#r$r%r)rr)r&r's r_savezContextStorage._save!s OdncG<<< D '15CCCC D D D D D D D D D D D D D D D D D D O O O LLMtMM!MM N N N N N N N N N Os7A ? A AA AA B%BBrolecontentmodelc|}||tjd}|r||d<||t||jkr||j d}||dS)z Add a message to the context. Args: role: 'user' or 'assistant' content: Message content model: Model identifier (only for assistant messages, currently always 'gemini') )r1r2 timestampr3N)r(timeappendlen MAX_MESSAGESr0)rr1r2r3r)msgs r add_messagezContextStorage.add_message(s**,,    ! CLs w<<$+ + +t00112G 7rc|}g}|D]&}||d|dd'|S)z Get conversation context in standard format. Returns messages as [{"role": "user", "content": "..."}, ...] r1r2)r1r2)r(r7)r full_contextfilteredr:s r get_contextzContextStorage.get_contextBsa zz||   C OOF y>     rc|jr|jtd|jdS)z#Clear all context for this channel.zCleared context for N)rrunlinkr$infor)rs rclearzContextStorage.clearTsO > " " $ N ! ! # # # <4?<<=====r)N)rN)__name__ __module__ __qualname____doc__r9strrrrrr(r0rr;r?rCrrrr sEEL<3<<<<tDcN+OT$sCx.1OdOOOOc(3-SW4T$sCx.1$>>>>>>rr)r!r6pathlibrtypingrrrrhealth.utils.logging_configrhealthr rDr$DATA_DIRrmkdirrrIrrrPs ,,,,,,,,,,,,444444 h  o/ $...K>K>K>K>K>K>K>K>K>K>r