refactor(examples/simple-app) - in make_llm_instance return LanguageM…

…odelLike type
ksachdeva · Oct 31, 2024 · 3a26146 · 3a26146
1 parent eb8799c
commit 3a26146
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/simple-app/app/common.py b/examples/simple-app/app/common.py
@@ -12,7 +12,7 @@
 from langchain_community.cache import SQLiteCache
 from langchain_community.storage import SQLStore
 from langchain_core.embeddings import Embeddings
-from langchain_core.language_models import BaseLLM
+from langchain_core.language_models import LanguageModelLike
 from langchain_ollama import OllamaEmbeddings, OllamaLLM
 from langchain_openai import (
     AzureChatOpenAI,
@@ -97,7 +97,7 @@ def make_llm_instance(
     ollama_num_context: int | None = None,
     temperature: float = 0.0,
     top_p: float = 1.0,
-) -> BaseLLM:
+) -> LanguageModelLike:
     if llm_type == LLMType.openai:
         check_if_necessary_openai_env_set()
         return ChatOpenAI(