File size: 523 Bytes
c0cd1dc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from typing import Optional, Union
from llama_index.llms.base import LLM
from langchain.base_language import BaseLanguageModel
from kron.llm_predictor.KronLangChainLLM import KronLangChainLLM
from llama_index.llms.openai import OpenAI
from llama_index.llms.utils import LLMType
def kron_resolve_llm(llm: Optional[LLMType] = None) -> LLM:
if isinstance(llm, BaseLanguageModel):
# NOTE: if it's a langchain model, wrap it in a LangChainLLM
return KronLangChainLLM(llm=llm)
return llm or OpenAI()
|