a ag@sddlmZmZddlmZmZmZddlmZddl m Z ddl m Z ddl mZddlmZmZGdd d eeZGd d d eZeed d dZeed ddZdS))ABCabstractmethod)CallableListTuple)BaseLanguageModel) BaseChatModel)BaseLLM)BasePromptTemplate) BaseModelFieldc@s$eZdZdZeeedddZdS)BasePromptSelectorz Base class for prompt selectors.llmreturncCsdS)z(Get default prompt for a language model.N)selfrrrn/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain/chains/prompt_selector.py get_promptszBasePromptSelector.get_promptN)__name__ __module__ __qualname____doc__rrr rrrrrr sr c@sReZdZUdZeed<eedZe e e e ge fefed<e edddZdS) ConditionalPromptSelectorz1Prompt collection that goes through conditionals.default_prompt)default_factory conditionalsrcCs&|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 rrcCs 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,s r$cCs t|tS)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_model8s r%N)abcrrtypingrrrZlangchain_core.language_modelsrZ*langchain_core.language_models.chat_modelsrZ#langchain_core.language_models.llmsr Zlangchain_core.promptsr Zpydanticr r r rr!r$r%rrrrs