L#{j bddlZddlmZddlmZddlmZddlmZm Z GddZ dZ dS) N)Document)OpenAIEmbeddings)settings)hotel_pinecone_servicehotel_private_pinecone_servicecpeZdZdefdZdefdZd deededee fdZ dee defd Z d S) HotelRAGServicereturnctjstdttjtjdS)Nz!OpenAI API key is not configured.i)modelopenai_api_key dimensions)rOPENAI_API_KEY ValueErrorrEMBEDDING_MODEL)selfs ?/var/www/html/ls-pms-bot-apis/app/services/hotel_rag_service.py_get_embeddingszHotelRAGService._get_embeddings sC& B@AA A*#2    cg}ttfD]7}|jr.|jr'||8|st d|S)NzZNo hotel Pinecone indexes configured. Set message-center and/or private index credentials.)rrpc index_nameappend get_indexr)rindexesservices r_available_indexesz"HotelRAGService._available_indexessv.0NO 4 4Gz 4g0 4w0022333 G rquerieskc Kd|D}|sgS||}||d{V}dttffd fd|D}t j|d{V}g}t} |D]} t| dgpg} | D]} t| ddpi} | dp| d }| dd|sX|| d p| d pd f}|| vr| || t|| Ō|S)zGBatch-embed queries, then search all configured hotel Pinecone indexes.cg|]?}t|t|+|@S) isinstancestrstrip).0qs r z8HotelRAGService._safe_retrieve_batch..!sAZZZAs9K9KZPQPWPWPYPYZ17799ZZZrNvectorcPKtj|j|ddd{VS)NTF)r*top_kinclude_metadatainclude_values)asyncio to_threadquery)indexr*r s r _query_indexz:HotelRAGService._safe_retrieve_batch.._query_index)sP * !%$  rc2g|]}D]}||Sr#r#)r'r*r2r3rs rr)z8HotelRAGService._safe_retrieve_batch..2sL       L ' '    rmatchesmetadatacontenttextsourceurl) page_contentr6)rraembed_documentslistfloatr/gathersetgetattrgetpopaddrr)rrr cleaned_queries embeddings query_vectorstasksresults documents seen_keysresultr5matchr6r7keyr3rs ` @@r_safe_retrieve_batchz$HotelRAGService._safe_retrieve_batchsZZgZZZ I))++))++ (99/JJJJJJJJ  d5k           '     .......$& *-%%  T TFfi44:G T T"5*d;;Ar",,y11IX\\&5I5I Y--- X 6 6 S(,,u:M:M SQST)## c"""  w!R!R!RSSSS Trdocsc d}|D]w}|jpi}|dd}|jpd}|ddp|dd}|d|d|d|d z }x|S) Nr;titlezN/Ar9r:z z z z )r6rCr<r&)rrQmultiline_stringdocr6rSr7r:s rformat_documentsz HotelRAGService.format_documentsNs  C \/RHLL%00E'-24466G,,x..K(,,ud2K2KC #    rN)r) __name__ __module__ __qualname__rrr>rr%intrrPrVr#rrr r s !1     D    --$s)--DQYN----^ T(^       rr ctS)N)r r#rrget_hotel_rag_servicer\cs   r) r/langchain_core.documentsrlangchain_openairapp.core.configrapp.services.pinecone_servicerrr r\r#rrras------------$$$$$$````````X X X X X X X X vr