sovai 0.2.27__tar.gz → 0.2.28__tar.gz

{sovai-0.2.27 → sovai-0.2.28}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sovai
-Version: 0.2.27
+Version: 0.2.28
 Summary: python for asset management
 License: Proprietary License
 Author: Derek Snow
@@ -22,6 +22,7 @@ Requires-Dist: joblib (>=1.0)
 Requires-Dist: lightgbm (>=4.5.0) ; extra == "full"
 Requires-Dist: matplotlib (>=3.0) ; extra == "full"
 Requires-Dist: mfles (>=0.2) ; extra == "full"
+Requires-Dist: networkx (>=3.3,<4.0)
 Requires-Dist: numba (>=0.50) ; extra == "full"
 Requires-Dist: numpy (>=1.20)
 Requires-Dist: openai (>=1.0) ; extra == "full"
@@ -33,6 +34,7 @@ Requires-Dist: polars-talib (==0.1.3) ; extra == "full"
 Requires-Dist: pyarrow (>=5.0)
 Requires-Dist: python-dateutil (>=2.8)
 Requires-Dist: python-dotenv (>=0.10)
+Requires-Dist: pyvis (>=0.3.2,<0.4.0)
 Requires-Dist: rapidfuzz (>=3.9.7,<4.0.0)
 Requires-Dist: requests (>=2.20)
 Requires-Dist: ruptures (>=1.0) ; extra == "full"
@@ -42,6 +44,7 @@ Requires-Dist: shap (>=0.40) ; extra == "full"
 Requires-Dist: skfolio (>=0.3) ; extra == "full"
 Requires-Dist: statsforecast (>=1.0) ; extra == "full"
 Requires-Dist: tensorly (>=0.6) ; extra == "full"
+Requires-Dist: yachalk (>=0.1.6,<0.2.0)
 Description-Content-Type: text/markdown
 
 [![Sovai Python SDK package](https://github.com/sovai-research/SovAI/actions/workflows/main.yml/badge.svg)](https://github.com/sovai-research/SovAI/actions/workflows/main.yml) 

{sovai-0.2.27 → sovai-0.2.28}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "sovai"
-version = "0.2.27"
+version = "0.2.28"
 description = "python for asset management"
 authors = ["Derek Snow <d.snow@outlook.com>"]
 license = "Proprietary License"
@@ -61,6 +61,9 @@ lightgbm = {version = ">=4.5.0", optional = true}
 ipywidgets = {version = ">=8.1.3", optional = true}
 polars-talib = {version = "0.1.3", optional = true}
 dash-bootstrap-components = {version = ">=1.6.0", optional = true}
+yachalk = "^0.1.6"
+pyvis = "^0.3.2"
+networkx = "^3.3"
 
 
 [tool.poetry.extras]

{sovai-0.2.27 → sovai-0.2.28}/sovai/__init__.py

@@ -25,7 +25,7 @@ try:
     from .get_compute import compute
     from .studies.nowcasting import nowcast
     from .extensions.pandas_extensions import CustomDataFrame as extension
-    from .get_tools import sec_search, sec_filing, code
+    from .get_tools import sec_search, sec_filing, code, sec_graph
     HAS_FULL_INSTALL = True
 except ImportError:
     HAS_FULL_INSTALL = False
@@ -33,4 +33,4 @@ except ImportError:
 __all__ = ['read_key', 'save_key', 'ApiConfig', 'data', 'basic_auth', 'token_auth']
 
 if HAS_FULL_INSTALL:
-    __all__ += ['plot', 'report', 'compute', 'nowcast', 'extension', 'sec_search', 'sec_filing', 'code']
+    __all__ += ['plot', 'report', 'compute', 'nowcast', 'extension', 'sec_search', 'sec_filing', 'code', 'sec_graph']

{sovai-0.2.27 → sovai-0.2.28}/sovai/get_data.py

@@ -594,7 +594,7 @@ def data(
 
     elif (
         normalized_endpoint in client_side_endpoints_s3
-        and tickers is not None
+        # and tickers is not None
         and frequency is None
     ):
         verbose_mode.vprint("Grabbing s3 client side")

{sovai-0.2.27 → sovai-0.2.28}/sovai/get_tools.py

@@ -4,6 +4,8 @@ from .tools.sec.sec_10_k_8_k_filings import large_filing_module
 
 from .tools.sec.llm_code_generator import generate_sovai_code
 
+from .tools.sec.graphs import analyze_10k_graph
+
 
 
 def sec_search(search="CFO Resgination"):
@@ -25,3 +27,6 @@ def code(prompt="get bankruptcy data for Tesla", verbose=False, run=False):
 
 
 
+def sec_graph(ticker="AAPL", date_input="2024-Q3", verbose=False, ontology_type="causal", oai_model="gpt-4o-mini", batch=True, batch_size=10, sentiment_filter=None, output_dir="./docs", use_cache=True):
+
+    return analyze_10k_graph(ticker, date_input, section_select=True, ontology_type=ontology_type, oai_model=oai_model, batch=batch, batch_size=batch_size, sentiment_filter=sentiment_filter, output_dir=output_dir, use_cache=use_cache, verbose=verbose)