a bg@sdZddlmZddlmZmZmZmZmZm Z ddl Z ddl m Z ddl mZddlmZmZmZmZddlmZmZdd lmZdd lmZeedd d d Zdeee eeedddZGdddeZdS)z0z&_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's*      r3c@seZdZUdZedddZeed<edddZeed<dZ e ed <iZ e e e fed <iZe e e fed <eee e fd d dZee d ddZdeeeee eee edddZdS)HumanInputChatModelz3ChatModel which returns user input as the response.cCstSr!)r3rrrrIr'zHumanInputChatModel.)default_factory input_funccCstSr!)rrrrrr5Jr' message_funcr separator input_kwargsmessage_kwargs)rcCs|jj|jjdS)N)r7r8)r7__name__r8selfrrr_identifying_paramsOsz'HumanInputChatModel._identifying_paramscCsdS)zReturns the type of LLM.zhuman-input-chat-modelrr=rrr _llm_typeVszHumanInputChatModel._llm_typeN)rr run_managerkwargsrcKs>|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)r8r;r7r:rr)r>rr rArBZ user_inputrrr _generate[szHumanInputChatModel._generate)NN)r< __module__ __qualname____doc__rr7r__annotations__r8r9r0r:rrr;propertyrr?r@rr rr rrCrrrrr4Fs&   r4)N)rFiortypingrrrrrrrZlangchain_core.callbacksr Z*langchain_core.language_models.chat_modelsr Zlangchain_core.messagesr r r rZlangchain_core.outputsrrZpydanticrZlangchain_community.llms.utilsrrr0r3r4rrrrs