tj<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.edddd-d"Z/eddd#d.d(Z0eddd)Gd*d+e Z1dS)/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_importtextstrreturnc^d}tj||tj}|r|dn|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)r"patternmatchess h/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langchain_community/chains/graph_qa/cypher.pyextract_cypherr,0s3Gj$ 22G *71::d*z7langchain_neo4j.chains.graph_qa.cypher.construct_schemastructured_schemaDict[str, Any] include_types List[str] exclude_typesc f dfd fd|diD fd|d iD fd |d gDd }g}|dD]?\}}d d|D}||d|d@g}|d D]?\} }d d|D}|| d|d@d|d D} ddd|dd|dd| gS)z5Filter the schema based on included or excluded typesxr#r$boolcr|vn|vSN)r4r2r0s r+ filter_funcz%construct_schema..filter_funcSs%2NqM!!8NNr-c2i|]\}}|||Sr8r8.0kvr9s r+ z$construct_schema..Ws=   1{1~~ q   r- node_propsc2i|]\}}|||Sr8r8r;s r+r?z$construct_schema..\s=   1{1~~ q   r- rel_propscNg|] tfddD!S)c3:K|]}|VdSr7r8)r<tr9rs r+ z.construct_schema...ds1GG;;qt$$GGGGGGr-)startendtype)all)r<rFr9s @r+ z$construct_schema..asQ   GGGGG.FGGGGG    r- relationships)r@rBrMz, c4g|]}|dd|dSpropertyz: rJr8r<props r+rLz$construct_schema..l/ J J JtZ 2 2DL 2 2 J J Jr-z {}c4g|]}|dd|dSrOr8rQs r+rLz$construct_schema..trSr-c Jg|] }d|dd|dd|dd!S)z(:rHz)-[:rJz]->(:rI)r8r<els r+rLz$construct_schema..ysO  >DDFF       &**?B??   ''O&,\:@@BB@@zII J Jz J J J   ##u$>$>$>$>$>???? / < B B D DBB*II J Jz J J J   ""h#@#@9#@#@#@AAAA!/2N 99 0 HH) * * 8 HH( ) ) 2 HH^ $ $     r-zddddggdddded% d?d5Z d>d@d:ZxZS)AGraphCypherQAChainaChain 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_chainr# graph_schemaquery input_keyresult output_key inttop_kFr5return_intermediate_steps return_directNzOptional[CypherQueryCorrector]cypher_query_correctoruse_function_responseallow_dangerous_requestskwargsrr$Nonec jtjdi||jdurtddS)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.Nr8)super__init__r ValueError)selfr __class__s r+rzGraphCypherQAChain.__init__sN""6"""  ( 4 4W    5 4r-r1c|jgS)zReturn the input keys.)rrs r+ input_keyszGraphCypherQAChain.input_keyssr-c|jg}|S)zReturn the output keys.)r)r _output_keyss r+ output_keyszGraphCypherQAChain.output_keyss( r-cdS)Ngraph_cypher_chainr8rs r+ _chain_typezGraphCypherQAChain._chain_types##r-) qa_prompt cypher_prompt cypher_llmqa_llmr2r0validate_cypher qa_llm_kwargscypher_llm_kwargsrfunction_response_systemllmOptional[BaseLanguageModel]rOptional[BasePromptTemplate]rrr'Optional[Union[BaseLanguageModel, Any]]r2r0rrOptional[Dict[str, Any]]rrc |s|std|s|std|r|r|rtd|r| rtd|r| rtd| | ni}| | ni}d|vr||nt|d<d|vr||nt|d<|p|}| r |it jt | tjd td g}||ztz}n2#ttf$rtd wxYwtdd |i|}tdd |p|i|}|r|rtdt| dj||}d}|r9d| djdD}t%|}|d||||| d| S)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)rvz {question}function_response) variable_namez4Provided LLM does not support native tools/functionsrzGEither `exclude_types` or `include_types` can be provided, but not bothrcTg|]%}t|d|d|d&S)rHrJrI)rrXs r+rLz/GraphCypherQAChain.from_llm..Os@   r'{BvJ5 ::   r-rM)rrrrrr8)rrr bind_toolsr from_messagesrr from_templaterrNotImplementedErrorAttributeErrorr rhget_structured_schemar.r\r)clsrrrrrr2r0rrrrrruse_qa_llm_kwargsuse_cypher_llm_kwargsresponse_promptrrrrcorrector_schemas r+from_llmzGraphCypherQAChain.from_llms ( Y# YWXX X Uc USTT T  & S A   . J    F .;-FMMB!2!>  B  , , ,&2 8H h ' 0 0 0!.!: @X "( +3  A Y!!"%%%"4"B%.FGGG2@NN+:MNNN##+V3o6G6GG'8 Y Y Y !WXXX Y @@F@.?@@H"*# # !c# ## #   ] 0 ( 7O 1=-  "&  L   /;??PP    &::J%K%K "s %$;#9"7       s A.D!D)inputsr/ run_manager$Optional[CallbackManagerForChainRun]c|ptj}|}||j}||jd}||g}|j||}t|}|j r| |}| dd|j | |dd|j | d|i|r(|j |d |j} ng} |jr| } n| d d|j | t#| dd|j | d | i|jr.t'|| } |j|| d } n1|j|| d |} | |jj} |j| i} |jr || t0<| S)zGGenerate Cypher statement, use it to look up in db and answer question.)rischema) callbackszGenerated Cypher:rZ)rIverbosegreen)colorrIrrNz Full Context:rj)rir)rirj)r get_noop_manager get_childrrupdaterrunr,ron_textrr_rrrrr#rr{rinvokerrINTERMEDIATE_STEPS_KEY)rrr _run_managerrriargsrgenerated_cypherrj final_resultrr chain_results r+_callzGraphCypherQAChain._call^sh #S&@&Q&S&S  **,, $.) '   F#%7;;DI;VV**:;;  & M#::;KLL 0dDLQQQ Gt|     !!7,<"=>>>  j&&'7884:FGGG   @"LL  dDL Q Q Q  G Gt| !     % %y'&: ; ; ;) @$9(G$L$L!#}33!)@QRR   --!)g>>'. &dm&>? (,'F  ) F3EL/ 0r-)rrr$r)r$r1)r$r#r7)rrrrrrrrrrr2r1r0r1rr5rrrrrr5rr#rrr$r})rr/rrr$r/)__name__ __module__ __qualname____doc__rr__annotations__rrrrrrrrrrPrrr classmethodFUNCTION_RESPONSE_SYSTEMrr __classcell__)rs@r+r}r}s  d+++E++++%%%%''''IJEOOOO4&+++++VMM=AAAAA)"'''''H%*****     X X $$$X$,0a 376:26:>#%#% %266:&+(@a a a a a [a L=AAAAAAAAAAr-r})r"r#r$r#)r.r/r0r1r2r1r$r#)rir#rjrkr$rl)2r __future__rr&typingrrrrrlangchain_classic.chains.baser langchain_classic.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,rhr{r}r8r-r+rs&&"""""" 33333333333333//////111111666666??????<<<<<< :99999 .----->=====-   N +++  +$   P ;;;  ;|   U   2   ; rrrrrrr  rrrr-