lumis-ai 0.1.0a1__py3-none-any.whl → 0.1.1a1__py3-none-any.whl

lumis/core/django/__init__.py

@@ -1,5 +1,8 @@
 from __future__ import annotations
 
-from .models.chat_memory import ChatMemory
+try:
+    from .models.chat_memory import ChatMemory
 
-__all__ = ["ChatMemory"]
+    __all__ = ["ChatMemory"]
+except ImportError:
+    __all__ = []

lumis/nlp/information_extraction/FactExtractor.py

@@ -1,9 +1,6 @@
-from typing import Optional, Set, Tuple
+from __future__ import annotations
 
-import spacy
-from spacy.language import Language
-from spacy.matcher import Matcher
-from spacy.tokens import Doc, Span, Token
+from typing import Optional, Set, Tuple
 
 
 # TODO: Convert this to a spacy factory
@@ -13,9 +10,19 @@ class FactExtractor:
         Initialize the FactExtractor with a spaCy NLP pipeline.
         :param model: The name of the spaCy model to load.
         """
+        try:
+            import spacy
+            from spacy.matcher import Matcher
+            from spacy.tokens import Doc
+        except ImportError:
+            raise ImportError(
+                "spacy is required for FactExtractor. "
+                "Install it with: pip install lumis-ai[spacy]"
+            )
+
         # spaCy handles caching and singleton behavior automatically
-        self.nlp: Language = spacy.load(model)
-        self.matcher: Matcher = Matcher(self.nlp.vocab)
+        self.nlp = spacy.load(model)
+        self.matcher = Matcher(self.nlp.vocab)
         self._init_patterns()
 
         # Register custom extension if not already registered

lumis/nlp/ner/ner.py

@@ -1,13 +1,23 @@
+from __future__ import annotations
+
 from pathlib import Path
+from typing import TYPE_CHECKING
 
-from spacy.tokens import Doc
-from spacy_llm.util import assemble
+if TYPE_CHECKING:
+    from spacy.tokens import Doc
 
 relative_path = Path(__file__).parent / "config" / "entity_rel_extraction.cfg"
 
 
 class Ner:
     def __init__(self, config: str = str(relative_path)):
+        try:
+            from spacy_llm.util import assemble
+        except ImportError:
+            raise ImportError(
+                "spacy and spacy-llm are required for Ner. "
+                "Install them with: pip install lumis-ai[spacy]"
+            )
         self.nlp = assemble(config)
 
     def get_entities(self, text: str) -> list[tuple[str, str]]:

{lumis_ai-0.1.0a1.dist-info → lumis_ai-0.1.1a1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lumis-ai
-Version: 0.1.0a1
+Version: 0.1.1a1
 Summary: An AI agent framework for building LLM-powered applications
 Project-URL: Homepage, https://github.com/tareksanger/lumis
 Project-URL: Repository, https://github.com/tareksanger/lumis
@@ -23,12 +23,7 @@ Requires-Python: >=3.11
 Requires-Dist: arxiv>=0.5.0
 Requires-Dist: asgiref>=3.8.1
 Requires-Dist: beautifulsoup4>=4.12.0
-Requires-Dist: coreferee-model-en>=1.0.0
-Requires-Dist: coreferee>=1.3.0
 Requires-Dist: deepmerge>=2.0.0
-Requires-Dist: django>=4.2
-Requires-Dist: en-core-web-lg<3.8.0,>=3.7.0
-Requires-Dist: en-core-web-trf<3.8.0,>=3.7.0
 Requires-Dist: faiss-cpu>=1.7.0
 Requires-Dist: googlesearch-python>=1.2.0
 Requires-Dist: httpx>=0.27.0
@@ -54,8 +49,6 @@ Requires-Dist: redis>=5.0.0
 Requires-Dist: requests>=2.28.0
 Requires-Dist: scikit-learn>=1.0.0
 Requires-Dist: sentence-transformers>=2.2.2
-Requires-Dist: spacy-llm>=0.7.2
-Requires-Dist: spacy<3.8.0,>=3.7.0
 Requires-Dist: tavily-python>=0.3.0
 Requires-Dist: tenacity>=9.0.0
 Requires-Dist: textstat>=0.7.0
@@ -65,6 +58,15 @@ Requires-Dist: transformers>=4.0.0
 Requires-Dist: typing-extensions>=4.0.0
 Requires-Dist: wikipedia>=1.4.0
 Requires-Dist: yfinance>=0.2.0
+Provides-Extra: django
+Requires-Dist: django>=4.2; extra == 'django'
+Provides-Extra: spacy
+Requires-Dist: coreferee-model-en>=1.0.0; extra == 'spacy'
+Requires-Dist: coreferee>=1.3.0; extra == 'spacy'
+Requires-Dist: en-core-web-lg<3.8.0,>=3.7.0; extra == 'spacy'
+Requires-Dist: en-core-web-trf<3.8.0,>=3.7.0; extra == 'spacy'
+Requires-Dist: spacy-llm>=0.7.2; extra == 'spacy'
+Requires-Dist: spacy<3.8.0,>=3.7.0; extra == 'spacy'
 Description-Content-Type: text/markdown
 
 # lumis
@@ -88,7 +90,16 @@ An AI agent framework for building LLM-powered applications with multi-provider
 pip install lumis-ai
 ```
 
-After installing, download the required spaCy language models:
+### Optional extras
+
+Some integrations are opt-in to keep the base install lighter:
+
+| Extra      | What it adds                                  | Install                           |
+|------------|-----------------------------------------------|-----------------------------------|
+| `spacy`    | NER, fact extraction, coreference resolution  | `pip install lumis-ai[spacy]`     |
+| `django`   | Django ORM memory backend                     | `pip install lumis-ai[django]`    |
+
+The `spacy` extra requires language models. After installing, download them:
 
 ```bash
 python -m spacy download en_core_web_lg

{lumis_ai-0.1.0a1.dist-info → lumis_ai-0.1.1a1.dist-info}/RECORD

@@ -24,7 +24,7 @@ lumis/core/document.py,sha256=bat9TjbHigFf5TLVy9r0Ft67zjIMPJMwmQrsE4gP-IA,1080
 lumis/core/event_emitter.py,sha256=2LwAo0gLIIKMqKocjrZ3g86BRjrtnplTQth0S8iDw_U,2094
 lumis/core/common/coloured_logger.py,sha256=Pc_WcpPXfMIGHagrnJQ2MjY6x8MI1U5ZbTt5xtb8mgg,943
 lumis/core/common/logger_mixin.py,sha256=TmswwCfgEi8BeatoWsa3gMk3qhCEMEmZyLK4rkZWLUU,1013
-lumis/core/django/__init__.py,sha256=1eLq5GHCa7iCrlb3eg85Ifx9TtnnsYgMeZ6rakyzkWY,105
+lumis/core/django/__init__.py,sha256=yOr6iSaA7VSlIpSwryrwtL5zIHuv5HcY-9Fa9czoyRI,155
 lumis/core/django/models/chat_memory.py,sha256=j5E36hza2nbIKjVlv0j0WjttmRvOLjvTnoMFgNyUVVQ,6612
 lumis/core/utils/__init__.py,sha256=2hfqwepX9pCGlsPHZ3P1Zv_7jW0BCuW5r9P0aCu-xvU,246
 lumis/core/utils/coroutine.py,sha256=B_KQfwa6_wWI7a3hVPWv8GLAyuDPQywPgBYDd0iI4cI,665
@@ -65,8 +65,8 @@ lumis/nlp/semantic_parser.py,sha256=ydJjGy5h-gzaG0AJkBZOwj8JagoPxefm8RZx9D9FnsM,
 lumis/nlp/spacy_summarizer.py,sha256=6_ZMOYJ0gs7bcsef-x6GEhzsssitlpf4Ovw58SfEllw,2762
 lumis/nlp/summarizer.py,sha256=R-oTjtS86TnySTXjVhNf_NaTNJNbrfX8p964V5YfEIA,6920
 lumis/nlp/vector_similarity_retriever.py,sha256=mR2FjISAkAgS5bzSBwZIdnSr3q9s_r0GhCgrt5PlMAA,1714
-lumis/nlp/information_extraction/FactExtractor.py,sha256=3gA2LbAO9xOdcguoZQVssQYfDCOT6O9glWqBCUWxu_o,9355
-lumis/nlp/ner/ner.py,sha256=Dfxz2fuDedrd5kwLTHnpQQ5KnFOsmHEe1IfmvgQMYcI,504
+lumis/nlp/information_extraction/FactExtractor.py,sha256=Chu-fMoyBJe0axM-aH6T_LWcWCVyh_lzZOSRzePszJY,9564
+lumis/nlp/ner/ner.py,sha256=CVTft8EKgl3kvy9gzM0I2ANnLNmAGnYz0SdXXdf34II,819
 lumis/nlp/ner/config/entity_rel_extraction.cfg,sha256=rQKm-Agl_qse0EwI9XgLy0kjU4hdLe4B15bvspny5qY,923
 lumis/nlp/ner/config/fill-config.cfg,sha256=iMhF-QfZs15UtDHTBC3OPAhfaqyHMpnvqRAcWVAMU6U,4130
 lumis/reader/__init__.py,sha256=707frfd3cLTb7LTjhsWyFhb-u21YqaifWcyiCpVgYvg,52
@@ -84,7 +84,7 @@ lumis/tools/search/search_engine_client.py,sha256=5Q0nb7Lztj0mjaN3EwZQfpRMNUbxcH
 lumis/tools/search/vector_search_retrieval_engine.py,sha256=RGZyFah2F-ojFzH9xiApILK5oQWWZEuihV2EN3Ae0cc,7889
 lumis/tools/search/wiki.py,sha256=gGm6I273KOLvDN0E7boAezjPq6cywRhbZEBqB_Pem4U,7864
 lumis/tools/search/yahoofinance.py,sha256=xQTTMJU7IDfH2WkzIWPjWwwLao0unfMHGrbhVEJJSJg,7548
-lumis_ai-0.1.0a1.dist-info/METADATA,sha256=bVQmNhCCxjia24iGCDc-brqlxqfrp2Poy74ubstINpU,3783
-lumis_ai-0.1.0a1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-lumis_ai-0.1.0a1.dist-info/licenses/LICENSE,sha256=cKZn_mblbydX1fOrVMvUIrelUE9AElKLAR_7-VYRDow,1068
-lumis_ai-0.1.0a1.dist-info/RECORD,,
+lumis_ai-0.1.1a1.dist-info/METADATA,sha256=91heIxT2Jsuyf9cVXQtZG6hSugavVlX0FMOk6oAjjL8,4451
+lumis_ai-0.1.1a1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+lumis_ai-0.1.1a1.dist-info/licenses/LICENSE,sha256=cKZn_mblbydX1fOrVMvUIrelUE9AElKLAR_7-VYRDow,1068
+lumis_ai-0.1.1a1.dist-info/RECORD,,