Reorganize dependencies in pyproject.toml

The heaviest dependencies, such as torch and huggingface, are left in optional groups to improve the size of the installation.
GPTStonks · Mar 28, 2024 · 444c7ff · 444c7ff
1 parent cf389a7
commit 444c7ff
Show file tree

Hide file tree

Showing 3 changed files with 158 additions and 150 deletions.
diff --git a/openbb_chat/kernels/auto_llama_index.py b/openbb_chat/kernels/auto_llama_index.py
@@ -16,7 +16,11 @@
 from llama_index.core.query_engine import RetrieverQueryEngine
 from llama_index.core.retrievers import VectorIndexRetriever
 from llama_index.core.schema import NodeWithScore
-from llama_index.llms.huggingface import HuggingFaceLLM
+
+try:
+    from llama_index.llms.huggingface import HuggingFaceLLM
+except Exception as e:
+    print("Warning: HuggingFaceLLM not imported, so HuggingFace models are not available")
 from llama_index.llms.openai import OpenAI
 from llama_index.retrievers.bm25 import BM25Retriever
 from pydantic import BaseModel