o {q i@sddlmZmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZmZGdd d eeZGd d d eZd ed efddZd ed efddZdS))ABCabstractmethod)Callable)BaseLanguageModel) BaseChatModel)BaseLLM)BasePromptTemplate) BaseModelFieldc@s&eZdZdZededefddZdS)BasePromptSelectorz Base class for prompt selectors.llmreturncCsdS)z(Get default prompt for a language model.N)selfr rrg/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/langchain/chains/prompt_selector.py get_promptszBasePromptSelector.get_promptN)__name__ __module__ __qualname____doc__rrrrrrrrr sr c@sXeZdZUdZeed< eedZee e e ge fefed< de defddZ d S) ConditionalPromptSelectorz1Prompt collection that goes through conditionals.default_prompt)default_factory conditionalsr r cCs&|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 listrtuplerrboolrrrrrrs rr r cC 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. ) isinstancerr rrris_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)abcrrtypingrZlangchain_core.language_modelsrZ*langchain_core.language_models.chat_modelsrZ#langchain_core.language_models.llmsrZlangchain_core.promptsrZpydanticr r r rrr#r%rrrrs