j.=dZddlmZddlZddlmZmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZmZmZmZdd lmZdd lmZmZmZmZdd l m!Z!ddl"m#Z#ddl$m%Z%m&Z&ddl'm(Z(m)Z)ddl*m+Z+dZ,dZ-edddd dZ.eddd d!dZ/eddd d"dZ0edddGdde Z1y)#z Question answering over a graph.) annotationsN)AnyDictListOptionalUnion)Chain)LLMChain) deprecated)CallbackManagerForChainRun)BaseLanguageModel) AIMessage BaseMessage SystemMessage ToolMessage)StrOutputParser)BasePromptTemplateChatPromptTemplateHumanMessagePromptTemplateMessagesPlaceholder)Runnable)Field)CypherQueryCorrectorSchema)CYPHER_GENERATION_PROMPTCYPHER_QA_PROMPT) GraphStoreintermediate_stepszYou are an assistant that helps to form nice and human understandable answers based on the provided information from tools. Do not add any other information that wasn't present in the tools, and use very concise style in interpreting results! z0.3.8z1.0z5langchain_neo4j.chains.graph_qa.cypher.extract_cypher)sinceremovalalternative_importcbd}tj||tj}|r|dS|S)zExtract Cypher code from a text. Args: text: Text to extract Cypher code from. Returns: Cypher code extracted from the text. z ```(.*?)```r)refindallDOTALL)textpatternmatchess t/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain_community/chains/graph_qa/cypher.pyextract_cypherr*0s2Gjj$ 2G 71:*d*z7langchain_neo4j.chains.graph_qa.cypher.construct_schemac dfd |jdijDcic]\}}|r||c}}|jdijDcic]\}}|r||c}}|jdgDcgc]tfddDrc}d}g}|djD]F\}} dj| D cgc]} | d d | d c} } |j |d | d Hg} |djD]F\} } dj| D cgc]} | d d | d c} } | j | d | d H|dDcgc]}d|dd|d d|dd}}djddj|ddj| ddj|gScc}}wcc}}wcc}wcc} wcc} wcc}w)z5Filter the schema based on included or excluded typescr|vS|vSN)x exclude_types include_typess r) filter_funcz%construct_schema..filter_funcSs%2qM!N8NNr+ node_props rel_props relationshipsc34K|]}|ywr.r/).0tr3rs r) z#construct_schema..dsG.F;qt$.Fs)startendtype)r4r5r6z, propertyz: r>z {}z(:r<z)-[:z]->(:r=) z"Node properties are the following:,z*Relationship properties are the following:z$The relationships are the following:)r0strreturnbool)getitemsalljoinappend)structured_schemar2r1kvr:filtered_schemaformatted_node_propslabel propertiesprop props_strformatted_rel_propsrel_typeelformatted_relsr3s `` ` @r)construct_schemarYGsmO *--lB?EEG G11~ qDG *--k2>DDF F11~ qDF '**?B? ?G.FGG ? 'O&,\:@@BzII?I JztZ !DL> 2z J  ##ugS 2$>? C / < B B D*II?I JztZ !DL> 2z J  ""hZs9+R#@A !E"/22B R[Mbj\r%yk;2 99 0 HH) * 8 HH( ) 2 HH^ $    O    K K s#G*G%G+$G0 G5 =G:z)contentadditional_kwargs)rc tool_call_id)rrrD)questioncontextTOOL_IDmessagess r)get_function_responserjsa.G%)88)Cd)J$4%!+     CLw?!H$ Or+z"langchain_neo4j.GraphCypherQAChainceZdZUdZedZded<ded<ded <d ed <d Zd ed <dZd ed<dZ ded< dZ ded< dZ ded< dZ ded< dZ ded< dZded< d#fd Zed$dZed$dZed%dZe d&ddddggdddded d'd!Z d& d(d"ZxZS))GraphCypherQAChainaChain for question-answering against a graph by generating Cypher statements. *Security note*: Make sure that the database connection uses credentials that are narrowly-scoped to only include necessary permissions. Failure to do so may result in data corruption or loss, since the calling code may attempt commands that would result in deletion, mutation of data if appropriately prompted or reading sensitive data if such data is present in the database. The best way to guard against such negative outcomes is to (as appropriate) limit the permissions granted to the credentials used with this tool. See https://python.langchain.com/docs/security for more information. T)excludergraphr cypher_generation_chainzUnion[LLMChain, Runnable]qa_chainrD graph_schemaquery input_keyresult output_key inttop_kFrFreturn_intermediate_steps return_directNzOptional[CypherQueryCorrector]cypher_query_correctoruse_function_responseallow_dangerous_requestsc Vt|di||jdur tdy)zInitialize the chain.TaIn order to use this chain, you must acknowledge that it can make dangerous requests by setting `allow_dangerous_requests` to `True`.You must narrowly scope the permissions of the database connection to only include necessary permissions. Failure to do so may result in data corruption or loss or reading sensitive data if such data is present in the database.Only use this chain if you understand the risks and have taken the necessary precautions. See https://python.langchain.com/docs/security for more information.Nr/)super__init__r} ValueError)selfkwargs __class__s r)rzGraphCypherQAChain.__init__s9 "6"  ( ( 4W   5r+c|jgS)z7Return the input keys. :meta private: )rsrs r) input_keyszGraphCypherQAChain.input_keyss r+c |jg}|S)z8Return the output keys. :meta private: )ru)r _output_keyss r) output_keyszGraphCypherQAChain.output_keyss ( r+cy)Ngraph_cypher_chainr/rs r) _chain_typezGraphCypherQAChain._chain_types#r+) qa_prompt cypher_prompt cypher_llmqa_llmr1r2validate_cypher qa_llm_kwargscypher_llm_kwargsr|function_response_systemc X|s |s td|s |s td|r|r |r td|r | r td|r | r td| | ni}| | ni}d|vr ||nt|d<d|vr ||nt|d<|xs|}| rb |jit j t | tjd td g}||ztz}n tdd |i|}tdd |xs|i|}|r |r tdt| dj||}d}|rJ| dj j#dDcgc]}t%|d|d|d}}t'|}|d||||| d| S#ttf$r td wxYwcc}w)zInitialize from LLM.z8Either `llm` or `cypher_llm` parameters must be providedz4Either `llm` or `qa_llm` parameters must be providedzaYou can specify up to two of 'cypher_llm', 'qa_llm', and 'llm', but not all three simultaneously.zqSpecifying cypher_prompt and cypher_llm_kwargs together is not allowed. Please pass prompt via cypher_llm_kwargs.zeSpecifying qa_prompt and qa_llm_kwargs together is not allowed. Please pass prompt via qa_llm_kwargs.Nprompt)rcz {question}function_response) variable_namez4Provided LLM does not support native tools/functionsllmzGEither `exclude_types` or `include_types` can be provided, but not bothrnr6r<r>r=)rqrpror{r|r/)rrr bind_toolsr from_messagesrr from_templaterrNotImplementedErrorAttributeErrorr rYget_structured_schemarLrGrr)clsrrrrrr1r2rrrr|rruse_qa_llm_kwargsuse_cypher_llm_kwargsresponse_promptrprorqr{rWcorrector_schemas r)from_llmzGraphCypherQAChain.from_llmsb(#WX XcST T &SA  .J  F .;-FMB!2!> B  , ,&2 8H h ' 0 0!.!: @X "( +3  Y!!"%"4"B"B%.FG2@@N+:MN#+V3o6GG @F@.?@H"*# !c# ##  ]0 ( 7O 1 1=- "& !/;;??P PBr'{BvJ5 :P  &::J%K " %$;#9"7     7(8 Y !WXX Y* sA F F' F$c2|xstj}|j}||j}||jd}|j |g}|j j||}t|}|jr|j|}|jdd|j|j|dd|j|jd|i|r)|jj|d |j} ng} |j r| } n|jd d|j|jt#| dd|j|jd | i|j$r+t'|| } |j(j+|| d } n9|j(j+|| d |} | |j(j,} |j,| i} |j.r || t0<| S)zGGenerate Cypher statement, use it to look up in db and answer question.)rfschema) callbackszGenerated Cypher:rB)r=verbosegreen)colorr=rrrNz Full Context:rg)rfr)rfrg)r get_noop_manager get_childrsrqupdaterorunr*r{on_textrrKrnrrrxrzrDr|rjrpinvokeruryINTERMEDIATE_STEPS_KEY)rinputs run_manager _run_managerrrfargsrgenerated_cypherrg final_resultrrt chain_results r)_callzGraphCypherQAChain._callds #S&@&Q&Q&S  **, $..) ''  F#%77;;DI;V**:;  & &#::;KL 0dDLLQ Gt||   !!7,<"=> jj&&'784::FGG   "L  dDLL Q  G Gt|| !   % %y'&: ;))$9(G$L!#}}33!)@QR  --!)g>'. &dmm&>&>? (,'F  ) )3EL/ 0r+)rrrENone)rE List[str])rErDr.)rOptional[BaseLanguageModel]rOptional[BasePromptTemplate]rrrrrz'Optional[Union[BaseLanguageModel, Any]]r1rr2rrrFrOptional[Dict[str, Any]]rrr|rFrrDrrrErl)rDict[str, Any]rz$Optional[CallbackManagerForChainRun]rEr)__name__ __module__ __qualname____doc__rrn__annotations__rsrurxryrzr{r|r}rr?rrr classmethodFUNCTION_RESPONSE_SYSTEMrr __classcell__)rs@r)rlrls d+E:+%%''IsJE3O4&+t+VM4M=A:A)"'4'H%*d*    $$,0a 376:26:>#%#% %266:&+(@a (a 0 a 4 a 0 a 8a !a !a a 0a 4a  $a #&a a !a a L=AAA:A  Ar+rl)r&rDrErD)rLrr2rr1rrErD)rfrDrgzList[Dict[str, Any]]rEzList[BaseMessage])2r __future__rr#typingrrrrrlangchain.chains.baser langchain.chains.llmr langchain_core._api.deprecationr langchain_core.callbacksr langchain_core.language_modelsr langchain_core.messagesrrrrlangchain_core.output_parsersrlangchain_core.promptsrrrrlangchain_core.runnablesrpydanticr0langchain_community.chains.graph_qa.cypher_utilsrr+langchain_community.chains.graph_qa.promptsrr&langchain_community.graphs.graph_storerrrr*rYrjrlr/r+r)rs0&" 33')6?< : .>-  N +  +$  P ;%;;; ;  ;|  U 0  2  ; xx  xr+