o Zh@sdZddlmZddlmZmZmZmZmZm Z ddl Z ddl m Z ddl mZddlmZmZmZmZddlmZmZdd lmZdd lmZd eed dfd dZ dd eede eed efddZGdddeZdS)z0sz&_collect_yaml_input..rN)contentz'Cannot use when output is not a string.zInvalid YAML string entered.zInvalid message entered.)inputstripanyappendjoinr rZ safe_loadrr isinstancer(strr ValueErrorZ YAMLError)rr linesrrrr$r_collect_yaml_input's2     r2c @seZdZUdZedddZeed<edddZeed<dZ e ed <iZ e e e fed <iZe e e fed <ed ee e ffd dZed e fddZ  ddeedeee deede d ef ddZdS)HumanInputChatModelz3ChatModel which returns user input as the response.cCtSr!)r2rrrrIzHumanInputChatModel.)default_factory input_funccCr4r!)rrrrrr5Jr6 message_funcr separator input_kwargsmessage_kwargsrcCs|jj|jjdS)N)r8r9)r8__name__r9selfrrr_identifying_paramsOsz'HumanInputChatModel._identifying_paramscCsdS)zReturns the type of LLM.zhuman-input-chat-modelrr>rrr _llm_typeVszHumanInputChatModel._llm_typeNrr run_managerkwargscKs>|j|fi|j|j|fd|i|j}tt|dgdS)a Displays the messages to the user and returns their input as a response. Args: messages (List[BaseMessage]): The messages to be displayed to the user. stop (Optional[List[str]]): A list of stop strings. run_manager (Optional[CallbackManagerForLLMRun]): Currently not used. Returns: ChatResult: The user's input as a response. r )r)Z generations)r9r<r8r;rr)r?rr rBrCZ user_inputrrr _generate[szHumanInputChatModel._generate)NN)r= __module__ __qualname____doc__rr8r__annotations__r9r:r/r;rrr<propertyrr@rArr rr rrDrrrrr3Fs0   r3r!)rGiortypingrrrrrrrZlangchain_core.callbacksr Z*langchain_core.language_models.chat_modelsr Zlangchain_core.messagesr r r rZlangchain_core.outputsrrZpydanticrZlangchain_community.llms.utilsrrr/r2r3rrrrs(