feat(monke): clean up monke and make ready for use

Author: orhanrauf

backend/airweave/billing/service.py

@@ -24,7 +24,6 @@
     is_paid_plan,
 )
 from airweave.billing.transactions import billing_transactions
-from airweave.core.config import settings
 from airweave.core.exceptions import InvalidStateError, NotFoundException
 from airweave.core.logging import ContextualLogger, logger
 from airweave.db.unit_of_work import UnitOfWork
@@ -898,4 +897,4 @@ async def get_subscription_info(
 
 
 # Singleton instance
-billing_service = BillingService() if settings.STRIPE_ENABLED else None
+billing_service = BillingService()

backend/airweave/platform/sources/onedrive.py

@@ -330,7 +330,8 @@ def _build_file_entity(
         )
 
         # Add additional properties for file processing
-        entity.total_size = item.get("size", 0)
+        if entity.airweave_system_metadata:
+            entity.airweave_system_metadata.total_size = item.get("size", 0)
 
         return entity
 

monke/auth/broker.py

@@ -49,18 +49,14 @@ def __init__(
         account_id: Optional[str] = None,
     ) -> None:
         self.logger = get_logger("composio_broker")
-        self.api_key = api_key or os.getenv("DM_AUTH_PROVIDER_API_KEY")
-        self.auth_config_id = auth_config_id or os.getenv(
-            "DM_AUTH_PROVIDER_AUTH_CONFIG_ID"
-        )
-        self.account_id = account_id or os.getenv("DM_AUTH_PROVIDER_ACCOUNT_ID")
+        self.api_key = api_key or os.getenv("COMPOSIO_API_KEY")
+        self.auth_config_id = auth_config_id
+        self.account_id = account_id
 
         if not self.api_key:
-            raise ValueError("Missing Composio api key (DM_AUTH_PROVIDER_API_KEY)")
+            raise ValueError("Missing Composio API key (COMPOSIO_API_KEY)")
 
-    async def _get(
-        self, path: str, params: Optional[Dict[str, Any]] = None
-    ) -> Dict[str, Any]:
+    async def _get(self, path: str, params: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
         async with httpx.AsyncClient() as client:
             r = await client.get(
                 f"{self.BASE_URL}{path}",
@@ -112,10 +108,7 @@ async def get_credentials(
         sensitive_fields = [
             k
             for k in creds.keys()
-            if any(
-                sensitive in k.lower()
-                for sensitive in ["token", "key", "secret", "password"]
-            )
+            if any(sensitive in k.lower() for sensitive in ["token", "key", "secret", "password"])
         ]
         non_sensitive_fields = [k for k in creds.keys() if k not in sensitive_fields]
 

monke/auth/credentials_resolver.py

@@ -1,8 +1,8 @@
 """Centralized credential resolution for connectors.
 
 Priority:
-1) Explicit auth_fields provided in test config
-2) Auth broker (e.g., Composio) if configured
+1) Explicit auth_fields provided in test config (direct auth)
+2) Composio auth broker if API key is configured
 """
 
 from __future__ import annotations
@@ -13,22 +13,15 @@
 from monke.auth.broker import BaseAuthBroker, ComposioBroker
 
 
-def _make_broker(source_short_name: str) -> Optional[BaseAuthBroker]:
-    provider = os.getenv("DM_AUTH_PROVIDER")  # e.g., "composio"
-    if not provider:
+def _make_broker() -> Optional[BaseAuthBroker]:
+    """Create a Composio broker if API key is available."""
+    # Check if Composio API key exists
+    if not os.getenv("COMPOSIO_API_KEY"):
         return None
-    if provider == "composio":
-        # Check for source-specific account/config IDs
-        source_upper = source_short_name.upper()
-        account_id = os.getenv(f"{source_upper}_AUTH_PROVIDER_ACCOUNT_ID")
-        auth_config_id = os.getenv(f"{source_upper}_AUTH_PROVIDER_AUTH_CONFIG_ID")
 
-        # If source-specific IDs exist, use them
-        if account_id and auth_config_id:
-            return ComposioBroker(account_id=account_id, auth_config_id=auth_config_id)
-        # Otherwise use default broker
-        return ComposioBroker()
-    raise ValueError(f"Unsupported auth provider: {provider}")
+    # Return a broker without specific account/auth config
+    # These will be provided from the YAML config when needed
+    return ComposioBroker()
 
 
 async def resolve_credentials(
@@ -46,11 +39,11 @@ async def resolve_credentials(
     if provided_auth_fields:
         return provided_auth_fields
 
-    broker = _make_broker(connector_short_name)
+    broker = _make_broker()
 
     if broker:
         return await broker.get_credentials(connector_short_name)
 
     raise ValueError(
-        f"No credentials provided and no DM_AUTH_PROVIDER configured for {connector_short_name}"
+        f"No credentials provided and no Composio API key configured for {connector_short_name}"
     )