sovai 0.2.6__tar.gz → 0.2.8__tar.gz

{sovai-0.2.6 → sovai-0.2.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sovai
-Version: 0.2.6
+Version: 0.2.8
 Summary: python for asset management
 License: Proprietary License
 Author: Derek Snow
@@ -11,35 +11,35 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Provides-Extra: data
-Requires-Dist: boto3 (>=1.20)
-Requires-Dist: dash (>=2.0)
-Requires-Dist: dash-bootstrap-components (>=1.6.0,<2.0.0)
-Requires-Dist: great-tables (>=0.9)
-Requires-Dist: ipywidgets (>=8.1.3,<9.0.0)
-Requires-Dist: joblib (>=1.0)
-Requires-Dist: lightgbm (>=4.5.0,<5.0.0)
-Requires-Dist: matplotlib (>=3.0)
-Requires-Dist: mfles (>=0.2)
-Requires-Dist: numba (>=0.50)
-Requires-Dist: numpy (>=1.20) ; extra == "data"
-Requires-Dist: openai (>=1.0)
-Requires-Dist: pandas (>=1.0) ; extra == "data"
-Requires-Dist: pexpect (>=4.9.0,<5.0.0)
-Requires-Dist: plotly (>=5.0)
-Requires-Dist: polars (>=0.20.30) ; extra == "data"
-Requires-Dist: polars-talib (==0.1.3)
-Requires-Dist: pyarrow (>=5.0) ; extra == "data"
-Requires-Dist: python-dateutil (>=2.8) ; extra == "data"
-Requires-Dist: python-dotenv (>=0.10) ; extra == "data"
-Requires-Dist: requests (>=2.20) ; extra == "data"
-Requires-Dist: ruptures (>=1.0)
-Requires-Dist: scikit-learn (>=1.0)
-Requires-Dist: scipy (>=1.0)
-Requires-Dist: shap (>=0.40)
-Requires-Dist: skfolio (>=0.3,<0.4)
-Requires-Dist: statsforecast (>=1.0)
-Requires-Dist: tensorly (>=0.6)
+Provides-Extra: full
+Requires-Dist: boto3 (>=1.20) ; extra == "full"
+Requires-Dist: dash (>=2.0) ; extra == "full"
+Requires-Dist: dash-bootstrap-components (>=1.6.0) ; extra == "full"
+Requires-Dist: great-tables (>=0.9) ; extra == "full"
+Requires-Dist: ipywidgets (>=8.1.3) ; extra == "full"
+Requires-Dist: joblib (>=1.0) ; extra == "full"
+Requires-Dist: lightgbm (>=4.5.0) ; extra == "full"
+Requires-Dist: matplotlib (>=3.0) ; extra == "full"
+Requires-Dist: mfles (>=0.2) ; extra == "full"
+Requires-Dist: numba (>=0.50) ; extra == "full"
+Requires-Dist: numpy (>=1.20)
+Requires-Dist: openai (>=1.0) ; extra == "full"
+Requires-Dist: pandas (>=1.0)
+Requires-Dist: pexpect (>=4.9.0) ; extra == "full"
+Requires-Dist: plotly (>=5.0) ; extra == "full"
+Requires-Dist: polars (>=0.20.30)
+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: requests (>=2.20)
+Requires-Dist: ruptures (>=1.0) ; extra == "full"
+Requires-Dist: scikit-learn (>=1.0) ; extra == "full"
+Requires-Dist: scipy (>=1.0) ; extra == "full"
+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"
 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.8/pyproject.toml

@@ -0,0 +1,90 @@
+[tool.poetry]
+name = "sovai"
+version = "0.2.8"
+description = "python for asset management"
+authors = ["Derek Snow <d.snow@outlook.com>"]
+license = "Proprietary License"
+readme = "README.md"
+include = ["sovai/**/*", "data/**/*"]
+exclude = [
+ "notebooks",
+ "notebooks/**/*",
+ "tests",
+ "tests/**/*",
+ "docs",
+ "docs/**/*",
+ ".github",
+ ".github/**/*",
+ ".venv",
+ ".venv/**/*",
+ "dist",
+ "dist/**/*",
+ "*.pyc",
+ "**/*.pyc",
+ "*.pyo",
+ "**/*.pyo",
+ "__pycache__",
+ "**/__pycache__",
+ ".ipynb_checkpoints",
+ "**/.ipynb_checkpoints",
+]
+
+[tool.poetry.dependencies]
+python = "^3.10"
+pandas = ">=1.0"
+numpy = ">=1.20"
+pyarrow = ">=5.0"
+requests = ">=2.20"
+python-dateutil = ">=2.8"
+python-dotenv = ">=0.10"
+polars = ">=0.20.30"
+scipy = {version = ">=1.0", optional = true}
+joblib = {version = ">=1.0", optional = true}
+matplotlib = {version = ">=3.0", optional = true}
+plotly = {version = ">=5.0", optional = true}
+scikit-learn = {version = ">=1.0", optional = true}
+numba = {version = ">=0.50", optional = true}
+boto3 = {version = ">=1.20", optional = true}
+dash = {version = ">=2.0", optional = true}
+great-tables = {version = ">=0.9", optional = true}
+ruptures = {version = ">=1.0", optional = true}
+shap = {version = ">=0.40", optional = true}
+skfolio = {version = ">=0.3", optional = true}
+statsforecast = {version = ">=1.0", optional = true}
+tensorly = {version = ">=0.6", optional = true}
+openai = {version = ">=1.0", optional = true}
+mfles = {version = ">=0.2", optional = true}
+pexpect = {version = ">=4.9.0", optional = true}
+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}
+
+[tool.poetry.extras]
+full = [
+ "scipy",
+ "joblib",
+ "matplotlib",
+ "plotly",
+ "scikit-learn",
+ "numba",
+ "boto3",
+ "dash",
+ "great-tables",
+ "ruptures",
+ "shap",
+ "skfolio",
+ "statsforecast",
+ "tensorly",
+ "openai",
+ "mfles",
+ "pexpect",
+ "lightgbm",
+ "ipywidgets",
+ "polars-talib",
+ "dash-bootstrap-components"
+]
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

sovai-0.2.8/sovai/__init__.py

@@ -0,0 +1,29 @@
+"""
+Main SovAI SDK Tool Kit package
+"""
+from .api_config import read_key, save_key, ApiConfig
+from .get_data import data
+from .basic_auth import basic_auth
+from .token_auth import token_auth
+from .utils.file_management import save_or_update_tickers
+
+__version__ = "0.1.25"
+
+# Call the function to ensure tickers data is up-to-date
+save_or_update_tickers()
+
+try:
+    from .get_plots import plot
+    from .get_reports import report
+    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
+    HAS_FULL_INSTALL = True
+except ImportError:
+    HAS_FULL_INSTALL = False
+
+__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']

{sovai-0.2.6 → sovai-0.2.8}/sovai/get_data.py

@@ -2,32 +2,35 @@ from typing import Optional, Union, Tuple, List, Dict
 import re
 from datetime import datetime
 import pandas as pd
-import requests  # type: ignore
+import requests
+import json
+import hashlib
+import numpy as np
+import pyarrow.parquet as pq
+from io import BytesIO
+import boto3
+import polars as pl
+
+# from fastapi import HTTPException ## Dee
+
+import pyarrow.parquet as pq
 
 from sovai.api_config import ApiConfig
 from sovai.errors.sovai_errors import InvalidInputData
 from sovai.utils.converter import convert_data2df
 from sovai.utils.stream import stream_data, stream_data_pyarrow
 from sovai.utils.datetime_formats import datetime_format
-from sovai.utils.plot import plotting_data
 from sovai.utils.client_side import client_side_frame
 
-from sovai.extensions.pandas_extensions import CustomDataFrame
+try:
+    from sovai.extensions.pandas_extensions import CustomDataFrame
+    from sovai.utils.plot import plotting_data
+    HAS_FULL_INSTALL = True
+except ImportError:
+    HAS_FULL_INSTALL = False
 
-import pandas as pd
-import polars as pl
-
-import boto3
-from io import BytesIO
-import json
-
-# from fastapi import HTTPException ## Dee
-
-import pyarrow.parquet as pq
-import time
-import hashlib
-import json
-import numpy as np
+def is_full_installation():
+    return HAS_FULL_INSTALL
 
 # client_side_frame is the one for very quick public files.
 
@@ -40,35 +43,23 @@ import numpy as np
 # Wall time: 50s (direct via gcp)
 
 
-## This is not in use
-def load_df_from_wasabi(bucket_name, file_name, access_key, secret_key):
-    """
-    Load a Parquet file from a Wasabi bucket into a pandas DataFrame.
-
-    Args:
-    bucket_name (str): Name of the Wasabi bucket.
-    file_name (str): Name of the file in the bucket.
-    access_key (str): Wasabi access key.
-    secret_key (str): Wasabi secret key.
-
-    Returns:
-    pandas.DataFrame: DataFrame loaded from the Parquet file.
-    """
+# Global cache
+_query_cache = {}
 
-    # Set up Boto3 client
+def load_df_from_wasabi(bucket_name, file_name, access_key, secret_key):
+    import boto3
     s3_client = boto3.client(
         "s3",
         endpoint_url="https://s3.wasabisys.com",
         aws_access_key_id=access_key,
         aws_secret_access_key=secret_key,
     )
-
-    # Download the file
     parquet_buffer = BytesIO()
     s3_client.download_fileobj(bucket_name, file_name, parquet_buffer)
     parquet_buffer.seek(0)
+    df = pq.read_table(source=parquet_buffer).to_pandas()
+    return CustomDataFrame(df) if HAS_FULL_INSTALL else df
 
-    return pq.read_table(source=parquet_buffer).to_pandas()
 
 
 def is_all(tickers):
@@ -92,20 +83,12 @@ import polars as pl
 from typing import Union
 
 
-def read_parquet(
-    url: str, use_polars: bool = False
-) -> Union[pd.DataFrame, pl.DataFrame]:
-    """
-    Reads a Parquet file into a DataFrame using Pandas or Polars.
-
-    :param url: URL or file path of the Parquet file.
-    :param use_polars: If True, use Polars to read the file; otherwise, use Pandas.
-    :return: DataFrame loaded from the Parquet file.
-    """
+def read_parquet(url: str, use_polars: bool = False) -> Union[pd.DataFrame, pl.DataFrame]:
     if use_polars:
         return pl.read_parquet(url)
     else:
-        return CustomDataFrame(pd.read_parquet(url))
+        df = pd.read_parquet(url)
+        return CustomDataFrame(df) if HAS_FULL_INSTALL else df
     
 
 import pandas as pd
@@ -381,11 +364,10 @@ def data(
     frequency: str = None,
     verbose: bool = False,
     # **kwargs,                  ## kwargs if you want to allow for random
-) -> pd.DataFrame:
+) -> Union[pd.DataFrame, 'CustomDataFrame']:
     verbose_mode.toggle_verbose(verbose)
 
     params = params or {}
-    # params.update(kwargs)
     params = map_synonyms(params)
 
     params.update(
@@ -472,13 +454,18 @@ def data(
         if content_type == "application/octet-stream":
             if data_format == "pyarrow":
                 verbose_mode.vprint(f"header: {data_format}")
-                data = CustomDataFrame(stream_data_pyarrow(res))
+                data = stream_data_pyarrow(res)
             else:
                 verbose_mode.vprint(f"header: not pyarrow")
-                data = CustomDataFrame(stream_data(res))
+                data = stream_data(res)
+            
+            if HAS_FULL_INSTALL:
+                data = CustomDataFrame(data)
+            else:
+                data = pd.DataFrame(data)
+            
             _query_cache[cache_key] = data
             return data
-
  
         # Example usage in your existing code:
         if is_all(tickers):
@@ -492,15 +479,20 @@ def data(
 
         verbose_mode.vprint("It reached the DF")
         try:
-            data = CustomDataFrame(convert_data2df(res.json()))
-            _query_cache[cache_key] = data
-        except:
-            _query_cache[cache_key] = data
+            from sovai.extensions.pandas_extensions import CustomDataFrame
+            data = CustomDataFrame(data)
+        except ImportError:
+            data = pd.DataFrame(data)
+
+        _query_cache[cache_key] = data
 
         verbose_mode.vprint("It passed the DF")
 
         if plot:
-            _draw_graphs(data)
+            if HAS_FULL_INSTALL:
+                _draw_graphs(data)
+            else:
+                print("Plotting is only available with the full installation. Please install 'sovai[full]' to use this feature.")
             return None
         return data
     except Exception as err:

sovai-0.2.6/pyproject.toml

@@ -1,76 +0,0 @@
-[tool.poetry]
-name = "sovai"
-version = "0.2.6"
-description = "python for asset management"
-authors = ["Derek Snow <d.snow@outlook.com>"]
-license = "Proprietary License"
-readme = "README.md"
-include = ["sovai/**/*", "data/**/*"]
-exclude = [
- "notebooks",
- "notebooks/**/*",
- "tests",
- "tests/**/*",
- "docs",
- "docs/**/*",
- ".github",
- ".github/**/*",
- ".venv",
- ".venv/**/*",
- "dist",
- "dist/**/*",
- "*.pyc",
- "**/*.pyc",
- "*.pyo",
- "**/*.pyo",
- "__pycache__",
- "**/__pycache__",
- ".ipynb_checkpoints",
- "**/.ipynb_checkpoints",
-]
-
-[tool.poetry.dependencies]
-python = "^3.10"
-numpy = ">=1.20"
-scipy = ">=1.0"
-pandas = ">=1.0"
-python-dateutil = ">=2.8"
-python-dotenv = ">=0.10"
-requests = ">=2.20"
-joblib = ">=1.0"
-pyarrow = ">=5.0"
-matplotlib = ">=3.0"
-plotly = ">=5.0"
-scikit-learn = ">=1.0"
-numba = ">=0.50"
-boto3 = ">=1.20"
-dash = ">=2.0"
-great-tables = ">=0.9"
-polars = ">=0.20.30"
-ruptures = ">=1.0"
-shap = ">=0.40"
-skfolio = {version = "^0.3", extras = []}
-statsforecast = ">=1.0"
-tensorly = ">=0.6"
-openai = ">=1.0"
-mfles = ">=0.2"
-pexpect = "^4.9.0"
-lightgbm = "^4.5.0"
-ipywidgets = "^8.1.3"
-polars-talib = "0.1.3"
-dash-bootstrap-components = "^1.6.0"
-
-[tool.poetry.extras]
-data = [
-    "pandas",
-    "numpy",
-    "pyarrow",
-    "requests",
-    "python-dateutil",
-    "python-dotenv",
-    "polars",
-]
-
-[build-system]
-requires = ["poetry-core"]
-build-backend = "poetry.core.masonry.api"

sovai-0.2.6/sovai/__init__.py

@@ -1,36 +0,0 @@
-"""
-Main SovAI SDK Tool Kit package
-"""
-
-from .api_config import read_key, save_key
-from .api_config import ApiConfig
-from .get_data import data
-from .get_plots import plot
-from .get_reports import report
-from .get_compute import compute
-
-from .basic_auth import basic_auth
-from .token_auth import token_auth
-from .studies.nowcasting import nowcast
-
-from .extensions.pandas_extensions import CustomDataFrame as extension
-
-
-from .utils.file_management import save_or_update_tickers
-
-
-
-from .get_tools import sec_search
-
-from .get_tools import sec_filing
-
-
-
-
-
-
-# Call the function to ensure tickers data is up-to-date
-save_or_update_tickers()
-
-
-__version__ = "0.1.25"