removed dependency for ollama

assafelovic · Feb 15, 2025 · 953b758 · 953b758
1 parent 4f901fb
commit 953b758
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/gpt_researcher/llm_provider/generic/base.py b/gpt_researcher/llm_provider/generic/base.py
@@ -72,6 +72,7 @@ def from_provider(cls, provider: str, **kwargs: Any):
             llm = ChatFireworks(**kwargs)
         elif provider == "ollama":
             _check_pkg("langchain_community")
+            _check_pkg("langchain_ollama")
             from langchain_ollama import ChatOllama
 
             llm = ChatOllama(base_url=os.environ["OLLAMA_BASE_URL"], **kwargs)