lumis-ai 0.1.1a2__py3-none-any.whl → 0.1.1a4__py3-none-any.whl

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/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.dist-info → lumis_ai-0.1.1a4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lumis-ai
-Version: 0.1.1a2
+Version: 0.1.1a4
 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.dist-info → lumis_ai-0.1.1a4.dist-info}/RECORD

@@ -13,7 +13,7 @@ lumis/agents/research_agent/source_models.py,sha256=w5cIJcqQtVTH-nDMR8m152ZPvy3_
 lumis/agents/research_agent/source_tracker.py,sha256=m3SHXrV5mmg6wrDtqMGVUAHShTxJAV43hheLKAzinLQ,14361
 lumis/agents/research_agent/types.py,sha256=bRlXThmSdZez6aT1ZXqo6KFCz0jmCmWgA-GqCgbNEdI,1767
 lumis/agents/research_agent/tools/__init__.py,sha256=dSl8jV7BOvZzJ-1o81r25cf0Phl6PPJWAHsw71-h-Sk,166
-lumis/agents/research_agent/tools/search.py,sha256=JgwqZf3GvDWeH7sxBCOZie28qemqDh_ZdcYWqrSGDiQ,12151
+lumis/agents/research_agent/tools/search.py,sha256=Hbm1BPriQGdq9XU2IeYN6ILl5QTAqxQhA-GvZgQgn88,12109
 lumis/agents/storm/agent.py,sha256=wdZ4OcFuNsgBSlXaNhvz8kStEcUAnNxX8YEvV96OR8M,21899
 lumis/agents/storm/interview_graph.py,sha256=waFYUiMsT-SHCsxEDx1qkpbJ913MztILmnMhRKwGd6o,14585
 lumis/agents/storm/models/outline.py,sha256=4xEr_kfMlq5fbWf7mmerRYV_tKqzx-qeU1Inl1lwY4c,2152
@@ -80,11 +80,11 @@ lumis/tools/scraper.py,sha256=0iNQS2jptT093ZkOSKamvNDOQuJRBaAwM2_ztmS1w6U,13670
 lumis/tools/search/__init__.py,sha256=_MGFJAeizjcX0oI9Cr0B9efTfolRbL_3QQo2ZcNUklE,283
 lumis/tools/search/arxiv.py,sha256=ldRHo_SSyhKm7CwjCuOjxZlIhkutwu85_UD8YcXCAtI,10245
 lumis/tools/search/pytrends.py,sha256=_tTmGMNNqSUbN3s7Scrx771KzX9nLHuYQrxFPC_nXIo,1980
-lumis/tools/search/search_engine_client.py,sha256=5Q0nb7Lztj0mjaN3EwZQfpRMNUbxcHyhp-wGbmOZdiA,11766
+lumis/tools/search/search_engine_client.py,sha256=WSYOQcy1-6gQZUJXTT5lYNFJWjwLSruYuLD_GAiT0gM,11939
 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=0Oz7kfdWI2ehllo9NM2wKS0WSAUeGnPyuVnfobyBXHA,7851
-lumis_ai-0.1.1a2.dist-info/METADATA,sha256=Z0Ajv1lr3HYpO6izksSyOIU-79wK1KGzMgG5l0-6K4Y,4630
-lumis_ai-0.1.1a2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-lumis_ai-0.1.1a2.dist-info/licenses/LICENSE,sha256=cKZn_mblbydX1fOrVMvUIrelUE9AElKLAR_7-VYRDow,1068
-lumis_ai-0.1.1a2.dist-info/RECORD,,
+lumis_ai-0.1.1a4.dist-info/METADATA,sha256=w2TTLi05kh3rtjascf-xJ8Iym78sBAN5tH42_1LElgg,4473
+lumis_ai-0.1.1a4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+lumis_ai-0.1.1a4.dist-info/licenses/LICENSE,sha256=cKZn_mblbydX1fOrVMvUIrelUE9AElKLAR_7-VYRDow,1068
+lumis_ai-0.1.1a4.dist-info/RECORD,,