sovai 0.2.67__tar.gz → 0.2.70__tar.gz

{sovai-0.2.67 → sovai-0.2.70}/PKG-INFO

@@ -1,8 +1,9 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: sovai
-Version: 0.2.67
+Version: 0.2.70
 Summary: python for asset management
 License: Proprietary License
+License-File: LICENSE
 Author: Derek Snow
 Author-email: d.snow@outlook.com
 Requires-Python: >=3.10
@@ -10,6 +11,7 @@ Classifier: License :: Other/Proprietary License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Provides-Extra: full
 Requires-Dist: aiobotocore (==2.21.1)
 Requires-Dist: boto3 (==1.37.1)

{sovai-0.2.67 → sovai-0.2.70}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "sovai"
-version = "0.2.67"
+version = "0.2.70"
 description = "python for asset management"
 authors = [
     {name = "Derek Snow", email = "d.snow@outlook.com"}

{sovai-0.2.67 → sovai-0.2.70}/sovai/__init__.py

@@ -5,11 +5,18 @@ from .api_config import read_key, save_key, ApiConfig
 from .basic_auth import basic_auth
 from .token_auth import token_auth
 from .utils.file_management import update_data_files # Keep update_data_files import
+from .api_config import read_key, ApiConfig
+
 
 
 import warnings
 warnings.filterwarnings("ignore")
+update_data_files()
 
+try:
+    read_key(".env")
+except FileNotFoundError:
+    pass
 
 try:
     from importlib.metadata import version

{sovai-0.2.67 → sovai-0.2.70}/sovai/get_data.py

@@ -379,15 +379,14 @@ ENDPOINT_TO_TICKER = {
    "/news/polarity_score": "",
    "/macro/features": "",
    "/congress": "",
+
    "/market/closeadj": "",
-   "/lobbying/public": "",
+   "/market/prices": "",
+
+   "/lobbying": "",
    "/liquidity/price_improvement": "",
    "/liquidity/market_opportunity": "",
-   "/trials/predict": "",
-   "/trials/describe": "",
-   "/trials/all/predict": "",
-   "/trials/all/decribe": "",
-   "/trials/all": "",
+
 
     "/patents/applications": "",
     "/patents/grants": "",
@@ -395,11 +394,13 @@ ENDPOINT_TO_TICKER = {
     "/clinical/trials": "",
 
 
-
     "/spending/awards": "",
     "/spending/compensation": "",
     "/spending/competition": "",
     "/spending/contracts": "",
+
+    # "/spending/contract": "", has changed to details.
+
     "/spending/entities": "",
     "/spending/location": "",
     "/spending/product": "",
@@ -993,6 +994,7 @@ def _get_ticker_codes_df():
         return pd.read_parquet("data/codes.parq")
     except Exception as e:
         logger.error(f"Failed to load ticker mapping data (data/codes.parq): {e}")
+
         # Consider if this should raise a more specific error or if ValueError is appropriate
         raise ValueError(f"Cannot perform ticker mapping due to data loading failure: {e}")
 

sovai-0.2.70/sovai/utils/file_management.py

@@ -0,0 +1,58 @@
+import pandas as pd
+import os
+import time
+from datetime import timedelta
+from pathlib import Path
+
+# Cache configuration
+CACHE_DIR = Path("data")
+CACHE_TTL_SECONDS = timedelta(days=7).total_seconds()
+
+def _get_file_age_seconds(file_path: Path) -> float | None:
+    """Return file age in seconds, or None if it doesn't exist."""
+    try:
+        return time.time() - file_path.stat().st_mtime
+    except (FileNotFoundError, OSError):
+        return None
+
+
+def _cached_parquet(url: str, filename: str) -> pd.DataFrame:
+    """
+    Fetch a Parquet file from a URL into a local cache directory if missing or expired,
+    then load and return it as a pandas DataFrame.
+    """
+    CACHE_DIR.mkdir(parents=True, exist_ok=True)
+    path = CACHE_DIR / filename
+    age = _get_file_age_seconds(path)
+    if age is None or age > CACHE_TTL_SECONDS:
+        print(f"Downloading {url} to {path}")
+        df = pd.read_parquet(url)
+        df.to_parquet(path)
+    else:
+        print(f"Loading from cache: {path}")
+        df = pd.read_parquet(path)
+    return df
+
+
+def get_ticker_meta() -> pd.DataFrame:
+    """Return the tickers metadata, cached for 7 days."""
+    return _cached_parquet(
+        url="https://storage.googleapis.com/sovai-public/accounting/tickers_transformed.parq",
+        filename="tickers.parq"
+    )
+
+
+def get_codes_meta() -> pd.DataFrame:
+    """Return the codes metadata, cached for 7 days."""
+    return _cached_parquet(
+        url="https://storage.googleapis.com/sovai-public/sovai-master/output/df_codes.parquet",
+        filename="codes.parq"
+    )
+
+
+def update_data_files():
+    """
+    Manually pre-warm both ticker and code caches.
+    """
+    get_ticker_meta()
+    get_codes_meta()