google-adk-extras 0.2.6__py3-none-any.whl → 0.3.0__py3-none-any.whl

google_adk_extras/credentials/x_oauth2_credential_service.py

@@ -1,240 +0,0 @@
-"""X (Twitter) OAuth2 credential service implementation."""
-
-from typing import Optional, List
-import logging
-
-from google.adk.auth.credential_service.session_state_credential_service import SessionStateCredentialService
-from google.adk.auth.credential_service.base_credential_service import CallbackContext
-from google.adk.auth import AuthConfig, AuthCredential, AuthCredentialTypes
-from google.adk.auth.auth_credential import OAuth2Auth
-from fastapi.openapi.models import OAuth2
-from fastapi.openapi.models import OAuthFlowAuthorizationCode, OAuthFlows
-
-from .base_custom_credential_service import BaseCustomCredentialService
-
-logger = logging.getLogger(__name__)
-
-
-class XOAuth2CredentialService(BaseCustomCredentialService):
-    """X (Twitter) OAuth2 credential service for handling X API authentication flows.
-    
-    This service provides pre-configured OAuth2 flows for X API v2 including
-    reading tweets, posting content, and managing user data.
-    
-    Args:
-        client_id: The X OAuth2 client ID from X Developer Portal.
-        client_secret: The X OAuth2 client secret from X Developer Portal.
-        scopes: List of OAuth2 scopes to request. Common scopes include:
-            - "tweet.read" - Read tweets
-            - "tweet.write" - Write tweets
-            - "tweet.moderate.write" - Moderate tweets
-            - "users.read" - Read user information
-            - "follows.read" - Read follows information
-            - "follows.write" - Manage follows
-            - "offline.access" - Maintain access when user is offline
-            - "space.read" - Read Spaces information
-            - "mute.read" - Read muted accounts
-            - "mute.write" - Manage muted accounts
-            - "like.read" - Read likes information
-            - "like.write" - Manage likes
-            - "list.read" - Read list information
-            - "list.write" - Manage lists
-            - "block.read" - Read blocked accounts
-            - "block.write" - Manage blocked accounts
-        use_session_state: If True, stores credentials in session state. If False,
-            uses in-memory storage. Default is True for persistence.
-    
-    Example:
-        ```python
-        credential_service = XOAuth2CredentialService(
-            client_id="your-x-client-id",
-            client_secret="your-x-client-secret",
-            scopes=["tweet.read", "tweet.write", "users.read", "offline.access"]
-        )
-        await credential_service.initialize()
-        
-        # Use with Runner
-        runner = Runner(
-            agent=agent,
-            session_service=session_service,
-            credential_service=credential_service,
-            app_name="my_app"
-        )
-        ```
-    """
-
-    # X OAuth2 endpoints
-    X_AUTH_URL = "https://twitter.com/i/oauth2/authorize"
-    X_TOKEN_URL = "https://api.twitter.com/2/oauth2/token"
-    
-    # Common X OAuth2 scopes
-    COMMON_SCOPES = {
-        # Tweet scopes
-        "tweet.read": "Read tweets",
-        "tweet.write": "Write tweets",
-        "tweet.moderate.write": "Hide and unhide replies to your tweets",
-        
-        # User scopes
-        "users.read": "Read user profile information",
-        
-        # Follows scopes
-        "follows.read": "Read who a user is following or who is following a user",
-        "follows.write": "Follow and unfollow other users",
-        
-        # Offline access
-        "offline.access": "Maintain access to accounts when users are offline",
-        
-        # Space scopes
-        "space.read": "Read Spaces information",
-        
-        # Mute scopes
-        "mute.read": "Read muted accounts",
-        "mute.write": "Mute and unmute accounts",
-        
-        # Like scopes
-        "like.read": "Read liked tweets",
-        "like.write": "Like and unlike tweets",
-        
-        # List scopes
-        "list.read": "Read list information",
-        "list.write": "Create and manage lists",
-        
-        # Block scopes
-        "block.read": "Read blocked accounts",
-        "block.write": "Block and unblock accounts",
-        
-        # Bookmark scopes
-        "bookmark.read": "Read bookmarked tweets",
-        "bookmark.write": "Bookmark and unbookmark tweets",
-        
-        # Direct message scopes
-        "dm.read": "Read direct messages",
-        "dm.write": "Send direct messages",
-    }
-
-    def __init__(
-        self,
-        client_id: str,
-        client_secret: str,
-        scopes: Optional[List[str]] = None,
-        use_session_state: bool = True
-    ):
-        """Initialize the X OAuth2 credential service.
-        
-        Args:
-            client_id: X OAuth2 client ID.
-            client_secret: X OAuth2 client secret.
-            scopes: List of OAuth2 scopes to request.
-            use_session_state: Whether to use session state for credential storage.
-        """
-        super().__init__()
-        self.client_id = client_id
-        self.client_secret = client_secret
-        self.scopes = scopes or ["tweet.read", "users.read", "offline.access"]
-        self.use_session_state = use_session_state
-        
-        # Underlying credential service for storage
-        if use_session_state:
-            self._storage_service = SessionStateCredentialService()
-        else:
-            from google.adk.auth.credential_service.in_memory_credential_service import InMemoryCredentialService
-            self._storage_service = InMemoryCredentialService()
-
-    async def _initialize_impl(self) -> None:
-        """Initialize the X OAuth2 credential service.
-        
-        Validates the client credentials and sets up the OAuth2 auth scheme.
-        
-        Raises:
-            ValueError: If client_id or client_secret is missing.
-        """
-        if not self.client_id:
-            raise ValueError("X OAuth2 client_id is required")
-        if not self.client_secret:
-            raise ValueError("X OAuth2 client_secret is required")
-        if not self.scopes:
-            raise ValueError("At least one OAuth2 scope is required")
-            
-        # Validate scopes against known X scopes
-        unknown_scopes = set(self.scopes) - set(self.COMMON_SCOPES.keys())
-        if unknown_scopes:
-            logger.warning(f"Unknown X OAuth2 scopes: {unknown_scopes}")
-            
-        logger.info(f"Initialized X OAuth2 credential service with scopes: {self.scopes}")
-
-    def create_auth_config(self) -> AuthConfig:
-        """Create an AuthConfig for X OAuth2 authentication.
-        
-        Returns:
-            AuthConfig: Configured auth config for X OAuth2 flow.
-        """
-        self._check_initialized()
-        
-        # Create OAuth2 auth scheme
-        auth_scheme = OAuth2(
-            flows=OAuthFlows(
-                authorizationCode=OAuthFlowAuthorizationCode(
-                    authorizationUrl=self.X_AUTH_URL,
-                    tokenUrl=self.X_TOKEN_URL,
-                    scopes={
-                        scope: self.COMMON_SCOPES.get(scope, f"X API scope: {scope}")
-                        for scope in self.scopes
-                    }
-                )
-            )
-        )
-        
-        # Create OAuth2 credential
-        auth_credential = AuthCredential(
-            auth_type=AuthCredentialTypes.OAUTH2,
-            oauth2=OAuth2Auth(
-                client_id=self.client_id,
-                client_secret=self.client_secret
-            )
-        )
-        
-        return AuthConfig(
-            auth_scheme=auth_scheme,
-            raw_auth_credential=auth_credential
-        )
-
-    async def load_credential(
-        self,
-        auth_config: AuthConfig,
-        callback_context: CallbackContext,
-    ) -> Optional[AuthCredential]:
-        """Load X OAuth2 credential from storage.
-        
-        Args:
-            auth_config: The auth config containing credential key information.
-            callback_context: The current callback context.
-            
-        Returns:
-            Optional[AuthCredential]: The stored credential or None if not found.
-        """
-        self._check_initialized()
-        return await self._storage_service.load_credential(auth_config, callback_context)
-
-    async def save_credential(
-        self,
-        auth_config: AuthConfig,
-        callback_context: CallbackContext,
-    ) -> None:
-        """Save X OAuth2 credential to storage.
-        
-        Args:
-            auth_config: The auth config containing the credential to save.
-            callback_context: The current callback context.
-        """
-        self._check_initialized()
-        await self._storage_service.save_credential(auth_config, callback_context)
-        
-        logger.info(f"Saved X OAuth2 credential for user {callback_context._invocation_context.user_id}")
-
-    def get_supported_scopes(self) -> dict:
-        """Get dictionary of supported X OAuth2 scopes and their descriptions.
-        
-        Returns:
-            dict: Mapping of scope names to descriptions.
-        """
-        return self.COMMON_SCOPES.copy()

google_adk_extras-0.2.6.dist-info/RECORD

@@ -1,39 +0,0 @@
-google_adk_extras/__init__.py,sha256=cIaqZH7E__SmojqMjCqOA4FV6S6-k1pH9_mGCfkCfU4,830
-google_adk_extras/adk_builder.py,sha256=anTpd-UYtPugRSqdYNmWa_uesVjHGEKO2mntkcU-J6g,41179
-google_adk_extras/custom_agent_loader.py,sha256=6rQyBTAxvuwFPs3QGt8hWYZKsQVpbI4WIy6OS73tsac,6001
-google_adk_extras/enhanced_adk_web_server.py,sha256=rML_m4Um9QCcnnlHvCOyLGjLHDcyladEunX-x4JX43Q,5414
-google_adk_extras/enhanced_fastapi.py,sha256=TzIr159B0tnEqj2dXKmV_tJvkA66Fq2I-kL5cz9VGSk,28304
-google_adk_extras/enhanced_runner.py,sha256=b7O1a9-4S49LduILOEDs6IxjCI4w_E39sc-Hs4y3Rys,1410
-google_adk_extras/artifacts/__init__.py,sha256=_IsKDgf6wanWR0HXvSpK9SiLa3n5URKLtazkKyH1P-o,931
-google_adk_extras/artifacts/base_custom_artifact_service.py,sha256=O9rkc250B3yDRYbyDI0EvTrCKvnih5_DQas5OF-hRMY,9721
-google_adk_extras/artifacts/local_folder_artifact_service.py,sha256=Vb-KYExEcanHy9c1wAjB1n6hVAUw7pg9ZeahWE--Jx8,12533
-google_adk_extras/artifacts/mongo_artifact_service.py,sha256=z6x7zL4RsPuUYOgv02xkD_c3BNInjTa61ElgsvkZazc,7111
-google_adk_extras/artifacts/s3_artifact_service.py,sha256=HMOlHzs5Z2DicaTa6O3LJRwzhzPlpBpwftxWtmjwf1I,15669
-google_adk_extras/artifacts/sql_artifact_service.py,sha256=9wy-UP8rvFng_h66X7mU0y7DRlxI-AcqxRtg6Ph2VDU,11992
-google_adk_extras/credentials/__init__.py,sha256=sklM_jV5hM_zffABV5TWx-B99bFNrSLP8Ik7Z6mD7Zs,1274
-google_adk_extras/credentials/base_custom_credential_service.py,sha256=iYHacJAsZmDfpxLOPYx4tQpbtWTbwC75tRp6hlZFoSg,4014
-google_adk_extras/credentials/github_oauth2_credential_service.py,sha256=bQwVh6uLLH7w_9EtS_BIcEI3VpHS6G93KSzNPB-gsig,8677
-google_adk_extras/credentials/google_oauth2_credential_service.py,sha256=3zl3bZmzch7gD4FaQinlRzKfS0Z8GRMbn161kIqgwgg,8503
-google_adk_extras/credentials/http_basic_auth_credential_service.py,sha256=s_t98s1ESnTDMUwg6Z5dvxxZ6aPd5w4Qooz6ZtMNm2I,14305
-google_adk_extras/credentials/jwt_credential_service.py,sha256=S5JzD2gGrg9YGr8iu4zYsC1B8x6vUijglAVkACOCYOA,12604
-google_adk_extras/credentials/microsoft_oauth2_credential_service.py,sha256=ST1i3O7LCCOa2RnvTP0mPIFsj5aMtWgfMnCyW2kgyUg,10366
-google_adk_extras/credentials/x_oauth2_credential_service.py,sha256=zFESyKPjzJJnmdKcU3ZCLZvbHLMPifdKUDYTiN_16vc,8952
-google_adk_extras/memory/__init__.py,sha256=2FFJXw9CZHctKXmCuc-lrdETeQ5xqdivy3oarHJz5gs,994
-google_adk_extras/memory/base_custom_memory_service.py,sha256=TRQMaXiRg2LXFwYZnFHoL-yBVtecuX1ownyPBJf6Xww,3613
-google_adk_extras/memory/mongo_memory_service.py,sha256=M2FGzGaWuEkDILjEpUPx7OOzuek7d1OP9TXzKAeb9jE,6759
-google_adk_extras/memory/redis_memory_service.py,sha256=P6vYvP8gv6kCH1lRB0SQld3mzS_JVKMUDtKifXDfu38,7400
-google_adk_extras/memory/sql_memory_service.py,sha256=cmy2sL4bzVi4WWyEeVt7FNhioxSd1OGm7fQKzESvyO8,10766
-google_adk_extras/memory/yaml_file_memory_service.py,sha256=St1LMYwLE7HnW-hWhckuEKFadlULUPKkqRMwLpOf48I,9941
-google_adk_extras/sessions/__init__.py,sha256=VgHyPULLzjJD7ShsyABz98rWVND0mOoM6qX74MrTJwA,915
-google_adk_extras/sessions/base_custom_session_service.py,sha256=npwrSNAtgqN6K7C8e4idiWkFNr_3pcOAiFYpGXu3NnI,8912
-google_adk_extras/sessions/mongo_session_service.py,sha256=r3jZ3PmDpbZ0veNzXzuAj_fqRWNZL8RO53ESEHrr7uQ,8329
-google_adk_extras/sessions/redis_session_service.py,sha256=yyfXZozeFWJ2S_kz7zXqz--f_ymE6HMMpT3MhcpFXIE,10434
-google_adk_extras/sessions/sql_session_service.py,sha256=TaOeEVWnwQ_8nvDZBW7e3qhzR_ecuGsjvZ_kh6Guq8g,14558
-google_adk_extras/sessions/yaml_file_session_service.py,sha256=SpTh8YHIALcoxzmturhcZ4ReHKQrJI1CxoYiJQ-baRc,11819
-google_adk_extras/streaming/__init__.py,sha256=rcjmlCJHTlvUiCrx6qNGw5ObCnEtfENkGTvzfEiGL0M,461
-google_adk_extras/streaming/streaming_controller.py,sha256=Qxg4yqakd3mBPJ9sISe4gO_pvBpYttvSq5SIK4W5ZPQ,10505
-google_adk_extras-0.2.6.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-google_adk_extras-0.2.6.dist-info/METADATA,sha256=NXbByeiIsdHl2UcZ5F-i22thmm-rFcTr1poTFdVY5cI,10707
-google_adk_extras-0.2.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-google_adk_extras-0.2.6.dist-info/top_level.txt,sha256=DDWgVkz8G8ihPzznxAWyKa2jgJW3F6Fwy__qMddoKTs,18
-google_adk_extras-0.2.6.dist-info/RECORD,,