File size: 213 Bytes
c6138d0 |
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]
|