*QjKp FUdZddlmZddlZddlZddlZddlZddZddZ ddZ ddZ ddZ d dd Z d dd Zd Zd ZddZdaddZddZddZdddZddZddZhdZddZ d ddZedddddd d!dd"d#ged$d%dd&ddd'dd(dged)d*dd+ddd,dd-dged.d/ied0d1d2dd3did2ged4d5dd6ddd7ddgd8d9d:d;gedd?ied@dAdBd dCdigedDdEddFdddGddHdIdJgedKdLdIddMdidIgedNdOiedPdQddRdddSdddTddUddgedVdWdddXdidgedYdZied[d\d]dd^did]ged_d`d]ddadid]gedbdciedddeddfdd dgdgdhdiddjddkdldidmdnddoddpdlgdqd drdgdsdiddtdddugdvd:dwdlgdqdxgdxedydzied{d|ddd}didggZd~ed<ddZddZddZ ddZ!ddZ"ddZ#ddZ$ddZ%ddZ&ddZ'ddZ(ddZ)ddZ*ddZ+ddZ,ddZ-ddZ.ddZ/ddZ0ddZ1ddZ2ddZ3ddZ4ddZ5ddZ6ddZ7ided$ed)e d.e!d0e"d4e$de&d@e'dDe)dKe+dNe,dPe-dVe/dYe0d[e1d_e2e3e5e6e7dZ8ddZ9y)uTool definitions + executors for the gateway brain. Two backends, both reached over plain async HTTP (httpx): * The memory service (Graphiti/FalkorDB on Railway) — same endpoints and bearer-header shape as ``voice/memory_control.py`` / ``voice/tasks_control.py`` (MEMORY_API_URL + MEMORY_API_KEY, group namespace MEMORY_GROUP). * Supabase (the Ultron master DB) via PostgREST RPCs — the exact call shape copied from ``dashboard/server.js``: POST ``{SUPABASE_REST}/rpc/`` with ``{apikey, Authorization: Bearer }`` headers. Tools exposed to the LLM (OpenAI function-calling schema, ``TOOL_SPECS``): memory_search, memory_remember, task_add, tasks_list, task_done, ultron_leads, ultron_team_activity, ultron_stats, ultron_contact_stats. The HTTP is funnelled through ``_mem_request`` and ``_rpc`` so tests can monkeypatch those two functions and never touch the network. ) annotationsNc`tjjddjdS)NMEMORY_API_URL/osenvirongetrstrip;/Users/ahmed/devFolder/Ultron/claude-voice/gateway/tools.py_mem_urlrs# ::>>*B / 6 6s ;;rcBtjjddS)NMEMORY_API_KEYrr r r r rr_mem_keyr!s ::>>*B //rcBtjjddS)N MEMORY_GROUPahmedrr rr_groupr%s ::>>.' 22rc`tjjddjdS)N SUPABASE_RESTrrrr rr_restr)s" ::>>/2 . 5 5c ::rcBtjjddS)NSUPABASE_SECRETrrr rr_sup_keyr-s ::>>+R 00rcKddl}t}|s td||z}|r%|dtjj |zz }ddi}t } | rd| |d<|j| 4d{} | j|||||nd d{} | j| j} | rtj| nicdddd{S7k7L7 #1d{7swYyxYww) z@Call the memory service. Returns parsed JSON ({} on empty body).rNzMEMORY_API_URL not set? Content-Typeapplication/jsonBearer Authorizationtimeoutheadersjson) httpxr RuntimeErrorurllibparse urlencoder AsyncClientrequestraise_for_statuscontentr)loads) methodpathparamsbodyr&r*baseurlr(keyclientrespraws r _mem_requestr>2s  :D 344 +C  sV\\++F33312G *C %,SE?    11V^^FC151At$MM ll"%tzz#2 211M2111sZA6C?8C$9C?<C*C&8C* C?C(C?&C*(C?*C<0C3 1C<8C?objectc~Kddl}t}|s tdt}|d|dd}|j |4d{}|j |d|||xsi d{}|j |jcdddd{S7Z777 #1d{7swYyxYww) uDCall a Supabase PostgREST RPC — same shape as dashboard/server.js.rNzSUPABASE_REST not setr#r")apikeyr$r!r%z/rpc/r')r*rr+rr/postr1r)) fnargsr&r*restr:r(r;r<s r_rpcrFJs 7D 233 *C13G   11V[[D6rd!3W&*jb!22 yy{ 21122111sZAB=B"B= !B(,B$-#B( B=B&B=$B(&B=(B:.B1 /B:6B=) maybemightpossiblyperhapszi thinkznot surearoundroughly approximatelyzor soprobablyzcould bezi guess) signed confirmed definitelyzfor sureagreedpaid committed finalizedz locked intextc|xsdjtfdtDrytfdtDryy)Nrc3&K|]}|v ywNr .0wts r z_confidence..es !5a165g?c3&K|]}|v ywrYr rZs rr^z_confidence..gs "6a166r_g?g?)lowerany_FIRM_HEDGE)rVr]s @r _confidencerecs: A !5 !! "6 "" rc|ay)zBmain registers an async callback fired after any routine mutation.N) _ROUTINE_HOOK)rCs rset_routine_hookrhss MrcbKt td{yy7#t$rYywxYwwrY)rg Exceptionr rr_notify_routine_changerkys5  / ! !! !   s(/  / ,/,/numbercdjdt|xsdD}|syddj|ddzS)uA phone number rendered for the model/logs so a full number is never echoed or read aloud — only the last three digits survive.rc3BK|]}|js|ywrY)isdigit)r[chs rr^zmask_number..sD"3Brzz|R"3sz (no number)zending  N)joinstr)rldigitss r mask_numberrvsAWWD#fl"3D DF  sxxrs , ,,rc^K tddt|dd{}g}|jdgD]O}|jdxs|jd xsdj }|s<|j d |Qd j |S7{#t$rYywxYww) uA short 'index' of recent long-term memories for the system prompt. The desktop reads a local memory/MEMORY.md; the phone has no filesystem memory, so we summarise the newest facts from the service instead. Never raises — returns "" on any failure.GETz /memories)grouplimitr6NrmemoriesrVfact-  )r>rrjr stripappendrs)rzoutlinesmtxts rmemory_index_textrs   28(U(KMM E WWZ $uuV}3f 3::<  LL2cU $% 99U M s>B-BBBAB-5'B-B B*'B-)B**B-cK tdddtid{}|jdxsdj S7'#t$rYywxYww)zBAhmed's north-star profile from the memory service. "" on failure.rxz/valuesryr{NrrV)r>rrjr r)rs r values_textrsX  7FH:MNN GGFO !r ( ( **O s7AA AA #AA AAAAc K|xsdj}t|sy tdd|tddt |dd{y7#t $rYywxYww) zFire-and-forget passive save after an exchange, with a simple v1 salience gate. Mirrors the desktop's auto-remember intent: only durable- looking utterances are stored, as observations. Never raises.rNPOST /remembergateway observationrVrysourcekind confidencer7)r _is_salientr>rrerj) user_textr]s rpassive_rememberrsm b!A q> 6;)*VX+4m/:1~!?@ @ @    s:!A#(A A AA#A A A#A  A#> thank you good night good morninghinookyobyeheyhmmsupyescoolnicenopeokaystopwaityeahhellocheersthanks goodnightc|j}|jjdt|dkrytvry|j drd|vrt fddDsyy) zCheap v1 gate: keep declarative, substantive utterances; drop greetings, one-liners, and bare questions (those are asks, not facts to store).z!.Fr rqc3&K|]}|v ywrYr )r[clows rr^z_is_salient..s0KIAHIr_)rememberz note thatzfor the recordT)rrar len_TRIVIALendswithrb)rVr]rs @rrrsl A '')  4 C 1v{ hzz#3!8C0KI0K-K rnamec$d||d||xsgdddS)Nfunctionr?)type propertiesrequired)r description parameters)rrr )rrrrs r_fnrs.& ($N   r memory_searchu Search Ahmed's long-term memory by meaning and get back connected facts. Use whenever the answer might depend on something he told you before — a person, client, meeting, number, price, project or plan. Read every fact returned; the specific one is often not first.stringz$What to recall, in natural language.)rrintegerz!Max facts to return (default 10).)querykrmemory_rememberuStore a durable new fact Ahmed just told you (a preference, a person, a decision, a number). Saving is quick — acknowledge and move on. Do NOT use for transient chit-chat.zOne clear fact to remember.z1fact|preference|event|observation (default fact).)rVrtask_adduuAdd a task or reminder to Ahmed's shared, cross-device list. Use for 'remind me to…', 'add a task', 'I need to…'.z0Short imperative task, e.g. 'Call the supplier'.zdISO local time for a timed reminder, e.g. 2026-07-08T17:00:00 (Asia/Riyadh). Omit for a plain to-do.)rVdue tasks_listz[List Ahmed's open tasks and reminders. Use for 'what are my tasks', 'what do I have to do'. task_donez=Mark a task or reminder complete by its id (from tasks_list).idz The task id. ultron_leadszSearch the Ultron lead database (Saudi business leads scraped from Google Maps). Use when Ahmed asks about leads, businesses, or prospects in a city.z.Free-text search (company/category), optional.z+City filter, e.g. 'Al Khobar'. Default all.)allrmobileu?'mobile' = has a WhatsApp-able Saudi 05… number. Default all.)renumr)rcity phone_filterultron_team_activityzWho on the agency team contacted which leads recently (last 30 days). Use for 'what has the team been doing', outreach activity. ultron_statszThe Ultron dashboard funnel/summary numbers (total leads, verified emails, phones, recent haul). Use for 'how many leads do we have'.ultron_contact_statszsOutreach contact volume per day over the last N days (per client). Use for 'how much outreach did we do this week'.daysz&Look-back window in days (default 30). save_placeuRemember a place by a short label so Ahmed can later say 'navigate to the office'. Use when he names a location — 'this is my gym', 'save home as…'. Store an address, or GPS as 'lat,lng' (e.g. from a new-location prompt).z'Short name, e.g. 'work office', 'home'.z>A street address, OR 'lat,lng' coordinates like '26.30,50.20'.)labeladdress_or_coordsrr get_placez]Look up a saved place by its label. Use before navigating or when he refers to a named place.zThe place label. list_placesz;List every place Ahmed has saved (labels + where they are). save_contactuRemember a person's phone number so Ahmed can later say 'call my father'. Use when he gives you a name + number. Numbers are private — just confirm you saved it, never read the number back.z#Who this is, e.g. 'father', 'Sara'.zTheir phone number.z1Optional relationship, e.g. 'family', 'supplier'.)rrlrelation get_contactuLook up a saved contact by name. Returns the name/relation and a MASKED number (last digits only) — to actually dial, use call_contact.zWho to look up. list_contactszRList everyone in Ahmed's saved phonebook (names + relations; numbers stay masked). sms_watch_adduAdd a sender to the SMS watch-list — texts from it get forwarded to you to summarise/act on. Senders like 'SNB-AlAhli', 'Barq', or '*' for ALL senders.senderz#Sender id, or '*' for every sender.sms_watch_removez(Remove a sender from the SMS watch-list.zSender id to stop watching.sms_watch_listz1List the SMS senders currently on the watch-list.routine_createuCreate a routine that fires automatically — e.g. 'each morning at seven give me a brief of my tasks' (time 07:00 daily → speak_brief tasks) or 'when I get to the office, read my tasks' (place trigger). Resolve times yourself in 24h Asia/Riyadh.z(Short unique name, e.g. 'morning brief'.zWhen it fires.timeplaceboth)rrz24h time 'HH:MM' (time/both).arrayrz1Weekdays mon..sun; omit or ['daily'] = every day.)ritemsrz!A saved place label (place/both).)ratrr)rrrrzWhat Jarvis does when it fires. speak_briefsay_textrunz!For say_text/run: what to say/do.taskszFor speak_brief: which brief.)rrVbriefrtriggeraction routine_listz;List Ahmed's routines (name, when they fire, what they do).routine_cancelz Cancel/delete a routine by name.zThe routine name.z list[dict] TOOL_SPECSc VKt|jddj}|sytdt t |jdxsdd}t dd ||td  d{}|jd g}|sy g}|D]w}|jdrdnd}|jdxsddd}|jd|jdd||rd|dndzd|jdddzydj|S7w)Nrrzmemory_search failed: no query.r rxz/search)qrryr{factsz (nothing relevant in memory yet)linkedz [connected] refers_tor~r}z (re: ) [id rr ]r) rtr rmaxminintr>rrrs) rDrrrrrftagwhens r_memory_searchrs. DHHWb !"((*A 0 As3txx}*+R01AUI*+!fh$GI IC GGGR E 1 E  !hnRk"(b#2. r!%%+,SE2-1'$q)r;dC 0134 5 99U  IsAD'?B)D)c "Kt|jddj}|syt|jdxsdjxsd}tdd|t d|t |d  d{y 7w) NrVrz memory_remember failed: no text.rr}rrrrrz Remembered.)rtr rr>rre)rDrVrs r_memory_rememberrs txx# $ * * ,D 1 txx)6 * 0 0 2  ???  ?sBBB Bc:Kt|jddj}|sy|tdd}|jdrt|d|d<t dd| d{d ||jdr d |dd zSdzS7(w) NrVrztask_add failed: no task text.jarvis)rVryrrrz/taskrzAdded: z (reminder set for r)rtr rrr>)rDrVr7s r _task_addr s txx# $ * * ,D /68x @D xx$u+&U vwT 222dV 7;xx$T%[M3 PQLN PQ3sA.B0B1)Bc ~Ktdddtid{}|jdg}|syg}|D]k}|jdrd|ddd jd d d nd }|j d|jdd |d|jdddmdj |S7w)Nrxz/tasksryr{rzNo open tasks.rz (due Trqrrr~rVrrr rr)r>rr replacerrs)_argsrrrr]rs r _tasks_listrsUHgvx5HI IC GGGR E  E %%,5#2..sC89;$&  r!%%+,SEdC8H7IKL 99U  JsB=B;BB=cKt|jddj}|sytdd|t dd{y7w) Nrrztask_done failed: no task id.rz /task/done)rryrz Marked done.)rtr rr>r)rDtids r _task_donersL dhhtR ! ' ' )C . v|vx2P QQQ RsA A A AcPt|tr|St|tr|gSgS)zERPCs return either a list of rows or a single JSON object. Normalise.) isinstancelistdict)results r_rowsrs'&$ &$x IrcKt|jddj}t|jdxsdjxsd}t|jdxsdjxsd}td|d|dd|dddd d d{}t |}|sy g}|ddD]}|jd xs|jd xsd}|g} |jdr| j t|d|jdr| j t|d|jdr| j t|d|j ddj | zt|ddj |zS7w)Nrrrrr search_leadsF r) r p_categoryp_cityp_fresh p_verifiedp_phonep_clientp_assigned_onlylimoffzNo matching leads.companyrz (unnamed)categoryphoner~u · z leads: r)rtr rrFrrrsr) rDrrr)resrowsrrrpartss r _ultron_leadsr.s DHHWb !"((*A txx(5 ) / / 1 :UD (1E 2 8 8 : CeE^ et%U! & C :D # E #2YuuY?155=?K 55  LLQz]+ , 55= LLQvY ( 55> LLQwZ ) TFKK../$i[ "TYYu%5 55) sB$G&F='DGcKtdddid{}t|}|sytj|dddS7,w)Nget_team_activityp_range30dz%No team activity in the last 30 days.F ensure_ascii)rFrr)dumps)rr*r+s r_ultron_team_activityr7sF(9e*<= =C :D 6 ::d3Bie 44 >sAA-AcnKtdddid{}tj|dddS7w)Nget_dashboard_statsr1rFr4)rFr)r6)rr*s r _ultron_statsr;s9*Y,>? ?C ::c .u 55 @s 535cKt|jdxsd}tdd|did{}tj|dddS7w) Nrget_contact_statsr1dFr4r:)rr rFr)r6)rDrr*s r_ultron_contact_statsr@sW txx%2 &D(9aj*AB BC ::c .u 55 Cs3AAAc>t|xsdjddjdDcgc]}|s| }}t|dk7ry t |dt |d}}d|cxkrd krnyd |cxkrd kr||fSyycc}w#t $rYywxYw) z6Parse 'lat,lng' (or 'lat lng') into floats, else None.rrq,NrriZiL)rtrsplitrfloat ValueError)spr-latlngs r _parse_coordsrMsAG ,,S#6<save_place failed: need a label and an address or coordinates.rKrLaddresszSaved ''.)rtr rrM phone_storeput_item KIND_PLACE)rDrwherepayloadcoordss r _save_placerWs "% & , , .E ,b1 2 8 8 :E Oe$G 5 !F )/&"    {55w ??? UG2 @sB!B0#B.$ B0c|jdr|jdd|dSd|vr#d|vr|jdd|dd|dSt|jddS)NrOrz: rKrL, r)r rt)rJs r _fmt_placerZ!swuuY%%.!AiL>22 zeqj%%.!AeH:R%z:: quuWb! ""rcKt|jddj}tjtj |d{}|sd|dSt |S7w)NrrzNo place saved as 'rP)rtr rrQget_itemrSrZ)rDrrJs r _get_placer])sa "% & , , .E"";#9#95AAA $UG2.. a= BAA-A+A-cKtjtjd{}|sydjd|DS7w)NzNo places saved yet.rc38K|]}dt|zywr~N)rZ)r[rJs rr^z_list_places..5s95aTJqM)5)rQ list_itemsrSrsrrs r _list_placesre1s@(()?)?@ @E % 99959 99 A'A A  A cKt|jddj}t|jddj}|r|sy||d}t|jddj}|r||d<tjtj |d{d|dS7 w) Nrrrlz.save_contact failed: need a name and a number.)rrlrzSaved .)rtr rrQrR KIND_CONTACT)rDrrlrUrels r _save_contactrk9s txx# $ * * ,D (B' ( . . 0F v?v.G dhhz2& ' - - /C !    {77 AAA D6 BsB4C6C7 Cc |jdr d|ddnd}|jd|dt|jddS)Nrz (rrru — rl)r rv)rrjs r _fmt_contactrmFsO#$55#4Bq}oQ "CeeFm_SE{15523F'G&H IIrcKt|jddj}tjtj |d{}|sd|dSt |S7w)NrrzNo contact saved for 'rP)rtr rrQr\rirm)rDrrs r _get_contactroKsa txx# $ * * ,D"";#;#;TBBA 'vR00 ? Cr^cKtjtjd{}|sydjd|DS7w)NzNo contacts saved yet.rc38K|]}dt|zywra)rm)r[rs rr^z!_list_contacts..W;UTLO+Urb)rQrcrirsrds r_list_contactsrsSs@(()A)AB BE ' 99;U; ;; CrfcKt|jddj}|sytjtj d|id{|dk(ryd|dS7w)Nrrz sms_watch_add failed: no sender.*zWatching all SMS senders now.zWatching texts from rh)rtr rrQrRKIND_SMS_WATCH)rDrs r_sms_watch_addrw[sj (B' ( . . 0F 1   {99Hf;M NNN }. !& ++OsAA+A)A+cKt|jddj}|sytjtj |d{}|rd|dS|dS7w)Nrrz#sms_watch_remove failed: no sender.zStopped watching rhz wasn't on the list.)rtr rrQ delete_itemrv)rDrns r_sms_watch_remover{esj (B' ( . . 0F 4%%k&@&@&IIA,- vha (RfX=Q3RR JsAA*A(A*cKtjtjd{}|syddjd|DzS7"w)NzThe SMS watch-list is empty.z Watching: rYc3RK|]}t|jdd!yw)rrN)rtr )r[is rr^z"_sms_watch_list..qs!#LeCh(;$""  #c&k#2&'r**eeFm_JtfE$ 88rcKt|jddj}|jd}|jd}|r t|trt|tsy|jddvry|jdd vry |dd vr*t|jd djsy |ddvr*t|jddjsyt j t j|||dd{td{d|dS77 w)Nrrrrz.rrrb)rQrcrrsrds r _routine_listrs@(()A)AB BE ! 99;U; ;; CrfcKt|jddj}|sytjtj |d{}t d{|rd|dSd|dS7$7w)Nrrzroutine_cancel failed: no name.z Cancelled 'rPzNo routine named ')rtr rrQryrrk)rDrrzs r_routine_cancelrs{ txx# $ * * ,D 0%%k&>&>EEA """%&[b !I.@b,II F"s$AA?A;A?(A=)A?=A?)rrrrcKtj|}|d|dS ||xsid{S7#t$r}|dt|ddcYd}~Sd}~wwxYww)zEDispatch a tool call. Returns a compact string for the model to read.Nzunknown tool ''z failed: ) _EXECUTORSr rjrt)rrDrCes rexecutersk  B zvQ''0 ^### 0yQ//0s=A 757A 7 AAAA AA )returnrt)NNg4@) r4rtr5rtr6 dict | Noner7rr&rGrr)Ng>@)rCrtrDrr&rGrr?)rVrtrrG)rNone)rlrtrrt)r)rzrrrt)rrtrr)rVrtrboolrY) rrtrrtrrrzlist[str] | Nonerr)rDrrrt)rrrrt)rr?rr)rIrtrztuple[float, float] | None)rJrrrt)rrrrt)r,rrrt)rrtrDrrrt):__doc__ __future__rr)r urllib.parser,rQrrrrrr>rFrdrcrergrhrkrvrrrrrrr__annotations__rrr rrrr.r7r;r@rMrWrZr]rerkrmrorsrwr{rrrrrrrr rrrs.&# <03;1 .2+/(,.*.(.!&.26.0-1 $)/$ ! 0  -&+    (&* " .2 " I$"HJ AC D    +#!>@"!TV W   ;"!SU!!/0 1    "   G @A  $"RT"!NP"*":*FG  H    E   M   ; )!IK L    $"KM'//PQ R %& (  # 84FGH    E   A#!FH$5JK&&34 5    L (3DEF        H#HJ K  2 H5RST  ;   '#!KM%$4*2*C"E(0/N P*1VX