3fiddlZddlmZmZmZmZddlmZddl m Z ddl m Z ddl mZddlmZddlmZmZdd lmZerdd lmZerdd lmZGd d eZy) N) TYPE_CHECKINGAnyOptionalSequence) exceptions) Credentials) Callbacks)Document)BaseDocumentCompressor) ConfigDictField)get_client_infodiscoveryengine_v1alphac eZdZUdZedZeed< edZeed< edZ eed< ed Z eed < ed Z e ed < ed Z eed< edZeeed< edZeeed< edZeeed< edZeeed< edZee ed< dZeed<deffd Zd!dZdedeedeefdZ d"deededeedeefdZedd Z xZ!S)# VertexAIRankaDocument compressor using Vertex AI Ranking API. Inherits from [`BaseDocumentCompressor`][langchain_core.documents.compressor.BaseDocumentCompressor]. Reranks documents based on relevance to a query using Google's semantic ranking model. N)default project_idglobal location_iddefault_configranking_configzsemantic-ranker-512@latestmodel top_nF!ignore_record_details_in_responseid_field title_field credentialscredentials_pathtimeoutclientkwargsc t|di||jd|_|js|j |_yy)zwInitialize the Vertex AI Ranker. Configures ranking parameters and initializes Google Cloud services. r"N)super__init__getr"_get_rank_service_client)selfr# __class__s d/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_google_community/vertex_rank.pyr'zVertexAIRank.__init__AsA "6"jj* {{779DKreturnc ddlm}|j|jxs-|j rt j|j ndtdS#t$r}td|d}~wwxYw)zGet `RankServiceClient` for Vertex AI Ranking API calls. Returns: Client instance for ranking API. rrzCould not import google-cloud-discoveryengine python package. Please, install vertexaisearch dependency group: `pip install langchain-google-community[vertexaisearch]`Nzvertex-ai-search)module)r client_info) google.cloudr ImportErrorRankServiceClientrr rfrom_service_account_filer)r*rexcs r,r)z%VertexAIRank._get_rank_service_clientKs   <9&88((((55d6K6KL'/AB   K   sA A9( A44A9query documentsc ddlm} t|Dcgc]\}}|js1|jr|j j |jr|jd|jr%|j j |jn t||jd|jr'd|j j |jini}}}d|jd|jd |j}|j ||j"||j$|j& } |j(j+||j, } | j6D cgc]d} t9|j&s | j:nd| j<| j>d|jr|j| j@inifc} Scc}}w#t$r'tjd|jdY6wxYw#t.j0$r4} t3d t| t5d t| | d } ~ wwxYwcc} w)zRerank documents based on query relevance. Args: query: Query to use for reranking. documents: Documents to rerank. Returns: Reranked documents with relevance scores. rr)idcontenttitlez id_field 'z!' not found in document metadata.z projects/z /locations/z/rankingConfigs/)rrr7recordsrr)requestr!z%Error in Vertex AI Ranking API call: N)r:relevance_score) page_contentmetadatar%)!r2r enumeraterArrBr( RankingRecordrstrKeyErrorwarningswarnrrr RankRequestrrrr"rankr!core_exceptionsGoogleAPICallErrorprint RuntimeErrorr=r r;r:scorer<) r*r7r8ridxdocr=ranking_config_pathr>responseerecords r,_rerank_documentszVertexAIRank._rerank_documentscsQ 9 Y!*) 4 C##$$)9)9$:J:J)K6'55;?== ((7cRUh,,  ++!#,,"2"243C3C"DE  G $( D4D4D3Et223 5  6)55.****.2.T.T   X{{'''NH #**   ==$^^ ))'-||<@;K;K((&,,7QS   K  Y MMJt}}o5VW X Y&11 X 9#a&B C!Fs1vhOPVW W X  sBG3CG-)G3 'H&A)I0-G33,H#"H#&I-9/I((I- callbacksc&|j||S)aACompress documents using Vertex AI rerank API. Args: documents: Document instances to compress. query: Query string for document compression. callbacks: Callbacks to execute during compression. Returns: Compressed documents with relevance scores. )rV)r*r8r7rWs r,compress_documentszVertexAIRank.compress_documentss %%eY77r-ignoreT)extraarbitrary_types_allowed)r.z)discoveryengine_v1alpha.RankServiceClient)N)"__name__ __module__ __qualname____doc__r rrE__annotations__rrrrintrboolrrrrrr r!r"rr'r)rr rVr rYr model_config __classcell__)r+s@r,rrsD)J)"X.K..(89NC9*;rns>999/.-F&=44f)fr-