2c2e788 08cbdf8 2c2e788 08cbdf8 2c2e788 08cbdf8 2c2e788
1
2
3
4
5
6
7
8
9
10
11
12
from typing import Callable, Dict _LLMS: Dict[str, Callable] = {} def register_llm(name: str, llm_ask_fn: Callable): _LLMS[name] = llm_ask_fn def get_llm_fn(name: str) -> Callable: return _LLMS[name]
