3fi&ddlmZmZmZmZmZddlmZddl m Z ddl m Z ddl mZmZddlmZmZmZddlmZerddlmZGd d eed fZy ) ) TYPE_CHECKINGAnyDictListOptional) exceptions) Credentials)Document)RunnableConfigRunnableSerializable) BaseModel ConfigDictField)get_client_infodiscoveryengine_v1alphaceZdZUdZedZeed<edZeed<edZ eed<ed Z e e ed <dZ eed <edZe eed <edZe eed <GddeZdeffd Z ddZ d dede edefdZde edeefdZdeeeefdefdZedeefdZedefdZ e!ddZ"xZ#S)!VertexAICheckGroundingWrappera Initializes the Vertex AI CheckGroundingOutputParser with configurable parameters. Calls the Check Grounding API to validate the response against a given set of documents and returns back citations that support the claims along with the cited chunks. Output is of the type CheckGroundingResponse. Attributes: project_id: Google Cloud project ID location_id: Location ID for the ranking service. grounding_config: The resource name of the grounding config, such as `default_grounding_config`. It is set to `default_grounding_config` by default if unspecified citation_threshold: The threshold (in `[0, 1]`) used for determining whether a fact must be cited for a claim in the answer candidate. Choosing a higher threshold will lead to fewer but very strong citations, while choosing a lower threshold may lead to more but somewhat weaker citations. If unset, the threshold will default to `0.6`. credentials (Optional[Credentials]): Google Cloud credentials object. credentials_path (Optional[str]): Path to the Google Cloud service account credentials file. N)default project_idglobal location_iddefault_grounding_configgrounding_configg333333?citation_thresholdclient credentialscredentials_pathcfeZdZUdZeed<gZeee e fed<gZ eee e fed<dZ e ed<y)4VertexAICheckGroundingWrapper.CheckGroundingResponseg support_score cited_chunksclaimsanswer_with_citationsN) __name__ __module__ __qualname__r!float__annotations__r"rrstrrr#r%o/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_google_community/vertex_check_grounding.pyCheckGroundingResponser 3sE" u"-/ d4S>*/')T#s(^$)%'s'r-r/kwargsc t|di||jd|_|js|j |_yy)z Constructor for CheckGroundingOutputParser. Initializes the grounding check service client with necessary credentials and configurations. rNr,)super__init__getr#_get_check_grounding_service_client)selfr0 __class__s r.r3z&VertexAICheckGroundingWrapper.__init__9sA "6"jj* {{BBDDKr-returnc ddlm}|j|jxs-|j rt j|j ndtdS#t$r}td|d}~wwxYw)z Returns a GroundedGenerationServiceClient instance using provided credentials. Raises ImportError if necessary packages are not installed. Returns: A GroundedGenerationServiceClient instance. 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 ImportErrorGroundedGenerationServiceClientrrr from_service_account_filer)r6rexcs r.r5zAVertexAICheckGroundingWrapper._get_check_grounding_service_clientDs  <G&FF  ,, 99$:O:OP(/AB   K   sA A9( A44A9inputconfigcZddlm}|}|j|}|j|j}|D cgc]b}|j rT|j |j |jxsijD cic]\}} |dvr| || c} }d} }}} | s td|jd|jd |jd |j|| | } |j td  |jj| } | j(}| j*Dcgc]'}|j,|t/|j0d)}}| j2Dcgc]:}|j4|j6|j8t;|j<d<}}|j?|}|jA||||Scc} }wcc} }}w#t j"$r} t%dt'| | d} ~ wwxYwcc}wcc}w)av Calls the Vertex Check Grounding API for a given answer candidate and a list of documents (claims) to validate whether the set of claims support the answer candidate. Args: input: The candidate answer to be evaluated for grounding. config: Runnable configuration containing documents in `config["configurable"]["documents"]`. The documents are used as grounding facts and will be converted to `google.cloud.discoveryengine_v1alpha.types.GroundingFact` objects. Supports up to `200` facts. Returns: Response of the type CheckGroundingResponse Attributes: support_score (float): The support score for the input answer candidate. Higher the score, higher is the fraction of claims that are supported by the provided facts. This is always set when a response is returned. cited_chunks (MutableSequence[google.cloud.discoveryengine_v1alpha.types. FactChunk]): List of facts cited across all claims in the answer candidate. These are derived from the facts supplied in the request. claims (MutableSequence[google.cloud.discoveryengine_v1alpha.types. CheckGroundingResponse.Claim]): Claim texts and citation info across all claims in the answer candidate. answer_with_citations (str): Complete formed answer formatted with inline citations rr)r)idrelevance_scoreN) fact_text attributesz*No valid documents provided for grounding.z projects/z /locations/z/groundingConfigs/)ranswer_candidatefactsgrounding_speczClient not initialized.)requestz-Error in Vertex AI Check Grounding API call: ) chunk_textsource) start_posend_pos claim_textcitation_indices)r!r"r#r%)!r<rextract_documentsCheckGroundingSpecr page_content GroundingFactmetadataitems ValueErrorCheckGroundingRequestrrrrcheck_groundingcore_exceptionsGoogleAPICallError RuntimeErrorr+r!r"rLintrMr#rNrOrPlistrQcombine_claims_with_citationsr/)r6rArBrrH documentsrJdockeyvaluerIrKresponseer!chunkr"claimr#r%s r.invokez$VertexAICheckGroundingWrapper.invokebslR 9 **62 C0CC#66 !    2 # 1 1** *eg "U";;@Q J     IJ J?)??((9TEUEUDVVhimi~i~hA-)   ;; 67 7 {{2272CH !.. "..  $..#C $56   "  #__ ==#..$()?)?$@    !% B B6 J**'%"7 +  e  :11 ?AxH     s<A G)G# G) G0,H#zBVertexAICheckGroundingWrapper.extract_documents..s@ *-JsH %@ sz8Invalid documents. Each must be an instance of Document.z/This wrapper requires documents for processing.)rXr4rnr_all)r6rBpotential_documentss r.rRz/VertexAICheckGroundingWrapper.extract_documentsss9: :$jj<@@bQ-t4C@ 1D@ = WX X"NO O""r-r#c t|d}g}|D]M}|dr0dj|dDcgc]}d|d c}}|d|}n|d}|j|Odj|jScc}w) Nc |dS)NrNr,)xs r.zMVertexAICheckGroundingWrapper.combine_claims_with_citations..s Q{^r-)rcrQr$[]rP )sortedjoinappendstrip)r6r# sorted_claimsresultrhidx citationsrPs r.r`z;VertexAICheckGroundingWrapper.combine_claims_with_citationssv+CD " &E'(GG5AS;T$UCqQZ$UV %l 34YK@ "<0 MM* %  &xx%%'' %Vs B c gdS)N) langchain utilitiesrZr,clss r.get_lc_namespacez.VertexAICheckGroundingWrapper.get_lc_namespaces<>Bl l "*>":l l \ #(@ #T(^ # (Dc3h4H (S (=c==4 $Lr-rr N)typingrrrrrgoogle.api_corerr[google.auth.credentialsr langchain_core.documentsr langchain_core.runnablesr r pydanticr rr!langchain_google_community._utilsrr<rr+rr,r-r.rs@;;9/-I11=4gTTUgr-