Add mypy to check type

DEVELOPMENT.md

@@ -106,6 +106,12 @@ The project uses several tools to maintain code quality:
 
 All these tools are automatically run as pre-commit hooks when you commit code.
 
+### Type Checking
+
+```bash
+mypy .
+```
+
 ### Manual Pre-commit Checks
 
 To manually run all pre-commit checks:

MANIFEST.in

@@ -4,3 +4,4 @@ include requirements.txt
 recursive-include src *.py *.json
 recursive-include src/story_protocol_python_sdk/abi *.json
 recursive-include src/story_protocol_python_sdk/scripts *.json
+include src/story_protocol_python_sdk/py.typed

mypy.ini

@@ -0,0 +1,24 @@
+[mypy]
+python_version = 3.10
+warn_return_any = True
+warn_unused_configs = True
+disallow_untyped_defs = False
+disallow_incomplete_defs = False
+check_untyped_defs = True
+disallow_untyped_decorators = False
+no_implicit_optional = True
+warn_redundant_casts = True
+warn_unused_ignores = True
+warn_no_return = True
+warn_unreachable = True
+strict_equality = True
+exclude = story_protocol_python_sdk/abi|story_protocol_python_sdk/scripts
+# Due to previous code having any return types, we need to disable this error code
+disable_error_code = no-any-return
+
+# Ignore missing imports for external libraries
+[mypy-web3.*]
+ignore_missing_imports = True
+
+[mypy-ens.*]
+ignore_missing_imports = True

setup.py

@@ -10,6 +10,7 @@
     package_dir={"": "src"},
     install_requires=["web3>=7.0.0", "pytest", "python-dotenv", "base58"],
     include_package_data=True,  # Ensure package data is included
+    package_data={"story_protocol_python_sdk": ["py.typed"]},
     url="https://github.com/storyprotocol/python-sdk",
     license="MIT",
     author="Andrew Chung",

src/story_protocol_python_sdk/abi/CoreMetadataViewModule/CoreMetadataViewModule_client.py

@@ -15,7 +15,6 @@ def __init__(self, web3: Web3):
         )
         with open(config_path, "r") as config_file:
             config = json.load(config_file)
-
         contract_address = None
         for contract in config["contracts"]:
             if contract["contract_name"] == "CoreMetadataViewModule":

src/story_protocol_python_sdk/abi/DisputeModule/DisputeModule_client.py

@@ -59,5 +59,17 @@ def build_resolveDispute_transaction(self, disputeId, data, tx_params):
             disputeId, data
         ).build_transaction(tx_params)
 
+    def tagIfRelatedIpInfringed(self, ipIdToTag, infringerDisputeId):
+        return self.contract.functions.tagIfRelatedIpInfringed(
+            ipIdToTag, infringerDisputeId
+        ).transact()
+
+    def build_tagIfRelatedIpInfringed_transaction(
+        self, ipIdToTag, infringerDisputeId, tx_params
+    ):
+        return self.contract.functions.tagIfRelatedIpInfringed(
+            ipIdToTag, infringerDisputeId
+        ).build_transaction(tx_params)
+
     def isWhitelistedDisputeTag(self, tag):
         return self.contract.functions.isWhitelistedDisputeTag(tag).call()