idZddlmZddlZddlmZmZddlmZmZddl m Z ddl m Z ddl mZmZmZej"eZd Zd Zd Zdd ZGd dZy)u) Qdrant Retriever — RLM Nervous System Story 2.2 ================================================ Wraps core/kb/qdrant_store.py and core/kb/embedder.py to expose IRetriever-compatible semantic search over the genesis_memories collection. PRD: _bmad-output/RLM_NERVOUS_SYSTEM_PRD.md (Story 2.2) ) annotationsN)datetimetimezone)ListOptional) embed_text) _get_client) IRetrieverRetrievalRequestRetrievedChunkgenesis_memories)contenttext created_atc|jt}|sy t|ttfr/t j t |tj}n.t jt|jdd}t jtj|z }td|jS#t$rYywxYw)z Compute how many days old this point is. Reads payload['created_at'] (ISO-8601 string or Unix epoch float). Returns 0 if the field is absent or unparseable. r)tzZz+00:00)get_CREATED_AT_FIELD isinstanceintfloatr fromtimestamprutc fromisoformatstrreplacenowmaxdays Exception)payloadrawcreateddeltas =/mnt/e/genesis-system/core/nervous_system/qdrant_retriever.py_freshness_daysr's ++' (C  cC< (,,U3ZHLLIG,,SX-=-=c8-LMG  -71ejj!! sB/C CCc>eZdZdZefddZddZd dZed dZ y) QdrantRetrievera] IRetriever implementation backed by Qdrant's genesis_memories collection. Lifecycle: r = QdrantRetriever() # collection defaults to genesis_memories r.health_check() # True if collection reachable with >0 points chunks = r.retrieve(request) # semantic search, filtered by min_relevance c0||_t|_yN) _collectionr _client)self collections r&__init__zQdrantRetriever.__init__<s%"} c j|jdd} t|} |j j |j||jd}t|dr |jn t|}g}|D]}t|dd } | |jkr |jxsi} d } tD]"} | j!| } | st#| } n| j%Dcic]\}}|tvr||}}}|j't)| |j*t-| d t/| | |j1d d|S#t$r"}tj d|gcYd}~Sd}~wwxYw#t$r"}tj d|gcYd}~Sd}~wwxYwcc}}w)a Embed query text and search Qdrant for the top_k most similar points. Filters out any points whose score is below request.min_relevance. Results are returned sorted descending by relevance_score. BB test targets: - Returns non-empty list for a known query ("GHL integration") - All returned chunks have relevance_score >= request.min_relevance - Returns [] for an absurdly high min_relevance (e.g. 0.9999) - Truncates query at 512 chars before embedding WB test targets: - _freshness_days(payload) called and value stored in freshness_days - metadata dict excludes content/text keys Niz/QdrantRetriever.retrieve: embed_text failed: %sT)collection_namequerylimit with_payloadz1QdrantRetriever.retrieve: Qdrant query failed: %spointsscoreg)rsourcerelevance_scorefreshness_daysmetadatac|jSr+)r<)cs r&z*QdrantRetriever.retrieve..s !"3"3r1)keyreverse)r4rr!loggererrorr- query_pointsr,top_khasattrr7listgetattr min_relevancer"_CONTENT_FIELDSrritemsappendr source_nameroundr'sort)r.request query_textvecexcresponser7chunkspointr8r"rfieldvalkvr>s r&retrievezQdrantRetriever.retrieveBs"]]4C(  Z(C ||00 $ 0 0mm! 1H%,Hh$?T(^') E"5'37Ew,,,!MM/RGG( kk%(!#hG  ")AO+1H MM#++$)%O#27#;%  + @  3T B i  LLJC PI   LLLc RI 0s@ E3F'F/ E>E93E>9E> F, F'!F,'F,c |jj|j}|j}|duxr|dkDS#t$r }t j d|Yd}~yd}~wwxYw)z Return True if genesis_memories collection is reachable and non-empty. BB test: returns True against live Elestio Qdrant (28,435+ points). WB test: catches exception and returns False without raising. )r3Nrz'QdrantRetriever.health_check failed: %sF)r-get_collectionr, points_countr!rDwarning)r.infocountrUs r& health_checkzQdrantRetriever.health_checkse <<..t?O?O.PD#'#4#4E$2 2  NNDc J srxsX#'!',VV   8 $(% *l,l,r1