This commit is contained in:
seehi 2024-03-08 20:22:48 +08:00
parent 9fe9a4a2d1
commit 451cbd7e46

View file

@ -43,6 +43,6 @@ class RAGLLM(CustomLLM):
...
def get_rag_llm(model_infer: BaseLLM = None):
def get_rag_llm(model_infer: BaseLLM = None) -> RAGLLM:
"""Get llm that can be used by LlamaIndex."""
return RAGLLM(model_infer=model_infer or LLM())