lumis-ai 0.1.1a2__tar.gz → 0.1.1a3__tar.gz

{lumis_ai-0.1.1a2 → lumis_ai-0.1.1a3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lumis-ai
-Version: 0.1.1a2
+Version: 0.1.1a3
 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
@@ -24,6 +24,7 @@ Requires-Dist: asgiref>=3.8.1
 Requires-Dist: beautifulsoup4>=4.12.0
 Requires-Dist: deepmerge>=2.0.0
 Requires-Dist: faiss-cpu>=1.7.0
+Requires-Dist: google-genai>=1.61.0
 Requires-Dist: googlesearch-python>=1.2.0
 Requires-Dist: httpx>=0.27.0
 Requires-Dist: huggingface-hub>=0.24.0
@@ -31,7 +32,6 @@ Requires-Dist: langchain-core>=0.3.13
 Requires-Dist: langchain>=0.3.4
 Requires-Dist: lxml>=4.9.0
 Requires-Dist: minify-html>=0.11.0
-Requires-Dist: neo4j>=5.25.0
 Requires-Dist: nltk>=3.9.0
 Requires-Dist: numpy>=1.24
 Requires-Dist: ollama>=0.3.0
@@ -42,12 +42,11 @@ Requires-Dist: pymupdf>=1.22.0
 Requires-Dist: pypdf2>=3.0.0
 Requires-Dist: python-dateutil>=2.8.0
 Requires-Dist: pyvis>=0.3.0
-Requires-Dist: readability-lxml>=3.3.0
+Requires-Dist: readability-lxml>=0.8.0
 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: tavily-python>=0.3.0
 Requires-Dist: tenacity>=9.0.0
 Requires-Dist: textstat>=0.7.0
 Requires-Dist: tiktoken>=0.5.0
@@ -59,13 +58,11 @@ Provides-Extra: django
 Requires-Dist: django>=4.2; extra == 'django'
 Provides-Extra: search
 Requires-Dist: arxiv>=0.5.0; extra == 'search'
-Requires-Dist: pytrends>=5.1.0; extra == 'search'
+Requires-Dist: pytrends>=4.5.0; extra == 'search'
+Requires-Dist: tavily-python>=0.3.0; extra == 'search'
 Requires-Dist: yfinance>=0.2.0; extra == 'search'
 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: pytextrank>=3.2.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_ai-0.1.1a2 → lumis_ai-0.1.1a3}/lumis/agents/research_agent/tools/search.py

@@ -7,8 +7,6 @@ from typing import Any, Optional
 from lumis.llm import Gemini
 from lumis.tools import WikipediaSearcher
 from lumis.tools.search.arxiv import ArxivResult, ArxivSearcher
-
-from arxiv import SortCriterion, SortOrder
 from tenacity import retry, retry_if_exception_type, stop_after_attempt, wait_exponential
 
 logger = logging.getLogger(__name__)
@@ -173,7 +171,8 @@ async def search_arxiv(  # noqa: C901
     summary_semaphore = asyncio.Semaphore(4)
 
     try:
-        # Convert string sort parameters to enums
+        from arxiv import SortCriterion, SortOrder
+
         sort_criterion = SortCriterion(sort_by)
         sort_order_enum = SortOrder(sort_order)
 

{lumis_ai-0.1.1a2 → lumis_ai-0.1.1a3}/lumis/tools/search/search_engine_client.py

@@ -8,9 +8,16 @@ from typing import Literal, Optional, Sequence
 
 from googlesearch import search as google_search
 import redis.asyncio as aioredis
-from tavily import TavilyClient, UsageLimitExceededError
 from typing_extensions import TypedDict
 
+try:
+    from tavili import TavilyClient, UsageLimitExceededError
+except ImportError:
+    TavilyClient = None  # type: ignore[assignment,misc]
+
+    class UsageLimitExceededError(Exception):  # type: ignore[no-redef]
+        pass
+
 logger = logging.getLogger(__name__)
 
 

{lumis_ai-0.1.1a2 → lumis_ai-0.1.1a3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lumis-ai"
-version = "0.1.1a2"
+version = "0.1.1a3"
 description = "An AI agent framework for building LLM-powered applications"
 readme = "README.md"
 license = "MIT"
@@ -33,7 +33,6 @@ dependencies = [
     "langchain-core>=0.3.13",
     "lxml>=4.9.0",
     "minify-html>=0.11.0",
-    "neo4j>=5.25.0",
     "nltk>=3.9.0",
     "numpy>=1.24",
     "ollama>=0.3.0",
@@ -44,12 +43,11 @@ dependencies = [
     "pydantic>=2.9.2",
     "python-dateutil>=2.8.0",
     "pyvis>=0.3.0",
-    "readability-lxml>=3.3.0",
+    "readability-lxml>=0.8.0",
     "redis>=5.0.0",
     "requests>=2.28.0",
     "scikit-learn>=1.0.0",
     "sentence-transformers>=2.2.2",
-    "tavily-python>=0.3.0",
     "tenacity>=9.0.0",
     "textstat>=0.7.0",
     "tiktoken>=0.5.0",
@@ -57,6 +55,7 @@ dependencies = [
     "transformers>=4.0.0",
     "typing-extensions>=4.0.0",
     "wikipedia>=1.4.0",
+    "google-genai>=1.61.0",
 ]
 
 [project.optional-dependencies]
@@ -65,14 +64,12 @@ django = [
 ]
 search = [
     "arxiv>=0.5.0",
-    "pytrends>=5.1.0",
+    "pytrends>=4.5.0",
+    "tavily-python>=0.3.0",
     "yfinance>=0.2.0",
 ]
 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",
+    "pytextrank>=3.2.0",
     "spacy>=3.7.0,<3.8.0",
     "spacy-llm>=0.7.2",
 ]
@@ -96,8 +93,6 @@ dev = [
 torch = [
     { index = "pytorch", marker = "sys_platform != 'darwin'" },
 ]
-en-core-web-trf = { url = "https://github.com/explosion/spacy-models/releases/download/en_core_web_trf-3.7.1/en_core_web_trf-3.7.1-py3-none-any.whl" }
-en-core-web-lg = { url = "https://github.com/explosion/spacy-models/releases/download/en_core_web_lg-3.7.0/en_core_web_lg-3.7.0-py3-none-any.whl" }
 
 [[tool.uv.index]]
 name = "pytorch"