gh>dZddlmZddlZddlZddlZddlmZmZddl m Z ddl m Z ddl mZddlmZmZmZmZmZmZmZddlZdd lmZmZmZmZdd lmZmZdd lm Z m!Z!dd l"m#Z#dd l$m%Z%ddl&m'Z'm(Z(m)Z)ddl*m+Z+m,Z,ddl-m.Z.ddl/m0Z0erddl1m2Z2edZ3Gdde,e4e(fee3eZ5d dZ6d!dZ7d!dZ8dS)"z Base class for prompt templates.) annotationsN)ABCabstractmethod)Mapping)cached_property)Path) TYPE_CHECKINGAnyCallableGenericOptionalTypeVarUnion) BaseModel ConfigDictFieldmodel_validator)Selfoverride) ErrorCodecreate_messagedumpd)BaseOutputParser)ChatPromptValueConcrete PromptValueStringPromptValue)RunnableConfigRunnableSerializable) ensure_config)create_model_v2)DocumentFormatOutputTypec`eZdZUdZded< egZded< eedZded < d Z d ed < ee Z ded< d Z ded< d Z ded< e dd>dZed?dZed@dZedZedAdZeedBd Ze dCdDd$ZdEd'ZdFd)ZdFd*Ze dCdGd-Ze dCdGd.ZedHd/Z dHd0Z!dId2Z"dJd3Z#edKd5Z$dKd6Z%edLd8Z&dMfd9 ZdNd=Z'xZ(S)OBasePromptTemplatez8Base class for all prompt templates, returning a prompt. list[str]input_variables)defaultoptional_variablesT)default_factoryexcludeztyping.Dict[str, Any] input_typesNzOptional[BaseOutputParser] output_parser)r*zMapping[str, Any]partial_variableszOptional[typing.Dict[str, Any]]metadatazOptional[list[str]]tagsafter)modereturnrcd|jvr*d}tt|tjd|jvr*d}tt|tjt |j|j}|r-d|}tt|tj|S)z8Validate variable names do not include restricted names.stopzTCannot have an input variable named 'stop', as it is used internally, please rename.message error_codezVCannot have an partial variable named 'stop', as it is used internally, please rename.z/Found overlapping input and partial variables: )r' ValueErrorrrINVALID_PROMPT_INPUTr.set intersection)selfmsgoveralls X/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/langchain_core/prompts/base.pyvalidate_variable_namesz*BasePromptTemplate.validate_variable_namesHs T) ) )" sy7UVVV  T+ + +- sy7UVVV d*++889OPP  MGMMCsy7UVVV  c gdS)zpGet the namespace of the langchain object. Returns ["langchain", "schema", "prompt_template"]. ) langchainschemaprompt_templateclss r@get_lc_namespacez#BasePromptTemplate.get_lc_namespaceds :999rBboolcdS)zJReturn whether this class is serializable. Returns True. TrGrHs r@is_lc_serializablez%BasePromptTemplate.is_lc_serializablels trB)arbitrary_types_alloweddict[str, Any]c t|SNrr=s r@ _serializedzBasePromptTemplate._serializedxsT{{rBr c4tttfS)z%Return the output type of the prompt.)rrrrRs r@ OutputTypezBasePromptTemplate.OutputType|s&(??@@rBconfigOptional[RunnableConfig]type[BaseModel]czfdjD}fdjD}tdi||S)zGet the input schema for the prompt. Args: config: RunnableConfig, configuration for the prompt. Returns: Type[BaseModel]: The input schema for the prompt. cVi|]%}|j|tdf&S).r,getstr.0kr=s r@ z7BasePromptTemplate.get_input_schema..s@$ $ $ 78A $$Q,,c2$ $ $ rBcVi|]%}|j|tdf&SrQr[r^s r@raz7BasePromptTemplate.get_input_schema..s@$ $ $ 89A $$Q,,d3$ $ $ rB PromptInput)field_definitions)r'r)r!)r=rVrequired_input_variablesoptional_input_variabless` r@get_input_schemaz#BasePromptTemplate.get_input_schemas$ $ $ $ <@missing example_keys r@_validate_inputz"BasePromptTemplate._validate_inputs+t,, 4'((A--/2'5 59P55 $[ 1 1555 " # 0N d*++66{CC  9DN399799"299";#3#3#5#56699  "++--K ,[,,#,,, C sy7UVVV rBrcF||}|jdi|SNrG)r{ format_promptr=rh _inner_inputs r@"_format_prompt_with_error_handlingz5BasePromptTemplate._format_prompt_with_error_handlings.++K88 !t!11L111rBcVK||}|jdi|d{VSr})r{aformat_promptrs r@#_aformat_prompt_with_error_handlingz6BasePromptTemplate._aformat_prompt_with_error_handlingsF++K88 (T(88<888888888rBinputkwargsc t|}|jri|d|j|d<|jr|d|jz|d<||j||d|jS)zInvoke the prompt. Args: input: Dict, input to the prompt. config: RunnableConfig, configuration for the prompt. Returns: PromptValue: The output of the prompt. r/r0promptrun_type serialized)r r/r0_call_with_configrrSr=rrVrs r@invokezBasePromptTemplate.invokesv&& = I!HF:$6!H$-!HF:  9 8#F^di7F6N%%  3  ' &   rBcKt|}|jr |d|j|jr |d|j||j||d|jd{VS)zAsync invoke the prompt. Args: input: Dict, input to the prompt. config: RunnableConfig, configuration for the prompt. Returns: PromptValue: The output of the prompt. r/r0rrN)r r/updater0extend_acall_with_configrrSrs r@ainvokezBasePromptTemplate.ainvokesv&& = 5 :  % %dm 4 4 4 9 - 6N ! !$) , , ,,,  4  ' -         rBc dS)zCreate Prompt Value. Args: kwargs: Any arguments to be passed to the prompt template. Returns: PromptValue: The output of the prompt. NrGr=rs r@r~z BasePromptTemplate.format_promptrBc K|jdi|S)zAsync create Prompt Value. Args: kwargs: Any arguments to be passed to the prompt template. Returns: PromptValue: The output of the prompt. rG)r~rs r@rz!BasePromptTemplate.aformat_prompts!"t!++F+++rBUnion[str, Callable[[], str]]c |j}tt|j||d<i|j||d<t|di|S)zReturn a partial of the prompt template. Args: kwargs: Union[str, Callable[[], str]], partial variables to set. Returns: BasePromptTemplate: A partial of the prompt template. r'r.rG)__dict__copyrtr;r'rsr.rq)r=r prompt_dicts r@partialzBasePromptTemplate.partialsm((** )- $ % % 0 0 8 8* * %&,Pd.D+O+O '(tDzz((K(((rBc Td|jD}i||S)NcLi|]!\}}|t|s|n |"SrG)callable)r_r`vs r@razHBasePromptTemplate._merge_partial_and_user_variables.."s@   15AA ,qq   rB)r.items)r=rpartial_kwargss r@!_merge_partial_and_user_variablesz4BasePromptTemplate._merge_partial_and_user_variables sA  9=9O9U9U9W9W   ,.+F++rBr#c dS)aFormat the prompt with the inputs. Args: kwargs: Any arguments to be passed to the prompt template. Returns: A formatted string. Example: .. code-block:: python prompt.format(variable1="foo") NrGrs r@formatzBasePromptTemplate.format'rrBc K|jdi|S)aAsync format the prompt with the inputs. Args: kwargs: Any arguments to be passed to the prompt template. Returns: A formatted string. Example: .. code-block:: python await prompt.aformat(variable1="foo") rG)rrs r@aformatzBasePromptTemplate.aformat8s t{$$V$$$rBr]ct)zReturn the prompt type key.)NotImplementedErrorrRs r@ _prompt_typezBasePromptTemplate._prompt_typeIs "!rBc tjdi|}tjt5|j|d<dddn #1swxYwY|S)a1Return dictionary representation of prompt. Args: kwargs: Any additional arguments to pass to the dictionary. Returns: Dict: Dictionary representation of the prompt. Raises: NotImplementedError: If the prompt type is not implemented. _typeNrG)super model_dump contextlibsuppressrr)r=rrros r@rizBasePromptTemplate.dictNs)egg(22622  !4 5 5 5 5#'#4K  5 5 5 5 5 5 5 5 5 5 5 5 5 5 5s A  AA file_pathUnion[Path, str]Nonect|jrd}t||}d|vrd|d}t|t |}|j}|dd|jdkrG|d5}tj ||d d d d d S#1swxYwYd S|j d rG|d5}tj ||d d d d d S#1swxYwYd S|d}t|)aSave the prompt. Args: file_path: Path to directory to save prompt to. Raises: ValueError: If the prompt has partial variables. ValueError: If the file path is not json or yaml. NotImplementedError: If the prompt type is not implemented. Example: .. code-block:: python prompt.save(file_path="path/prompt.yaml") z*Cannot save prompt with partial variables.rzPrompt z does not support saving.T)parentsexist_okz.jsonw)indentN)z.yamlz.ymlF)default_flow_stylez must be json or yaml) r.r9rirrparentmkdirsuffixopenjsondumpendswithyaml)r=rr>r save_pathdirectory_pathfs r@savezBasePromptTemplate.save_s  ! ">CS// !iikk + % %;D;;;C%c** *OO ")TD999  w & &$$ 4 +q3333 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4   & &'8 9 9 "$$ D +qUCCCC D D D D D D D D D D D D D D D D D D555CS// !s$B88B<?B<4DD D)r3r)r3r&)r3rK)r3rO)r3r rQ)rVrWr3rX)rhr r3ri)rhrir3r)rrirVrWrr r3r)rr r3r)rrr3r%)rr r3rO)rr r3r#)r3r])rr r3ri)rrr3r))rp __module__ __qualname____doc____annotations__rr)rir,r-r.r/r0rrA classmethodrJrMr model_configrrSpropertyrrUrgr{rrrrrr~rrrrrrr __classcell__)ros@r@r%r%-sUCB$)E"$5$5$55555:*/tT)R)R)RKRRRRA04M4444M+05+F+F+FFFFF415H4444* $D$$$$&_'"""#"6:::[:[: $L_ AAAXXA15    X .    D22229999 >B    X 2>B    X 2   ^  , , , ,)))) ,,,,   ^ %%%%""""X""("("("("("("("("rBr%docr"rBasePromptTemplate[str]r3ric\d|ji|jt|j}t |dkrOd|jD}d|dt |d}tt|tj fd|jDS) N page_contentrcg|] }|dk| S)rrG)r_ivs r@ z&_get_document_info..s)   23G3GB3G3G3GrBz?Document prompt requires documents to have metadata variables: z+. Received document with missing metadata: rlr6c"i|] }|| SrGrG)r_r` base_infos r@raz&_get_document_info..s < < <Ay| < < rs @r@_get_document_infors!1BS\BI6122==iHH q    /    )  ) )$%% ) ) )  393Q R R R    = < < >> "Page 1: This is a joke" rG)rrrrs r@format_documentrs(D 6= ; ;-c6:: ; ;;rBcHK|jdit||d{VS)a Async format a document into a string based on a prompt template. First, this pulls information from the document from two sources: 1. page_content: This takes the information from the `document.page_content` and assigns it to a variable named `page_content`. 2. metadata: This takes information from `document.metadata` and assigns it to variables of the same name. Those variables are then passed into the `prompt` to produce a formatted string. Args: doc: Document, the page_content and metadata will be used to create the final string. prompt: BasePromptTemplate, will be used to format the page_content and metadata into the final string. Returns: string of the document formatted. NrG)rrrs r@aformat_documentrs=. BB"4S&"A"ABB B B B B B BBrB)rr"rrr3ri)rr"rrr3r])9r __future__rrrtypingabcrrcollections.abcr functoolsrpathlibrr r r r r rrrpydanticrrrrtyping_extensionsrrlangchain_core.exceptionsrrlangchain_core.loadr"langchain_core.output_parsers.baserlangchain_core.prompt_valuesrrrlangchain_core.runnablesrrlangchain_core.runnables.configr langchain_core.utils.pydanticr!langchain_core.documentsr"r#rir%rrrrGrBr@rs&&"""""" ##############%%%%%% BBBBBBBBBBBB,,,,,,,,????????%%%%%%?????? JIIIIIII99999999999921111117-..Z"Z"Z"Z"Z"{*+W5E-FZ"Z"Z"z ====$"<"<"<"