6jSDdZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddl mZmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZedZ dZ!ej"dZ#dZ$ddZ%GddZ&GddZ'dS)uWerewolfMind:LLM 角色脑,实现引擎那套 DecisionProvider 接口(无缝替下 RandomProvider)。 设计延续平台理念:人设 + 如实状态(对局板)+ 软引导,少规则、让博弈智能涌现; 但比命案局更吃逻辑与欺骗。每个决策点 = 一次结构化输出调用;失败/越界 → 随机兜底,世界不停摆。 模型工厂 make_model 注入(真实=lc_adapter.build_chat_model;测试=FakeModel),便于零真实调用单测。 ) annotationsN)Path)CallableOptional) HumanMessage SystemMessage) get_logger) perception)schema) BoardMemory)RandomProvider)lookup_transcript)Role) GameStatezwerewolf.brainu`9 人预女猎·屠边:狼方杀光所有神 或 所有民 即胜;好人放逐所有狼即胜。 world_dir str | Pathreturndict[int, str]ct|dz }|siS tj|d}nU#tjt f$r<}tdt|ddicYd}~Sd}~wwxYwi}| D]M\}}t| r't|tr||t|<N|S)ur从世界目录读 personas.json → {seat: 人设文本};无文件或解析失败则返回空(退回无人设)。z personas.jsonzutf-8)encodingu读取座位人设失败:%sNx)rexistsjsonloads read_textJSONDecodeErrorOSErrorloggerwarningstritemsisdigit isinstanceint)rpraweoutkvs 1/root/genesis-world/src/genesis/werewolf/brain.py load_personasr.*s Y/)A 88:: jg6677  ' *4c!ffTcTlCCC C 1 q66>>    1c 2 2 CAK Js(AB%)1B B% B%ceZdZdZdddddd5dZd6d7dZd8d9d"Zd#Zd$Zd%Z d&Z d'Z d(Z d)Z d*Zd+Zd,Zd:d-Zd.Zd;d0Zedsq   $ 0   && &s'AfmD.A.ABB T structuredboolc |jjdkrdnd}d|jd|jjd|d|jd g}|jr|d |jz|jjr||jj|d tz|r|d n|d |d d|S)Nwolfu 狼人阵营u 好人阵营u你是 u 号玩家,你的身份是【u】(u)。全程牢记你就是 u 号本人——绝不会查验/投票/怀疑/带走你自己;提到别人时一律用对方的座位号,别把自己的号说成别人。u'你的性子(用它决定说话风格):u你在玩 9 人狼人杀。uj凭你的身份、性子与立场,基于下面如实的对局信息做决策。只输出结构化结果。u+发言时像真人一样自然口语:可以有停顿、迟疑、『嗯…』『我想想』、改口、带口头禅和情绪,别像念稿。高手不耿直——会铺垫、试探、藏信息、找时机、配合队友,而不是一上来就摊牌或说『我不跳预言家』这种没信息量的废话。该悍跳、对跳、隐藏实力、结盟、带节奏、撒谎就来(尤其你若是狼:和狼队口径一致、互相抬轿、别公开拆队友的台)。每次只说有信息量或有目的的一段话,只说这段话本身,不要前缀/解释。u注意:你只能基于『你确实掌握到的信息』推理。可以伪装身份、可以撒谎带节奏,但别把『纯靠猜的』说成『我知道的事实』——尤其首夜你对别人的身份一无所知,不要凭空断言谁是预言家/女巫/神。 ) r8teamr7namer4append persona_hintWIN_RULEjoin)rFrIrNpartss r-_systemzWerewolfMind._systemLsR!%6!9!9~~~@49@@49>@@W[@@*.)@@@A < S LLBT\Q R R R 9 ! 1 LL/ 0 0 0 2X=>>>  q LLF G G G G LLp q q q  t u u uyyrHFtaskstaterextrathinkc |jtj||j|jt dz|z|z}t|t|g}|r|j r|j n|j }t|dd} t|jdzD]} t5||d|} d d d n #1swxYwY| | cSt$d |j| | dznR#t($rE} t$d |j| | dzt+| d d Yd } ~ nd } ~ wwxYw| |jkrt-jd | dzzt$d|j| |jdzd S)u渲染对局板 + 本次任务 → 结构化输出;失败重试 self.retries 次,仍失败才返回 None(随机兜底)。 think=True 用思考模式模型(发言/复盘,质量优先)。重试是稳定性关键:治"LLM偶发失败→3号发言这种傻瓜兜底"。transcript_tailz content__name__?function_calling)methodNu[%d号 %s] 第%d次返回空u[%d号 %s] 第%d次失败:%srg?u.[%d号 %s] 重试 %d 次仍失败,随机兜底)rArenderr viewr7r9TRANSCRIPT_TAILrrUrr3r;getattrranger6_LLM_SEMwith_structured_outputinvoker r! Exceptionr"timesleep) rFr rVrWrXrYusermsgsmakerOattemptr*r)s r-_askzWerewolfMind._askdsM !!*/%DJ"O"Oap!qq "'(dllnn555|D7Q7Q7QR).^43H^t$$tvz3//T\A-.. 0 0G khh$&&77GY7ZZaabfggChhhhhhhhhhhhhhh?JJJ=ty$PWZ[P[\\\\ k k k=ty$PWZ[P[]`ab]c]cdhehdh]ijjjjjjjj k%% 3'A+.///GTXZ^ZfijZjkkktsB=D:3D7 D:D D: D D:%D:: F ;FF c|jdkrdnd}|tjd|d|d|}| |j|vr|j|||S|j|jdS)Nrau今天是首夜,你们对所有人的身份一无所知——按直觉/座位/谁威胁大盲刀即可,别在理由里编造『他是预言家』这种你根本没法知道的话。r1u天黑,你在狼队私聊频道里和队友商量今晚刀谁(你说的只有狼队友看得到;前面队友的提议见上方)。u可刀:u=。给出你提议的 target 和一句给队友的 message。)targetmessage)dayrsSCWolfDiscussOutrurD wolf_discussrv)rFrWr7 candidates firstnightr*s r-rzzWerewolfMind.wolf_discuss|sjojswxjxjxggA ii)x%xx.8xxxy~@@ ;#*J66=--eT:FF F*===rHc|tjd|d|}|r|j|vr|jn|j|||S)Nu5天黑了,和狼队商定今晚刀谁。可刀对象:。)rsrx NightKillOutrurD decide_killrFrWr7r{r*s r-rzWerewolfMind.decide_kills[ii)pak)p)p)prwxx!wcjJ&>&>szzT]E^E^_dfjlvEwEwwrHc|tjd|d|}|r|j|vr|jn|j|||S)Nu(你要查验一人验明好坏。可查:r~)rsrx NightCheckOutrurD decide_checkrs r-rzWerewolfMind.decide_checks\ii(*dU_*d*d*dfkll!xcjJ&>&>szzT]E_E_`egkmwExExxrHc |d|dnd}|tj|d|d|d|d|}||j||||||S|j|vr|jnd} t |j| dS) Nu今夜 u 号倒在刀下。u今夜你未获知刀型。u 解药可用=u,毒药可用=u&。决定救不救、毒不毒(可毒:u)。)savepoison)rsrxWitchOutrD decide_witch poison_targetrJr) rFrWr7knifecan_save can_poisonpoison_candidatesknr*rs r-rzWerewolfMind.decide_witchs5:5F 1u 1 1 1 1Liii NNHNNJNNwHNNNPUVV ;=--eT5(JXijj j&)&7;L&L&L""RVSX&999rHc|tjd|}|rt|jn|j||S)NuD警长竞选:是否上警竞选警徽(1.5 票、可定发言顺序)?)rsrx RunSheriffOutrJrunrDdecide_run_sheriffrFrWr7r*s r-rzWerewolfMind.decide_run_sheriffsFii(*prwxx #VtCG}}})I)I%QU)V)VVrHct|tjd|}|r|jdvr|jnd}d|iS)Nu你当选警长,握有发言定序权:决定本轮从你左手边(cw,座位顺位)还是右手边(ccw,逆位)起依次发言,你自己将压轴最后发言。把你最想压制/想后置观察的可疑位置安排到对你最有利的发言序。)cwccwr direction)rsrx SpeechDirOutr)rFrWr7r*ds r-decide_speech_dirzWerewolfMind.decide_speech_dirsQiiTV[\\" Ocm}&D&DCMM4QrHct|tjd|}|rt|jndS)Nu竞选发言结束,进入退水环节:综合场上局势与你的目的,是否退水(退出警长竞选)?(若你是悍跳/警上无优势/想把警徽让给队友等,可考虑退;否则不退)F)rsrx WithdrawOutrJwithdrawrs r-decide_withdrawzWerewolfMind.decide_withdraws@iirsxzz&)3tCL!!!e3rHct|tjd|d|}|r|j|vr|jSdS)Nu[你(警长)即将出局,临终可把警徽移交给一名存活玩家或撕毁。可移交:uP。移交给你最信任、对好人/你阵营最有利的人(撕毁填 null)。)rsrx BadgePassOutrurs r-decide_badge_passzWerewolfMind.decide_badge_passs`iihwAhhhinpp  3:++: trHct|tjd|}|rt|jndS)Nu"轮到你发言。你是狼,可选择自爆(掀桌亮身份,中止今天发言投票直接入夜)。自爆是极强但极少用的手段——只在能打断好人关键归票/查杀传递、或保住更重要队友时才用;通常不自爆,正常发言带节奏即可。现在要自爆吗?F)rsrx ExplodeOutrJexploders r-decide_explodezWerewolfMind.decide_explodes@ii \]bdd%(2tCK   U2rHc~tj||j|j}d}|jdkrrt |jdkrZ|tj d|}|r7|j pd rt||j }|rd|z}t|jdzD]C}|||}|r|cS||jkrt!jd|dzzDt$d|j|jdz|j||S) Nr1r2 u发言前,你想调阅哪段原始记录来佐证判断?给一个 query(某座位号 / 'day:N' / 关键词);不需要就留空。u$ 【你刚调阅的原始记录】 ra?u.[%d号] 发言重试 %d 次仍空,随机兜底)r rer7r9rwlen transcriptrsrx RecallOutquerystriprrhr6 _speak_textrmrnr r!rDspeak) rFrWr7r,rXrechitrrtexts r-rzWerewolfMind.speaksR OE49dj 9 9 5A::#al++b00))BL`bghhC L R..00 L'3955LEKET\A-.. 0 0G##E511D  %% 3'A+.///GTXT`cdTdeee}""5$///rHc|jdkrdnd}|jtj||j|jtd|jdz|zdz|z} t5| t| d  t| g}d d d n #1swxYwYt|d d pd pd S#t $rA}t"d|jt'|d dYd }~d Sd }~wwxYw)uI纯文本生成一段发言(非结构化,鲁棒);失败/空返回 None。rau/现在是首日,信息少,可以略作铺垫。uu已是第二天之后,务必言简意赅、抓重点,一两句到一小段就好,别长篇复述别人说过的话。r[u 轮到你(u号)发言。以你的身份和立场,像真人一样自然口语地说(可带停顿/迟疑/口头禅/情绪,别念稿;有套路、有目的,别耿直摊牌或说废话)。ux过身份/报查验/站边/归票/试探/隐藏都行。只输出这段话本身,不要任何前缀、格式或解释。FrIr]Nr^r1u[%d号] 发言生成失败:%sr)rwrArdr rer7r9rfrir;rkrrUrrgrrlr r!r")rFrWrXbrevityror*r)s r-rzWerewolfMind._speak_textsIN UVDDP  !!*/%DJ"O"Oap!qqz49zzzzYY\aa  M Moo''.. dll^clFdFd0e0e0e0>F$ F    NN:DIs1vvdsd| T T T44444 s=%D,AC DCDC)D E6E  Ec>t|jdzD][}||}|r|j|dS||jkrt jd|dzz\td|j |jdzdS)u每天收尾的记忆压缩(每个存活 agent 各跑一次;引擎 _digest_day 钩子触发)。 多回合、信息量大 → 双层压缩防遗忘+控长: ① key_facts = 铁的客观事实清单(死亡+死法/公开身份声明/对跳/查验/警长/票型异常),累积保留别丢旧的; ② notes = 滚动主观分析+计划,在旧 notes 上重写。 触发规则:每天收尾压一次(此处);常驻事实/票型/查验等硬信息另由 PlayerView 每回合如实下发,二者互补。 ★实现:走纯文本+宽松解析(非结构化)——实测 MiniMax 对 DayDigestOut 这种多字段 function_calling 首日约 40% 吐空/回显字段名,导致 agent 次日记忆全空;纯文本 invoke 无函数调用脆弱性,稳。失败重试。raNru:[%d号] 记忆压缩重试 %d 次仍失败,保留旧局记) rhr6 _digest_textrA apply_digestrmrnr r!r7)rFrWr7rrrs r- digest_dayzWerewolfMind.digest_daysT\A-.. 0 0G!!%((A  ''***%% 3'A+.///SUYU^`d`lop`pqqqqqrHOptional[dict]cd}|jjs |jjrd|jjd|jj}|jt j||j|jt|zdz} t5| t| dt|g}d d d n #1swxYwYt|d dpd}nN#t"$rA}t$d |jt)|d d Yd }~d Sd }~wwxYw||S) uu纯文本生成记忆压缩 → 宽松解析成 {key_facts,notes,prime_suspect,most_trusted};失败/空返回 None。r1u 你已有的【事实】:u 你已有的【局记】:r[u= 今天结束,做记忆压缩。严格按下面 4 行纯文本输出(每行一项、冒号后写内容、没有就留空,不要任何别的话): 事实:<铁的客观事实清单,累积旧的+今天新增——谁死了+死法(刀/毒/枪/放逐)、谁公开跳了什么身份/谁对跳、公开查验结果、警长是谁、票型异常;只记客观事实不下判断> 分析:<基于事实,你这个身份/立场的好狼判断与下一步计划,在旧局记上重写,几句话> 最疑:<你最怀疑是狼的座位号,或留空> 最信:<你最信任的座位号,或留空>Frr]Nr^u#[%d号] 记忆压缩生成失败:%sr)rA key_factsnotesrdr rer7r9rfrir;rkrrUrrgrrlr r!r" _parse_digest)rFrWoldror*rr)s r-rzWerewolfMind._digest_texts :  v4:#3 vu$*2FuucgcmcsuuC !!*/%DJ"O"Oap!qq>>  M Moo''.. dll^clFdFd0e0e0e0>DD    NN@$)SQRVVTXUXTX\ Z Z Z44444 !!$'''s==DAC* D*C..D1C.2(D E&%6E!!E&rcddddd}i}|D]}|ddd}|D]f\}}||d zs||d zr1|t |d zd ||<ng|ds|dsd Sd D]T}tj d||pd}|r!t| nd ||<U|S)uy把『事实/分析/最疑/最信』四行纯文本宽松解析成 digest dict;至少要有事实或分析才算有效。rr prime_suspect most_trusted)u事实u分析u最疑u最信u-*【 u】r1:u：raN)rrz\d+) splitlinesrlstripreplacer# startswithrgetresearchr&group) rlabelsgotlineslabelkeyr+ms r-rzWerewolfMind._parse_digests`(7oaoppOO%%  D ##H--55eR@@A$llnn   s<< ,, UU]0K0K Ua17799CHEww{## CGGG,<,< 42 3 3A &#''!**"233A'(2S^^^dCFF rHc|tjd|d|}||j|||S|j|vr|jndS)Nu!放逐投票:投谁出局?可投:u(弃票填 null)。)rsrxVoteOutrD decide_voterurs r-rzWerewolfMind.decide_votes]ii $g $g$g$ginoo ;=,,UD*EE E Z:55szz4?rHc|tjd|d|}||j|||S|j|vr|jndS)Nu1你(猎人)出局,可开枪带走一人。可带:u(放弃填 null)。)rsrxHunterShootOutrDdecide_hunter_shootrurs r-rz WerewolfMind.decide_hunter_shootsiii)+~_i+~+~+~AFGG ;=44UD*MM M Z:55szz4?rH)r7r&r8rr9r:r;rr3r<r4r"r5r=r6r&rr>)T)rIrJrr")r1F)rVr"rWrrXr"rYrJ)rXr")rr)rr"rr)r_ __module__ __qualname____doc__rGrUrsrzrrrrrrrrrrrr staticmethodrrrrHr-r0r0;sEERV "4XY            00>>>xxxyyy:::WWW   444 333000*$rrr"((((.\"@@@ @@@@@rHr0ceZdZdZddZddZdZd Zd Zd Z d Z d Z dZ dZ dZdZdZdZdZe d d!dZdS)" MindProvideru]把每座位的 WerewolfMind 组装成引擎用的单一 DecisionProvider(按 seat 路由)。mindsdict[int, WerewolfMind]rr>c||_dSr@)r)rFrs r-rGzMindProvider.__init__(s  rHcL|jD] }||_ dS)ua给所有 AI 脑装上流式发言回调(网页 live 用);不设则发言走非流式结构化。N)rvaluesrE)rFsinkrs r-set_stream_sinkzMindProvider.set_stream_sink+s2""$$ ! !A AMM ! !rHcF|j||||Sr@)rrzrFrr7cs r-rzzMindProvider.wolf_discuss0"tz$/?/L/LQPTVW/X/X(XrHcF|j||||Sr@)rrrs r-rzMindProvider.decide_kill1!dj.>.J.J1dTU.V.V'VrHcF|j||||Sr@)rrrs r-rzMindProvider.decide_check2rrHcL|j|||||||Sr@)rr)rFrr7r+cscppcs r-rzMindProvider.decide_witch3s,4:d;K;X;XYZ\`bcegikmo;p;p4prHcD|j|||Sr@)rrrFrr7s r-rzMindProvider.decide_run_sheriff4s $*T2B2U2UVWY]2^2^+^rHcD|j|||Sr@)rrrs r-rzMindProvider.decide_speech_dir5s D1A1S1STUW[1\1\*\rHcD|j|||Sr@)rrrs r-rzMindProvider.decide_withdraw6s tz$/?/O/OPQSW/X/X(XrHcF|j||||Sr@)rrrs r-rzMindProvider.decide_badge_pass7s#DJt4D4V4VWXZ^`a4b4b-brHcD|j|||Sr@)rrrs r-rzMindProvider.decide_explode8sdj.>.M.MaQU.V.V'VrHcD|j|||Sr@)rrrs r-rzMindProvider.speak9sTZ%5%;%;At%D%DDrHcF|j||||Sr@)rrrs r-rzMindProvider.decide_vote:rrHcF|j||||Sr@)rrrs r-rz MindProvider.decide_hunter_shoot;s#dj6F6Z6Z[\^bde6f6f/frHcD|j|||Sr@)rrrs r-rzMindProvider.digest_day<s$*T*:*E*Ea*N*N#NrHNr;rpersonasrr3r<'MindProvider'c^pifdjjD}||S)u按已发牌的引擎给每个座位建一个对应身份的脑。make_think_model:发言/复盘用的思考模式模型(可选)。ci|]g}|t|jj|j|dt j|hS)r1)r3r4r5)r0r9rWrole_ofrrBrC).0r7enginer;r3rs r- z)MindProvider.for_game..Cs    ,tV\&,2F2Ft2L2L%Mv|]g0@'/||D"'='=6=QUCVCVXXX   rH)rWr9)clsrr;rr3rs ```` r-for_gamezMindProvider.for_game>s_>r        *     s5zzrH)rrrr>)rr>)NN)r;rrrr3r<rr)r_rrrrGrrzrrrrrrrrrrrr classmethodrrrHr-rr%sgg!!!! YXXVVVXXXppp^^^\\\XXXbbbVVVDDDVVVfffNNNOS8<    [   rHr)rrrr)(r __future__rrrBr threadingrmpathlibrtypingrrlangchain_core.messagesrrgenesis.obs.logging_setupr genesis.werewolfr r rxgenesis.werewolf.boardmemoryr genesis.werewolf.enginer genesis.werewolf.recallrgenesis.werewolf.rolesrgenesis.werewolf.staterr rR Semaphorerirfr.r0rrrHr-rs#"""""  %%%%%%%%????????000000''''''))))))444444222222555555'''''',,,,,, $ % % n 9 q ! !"g@g@g@g@g@g@g@g@T$$$$$$$$$$rH