moose-lib 0.4.231__py3-none-any.whl → 0.4.233__py3-none-any.whl

moose_lib/dmv2/__init__.py

@@ -44,6 +44,8 @@ from .ingest_pipeline import (
 from .consumption import (
     EgressConfig,
     ConsumptionApi,
+    get_moose_base_url,
+    set_moose_base_url,
 )
 
 from .sql_resource import (
@@ -115,6 +117,8 @@ __all__ = [
     # Consumption
     'EgressConfig',
     'ConsumptionApi',
+    'get_moose_base_url',
+    'set_moose_base_url',
 
     # SQL
     'SqlResource',
@@ -141,4 +145,4 @@ __all__ = [
     'get_sql_resource',
     'get_workflows',
     'get_workflow',
-]
+]

moose_lib/dmv2/consumption.py

@@ -4,13 +4,42 @@ Consumption (Egress) API definitions for Moose Data Model v2 (dmv2).
 This module provides classes for defining and configuring consumption APIs
 that allow querying data through user-defined functions.
 """
-from typing import Any, Callable, Optional, Tuple
+import os
+from typing import Any, Callable, Optional, Tuple, Generic
+
+import requests
 from pydantic import BaseModel
 from pydantic.json_schema import JsonSchemaValue
 
 from .types import BaseTypedResource, T, U
 from ._registry import _egress_apis
 
+# Global base URL configuration
+_global_base_url: Optional[str] = None
+
+
+def set_moose_base_url(url: str) -> None:
+    """Set the global base URL for consumption API calls.
+    
+    Args:
+        url: The base URL to use for API calls
+    """
+    global _global_base_url
+    _global_base_url = url
+
+
+def get_moose_base_url() -> Optional[str]:
+    """Get the configured base URL from global setting or environment variable.
+    
+    Returns:
+        The base URL if configured, None otherwise
+    """
+    # Priority: programmatically set > environment variable
+    if _global_base_url:
+        return _global_base_url
+    return os.getenv('MOOSE_BASE_URL')
+
+
 class EgressConfig(BaseModel):
     """Configuration for Consumption (Egress) APIs.
 
@@ -21,7 +50,8 @@ class EgressConfig(BaseModel):
     version: Optional[str] = None
     metadata: Optional[dict] = None
 
-class ConsumptionApi(BaseTypedResource):
+
+class ConsumptionApi(BaseTypedResource, Generic[U]):
     """Represents a Consumption (Egress) API endpoint.
 
     Allows querying data, typically powered by a user-defined function.
@@ -98,4 +128,50 @@ class ConsumptionApi(BaseTypedResource):
         from pydantic.type_adapter import TypeAdapter
         return TypeAdapter(self.return_type).json_schema(
             ref_template='#/components/schemas/{model}'
-        )
+        )
+
+    def call(self, params: T, base_url: Optional[str] = None) -> U:
+        """Call the consumption API with the given parameters.
+        
+        Args:
+            params: Parameters matching the input model T
+            base_url: Optional base URL override. If not provided, uses the global
+                     base URL set via set_base_url() or MOOSE_BASE_URL environment variable.
+            
+        Returns:
+            Response data matching the output model U
+            
+        Raises:
+            ValueError: If no base URL is configured
+        """
+        # Determine which base URL to use
+        effective_base_url = base_url or get_moose_base_url()
+        if not effective_base_url:
+            raise ValueError(
+                "No base URL configured. Set it via set_base_url(), "
+                "MOOSE_BASE_URL environment variable, or pass base_url parameter."
+            )
+
+        # Construct the API endpoint URL
+        url = f"{effective_base_url.rstrip('/')}/consumption/{self.name}"
+
+        # Convert Pydantic model to dictionary
+        params_dict = params.model_dump()
+
+        # Build query parameters, handling lists as repeated params
+        query_params = []
+        for key, value in params_dict.items():
+            if isinstance(value, list):
+                # For list values, add each item as a separate query param
+                for item in value:
+                    query_params.append((key, str(item)))
+            elif value is not None:
+                query_params.append((key, str(value)))
+
+        # Make the HTTP request
+        response = requests.get(url, params=query_params)
+        response.raise_for_status()  # Raise an exception for bad status codes
+
+        # Parse JSON response and return as the expected type
+        response_data = response.json()
+        return self._u.model_validate(response_data)

{moose_lib-0.4.231.dist-info → moose_lib-0.4.233.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: moose_lib
-Version: 0.4.231
+Version: 0.4.233
 Home-page: https://www.fiveonefour.com/moose
 Author: Fiveonefour Labs Inc.
 Author-email: support@fiveonefour.com

{moose_lib-0.4.231.dist-info → moose_lib-0.4.233.dist-info}/RECORD

@@ -12,9 +12,9 @@ moose_lib/clients/redis_client.py,sha256=UBCdxwgZpIOIOy2EnPyxJIAYjw_qmNwGsJQCQ66
 moose_lib/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 moose_lib/config/config_file.py,sha256=NyjY6YFraBel7vBk18lLkpGaPR1viKMAEv4ZldyfLIA,2585
 moose_lib/config/runtime.py,sha256=gF1wrUCbp0MxRU92k0vyItvWX2x6P1FuBRzM61nJE_U,2776
-moose_lib/dmv2/__init__.py,sha256=zyGK2YDN4zQ2SMn9-SAiOuhpU46STePuJjK18LEserg,2269
+moose_lib/dmv2/__init__.py,sha256=Po8jzSI-lc9LTSKo_F3rURRcDdraH-vzhkuEdpzakDY,2370
 moose_lib/dmv2/_registry.py,sha256=agdZ7xzS99Caou60Q2pEErzEwyNYHqwy6XqV79eEmwg,504
-moose_lib/dmv2/consumption.py,sha256=71wdv6ZuEi8Om7aX3Lq-d6bAoc1-iw3Wudb8dHESJKI,4072
+moose_lib/dmv2/consumption.py,sha256=vt1APul1O21yPHaB0MRjF_9ohz5MJs5QIpDCnzFwyJA,6656
 moose_lib/dmv2/ingest_api.py,sha256=Snek9NGwaJl_BuImSWGtQq91m9D3AJ4qBoGiKZ-9yTQ,2323
 moose_lib/dmv2/ingest_pipeline.py,sha256=Y1gsvHZjlW07gMapLnBRJEsoAPv7ThvLABoLmVV7BHE,6714
 moose_lib/dmv2/materialized_view.py,sha256=kcx-sJFTM-cH3Uc1GoldgFGodjoz0AegAQEMmohdS38,3826
@@ -31,7 +31,7 @@ tests/__init__.py,sha256=0Gh4yzPkkC3TzBGKhenpMIxJcRhyrrCfxLSfpTZnPMQ,53
 tests/conftest.py,sha256=ZVJNbnr4DwbcqkTmePW6U01zAzE6QD0kNAEZjPG1f4s,169
 tests/test_moose.py,sha256=mBsx_OYWmL8ppDzL_7Bd7xR6qf_i3-pCIO3wm2iQNaA,2136
 tests/test_redis_client.py,sha256=d9_MLYsJ4ecVil_jPB2gW3Q5aWnavxmmjZg2uYI3LVo,3256
-moose_lib-0.4.231.dist-info/METADATA,sha256=tmXsMESoMHk60NCcb7j84Xb9px2-tYmpA8I2sSsVSlI,729
-moose_lib-0.4.231.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-moose_lib-0.4.231.dist-info/top_level.txt,sha256=XEns2-4aCmGp2XjJAeEH9TAUcGONLnSLy6ycT9FSJh8,16
-moose_lib-0.4.231.dist-info/RECORD,,
+moose_lib-0.4.233.dist-info/METADATA,sha256=s4ePkxrMErxLqiD7vUPgBoTaPLaSs0GCaMKaGZN7h5w,729
+moose_lib-0.4.233.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+moose_lib-0.4.233.dist-info/top_level.txt,sha256=XEns2-4aCmGp2XjJAeEH9TAUcGONLnSLy6ycT9FSJh8,16
+moose_lib-0.4.233.dist-info/RECORD,,