TjY)dZddlmZddlZddlZddlZddlmZddlZddl m Z ddl m Z ddl mZddlZddlZddlZej$j'dd Zd eZer1ej,ej/j1dd nd Ze d d d e ddZej7d*d+dZej7 d, d-dZej7 d. d/dZej7d0dZej7d1dZ ej7d2dZ!ej7d3dZ"ej7d2dZ#ej7d1dZ$ej7d4dZ%ej7d1dZ&ej7d5dZ'ej7d6dZ(ej7d7dZ)ej7d8d Z*ej7d9d:d!Z+ej7d9d;d"Z,ej7d9dd&Z3Gd'd(Z4d>d)Z5y)?udMCP (Model Context Protocol) endpoint for the Jarvis memory service. Mounts a SINGLE streamable-HTTP MCP endpoint onto the same FastAPI app that serves the REST API, so Claude Code and the claude.ai app can share the one memory brain. The tools are THIN wrappers that call app.py's OWN functions in process (never over HTTP to ourselves, no logic duplicated). Two doors, same endpoint (auth enforced HERE, before the MCP handler runs): • POST /mcp — Authorization: Bearer $MEMORY_API_KEY (Claude Code --header) • POST /mcp/ — token = first 32 hex of sha256(MEMORY_API_KEY) (path-secret door for claude.ai custom connectors, whose UI can't send custom headers; OAuth is overkill for a personal service). If MEMORY_API_KEY is empty, BOTH doors are open (matches app.py `_auth`). Wiring (in app.py): `import mcp_server`, run `mcp_server.lifespan()` from the FastAPI lifespan (the session manager needs its task group running), and call `mcp_server.mount(app)` to attach the two routes. Nothing else changes. ) annotationsN)asynccontextmanager)FastMCP)TransportSecuritySettings)RouteMEMORY_API_KEYzBearer z jarvis-memoryTF)enable_dns_rebinding_protectionuAhmed's shared long-term memory (a fact graph). Search it before answering from scratch, save durable facts, and checkpoint projects so any Claude — Code or app — can pick up where the last one stopped.)stateless_http json_responsetransport_security instructionsctKddltjjfdd{S7w)zISearch shared memory: salience-ranked facts plus their connected context.rNc:jdddtS)NF)qkgrouphops include_stale authorization)search_BEARER)apprquerysG/Users/ahmed/devFolder/Ultron/claude-voice/memory-service/mcp_server.pyzmemory_search..Gs! Uat!).gGranyio to_threadrun_sync)rrrs``@r memory_searchr$Bs7)) GH HH Hs ,868cKddl|xsgDcgc] }t|jsd|i"}}j||xsdd|xsdtj j fdd{Scc}w7 w)zRSave one durable fact; `entities` are names (people/companies/projects) to attach.rNnamefactmcptextkindsourceentitiesc2jtSN)rrememberrrbodysrrz!memory_remember..T T 9r)rstrstripRememberr!r"r#)r*r+r-eentsrr3s @@rmemory_rememberr:Ks"*.b. D.ASV\\^VQK.D D <.br4r)rr6r7r!r"r#)r?done next_stepsdetailsr*rr3s @@rproject_checkpointrEWs gY0gj\ KD7==? !GMMO$%% <.js D Hrr )r&rs`@r entity_lookuprIes4))HJ JJ Js +646cxKtjjtjd{S7w)z1List the files in the shared Jarvis Drive folder.N)r!r"r#drive list_filesrr drive_listrNns)))%*:*:; ;; ; 1:8:cdKtjjfdd{S7w)zSRead a text/csv/json file from the shared Jarvis Drive folder. `file` = name or id.c.tjSN)rK read_filefilesrrzdrive_read..ws%//$2GrNr!r"r#rTs`r drive_readrWts%))*GH HH H &0.0chKtjjfdd{S7w)zBCreate or overwrite a text file in the shared Jarvis Drive folder.c0tjSrR)rK write_file)contentr&srrzdrive_write..}s%2B2B42QrNrV)r&r\s``r drive_writer]zs%))*QR RR R '202cdKtjjfdd{S7w)zORemove (trash) a file from the shared Jarvis Drive folder. `file` = name or id.c.tjSrR)rK delete_filerTsrrzdrive_delete..s%2C2CD2IrNrVrTs`r drive_deleterbs%))*IJ JJ JrXcxKtjjtjd{S7w)z=List Ahmed's Smartlead cold-email campaigns and their status.N)r!r"r# smartlead campaignsrMrrsmartlead_campaignsrfs))))*=*=> >> >rOcdKtjjfdd{S7w)zMAnalytics for one campaign (sent/open/reply/bounce). `campaign` = id or name.c.tjSrR)rdcampaign_statscampaignsrrz*smartlead_campaign_stats..s)2J2J82TrNrVrjs`rsmartlead_campaign_statsrls%))*TU UU UrXcxKtjjtjd{S7w)zCList sender inboxes and their warmup reputation / daily send limit.N)r!r"r#rdsendersrMrrsmartlead_sendersros))))*;*;< << .s)2K2KD2QrNrVrss`rsmartlead_create_campaignrts%))*QR RR RrXchKtjjfdd{S7w)zSet a campaign's email sequence. sequence_json = JSON array of steps {seq_number, seq_delay_details:{delay_in_days}, subject, email_body}.c0tjSrR)rd set_sequencerk sequence_jsonsrrz(smartlead_set_sequence..s &&x?rNrVrxs``rsmartlead_set_sequencerzs1))?A AA Ar^chKtjjfdd{S7w)z|Add leads to a campaign. leads_json = JSON array of {email, first_name, last_name, company_name, ...}. Max 100 per call.c0tjSrR)rd add_leadsrk leads_jsonsrrz%smartlead_add_leads..s ##Hj9rNrVr~s``rsmartlead_add_leadsrs-))9; ;; ;r^chKtjjfdd{S7w)zEStart / pause / stop a campaign. action = 'start' | 'pause' | 'stop'.c0tjSrR)rdcampaign_control)actionrksrrz,smartlead_campaign_control..s **8V >> >r^clKtjjfdd{S7w)uEscape hatch — call ANY Smartlead REST endpoint. method=GET/POST/DELETE, path after /api/v1 (api_key added for you), body_json = JSON body for writes.c2tjSrR)rdraw) body_jsonmethodpathsrrzsmartlead_api..s fdI6rNrV)rrrs```r smartlead_apirs-))68 88 8s (424chKtjjfdd{S7w)uRun SQL against the Ultron master DB (18k+ leads + CRM: clients, contacts, users…). SELECT to read/aggregate, INSERT/UPDATE to write. Irreversible statements (drop/truncate, WHERE-less delete/update) need confirm='yes'.c0tjSrR) ultron_dbr)confirmsqlsrrzultron_query..s)//#w2OrNrV)rrs``r ultron_queryrs% ))*OP PP Pr^cdKtjjfdd{S7w)uxInspect the DB: no table → all tables with row counts; with a table → its columns. Use before composing a query.c.tjSrR)rschematablesrrzultron_schema..s)2B2B52IrNrVrs`r ultron_schemars%))*IJ JJ JrXcKtj4d{ddddd{y77#1d{7swYyxYww)zIEnter from the FastAPI lifespan so the session manager's task group runs.N)session_managerrunrMrrlifespanrs3""$$ %$$$$$$s<A6A: A8AAA A A AcFtsy|tj|tS|j dgDcic]2\}}|j j |j 4}}}tj|j ddtScc}}w)NTheadersrr )API_KEYhmaccompare_digest PATH_TOKENgetdecodelowerr)scopetokenrvrs r _authorizedrs  ""5*55 99Y35341axxz!188:-3 5   w{{?B? II5s7BcdK|dddgdd{|dddd{y77w)Nzhttp.response.starti)s content-typesapplication/json)r@statusrzhttp.response.bodys{"error":"unauthorized"})r@r3rM)sends r_rejectrsK -BCE FFF ,6QR SSSFSs0,0.00ceZdZdZddZdZy)_DoorzHASGI gate in front of the shared MCP handler. mode: 'header' or 'token'.c||_yrR)mode)selfrs r__init__z_Door.__init__s  rc2K|ddk7rt|d{y|jdk(r!|jdijdnd}t||st|d{ytj |||d{y7w7'7 w)Nr@httpr path_params)rrrrrhandle_request)rrreceiverrs r__call__z_Door.__call__s =F "$-   II(="-11':.2 5%($-   ,,UGTBBB Bs4BBAB+B,B B BBBN)rr5)__name__ __module__ __qualname____doc__rrrMrrrrsR Crrc|jjjtdt d|jjjtdt dy)z=Attach both MCP doors to an existing FastAPI / Starlette app.z/mcpheaderz /mcp/{token}rN)routerroutesappendrr)rs rmountrsHJJU65?;<JJU>5>BCr))rr5rintreturndict)r'N)r*r5r+r5r-zlist[str] | NonerrrR) r?r5rBr5rCr5rD str | Nonerr)r&r5rr)rr5)rUr5rr5)r&r5r\r5rr5)rkr5rr5)r&r5rr5)rkr5ryr5rr5)rkr5rr5rr5)rkr5rr5rr5)r )rr5rr5rr5rr5)rr5rr5rr5)rr5rr5)rrrbool)rNone)6r __future__rhashlibros contextlibrr!mcp.server.fastmcprmcp.server.transport_securityrstarlette.routingrrKrdrenvironrrrsha256encode hexdigestrr(toolr$r:rErIrNrWr]rbrfrlrortrzrrrrrstreamable_http_app_rrrrrrrMrrrs&# * &C#  **..)2 . G9 CJ^W^^GNN, - 7 7 9#2 >r   1(-/ Q.H H177;;$4;@D; ;37 ;&0 ;<@ ;  ;J J< < I I S S K K ? ? V V = = S S A A; ;> > 8 8Q QK K%% JT CC$Dr