Ni6rdZddlZddlZddlZddlZddlmZmZddlmZejdZ e dz dz Z e dz dz Z d Z d Zd Zd Zd!dZd eefdZd"ded efdZded eefdZdeded efdZdeded efdZdeded efdZd#deded efdZdeded efdZded efdZd!dZ e!d k(re yy)$a kg_query_before_execution.py ----------------------------- Auto-queries the Knowledge Graph before every Genesis task starts, and injects the top N relevant axioms into the agent system prompt. Usage (CLI): python3 core/kg_query_before_execution.py --query "build voice bridge" --top 5 python3 core/kg_query_before_execution.py --query "build voice bridge" --top 5 --inject Programmatic: from core.kg_query_before_execution import kg_preflight, inject_axioms_into_prompt N)datetimetimezone)Optionalz/mnt/e/genesis-systemKNOWLEDGE_GRAPHaxiomsentitiesz"qdrant-b3knu-u50607.vm.elestio.appigenesis_memorieszmodels/gemini-embedding-001returnctdz dz }|jrt|5}|D]}|j}|r|j dsd|vr+|j d\}}}|j}|jjdjd}|s|t jvs|t j|< dddyy#1swYyxYw)zILoad /mnt/e/genesis-system/config/secrets.env into os.environ if not set.configz secrets.env#="'N) _GENESIS_ROOTexistsopenstrip startswith partitionosenviron) secrets_pathfhrawlinekey_vals 7/mnt/e/genesis-system/core/kg_query_before_execution.py _load_envr!+s 8+m;L ,  *2 *yy{ts3s$"nnS1 Qiikiik'',22373bjj0&)BJJsO * * * * *sBC,C?CC(cptdD]'}tjj|}|s%|cSy)N)GEMINI_API_KEY_NEWGEMINI_API_KEYGOOGLE_API_KEY)r!rrget)varrs r _gemini_api_keyr(;s5 KIjjnnS! J defaultc t|ttfr t|St|tr;dddddd}|j j }||vr||S t|S|S#t tf$r|cYSwxYw)z>Convert val to float safely, mapping string confidence labels.gffffff?g?gffffff?g?g333333?) very_highhighmediumlowvery_low) isinstanceintfloatstrrlower ValueError TypeError)rr*mappinglowereds r _safe_floatr:Ds#U|$Sz#sscC ))+##% g 7# # :  NI& N s" A//BBtextct}|sy ddlm}|j|}|jj t |g}t|jdjS#t$r}td|Yd}~yd}~wwxYw)zCReturn embedding vector for text using Gemini gemini-embedding-001.Nr)genai)api_key)modelcontentsz[kg_query] Gemini embed error: ) r(googler=Clientmodels embed_content EMBED_MODELlist embeddingsvalues Exceptionprint)r;r> new_genaiclientresponseexcs r _embed_textrOZsG  -!!'!2==..V/  H''*1122  /u56sAA++ B 4BB vectortop_nc  ddlm}|dtdttj j dddd }|jt||d }t|d r |jn|}g}|D]}|jxsi}|j d xs|j dxsd} |j| |j ddt|j ddt|j|j dt!|j"d|S#t$$r} t'd| gcYd} ~ Sd} ~ wwxYw)z8Query Qdrant collection and return scored axiom records.r) QdrantClientzhttps://:QDRANT_API_KEY@7b74e6621bd0e6650789f6662bca4cbf4143d3d1d710a0002b3b563973ca6876g @F)urlr>timeout prefer_grpcT)collection_namequerylimit with_payloadpointscontentaxiomsourceunknown confidenceidr_rbrdrelevance_scorerfz [kg_query] Qdrant query failed: N) qdrant_clientrS QDRANT_HOST QDRANT_PORTrrr& query_pointsQDRANT_COLLECTIONhasattrr^payloadappendr:r3scorer4rfrIrJ) rPrQrSrLrMr^rhitror_rNs r _query_qdrantrsps=.;-q 6JJNN#35wx  &&- ' %,Hh$?X Ckk'RGkk),J G0DJG MM#*#*;;x#C#.w{{</M#N#(#3#*;;tS[#A     067 sD)D,, E5E E Er[ct|jj}t|jj}|syt||zt|z S)z8Simple keyword overlap score (0-1) for fallback ranking.re)setr5splitlen)r;r[ query_tokens text_tokenss r _keyword_scorerzsTu{{}**,-Ltzz|))+,K  |k) *S-> >>r)task_descriptionc g}tjsgSttjdD]} t |dd5}|D]}|j }|s t j|}|jdxs|jdxsd}|sW|j||jd|jt|jd d t|||jd dd  d d d |j#dd|d |S#t j$rYwxYw#1swY=xYw#t$r&}td|j d|Yd }~Ed }~wwxYw)z?Scan all JSONL axiom files and return top N by keyword overlap.*.jsonlutf-8replaceencodingerrorsr_r`rarbrdrerfrgNz[kg_query] Error reading z: c|d|dfS)Nrhrd)xs r z'_fallback_jsonl_query..s1%6#7<"Ir)T)rreverse) _AXIOMS_DIRrsortedglobrrjsonloadsJSONDecodeErrorr&rpstemr:rzrIrJnamesort) r{rQ candidates jsonl_pathrrrecordr_rNs r _fallback_jsonl_queryrs{J     [--i89  j79E D::bF G  sTE E#D-8BE E-E EE EE E FE<<Fcx ddlm}|||}|r=|Dcgc]1}|j|j|j|jd3c}S t|}|rt||}|r|St dn t d t||Scc}w#t $r}t d|dYd}~bd}~wwxYw) a8 Query the Knowledge Graph for axioms semantically relevant to task_description. Uses the unified Nervous System retriever (Qdrant + PG + KG files). Falls back to legacy pipeline if nervous system import fails. Returns: List of dicts: {content, source, confidence, relevance_score} r)query_knowledge_structured)top_k)r_rbrdrhz%[kg_query] Unified retriever failed (u) — falling back to legacy.NuD[kg_query] Qdrant returned 0 results — falling back to JSONL scan.u@[kg_query] Embedding unavailable — falling back to JSONL scan.) !core.nervous_system.runtime_queryrr_rbrhrIrJrOrsr)r{rQrchunkscrNrPresultss r get_relevant_axiomsrsZP+,zkg_preflight..s&DRq&DNz*.jsonrc3BK|]}|jsdywrrrs r rzkg_preflight...s'Sb 'Srrzentities.jsonlc3BK|]}|jsdywrrrs r rzkg_preflight..6s#A"bhhjA#Ar)r entity_count axiom_count timestamp)rrrrrsumrI _ENTITIES_DIRrFreadrrrr1rwr splitlinesrrnowrutc isoformat) r{rrrprrparsed entities_roots r kg_preflightr s(!!1 ;FKL!!), A !gi@EB3&D2&D#DDKE m((34tM F1#F>HG/ H;5GH2H&3H21F; 6F>> G  G G H6HHHH H#"H#&H/+H22 H>=H>c tjdtjd}|jdddd|jd d td d |jddd|j }t d|jd|jd|jd k(r?t|j}t d|dd|dt d|ddnSt|j|jtjtjjd}|d}|s t dyt d t!|d!t#|d"#D]A\}}t d$|d%|d&t d'|d(d)|d*d+|d,d-t C|j$r.t'd.|}t d/t |t d0yy)1Nz2Query Genesis Knowledge Graph for relevant axioms.z Examples: python3 core/kg_query_before_execution.py --query "voice bridge MCP server" --top 3 python3 core/kg_query_before_execution.py --query "build landing page" --top 5 --inject ) descriptionformatter_classepilogz--queryz-qTz/Task description to search for relevant axioms.)requiredhelpz--topz-nrz,Number of top axioms to return (default: 3).)typer*rz--inject store_truez6Print example of axioms injected into a sample prompt.)actionrz [kg_query] Querying KG for: 'z' (top z) uKG Stats — axioms: rz | entities: ruTimestamp — rrr)rrrzNo axioms found.zTop z axiom(s) found: rrz rr_z source: rbz | confidence: rdz | relevance: rhz.4fzBuild the voice bridge module.z--- Injected prompt preview ---z --- end ---)argparseArgumentParserRawDescriptionHelpFormatter add_argumentr2 parse_argsrJr[toprrrrrrrrwrinjectr)parserargsresultririnjecteds r _clirEs  $ $H <<F  4$NP CKM  <UW    D +DJJrs\  '  45  11H<  11J> 9  &, * #ee,chtn,!$!s!t!N??S?U? C   L$:#$:c$:$$:N3:53545v-` zFr)