llama-index

1
from string import Formatter
2
from typing import List
3

4
from llama_index.legacy.llms.base import BaseLLM
5

6

7
def get_template_vars(template_str: str) -> List[str]:
8
    """Get template variables from a template string."""
9
    variables = []
10
    formatter = Formatter()
11

12
    for _, variable_name, _, _ in formatter.parse(template_str):
13
        if variable_name:
14
            variables.append(variable_name)
15

16
    return variables
17

18

19
def is_chat_model(llm: BaseLLM) -> bool:
20
    return llm.metadata.is_chat_model
21