airbyte-agent-stripe 0.5.32__py3-none-any.whl → 0.5.37__py3-none-any.whl

airbyte_agent_stripe/_vendored/connector_sdk/observability/session.py

@@ -3,46 +3,40 @@
 import logging
 import uuid
 from datetime import UTC, datetime
-from pathlib import Path
 from typing import Any, Dict, Optional
 
+from .config import SDKConfig, load_config
+
 logger = logging.getLogger(__name__)
 
+# Cache the config at module level to avoid repeated reads
+_cached_config: Optional[SDKConfig] = None
+
+
+def _get_config() -> SDKConfig:
+    """Get cached SDK config or load from file."""
+    global _cached_config
+    if _cached_config is None:
+        _cached_config = load_config()
+    return _cached_config
+
+
+def _clear_config_cache() -> None:
+    """Clear the cached config. Used for testing."""
+    global _cached_config
+    _cached_config = None
+
 
 def get_persistent_user_id() -> str:
     """
-    Get or create an anonymous user ID stored in the home directory.
+    Get the persistent anonymous user ID.
 
-    The ID is stored in ~/.airbyte/ai_sdk_user_id and persists across all sessions.
-    If the file doesn't exist, a new UUID is generated and saved.
+    Now reads from ~/.airbyte/connector-sdk/config.yaml
 
     Returns:
         An anonymous UUID string that uniquely identifies this user across sessions.
     """
-    try:
-        # Create .airbyte directory in home folder if it doesn't exist
-        airbyte_dir = Path.home() / ".airbyte"
-        airbyte_dir.mkdir(exist_ok=True)
-
-        # Path to user ID file
-        user_id_file = airbyte_dir / "ai_sdk_user_id"
-
-        # Try to read existing user ID
-        if user_id_file.exists():
-            user_id = user_id_file.read_text().strip()
-            if user_id:  # Validate it's not empty
-                return user_id
-
-        # Generate new user ID if file doesn't exist or is empty
-        user_id = str(uuid.uuid4())
-        user_id_file.write_text(user_id)
-        logger.debug(f"Generated new anonymous user ID: {user_id}")
-
-        return user_id
-    except Exception as e:
-        # If we can't read/write the file, generate a session-only ID
-        logger.debug(f"Could not access anonymous user ID file: {e}")
-        return str(uuid.uuid4())
+    return _get_config().user_id
 
 
 def get_public_ip() -> Optional[str]:
@@ -65,6 +59,18 @@ def get_public_ip() -> Optional[str]:
         return None
 
 
+def get_is_internal_user() -> bool:
+    """
+    Check if the current user is an internal Airbyte user.
+
+    Now reads from ~/.airbyte/connector-sdk/config.yaml
+    Environment variable AIRBYTE_INTERNAL_USER can override.
+
+    Returns False if not set or on any error.
+    """
+    return _get_config().is_internal_user
+
+
 class ObservabilitySession:
     """Shared session context for both logging and telemetry."""
 
@@ -84,6 +90,7 @@ class ObservabilitySession:
         self.operation_count = 0
         self.metadata: Dict[str, Any] = {}
         self.public_ip = get_public_ip()
+        self.is_internal_user = get_is_internal_user()
 
     def increment_operations(self):
         """Increment the operation counter."""

airbyte_agent_stripe/_vendored/connector_sdk/schema/operations.py

@@ -61,7 +61,7 @@ class Operation(BaseModel):
         description=(
             "JSONPath expression to extract records from API response envelopes. "
             "When specified, executor extracts data at this path instead of returning "
-            "full response. Returns array for list/search actions, single record for "
+            "full response. Returns array for list/api_search actions, single record for "
             "get/create/update/delete actions."
         ),
     )

airbyte_agent_stripe/_vendored/connector_sdk/telemetry/events.py

@@ -1,6 +1,6 @@
 """Telemetry event models."""
 
-from dataclasses import asdict, dataclass
+from dataclasses import asdict, dataclass, field
 from datetime import datetime
 from typing import Any, Dict, Optional
 
@@ -13,6 +13,7 @@ class BaseEvent:
     session_id: str
     user_id: str
     execution_context: str
+    is_internal_user: bool = field(default=False, kw_only=True)
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert event to dictionary with ISO formatted timestamp."""

airbyte_agent_stripe/_vendored/connector_sdk/telemetry/tracker.py

@@ -59,6 +59,7 @@ class SegmentTracker:
                 session_id=self.session.session_id,
                 user_id=self.session.user_id,
                 execution_context=self.session.execution_context,
+                is_internal_user=self.session.is_internal_user,
                 public_ip=self.session.public_ip,
                 connector_name=self.session.connector_name,
                 connector_version=connector_version,
@@ -101,6 +102,7 @@ class SegmentTracker:
                 session_id=self.session.session_id,
                 user_id=self.session.user_id,
                 execution_context=self.session.execution_context,
+                is_internal_user=self.session.is_internal_user,
                 public_ip=self.session.public_ip,
                 connector_name=self.session.connector_name,
                 entity=entity,
@@ -130,6 +132,7 @@ class SegmentTracker:
                 session_id=self.session.session_id,
                 user_id=self.session.user_id,
                 execution_context=self.session.execution_context,
+                is_internal_user=self.session.is_internal_user,
                 public_ip=self.session.public_ip,
                 connector_name=self.session.connector_name,
                 duration_seconds=self.session.duration_seconds(),

airbyte_agent_stripe/_vendored/connector_sdk/types.py

@@ -22,7 +22,7 @@ class Action(str, Enum):
     UPDATE = "update"
     DELETE = "delete"
     LIST = "list"
-    SEARCH = "search"
+    API_SEARCH = "api_search"
     DOWNLOAD = "download"
     AUTHORIZE = "authorize"