llama-index

1
"""Default prompt selectors."""
2

3
from llama_index.legacy.prompts import SelectorPromptTemplate
4
from llama_index.legacy.prompts.chat_prompts import (
5
    CHAT_REFINE_PROMPT,
6
    CHAT_REFINE_TABLE_CONTEXT_PROMPT,
7
    CHAT_TEXT_QA_PROMPT,
8
    CHAT_TREE_SUMMARIZE_PROMPT,
9
)
10
from llama_index.legacy.prompts.default_prompts import (
11
    DEFAULT_REFINE_PROMPT,
12
    DEFAULT_REFINE_TABLE_CONTEXT_PROMPT,
13
    DEFAULT_TEXT_QA_PROMPT,
14
    DEFAULT_TREE_SUMMARIZE_PROMPT,
15
)
16
from llama_index.legacy.prompts.utils import is_chat_model
17

18
DEFAULT_TEXT_QA_PROMPT_SEL = SelectorPromptTemplate(
19
    default_template=DEFAULT_TEXT_QA_PROMPT,
20
    conditionals=[(is_chat_model, CHAT_TEXT_QA_PROMPT)],
21
)
22

23
DEFAULT_TREE_SUMMARIZE_PROMPT_SEL = SelectorPromptTemplate(
24
    default_template=DEFAULT_TREE_SUMMARIZE_PROMPT,
25
    conditionals=[(is_chat_model, CHAT_TREE_SUMMARIZE_PROMPT)],
26
)
27

28
DEFAULT_REFINE_PROMPT_SEL = SelectorPromptTemplate(
29
    default_template=DEFAULT_REFINE_PROMPT,
30
    conditionals=[(is_chat_model, CHAT_REFINE_PROMPT)],
31
)
32

33
DEFAULT_REFINE_TABLE_CONTEXT_PROMPT_SEL = SelectorPromptTemplate(
34
    default_template=DEFAULT_REFINE_TABLE_CONTEXT_PROMPT,
35
    conditionals=[(is_chat_model, CHAT_REFINE_TABLE_CONTEXT_PROMPT)],
36
)
37