Merge pull request #60 from moonpyt/feature/autonomys

feat: Implement Autonomys Auto Drive storage integration with comprehensive API supportFeature/autonomys

Author: moonpyt

spoon_toolkits/storage/__init__.py

@@ -1,39 +1,5 @@
-"""DEX tools module for SpoonAI"""
+"""Storage tools module for SpoonAI"""
 
-from .base import (
-    DexBaseTool,
-    DefiBaseTool,
-    BitqueryTool
-)
-
-from .price_data import (
-    GetTokenPriceTool,
-    Get24hStatsTool,
-    GetKlineDataTool,
-)
-
-from .price_alerts import (
-    PriceThresholdAlertTool,
-    LpRangeCheckTool,
-    SuddenPriceIncreaseTool,
-)
-
-from .lending_rates import (
-    LendingRateMonitorTool,
-)
-
-# from .lst_arbitrage import LstArbitrageTool
-
-__all__ = [
-    "GetTokenPriceTool",
-    "Get24hStatsTool",
-    "GetKlineDataTool",
-    "PriceThresholdAlertTool",
-    "LpRangeCheckTool",
-    "SuddenPriceIncreaseTool",
-    "LendingRateMonitorTool",
-    "DexBaseTool",
-    "DefiBaseTool",
-    "BitqueryTool",
-    # "LstArbitrageTool",
-] 
+# This file is intentionally minimal to avoid circular dependencies 
+# and incorrect imports. Specific storage tools are imported from their
+# respective submodules.

spoon_toolkits/storage/autonomys_auto_drive/__init__.py

@@ -0,0 +1,85 @@
+"""Autonomys Auto Drive tools module"""
+
+# Account tools (renamed from Subscription)
+from .subscriptions_tools import (
+    GetAccountInfoTool,
+)
+
+# Upload tools
+from .uploads_tools import (
+    UploadFileTool,
+    UploadFileSmallTool,
+    UploadFileLargeTool,
+    GetUploadStatusTool,
+)
+
+# Object tools
+from .objects_tools import (
+    GetRootObjectsTool,
+    SearchObjectsTool,
+    PublishObjectTool,
+    UnpublishObjectTool,
+    DeleteObjectTool,
+    RestoreObjectTool,
+    GetSharedRootObjectsTool,
+    GetDeletedRootObjectsTool,
+    GetObjectSummaryTool,
+    ShareObjectTool,
+    GetObjectStatusTool,
+    GetObjectMetadataTool,
+)
+
+# Download tools
+from .downloads_tools import (
+    DownloadObjectTool,
+    StreamDownloadTool,
+    DownloadPublicObjectTool,
+    StreamDownloadPublicObjectTool,
+    CreateAsyncDownloadTool,
+    GetAsyncDownloadStatusTool,
+    ListAsyncDownloadsTool,
+    DismissAsyncDownloadTool,
+)
+
+# Provider
+from .provider import AutoDriveProvider
+from .base import get_provider
+
+__all__ = [
+    # Account tools (1) - renamed from Subscription
+    "GetAccountInfoTool",
+    
+    # Upload tools (4)
+    "UploadFileTool",
+    "UploadFileSmallTool",
+    "UploadFileLargeTool",
+    "GetUploadStatusTool",
+    
+    # Object tools (12)
+    "GetRootObjectsTool",
+    "SearchObjectsTool",
+    "PublishObjectTool",
+    "UnpublishObjectTool",
+    "DeleteObjectTool",
+    "RestoreObjectTool",
+    "GetSharedRootObjectsTool",
+    "GetDeletedRootObjectsTool",
+    "GetObjectSummaryTool",
+    "ShareObjectTool",
+    "GetObjectStatusTool",
+    "GetObjectMetadataTool",
+    
+    # Download tools (8)
+    "DownloadObjectTool",
+    "StreamDownloadTool",
+    "DownloadPublicObjectTool",
+    "StreamDownloadPublicObjectTool",
+    "CreateAsyncDownloadTool",
+    "GetAsyncDownloadStatusTool",
+    "ListAsyncDownloadsTool",
+    "DismissAsyncDownloadTool",
+    
+    # Provider
+    "AutoDriveProvider",
+    "get_provider",
+]

spoon_toolkits/storage/autonomys_auto_drive/base.py

@@ -0,0 +1,23 @@
+"""Base module for Autonomys Auto Drive tools"""
+
+import os
+from .provider import AutoDriveProvider
+
+def get_provider() -> AutoDriveProvider:
+    """
+    Get Auto Drive provider instance.
+    Validates that the API key is configured before initialization.
+    """
+    api_key = os.environ.get('AUTONOMYS_AUTO_DRIVE_API_KEY', 'apikey')
+    auth_provider = os.environ.get('AUTONOMYS_AUTO_DRIVE_AUTH_PROVIDER', 'apikey')
+    
+    if not api_key:
+        raise RuntimeError(
+            "AUTONOMYS_AUTO_DRIVE_API_KEY is not set in environment variables. "
+            "Please configure it to use Autonomys Auto Drive tools."
+        )
+        
+    return AutoDriveProvider(
+        api_key=api_key,
+        auth_provider=auth_provider
+    )