o i8@sUddlmZddlZddlZddlmZddlmZddlm Z ddl m Z ddl mZddlmZmZmZmZd Zd Zd Zeejd d dZdZdZeeZe eefed<eZ de!fddZ"de#e$de!fddZ%dede#e$de#e$fddZ&de$ddfddZ'dS)) defaultdictN)Path)Lock) DefaultDict)context)logger) ChatRequestGenerationConfigLLMMessage(2zdeepseek-v4-flashllmpromptszmemory_point_gen_prompt.mdzhttps://api.deepseek.com_group_message_counterreturncCstjddS)Nzutf-8)encoding)MEMORY_PROMPT_PATH read_textstriprr7/home/dore/onebot/Bot_Ero/core/group_memory_pipeline.py_load_memory_promptsrrecordscCs&dd|D}ddd|DS)NcSsg|]}t|qSr)runtime_contextformat_chat_record).0recordrrr sz*_build_chat_plain_text.. css|]}|r|VqdS)Nr)rlinerrr sz)_build_chat_plain_text..)joinr)rlinesrrr_build_chat_plain_textsr%group_idc Cs|sgStdd}|stdgSzt}Wnty5}ztd||gWYd}~Sd}~wwt|}|s>gS|d|}t d||t |t d}t d |d g}t td d } d} td td D]a} z |t|| d} Wnty}ztd|| t|WYd}~qhd}~ww| j} t d|| | zt| } t| tr| WStd|| tWqhtjytd|| tYqhwtd|| gS)u9提取群聊记忆点，返回解析后的 JSON 数组。DEEPSEEK_API_KEYu?群聊记忆提取跳过：缺少环境变量 SIFLOW_API_KEY。u;群 %s 记忆提取失败：读取 prompt 文件异常：%sNz u群 %s 记忆提取 prompt: %s)api_keybase_urluser)rolecontenti)model max_tokens)messagesconfigu>群 %s 记忆提取调用失败，准备重试（%s/%s）：%su'群 %s 记忆提取结果(第%s次): %suD群 %s 记忆提取返回非数组 JSON，准备重试（%s/%s）。uB群 %s 记忆提取 JSON 解析失败，准备重试（%s/%s）。uB群 %s 记忆提取失败：超过重试次数。最后返回：%s)osgetenvrrwarningr Exceptionerrorr%infor MEMORY_LLM_BASE_URLr r MEMORY_LLM_MODELrangeMEMORY_MAX_RETRIESchatrmessagejsonloads isinstancelistJSONDecodeError)r&rr) prompt_prefixexc chat_textpromptrr1r2 last_textattemptresponseparsedrrrextract_group_memory_points"sf        rLrcCs|ddkr dS|ddkrdS|d}|durdSt|}d}tt|d7<t|td kr8d }Wdn1sBwY|rWtj|td }t||dSdS) u 插件执行后调用： - 仅处理群聊 message 事件 - 每累计 40 条消息触发一次回调 - 回调入参为：该群最新 50 条消息 post_typer>N message_typegroupr&Fr0rT)limit) getint _counter_lockrMEMORY_BATCH_SIZErget_recent_chat_recordsMEMORY_CONTEXT_SIZErL)rr&gidshould_triggerpayloadrrr after_pluginsYs& rZ)( collectionsrr?r3pathlibr threadingrtypingrcorerr core.loggerr core.llm.llmrr r r rTrVr:__file__resolveparentrr9r<rRr__annotations__rSstrrrBdictr%rLrZrrrrs,     7