fix: register_derivative method

src/story_protocol_python_sdk/__init__.py

@@ -7,6 +7,18 @@
 from .resources.Royalty import Royalty
 from .resources.WIP import WIP
 from .story_client import StoryClient
+from .types.common import AccessPermission
+from .utils.constants import (
+    DEFAULT_FUNCTION_SELECTOR,
+    MAX_ROYALTY_TOKEN,
+    ROYALTY_POLICY_LAP_ADDRESS,
+    ROYALTY_POLICY_LRP_ADDRESS,
+    ZERO_ADDRESS,
+    ZERO_FUNC,
+    ZERO_HASH,
+)
+from .utils.derivative_data import DerivativeDataInput
+from .utils.ip_metadata import IPMetadataInput
 
 __all__ = [
     "StoryClient",
@@ -16,4 +28,15 @@
     "IPAccount",
     "Dispute",
     "WIP",
+    "AccessPermission",
+    "DerivativeDataInput",
+    "IPMetadataInput",
+    # Constants
+    "ZERO_ADDRESS",
+    "ZERO_HASH",
+    "ROYALTY_POLICY_LAP_ADDRESS",
+    "ROYALTY_POLICY_LRP_ADDRESS",
+    "ZERO_FUNC",
+    "DEFAULT_FUNCTION_SELECTOR",
+    "MAX_ROYALTY_TOKEN",
 ]

src/story_protocol_python_sdk/abi/AccessController/AccessController_client.py

@@ -48,13 +48,13 @@ def build_setAllPermissions_transaction(
             ipAccount, signer, permission
         ).build_transaction(tx_params)
 
-    def setTransientBatchPermissions(self, permissions):
-        return self.contract.functions.setTransientBatchPermissions(
+    def setBatchTransientPermissions(self, permissions):
+        return self.contract.functions.setBatchTransientPermissions(
             permissions
         ).transact()
 
-    def build_setTransientBatchPermissions_transaction(self, permissions, tx_params):
-        return self.contract.functions.setTransientBatchPermissions(
+    def build_setBatchTransientPermissions_transaction(self, permissions, tx_params):
+        return self.contract.functions.setBatchTransientPermissions(
             permissions
         ).build_transaction(tx_params)
 

src/story_protocol_python_sdk/abi/DerivativeWorkflows/DerivativeWorkflows_client.py

@@ -35,3 +35,23 @@ def __init__(self, web3: Web3):
         with open(abi_path, "r") as abi_file:
             abi = json.load(abi_file)
         self.contract = self.web3.eth.contract(address=contract_address, abi=abi)
+
+    def registerIpAndMakeDerivative(
+        self, nftContract, tokenId, derivData, ipMetadata, sigMetadataAndRegister
+    ):
+        return self.contract.functions.registerIpAndMakeDerivative(
+            nftContract, tokenId, derivData, ipMetadata, sigMetadataAndRegister
+        ).transact()
+
+    def build_registerIpAndMakeDerivative_transaction(
+        self,
+        nftContract,
+        tokenId,
+        derivData,
+        ipMetadata,
+        sigMetadataAndRegister,
+        tx_params,
+    ):
+        return self.contract.functions.registerIpAndMakeDerivative(
+            nftContract, tokenId, derivData, ipMetadata, sigMetadataAndRegister
+        ).build_transaction(tx_params)