lumis-ai 0.1.0a1__tar.gz → 0.1.1a1__tar.gz

{lumis_ai-0.1.0a1 → lumis_ai-0.1.1a1}/PKG-INFO

@@ -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 → lumis_ai-0.1.1a1}/README.md

@@ -19,7 +19,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.1a1/lumis/core/django/__init__.py

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

{lumis_ai-0.1.0a1 → lumis_ai-0.1.1a1}/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_ai-0.1.0a1 → lumis_ai-0.1.1a1}/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 → lumis_ai-0.1.1a1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lumis-ai"
-version = "0.1.0a1"
+version = "0.1.1a1"
 description = "An AI agent framework for building LLM-powered applications"
 readme = "README.md"
 license = "MIT"
@@ -25,12 +25,7 @@ dependencies = [
     "arxiv>=0.5.0",
     "asgiref>=3.8.1",
     "beautifulsoup4>=4.12.0",
-    "coreferee>=1.3.0",
-    "coreferee-model-en>=1.0.0",
     "deepmerge>=2.0.0",
-    "django>=4.2",
-    "en-core-web-lg>=3.7.0,<3.8.0",
-    "en-core-web-trf>=3.7.0,<3.8.0",
     "faiss-cpu>=1.7.0",
     "googlesearch-python>=1.2.0",
     "httpx>=0.27.0",
@@ -56,8 +51,6 @@ dependencies = [
     "requests>=2.28.0",
     "scikit-learn>=1.0.0",
     "sentence-transformers>=2.2.2",
-    "spacy>=3.7.0,<3.8.0",
-    "spacy-llm>=0.7.2",
     "tavily-python>=0.3.0",
     "tenacity>=9.0.0",
     "textstat>=0.7.0",
@@ -69,6 +62,19 @@ dependencies = [
     "yfinance>=0.2.0",
 ]
 
+[project.optional-dependencies]
+django = [
+    "django>=4.2",
+]
+spacy = [
+    "coreferee>=1.3.0",
+    "coreferee-model-en>=1.0.0",
+    "en-core-web-lg>=3.7.0,<3.8.0",
+    "en-core-web-trf>=3.7.0,<3.8.0",
+    "spacy>=3.7.0,<3.8.0",
+    "spacy-llm>=0.7.2",
+]
+
 [project.urls]
 Homepage = "https://github.com/tareksanger/lumis"
 Repository = "https://github.com/tareksanger/lumis"

lumis_ai-0.1.0a1/lumis/core/django/__init__.py

@@ -1,5 +0,0 @@
-from __future__ import annotations
-
-from .models.chat_memory import ChatMemory
-
-__all__ = ["ChatMemory"]