ghUdZddlZddlZddlmZddlmZmZmZddl Z ddl m Z ddl m Z ddlmZddlmZdd lmZd ZejeZd ed e fd Zded ed efdZd ed efdZd ed efdZd ed efdZd ed efdZ ddeeefdeed e fdZ ddeeefdeed e fdZ!d ed efdZ"eee"dZ#eeeege ffe$d<dS)z Load prompts.N)Path)CallableOptionalUnion)StrOutputParser)BasePromptTemplate)ChatPromptTemplate)FewShotPromptTemplate)PromptTemplatezIhttps://raw.githubusercontent.com/hwchase17/langchain-hub/master/prompts/configreturncd|vrtd|dd}|tvrd|d}t |t|}||S)zLoad prompt from Config Dict. Args: config: Dict containing the prompt configuration. Returns: A PromptTemplate object. Raises: ValueError: If the prompt type is not supported. _typez-No `_type` key found, defaulting to `prompt`.promptzLoading z prompt not supported)loggerwarningpoptype_to_loader_dict ValueError)r config_typemsg prompt_loaders [/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/langchain_core/prompts/loading.pyload_prompt_from_configrswfFGGG**Wh//K---;;;;oo' 4M =  var_namec|d|vrm||vrd|d|d}t|t||d}|jdkr|}nt|||<|S)z*Load template from the path if applicable._pathzBoth `z _path` and `z` cannot be provided.z.txt)rrrsuffix read_text)rr r template_pathtemplates r_load_templater#,s V## v  P8PPPPPCS// !VZZ8(:(:(:;;<<  6 ) )$..00HH #x Mrct|dtrnt|dtrt|d}|5}|jdkrt j|}n/|jdvrtj |}nd}t|dddn #1swxYwY||d<nd}t||S)zLoad examples if necessary.examples.json>.yml.yamlz=Invalid file format. Only json or yaml formats are supported.Nz;Invalid examples format. Only list or string are supported.) isinstanceliststrropenrjsonloadyaml safe_loadr)r pathfr%rs r_load_examplesr3@s&$d++ F:& , , F:&'' YY[[ &A{g%%9Q<< 111>!,,U oo% & & & & & & & & & & & & & & &&zKoo Ms!AB==CCcd|vr^|drV|d}|d}|dkr tdi|}nd|}t|||d<|S)zLoad output parser. output_parserrdefaultzUnsupported output parser )rrr)r _configoutput_parser_typer5rs r_load_output_parserr:Us&  VO%< **_--$[[11  * *+66g66MMC/ACCCS// !"/ MrcDtd|}td|}d|vr;d|vrd}t|t|d|d<nt |d|d<t |}t |}tdi|S)z+Load the "few shot" prompt from the config.rprefixexample_prompt_pathexample_promptzGOnly one of example_prompt and example_prompt_path should be specified.r7)r#r load_promptrrr3r:r )r rs r_load_few_shot_promptr@csHf - -F Hf - -F&& v % %  S// !#.vzz:O/P/P#Q#Q  #:6BR;S#T#T F # #F  ( (F * *6 * **rctd|}t|}|dd}|dkrd|d}t|t di|S)z%Load the prompt template from config.r"template_formatzf-stringjinja2zLoading templates with 'z' format is no longer supported since it can lead to arbitrary code execution. Please migrate to using the 'f-string' template format, which does not suffer from this issue.r7)r#r:getrr )r rBrs r _load_promptrEysJ / /F  ( (Fjj!2J??O("" V V V V oo  # #F # ##rr1encodingct|tr&|drd}t|t ||S)a+Unified method for loading a prompt from LangChainHub or local fs. Args: path: Path to the prompt file. encoding: Encoding of the file. Defaults to None. Returns: A PromptTemplate object. Raises: RuntimeError: If the path is a Lang Chain Hub path. zlc://zLoading from the deprecated github-based Hub is no longer supported. Please use the new LangChain Hub at https://smith.langchain.com/hub instead.)r)r+ startswith RuntimeError_load_prompt_from_file)r1rFrs rr?r?sS$ !9!9   3 !$ 1 11rfilect|}|jdkrC||5}tj|}dddn #1swxYwYnv|jdrC||5}t j|}dddn #1swxYwYnd|j}t|t|S)zLoad prompt from file.r&)rFN)r(r'zGot unsupported file type ) rrr,r-r.endswithr/r0rr)rKrF file_pathr2r rs rrJrJsM T I7"" ^^X^ . . "!Yq\\F " " " " " " " " " " " " " " "   " "#4 5 5 ^^X^ . . '!^A&&F ' ' ' ' ' ' ' ' ' ' ' ' ' ' '>9+;==oo "6 * **s#AAAB//B36B3c|d}|r!|dddnd}|d|sd}t|tjdd|i|S) zLoad chat prompt from config.messagesrrr"Ninput_variablesz'Can't load chat prompt without templater7)rrr from_template)r rPr"rs r_load_chat_promptrSszz*%%H8@Jx{8$((444dH JJ !!! 7oo  + H HX H H HHr)rfew_shotchatr)N)%__doc__r-loggingpathlibrtypingrrrr/$langchain_core.output_parsers.stringrlangchain_core.prompts.baserlangchain_core.prompts.chatr langchain_core.prompts.few_shotr langchain_core.prompts.promptr URL_BASE getLogger__name__rdictrr+r#r3r:r@rEr?rJrSr__annotations__r7rrrds ,,,,,,,,,, @@@@@@::::::::::::AAAAAA888888 V  8 $ $!D!-?!!!!0S$4(4D*      +$++@++++,$$.$$$$*7;22 T 2&.sm2222247;++ T +&.sm+++++( Id I'9 I I I I% HHT#x0B(BCCDr