Update dependencies for fast installation (#811)

2023-10-17 20:51:30 +05:30
parent c336292346
commit 413b107b9a
9 changed files with 73 additions and 24 deletions
--- a/embedchain/llm/base.py
+++ b/embedchain/llm/base.py
@@ -129,8 +129,12 @@ class BaseLlm(JSONSerializable):
        :return: Search results
        :rtype: Unknown
        """
-        from langchain.tools import DuckDuckGoSearchRun
-
+        try:
+            from langchain.tools import DuckDuckGoSearchRun
+        except ImportError:
+            raise ImportError(
+                'Searching requires extra dependencies. Install with `pip install --upgrade "embedchain[dataloaders]"`'
+            ) from None
        search = DuckDuckGoSearchRun()
        logging.info(f"Access search to get answers for {input_query}")
        return search.run(input_query)