Pj_:8dZddlmZddlZddlZddlZddlmZmZm Z ejdZ dZ ejdejZdZd Zdd ZGd d ZGd dZGddeZe eegeefZGddZddZGddZy)uThe brain: a streaming, tool-calling chat loop over a swappable LLM vendor. Phase-1 vendor is DeepSeek (OpenAI-compatible chat completions), ported from the pattern in ``voice/deepseek.py`` and extended with SSE streaming + tool calling. The sentence-regrouping approach (start speaking sentence one while the rest still generates) is ported from ``voice/llm.py``. Vendor-swappable: ``Brain`` speaks a neutral event stream and OpenAI-shaped message history. To add Anthropic later, implement ``LLMAdapter.stream`` to translate that history to/from the Anthropic API and yield the same events — nothing else changes. Neutral adapter event contract (async iterator of dicts): {"type": "text", "text": ""} # streamed assistant text {"type": "tool_calls", "calls": [ {id,name,arguments}, ... ]} # 0..1 times {"type": "finish", "reason": "stop"|"tool_calls"|...} # exactly once ) annotationsN) AsyncIterator AwaitableCallableu[.!?؟…]['\")\]]?\s|\nu[*_`#~|]|^\s*[-•]\s+z<>cJtjd|jS)z=Remove markdown symbols before a sentence goes to the UI/TTS.) _MD_STRIPsubstrip)texts ;/Users/ahmed/devFolder/Ultron/claude-voice/gateway/brain.py clean_for_ttsr*s ==T " ( ( **c(eZdZdZddZddZddZy) SentenceRegrouperzAccumulates streamed token deltas and emits complete sentences. ``push(delta)`` returns a list of finished sentences (cleaned, [emotion] tags preserved); ``flush()`` returns whatever remains at end-of-turn. cd|_yNr )_bufselfs r__init__zSentenceRegrouper.__init__6s  rc|xj|z c_g} tj|j}|rj|jd|jj }|j|jd|_|r|j t |nt|jtkDr||jjddt}|dkDr|nt}|jd|j }|j|d|_|r|j t |nn&|Dcgc]}|s| c}Scc}w)N r() r _SENTENCE_ENDsearchendr appendrlen _MAX_CHUNKrfind)rroutmchunkcutss rpushzSentenceRegrouper.push9s T $$TYY/A )AEEG,224 IIaeegh/ JJ}U34TYY*,iiooc1j9 2Xc: $3--/ IIcdO JJ}U34 $3a!3$$$s E Ec>t|j}d|_|Sr)rr)rtails rflushzSentenceRegrouper.flushNsTYY'  rNreturnNone)rstrr.z list[str])r.r0)__name__ __module__ __qualname____doc__rr)r,rrrr/s %*rrc(eZdZdZ ddZddZy) LLMAdapterzCVendor interface. Implement ``stream`` to yield the neutral events.cKtwN)NotImplementedError)rmessagestoolss rstreamzLLMAdapter.streamXs !!s c Kywr9r5rs raclosezLLMAdapter.aclose\s  sNr; list[dict]r<list[dict] | Noner.zAsyncIterator[dict]r-)r1r2r3r4r=r?r5rrr7r7UsM"-"2E" rr7c8eZdZdZdddZdZd dZ d dZy) DeepSeekAdapterz8DeepSeek / OpenAI-compatible streaming chat completions.Nctjjddjd|_|xs tjjdd|_t tjjdd|_ttjjdd |_ ||_ |du|_ y) N DEEPSEEK_URLzhttps://api.deepseek.com/MODELzdeepseek-v4-proBRAIN_TEMPERATUREz0.6BRAIN_MAX_TOKENS1024) osenvirongetrstripbasemodelfloat temperatureint max_tokens_client _owns_client)rclientrQs rrzDeepSeekAdapter.__init__csJJNN 688>s  HbjjnnW6GH  0CU!KLbjjnn-?HI "dNrc|j*ddl}|j|jd|_|jS)Nrg^@)timeout)rVhttpx AsyncClientTimeout)rr[s r_ensure_clientzDeepSeekAdapter._ensure_clientls7 <<   ,,U]]55I,JDL||rcK|jr7|j*|jjd{d|_yyy7wr9)rWrVr?rs rr?zDeepSeekAdapter.aclosers@   !9,,%%' ' 'DL":  's6A AA c PKtjjdd}|s td|j|d|j |j d}|r ||d<d|d<|j}d |d d }i}d}|jd |jd z||4d{} | j| j23d{} | r| jds| tddj} | dk(rn' tj | } | jdxsig} | d}|jdxsi}|jdr d|dd|jdxsgD]}|jdd}|j%|dddd}|jdr|d|d<|jdxsi}|jdr|d|d<|jds|dxx|dz cc<|jdsa|d}hdddd{|r"dt'|Dcgc]}|| c}dd |xsd!d"y77#t"$rYwxYw6Y7M#1d{7swY]xYwcc}ww)#NDEEPSEEK_API_KEYr zDEEPSEEK_API_KEY not setT)rQr;r=rSrUr<auto tool_choicezBearer zapplication/json) Authorizationz Content-TypePOSTz/chat/completions)jsonheaderszdata:z[DONE]choicesrdeltacontentr)typer tool_callsindex)idname argumentsrnfunctionrorp finish_reason)rkcallsfinishstop)rkreason)rLrMrN RuntimeErrorrQrSrUr^r=rPraise_for_status aiter_lines startswithr!r rfloads ValueError setdefaultsorted)rr;r<keypayloadrXrgtool_bufrtresplinedataobjrhchoiceritcidxslotfnis rr=zDeepSeekAdapter.streamwsjjnn/49: :ZZ ++//   $GG %+GM "$$&&-cUO#57$&==5H)H&-w!@@CG  ! ! #"..0 5d4??7#;CLM*0028#**T*C''),4  7+1r99Y'#)53CDD 99\28b8B&&!,C#..BDFDvvd|%'XT  +1rBvvf~')&zV vvk*[)R _<)9::o.#O4F=@@> '282BC2BQXa[2BCE E6+;V<+E $!$J6 *H$+LLF$<&%=\)FF7O1rE#.t m 2 )! !K==?D >>#OO[X$NO OO##+t !*% 0 !1155;5E!+z*+%%*;/0uu[/A/IT"KL!1   "%(1::aeeK&8&@DAD%dD1!!DuuVR(E!%%)NN4$>>> &tf,<=!E%dV:c!fTcl^DFE12t1K%s ,2!45)sA J CB=CJ1B?8,J%J=C? C J C  A"J-A E:9"J:GJ G%"J$G%%J;!H)H H)(J) I2I J I2J)rr7rrBrzToolExecutor | Noner.r/r9)r;rArCallable[[], None] | Noner.zAsyncIterator[str])r1r2r3r4rrr5rrrrs9#/#48#9=B5B)Brrcg}|D]_}|jd}|jd}|dvs*t|ts;|jsL|j ||da|S)u.The fast lane's view of the SHARED history: plain user/assistant text turns only. Tool-call scaffolding produced by full-brain turns (assistant messages with ``content=None`` + ``tool_calls``, and ``tool`` results) is dropped — the fast model has no tools and just needs the conversation.rrj)userrr)rNrr0r r )historyr$r%rrjs r _fast_historyrsa C uuV}%% " ( (Z-EMMO JJ9 :  Jrc6eZdZdZddZ ddZ d d dZy) FastLaneuEThe cheap, tool-less front lane over a fast model (deepseek-v4-flash). Streams spoken sentences from a compact-persona prompt; if the model emits the escalation sentinel ``<>`` (per that prompt) it signals a handoff so the caller silently reruns the turn on the full ``Brain``. Ported from the desktop's ``voice/local_brain.py`` two-lane design. ``run`` yields tuples: ``("say", sentence)`` a spoken sentence (raw, [emotion] tag intact) ``("handoff", None)`` escalate — emitted ALONE and FIRST, then the generator stops (nothing spoken, nothing to commit) ``("final", text)`` LAST on a clean finish — the full assistant text (sentinel stripped) for the caller to commit to the shared history c ||_||_yr9)r system_prompt)rrrs rrzFastLane.__init__(s *rcHd|jdgt|zd|dgzS)Nsystemrr)rr)rr user_contents rbuild_messageszFastLane.build_messages,s5"t/A/ABC()"|<=> ?rNctK|j||}t}d}d}d}|jj|d} | 23d{} | j ddk7r| |s d} ||| dz }|s/t |vr'dt| dd} |  | d{yy|j| dD]$} t | vs| jsd}d| f&7#t $rYwxYw7S#t $rYywxYw6|j} |s?t |vr7d t| dd} | ! | d{7y#t $rYywxYwy| rt | vr d}d| f|s7d t| dd} | ! | d{7y#t $rYywxYwy|jt dj}|rd |ft| dd} | ! | d{7y#t $rYywxYwy#t| dd} | ! | d{7w#t $rYwwxYwwxYww) Nr FrkrT)handoffNr?sayfinal) rrrr=rNr FAST_HANDOFFgetattrr)r r,replace)rrrrr;rfullspokerr=rr?rr+ committeds rrz FastLane.run2s&&w =#%$$Xt4( " ,b66&>V+' E 6 "!5++*VXt4F! (NN"+!( RZ 8H#x/x~~7G  E (++ !9 ,%<# M#&==?D\T1''VXt4F! (NN " D0dm#'' VXt4F! (NN "  \26<<>I **VXt4F! (NN "VXt4F! (NN "s?H8HDC% D H)C'0HH8 C8"C6#C8'H8)G??H8H5 H%H!H%$H5% H1.H50H11H55H8)rr7rr0r.r/)rrArr0r.rAr9)rrArr0rrr.zAsyncIterator[tuple])r1r2r3r4rrrr5rrrrs3 +?&?9=353+3rr)rr0r.r0)rrAr.rA)r4 __future__rrfrLretypingrrrcompilerr" MULTILINEr rrrrr7rDr0r ToolExecutorrrrr5rrrs$# 55  67   BJJ0",, ?  + ""L  M=jM=bd Ys^34 PPh NNr