i] dZddlmZddlmZmZddlmZmZm Z m Z m Z m Z eGddZ eGddZeGd d Ze Gd d e Ze Gd de Ze Gdde Ze Gdde Zy)u Genesis Nervous System — Interface Contracts ============================================= All retrievers, orchestrators, injectors, and scorers implement these protocols. PRD: _bmad-output/RLM_NERVOUS_SYSTEM_PRD.md (Story 2.1) ) annotations) dataclassfield)AnyDictListOptionalProtocolruntime_checkablecZeZdZUdZded<ded<ded<dZded <ee Zd ed <y )RetrievedChunkz6A single piece of knowledge retrieved from any source.strcontentsourcefloatrelevance_scorerintfreshness_daysdefault_factoryzDict[str, Any]metadataN) __name__ __module__ __qualname____doc____annotations__rrdictr6/mnt/e/genesis-system/core/nervous_system/contracts.pyr r s.@ L KNC$T:Hn:rr cdeZdZUdZded<dZded<edZd ed <d Zd ed <dZ ded<y)RetrievalRequestz+Parameters for a knowledge retrieval query.rquery rtop_kc gdS)N)qdrantpgkgrrrr zRetrievalRequest.s7Mrr List[str]sourcesg333333?r min_relevanceNz Optional[int] max_age_days) rrrrrr%rr,r-r.rrr r"r"s75 JE3O/MNGYNM5"&L-&rr"c^eZdZUdZded<ded<ded<eeZded <ed d Z y ) RetrievalResultz2Aggregated result from the retriever orchestrator.List[RetrievedChunk]chunksr latency_msr+sources_queriedrsources_failedc,t|jS)N)lenr2selfs r total_chunkszRetrievalResult.total_chunks,s4;;rN)returnr) rrrrrrlistr5propertyr:rrr r0r0$s8<   %d ;NI;   rr0c2eZdZdZddZddZeddZy) IRetrieverz:A single knowledge retriever (Qdrant, PG, KG files, etc.).cy)z9Retrieve relevant knowledge chunks for the given request.Nrr9requests r retrievezIRetriever.retrieve7 rcy)z;Return True if this retriever's backing store is reachable.Nrr8s r health_checkzIRetriever.health_check;rDrcy)z=Unique name for this source (e.g. 'qdrant:genesis_memories').Nrr8s r source_namezIRetriever.source_name?s rN)rBr"r;r1)r;bool)r;r)rrrrrCrFr=rHrrr r?r?3s#D    rr?c(eZdZdZddZddZddZy) IRetrieverOrchestratorz7Fan-out query to multiple retrievers and merge results.cy)z0Execute retrieval across all registered sources.NrrAs r r#zIRetrieverOrchestrator.queryIrDrcy)zRegister a retriever backend.Nr)r9 retrievers r register_retrieverz)IRetrieverOrchestrator.register_retrieverMrDrcy)z2Return health status of all registered retrievers.Nrr8s r healthzIRetrieverOrchestrator.healthQrDrN)rBr"r;r0)rNr?r;None)r;zDict[str, bool])rrrrr#rOrQrrr rKrKEsA   rrKceZdZdZddZy)IKnowledgeInjectorz:Injects retrieved knowledge into an agent's system prompt.cy)z:Return an enriched prompt with knowledge context injected.Nr)r9 base_promptresults r injectzIKnowledgeInjector.injectZrDrN)rVrrWr0r;r)rrrrrXrrr rTrTVs D rrTceZdZdZddZy)IFreshnessScorerz7Scores knowledge chunks by freshness (age-based decay).cy)z2Return freshness score 0.0 (stale) to 1.0 (fresh).Nr)r9chunks r scorezIFreshnessScorer.scorecrDrN)r\r r;r)rrrrr]rrr rZrZ_s A rrZN)r __future__r dataclassesrrtypingrrrr r r r r"r0r?rKrTrZrrr ras#(II  ;; ; '' '          "  X        x  r