tj|z dZddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z mZddlZddlmZddlmZmZmZmZmZmZddlmZdd lmZmZmZmZdd l m!Z!dd l"m#Z#m$Z$m%Z%m&Z&m'Z'dd l(m)Z)dd l*m+Z+ddl,m-Z-ej.e/Z0de1fdZ2Gddee3e4e fe3e4e ffeZ5dS)z0Base interface that all chains should implement.N)ABCabstractmethod)Path)Anycast) deprecated)AsyncCallbackManagerAsyncCallbackManagerForChainRunBaseCallbackManagerCallbackManagerCallbackManagerForChainRun Callbacks)RunInfo)RunnableConfigRunnableSerializable ensure_configrun_in_executor) create_model) BaseModel ConfigDictFieldfield_validatormodel_validator)override) BaseMemory)RUN_KEYreturnc"ddlm}|S)Nr get_verbose)langchain_classic.globalsr rs [/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langchain_classic/chains/base.py_get_verbosityr#.s 555555 ;==cVeZdZUdZdZedzed< eddZe ed< ee Z e ed< dZ eedzed < dZejeefdzed < eddZedzed < ed Ze d?d edzdeefdZe d?d edzdeefdZe d?deeefd edzdedeeeffdZe d?deeefd edzdedeeeffdZe defdZ!e"de#dedefdZ$e%dde#de dzde fdZ&e e'deefdZ(e e'deefdZ)deddfdZ*deeefddfd Z+e' d?dejeefd!e,dzdejeeffd"Z- d?dejeefd!e.dzdejeeffd#Z/e0d$d%d&' d@dddd(d)deeefezd*e de d eedzd eeefdzd+edzd,e deeeffd-Z1e0d$d.d&' d@dddd(d)deeefezd*e de d eedzd eeefdzd+edzd,e deeeffd/Z2 dAdeeefdeeefd*e deeeffd0Z3 dAdeeefdeeefd*e deeeffd1Z4deeefezdeeeffd2Z5deeefezdeeeffd3Z6e defd4Z7e0d$d%d&'dddd5d6ede d eedzd eeefdzdedef d7Z8e0d$d.d&'dddd5d6ede d eedzd eeefdzdedef d8Z9dedeffd9 Zd:e:ezddfd;Z;e0d$dZ<xZ=S)BChainaAbstract base class for creating structured sequences of calls to components. Chains should be used to encode a sequence of calls to components like models, document retrievers, other chains, etc., and provide a simple interface to this sequence. The Chain interface makes it easy to create apps that are: - Stateful: add Memory to any Chain to give it state, - Observable: pass Callbacks to a Chain to execute additional functionality, like logging, outside the main sequence of component calls, - Composable: the Chain API is flexible enough that it is easy to combine Chains with other components, including other Chains. The main methods exposed by chains are: - `__call__`: Chains are callable. The `__call__` method is the primary way to execute a Chain. This takes inputs as a dictionary and returns a dictionary output. - `run`: A convenience method that takes inputs as args/kwargs and returns the output as a string or object. This method can only be used for a subset of chains and cannot return as rich of an output as `__call__`. NmemoryT)defaultexclude callbacks)default_factoryverbosetagsmetadatacallback_manager)arbitrary_types_allowedconfigrc dtdit|jtdfS)N ChainInput)r3)rdictfromkeys input_keysrselfr1s r"get_input_schemazChain.get_input_schemaos- XXDMM$/CQU;,W,WXXXr$c ft dit|jtdfS)N ChainOutput)r;)rr4r5 output_keysrr7s r"get_output_schemazChain.get_output_schemaws;    mmD,sDk::   r$inputkwargsc t|}|d}|d}|d}|dp|}|d}|dd} |dd} ||} t j||j|j||j||j } tj |j j d } | d| || } || | r| | | n| | }|| || }n(#t"$r}||d}~wwxYw||| rt)|j |t,<|S Nr*r-r.run_namerun_idinclude_run_infoFreturn_only_outputs run_manager)name)rF)rC)rgetget_name prep_inputsr configurer*r,r-r.inspect signature_call parameterson_chain_start_validate_inputs prep_outputs BaseExceptionon_chain_error on_chain_endrrCrr8r>r1r?r*r-r.rBrCrDrEinputsr/new_arg_supportedrFoutputs final_outputses r"invokez Chain.invokesv&&JJ{++ zz&!!::j))::j))FF!!%((*4  N L  I  M  $-dj99DHHWW&55    6     ! !& ) ) )%( 6{ ;;;ZZ''  -1,=,=#--MM      & &q ) ) )     )))  H%,K4F%G%G%GM' "sAF"" G,GGc XKt|}|d}|d}|d}|dp|}|d}|dd} |dd} ||d{V} t j||j|j||j||j } tj |j j d } | d| || d{V} || | r| | | d{Vn| | d{V}|| || d{V}n.#t"$r!}||d{Vd}~wwxYw||d{V| rt)|j |t,<|SrA)rrHrI aprep_inputsr rKr*r,r-r.rLrM_acallrOrPrQ aprep_outputsrSrTrUrrCrrVs r"ainvokez Chain.ainvokesv&&JJ{++ zz&!!::j))::j))FF((/////////9  N L  I  M  $-dk::EII-XX,;;    <           ! !& ) ) )%/dkk&kkBBBBBBBBB;;v........  372D2D#33------MM    ,,Q// / / / / / / /  &&w/////////  H%,K4F%G%G%GM' "sA,G G- G((G-c$d}t|)Nz)Saving not supported for this chain type.)NotImplementedErrorr8msgs r" _chain_typezChain._chain_types9!#&&&r$before)modevaluesc|d[|dd}t|tjdtd|dd|d<|S)z6Raise deprecation warning if callback_manager is used.r/Nr*zCannot specify both callback_manager and callbacks. callback_manager is deprecated, callbacks is the preferred parameter to pass in.z=callback_manager is deprecated. Please use callbacks instead.) stacklevel)rH ValueErrorwarningswarnDeprecationWarningpop)clsrires r""raise_callback_manager_deprecationz(Chain.raise_callback_manager_deprecations ::( ) ) 5zz+&&2, !oo% MO"     #)**-?"F"FF;  r$c&|tS|S)zgSet the chain verbosity. Defaults to the global setting if not specified by the user. )r#)rrr,s r" set_verbosezChain.set_verbose s ?!## #r$cdS)z'Keys expected to be in the chain input.Nr8s r"r6zChain.input_keysr$cdS)z(Keys expected to be in the chain output.Nrwrxs r"r<zChain.output_keysryr$rWcnt|tsbt|j}|j||jj}t|dkrd|d}t|t|j|}|rd|}t|dS)z"Check that all inputs are present.NzMA single string input was passed in, but this chain expects multiple inputs (zt). When a chain expects multiple inputs, please call it by passing in a dictionary, eg `chain({'foo': 1, 'bar': 2})`zMissing some input keys: ) isinstancer4setr6r' differencememory_variableslenrm)r8rW _input_keysre missing_keyss r"rQzChain._validate_inputs!s&$'' &do..K{&*44T[5QRR ;1$$7(3777 !oo%4?++66v>>  "z"Chain.__call__..#U#U#UTQq}Aq}}}r$rErD)r\ritems r8rWrEr*r-r.rBrDr1s r"__call__zChain.__call__pshN#    {{  !#U#UV\\^^#U#U#U V V 3-    r$rac K||||d}||tdd|D||d{VS)aAsynchronously execute the chain. Args: inputs: Dictionary of inputs, or single input if chain expects only one param. Should contain all inputs specified in `Chain.input_keys` except for inputs that will be set by the chain's memory. return_only_outputs: Whether to return only outputs in the response. If `True`, only new keys generated by this chain will be returned. If `False`, both input keys and new keys generated by this chain will be returned. callbacks: Callbacks to use for this chain run. These will be called in addition to callbacks passed to the chain during construction, but only these runtime callbacks will propagate to calls to other objects. tags: List of string tags to pass to all callbacks. These will be passed in addition to tags passed to the chain during construction, but only these runtime tags will propagate to calls to other objects. metadata: Optional metadata associated with the chain. run_name: Optional name for this run of the chain. include_run_info: Whether to include run info in the response. Defaults to False. Returns: A dict of named outputs. Should contain all outputs specified in `Chain.output_keys`. rrci|] \}}||| Srrwrs r"rzChain.acall..rr$rN)rarrrs r"acallz Chain.acallsN#     \\  !#U#UV\\^^#U#U#U V V 3- "         r$c|||j|j|||r|Si||S)Validate and prepare chain outputs, and save info about this run to memory. Args: inputs: Dictionary of chain inputs, including any inputs added by chain memory. outputs: Dictionary of initial chain outputs. return_only_outputs: Whether to only return the chain outputs. If `False`, inputs are also added to the final outputs. Returns: A dict of the final chain outputs. )rr' save_contextr8rWrYrEs r"rRzChain.prep_outputssU$ w''' ; " K $ $VW 5 5 5  N$&$G$$r$cK|||j!|j||d{V|r|Si||S)rN)rr' asave_contextrs r"r`zChain.aprep_outputssk$ w''' ; "+++FG<< < < < < < < <  N$&$G$$r$c<t|tsXt|j}|j||jj}tt||i}|j'|j |}t|fi|}|S)Prepare chain inputs, including adding inputs from memory. Args: inputs: Dictionary of raw inputs, or single input if chain expects only one param. Should contain all inputs specified in `Chain.input_keys` except for inputs that will be set by the chain's memory. Returns: A dictionary of all inputs, including those added by the chain's memory. ) r}r4r~r6r'rrnextiterload_memory_variablesr8rWrexternal_contexts r"rJzChain.prep_inputs s&$'' 7do..K{&*44T[5QRR 4 ,,--v6F ; "#{@@HH &55$455F r$cLKt|tsXt|j}|j||jj}tt||i}|j-|j |d{V}t|fi|}|S)rN) r}r4r~r6r'rrrraload_memory_variablesrs r"r^zChain.aprep_inputs!s&$'' 7do..K{&*44T[5QRR 4 ,,--v6F ; "%)[%G%G%O%OOOOOOO &55$455F r$ct|jdkrd|jd}t||jdS)Nr|B`run` not supported when there is not exactly one output key. Got .r)rr<rmrds r"_run_output_keyzChain._run_output_key9sS t A % %;'+'7;;; S// !""r$r*r-r.argsc"|j}|rA|s?t|dkrd}t|||d||||S|r|s||||||S|s|sd}t|d|d|d}t|) a%Convenience method for executing chain. The main difference between this method and `Chain.__call__` is that this method expects inputs to be passed directly in as positional arguments or keyword arguments, whereas `Chain.__call__` expects a single input dictionary with all the inputs Args: *args: If the chain expects a single input, it can be passed in as the sole positional argument. callbacks: Callbacks to use for this chain run. These will be called in addition to callbacks passed to the chain during construction, but only these runtime callbacks will propagate to calls to other objects. tags: List of string tags to pass to all callbacks. These will be passed in addition to tags passed to the chain during construction, but only these runtime tags will propagate to calls to other objects. metadata: Optional metadata associated with the chain. **kwargs: If the chain expects multiple inputs, they can be passed in directly as keyword arguments. Returns: The chain output. Example: ```python # Suppose we have a single-input chain that takes a 'question' string: chain.run("What's the temperature in Boise, Idaho?") # -> "The temperature in Boise is..." # Suppose we have a multi-input chain that takes a 'question' string # and 'context' string: question = "What's the temperature in Boise, Idaho?" context = "Weather report for Boise, Idaho on 07/03/23..." chain.run(question=question, context=context) # -> "The temperature in Boise is..." ``` r|,`run` supports only one positional argument.rrz^`run` supported with either positional arguments or keyword arguments, but none were provided.^`run` supported with either positional arguments or keyword arguments but not both. Got args: and kwargs: r)rrrm)r8r*r-r.rr? _output_keyres r"runz Chain.runCs^*   4yyA~~D oo%4Q94(SSS   $ 4)$RRR  "d "+ S// ! E(, E E;A E E E oor$cKt|jdkrd|jd}t||r\|sZt|dkrd}t|||d|||d{V|jdS|r2|s0|||||d{V|jdSd|d |d}t|) a4Convenience method for executing chain. The main difference between this method and `Chain.__call__` is that this method expects inputs to be passed directly in as positional arguments or keyword arguments, whereas `Chain.__call__` expects a single input dictionary with all the inputs Args: *args: If the chain expects a single input, it can be passed in as the sole positional argument. callbacks: Callbacks to use for this chain run. These will be called in addition to callbacks passed to the chain during construction, but only these runtime callbacks will propagate to calls to other objects. tags: List of string tags to pass to all callbacks. These will be passed in addition to tags passed to the chain during construction, but only these runtime tags will propagate to calls to other objects. metadata: Optional metadata associated with the chain. **kwargs: If the chain expects multiple inputs, they can be passed in directly as keyword arguments. Returns: The chain output. Example: ```python # Suppose we have a single-input chain that takes a 'question' string: await chain.arun("What's the temperature in Boise, Idaho?") # -> "The temperature in Boise is..." # Suppose we have a multi-input chain that takes a 'question' string # and 'context' string: question = "What's the temperature in Boise, Idaho?" context = "Weather report for Boise, Idaho on 07/03/23..." await chain.arun(question=question, context=context) # -> "The temperature in Boise is..." ``` r|rrrrrNrr)rr<rmr)r8r*r-r.rr?res r"arunz Chain.aruns{^ t A % %;'+'7;;; S// !  # #4yyA~~D oo%jjG'% ! q!# #  #$ #jj'% ! q!# # E(, E E;A E E E oor$c tjdi|}tjt5|j|d<dddn #1swxYwY|S)aDictionary representation of chain. Expects `Chain._chain_type` property to be implemented and for memory to be null. Args: **kwargs: Keyword arguments passed to default `pydantic.BaseModel.dict` method. Returns: A dictionary representation of the chain. Example: ```python chain.model_dump(exclude_unset=True) # -> {"_type": "foo", "verbose": False, ...} ``` _typeNrw)super model_dump contextlibsuppressrcrf)r8r?_dict __class__s r"r4z Chain.dicts&#",,V,,  !4 5 5 . .!-E'N . . . . . . . . . . . . . . . s A  AA file_pathc|jd}t||}d|vrd|d}t|t |t rt |n|}|j}|dd|j dkrG| d 5}tj ||d ddddS#1swxYwYdS|j d rG| d 5}tj ||d ddddS#1swxYwYdS|d}t|)a4Save the chain. Expects `Chain._chain_type` property to be implemented and for memory to be null. Args: file_path: Path to file to save the chain to. Example: ```python chain.save(file_path="path/chain.yaml") ``` Nz&Saving of memory is not yet supported.rzChain z does not support saving.T)parentsexist_okz.jsonwrk)indent)z.yamlz.ymlF)default_flow_stylez must be json or yaml)r'rmrrcr}strrparentmkdirsuffixopenjsondumpendswithyaml)r8rre chain_dict save_pathdirectory_pathfs r"savez Chain.saves ; ":CS// !__&& * $ $:4:::C%c** *(2)S'A'APDOOOy ")TD999  w & &$$ 3 *a2222 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3   & &'8 9 9 "$$ C *aEBBBB C C C C C C C C C C C C C C C C C C555CS// !s$*CCC D00D47D4batch input_listc$fd|DS)z)Call the chain on all inputs in the list.c*g|]}|S))r*rw)rrWr*r8s r" zChain.apply..&s(KKKfVy111KKKr$rw)r8rr*s` `r"applyz Chain.applys$LKKKK KKKKr$r)FN)F)>__name__ __module__ __qualname____doc__r'r__annotations__rr*rr#r,boolr-listrr.builtinsr4rr/r r model_configrrtyperr9r=r\rapropertyrfr classmethodrsrrurr6r<rQrr rNr r_rrrrRr`rJr^rrrrrr __classcell__)rs@r"r&r&4s ,!%FJ $$$ !5t<<   c3h' 0$6  sCx    ^ 2?C   c3h' 5t;  sCx     8ZXw???%*# 1 "&*.#!&1 1 1 S#X$1 "1  1 3i$ 1 sCx.4'1 *1 1  c3h1 1 1 @?1 fZY@@@%*# 0 "&*.#!&0 0 0 S#X$0 "0  0 3i$ 0 sCx.4'0 *0 0  c3h0 0 0 A@0 l%* %%S#X%c3h%" % c3h %%%%:%* %%S#X%c3h%" % c3h %%%%2$sCx.3"64S>0c3h#)=$sCx.0####X#ZXw??? $!%*. GGGGG3i$ G sCx.4' G  G GGG@?GRZY@@@ $!%*. OOOOO3i$ O sCx.4' O  O OOOA@ObST0&"dSj&"T&"&"&"&"PZWg>>> $LLsCx01LL hmCH% & LLL?>LLLLLr$r&)6rrrrLrloggingrnabcrrpathlibrtypingrrrlangchain_core._apirlangchain_core.callbacksr r r r r rlangchain_core.outputsrlangchain_core.runnablesrrrrlangchain_core.utils.pydanticrpydanticrrrrrtyping_extensionsrlangchain_classic.base_memoryrlangchain_classic.schemar getLoggerrloggerrr#r4rr&rwr$r"rse66 ######## ******+***** 766666'&&&&&444444,,,,,,  8 $ $ r Lr Lr Lr Lr L c3hc3h!? @#r Lr Lr Lr Lr Lr$