Ngp'dZddlmZddlZddlmZmZmZmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZmZmZdd lmZGd de ZdS)z Question answering over a graph.) annotationsN)AnyDictListOptional)ChainLLMChain)CallbackManagerForChainRun)BaseLanguageModel)BasePromptTemplate)Field)AQL_FIX_PROMPTAQL_GENERATION_PROMPT AQL_QA_PROMPT) ArangoGraphcNeZdZUdZedZded<ded<ded<ded <d Zd ed <d Zd ed<dZ ded<dZ d ed<dZ ded<dZ ded<dZ ded<dZded< d2fd Zed3d!Zed3d"Zed4d#Zeeeed$d5d+Z d6d7d1ZxZS)8ArangoGraphQAChainaChain for question-answering against a graph by generating AQL 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 aql_generation_chain aql_fix_chainqa_chainquerystr input_keyresult output_key inttop_k aql_examplesFboolreturn_aql_queryreturn_aql_resultmax_aql_generation_attemptsallow_dangerous_requestskwargsrreturnNonec 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.N)super__init__r) ValueError)selfr* __class__s h/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/chains/graph_qa/arangodb.pyr0zArangoGraphQAChain.__init__KsN""6"""  ( 4 4W    5 4 List[str]c|jgSN)rr2s r4 input_keyszArangoGraphQAChain.input_keys[s r5c|jgSr8)rr9s r4 output_keyszArangoGraphQAChain.output_keys_s   r5cdS)Ngraph_aql_chainr.r9s r4 _chain_typezArangoGraphQAChain._chain_typecs  r5) qa_promptaql_generation_promptaql_fix_promptllmr r@r rArBc t||}t||}t||}|d|||d|S)zInitialize from LLM.)rCprompt)rrrr.r ) clsrCr@rArBr*rrrs r4from_llmzArangoGraphQAChain.from_llmgsmI666'C8MNNN S@@@ s !5'      r5NinputsDict[str, Any] run_manager$Optional[CallbackManagerForChainRun]cV|ptj}|}||j}|j|jj|j|d|}d}d}d} d} | | |j dzkrsd} tj | |tj } | sM| dd|j | |d d|j td || d }| d| d|j| |dd|j d dlm}  |j||j} n~#| $rv}|j}| dd|j | |dd|j |j|jj||d|}Yd}~nd}~wwxYw| dz } | | |j dzks| d|d}t|| dd|j | t-| dd|j ||jj||| d|}|j||jji}|jr||d<|jr| |d<|S)aw Generate an AQL statement from user input, use it retrieve a response from an ArangoDB Database instance, and respond to the user input in natural language. Users can modify the following ArangoGraphQAChain Class Variables: :var top_k: The maximum number of AQL Query Results to return :type top_k: int :var aql_examples: A set of AQL Query Examples that are passed to the AQL Generation Prompt Template to promote few-shot-learning. Defaults to an empty string. :type aql_examples: str :var return_aql_query: Whether to return the AQL Query in the output dictionary. Defaults to False. :type return_aql_query: bool :var return_aql_result: Whether to return the AQL Query in the output dictionary. Defaults to False :type return_aql_result: bool :var max_aql_generation_attempts: The maximum amount of AQL Generation attempts to be made prior to raising the last AQL Query Execution Error. Defaults to 3. :type max_aql_generation_attempts: int ) adb_schemar# user_input) callbacksr"Nz```(?i:aql)?(.*?)```zInvalid Response:  )endverbosered)colorrRrSzResponse is Invalid: rz AQL Query (z):)rSgreen)AQLQueryExecuteErrorzAQL Query Execution Error: yellowz )rM aql_query aql_errorz Maximum amount of AQL Query Generation attempts reached. Unable to execute the AQL Query due to the following error: z z AQL Result:)rMrNrY aql_resultrYr[)r get_noop_manager get_childrrrunrschemar#r(refindallDOTALLon_textrSr1arangorWrr! error_messagerrrrr%r&)r2rHrJ _run_managerrOrNaql_generation_outputrYrZr[aql_generation_attemptpatternmatchesrWemrs r4_callzArangoGraphQAChain._call}sB#S&@&Q&S&S  **,, DN+ !% 9 = ="j/ $ 1(    !>! !    !"  &)IA)MMM.Gj*?KKG R$$(dDL%$$)D$,%!!P9N!P!PQQQ I  84888$, !     dDL !    4 3 3 3 3 3 !Z--iDD '   O $$1tT\%$$X64<% )-(:(>(>&*j&7%.%. ( )?))%%%%%% 0 #a ' "q  &)IA)MMMr  A Q-- ]dlKKK  OO7dl    "j/(&(       /6$-*B#CD   ,"+F;   ! .#-F<  s> EG$A,GG)r*rr+r,)r+r6)r+r) rCr r@r rAr rBr r*rr+rr8)rHrIrJrKr+rI)__name__ __module__ __qualname____doc__rr__annotations__rrr!r#r%r&r(r)r0propertyr:r<r? classmethodrrrrGrm __classcell__)r3s@r4rrs  t,,,E,,,,""""IJEOOOOL#""""$####()((((%*****     X !!!X!!!!X! )64I-;      [ 0=ATTTTTTTTTr5r)rq __future__rr`typingrrrrlangchain.chains.baserlangchain.chains.llmr langchain_core.callbacksr langchain_core.language_modelsr langchain_core.promptsr pydanticr+langchain_community.chains.graph_qa.promptsrrr)langchain_community.graphs.arangodb_graphrrr.r5r4rs;&&"""""" ,,,,,,,,,,,,''''''))))))??????<<<<<<555555 BAAAAAzzzzzzzzzzr5