o Jjg@sddlmZmZddlmZmZmZddlmZddl m Z ddl m Z ddl mZddlmZmZGdd d eeZGd d d eZd ed efddZd ed efddZdS))ABCabstractmethod)CallableListTuple)BaseLanguageModel) BaseChatModel)BaseLLM)BasePromptTemplate) BaseModelFieldc@s&eZdZdZededefddZdS)BasePromptSelectorz Base class for prompt selectors.llmreturncCsdS)z(Get default prompt for a language model.N)selfrrrX/var/www/html/zoom/venv/lib/python3.10/site-packages/langchain/chains/prompt_selector.py get_promptszBasePromptSelector.get_promptN)__name__ __module__ __qualname____doc__rrr rrrrrr sr c@sXeZdZUdZeed< eedZe e e e ge fefed< de defddZd S) ConditionalPromptSelectorz1Prompt collection that goes through conditionals.default_prompt)default_factory conditionalsrrcCs&|jD] \}}||r|Sq|jS)zGet default prompt for a language model. Args: llm: Language model to get prompt for. Returns: Prompt to use for the language model. )rr)rr conditionpromptrrrrs  z$ConditionalPromptSelector.get_promptN)rrrrr __annotations__r listrrrrrboolrrrrrrs rrrcC t|tS)zCheck if the language model is a LLM. Args: llm: Language model to check. Returns: True if the language model is a BaseLLM model, False otherwise. ) isinstancer rrrris_llm, r$cCr!)zCheck if the language model is a chat model. Args: llm: Language model to check. Returns: True if the language model is a BaseChatModel model, False otherwise. )r"rr#rrr is_chat_model8r%r&N)abcrrtypingrrrlangchain_core.language_modelsr*langchain_core.language_models.chat_modelsr#langchain_core.language_models.llmsr langchain_core.promptsr pydanticr r r rr r$r&rrrrs