This commit is contained in:
seehi
2024-03-08 20:22:48 +08:00
parent 9fe9a4a2d1
commit 451cbd7e46
+1 -1
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())