fix: Add proper type casts to fix remaining mypy errors

- Add cast imports to svm.py, evm.py, base.py, and simple.py
- Cast all response.data returns to their proper types
- This ensures mypy understands the return types from divine-requests

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>

Author: codebydivine

src/thegraph_token_api/base.py

@@ -5,6 +5,7 @@
 """
 
 import os
+from typing import cast
 
 # Import divine-requests (should be installed as a package)
 from requests import NetworkingManager  # type: ignore[attr-defined]
@@ -68,7 +69,7 @@ async def get_health(self) -> str:
             Health status string (should be "OK")
         """
         response = await self.manager.get(f"{self.base_url}/health", headers=self._headers)
-        return response.text
+        return cast(str, response.text)
 
     async def get_version(self) -> VersionResponse:
         """
@@ -80,7 +81,7 @@ async def get_version(self) -> VersionResponse:
         response = await self.manager.get(
             f"{self.base_url}/version", headers=self._headers, expected_type=VersionResponse
         )
-        return response.data
+        return cast(VersionResponse, response.data)
 
     async def get_networks(self) -> NetworksResponse:
         """
@@ -92,4 +93,4 @@ async def get_networks(self) -> NetworksResponse:
         response = await self.manager.get(
             f"{self.base_url}/networks", headers=self._headers, expected_type=NetworksResponse
         )
-        return response.data
+        return cast(NetworksResponse, response.data)

src/thegraph_token_api/evm.py

@@ -4,6 +4,8 @@
 Provides access to EVM blockchain data including NFTs, tokens, balances, transfers, and DEX data.
 """
 
+from typing import cast
+
 from .base import BaseTokenAPI
 from .types import (
     BalancesResponse,
@@ -101,7 +103,7 @@ async def get_nft_ownerships(
             params=params,
             expected_type=NFTOwnershipsResponse,
         )
-        return response.data
+        return cast(NFTOwnershipsResponse, response.data)
 
     async def get_nft_collection(self, contract: str) -> NFTCollectionsResponse:
         """
@@ -121,7 +123,7 @@ async def get_nft_collection(self, contract: str) -> NFTCollectionsResponse:
             params=params,
             expected_type=NFTCollectionsResponse,
         )
-        return response.data
+        return cast(NFTCollectionsResponse, response.data)
 
     async def get_nft_item(self, contract: str, token_id: str) -> NFTItemsResponse:
         """
@@ -142,7 +144,7 @@ async def get_nft_item(self, contract: str, token_id: str) -> NFTItemsResponse:
             params=params,
             expected_type=NFTItemsResponse,
         )
-        return response.data
+        return cast(NFTItemsResponse, response.data)
 
     async def get_nft_activities(
         self,
@@ -202,7 +204,7 @@ async def get_nft_activities(
             params=params,
             expected_type=NFTActivitiesResponse,
         )
-        return response.data
+        return cast(NFTActivitiesResponse, response.data)
 
     async def get_nft_holders(self, contract: str) -> NFTHoldersResponse:
         """
@@ -222,7 +224,7 @@ async def get_nft_holders(self, contract: str) -> NFTHoldersResponse:
             params=params,
             expected_type=NFTHoldersResponse,
         )
-        return response.data
+        return cast(NFTHoldersResponse, response.data)
 
     async def get_nft_sales(
         self,
@@ -283,7 +285,7 @@ async def get_nft_sales(
         response = await self.manager.get(
             f"{self.base_url}/nft/sales/evm", headers=self._headers, params=params, expected_type=NFTSalesResponse
         )
-        return response.data
+        return cast(NFTSalesResponse, response.data)
 
     # ===== Balance Methods =====
 
@@ -314,7 +316,7 @@ async def get_balances(
             expected_type=BalancesResponse,
         )
         # Return the full response object
-        return response.data
+        return cast(BalancesResponse, response.data)
 
     # ===== Transfer Methods =====
 
@@ -374,7 +376,7 @@ async def get_transfers(
         response = await self.manager.get(
             f"{self.base_url}/transfers/evm", headers=self._headers, params=params, expected_type=TransfersResponse
         )
-        return response.data
+        return cast(TransfersResponse, response.data)
 
     # ===== Token Methods =====
 
@@ -394,7 +396,7 @@ async def get_token(self, contract: str) -> TokensResponse:
             f"{self.base_url}/tokens/evm/{contract}", headers=self._headers, params=params, expected_type=TokensResponse
         )
         # Return the full response object
-        return response.data
+        return cast(TokensResponse, response.data)
 
     async def get_token_holders(
         self,
@@ -432,7 +434,7 @@ async def get_token_holders(
             params=params,
             expected_type=TokenHoldersResponse,
         )
-        return response.data
+        return cast(TokenHoldersResponse, response.data)
 
     # ===== Swap Methods =====
 
@@ -501,7 +503,7 @@ async def get_swaps(
             f"{self.base_url}/swaps/evm", headers=self._headers, params=params, expected_type=SwapsResponse
         )
         # Return the full response object, since response is already the validated dict
-        return response.data
+        return cast(SwapsResponse, response.data)
 
     # ===== Pool Methods =====
 
@@ -547,7 +549,7 @@ async def get_pools(
         response = await self.manager.get(
             f"{self.base_url}/pools/evm", headers=self._headers, params=params, expected_type=PoolsResponse
         )
-        return response.data
+        return cast(PoolsResponse, response.data)
 
     # ===== OHLC Methods =====
 
@@ -585,7 +587,7 @@ async def get_ohlc_pools(
         response = await self.manager.get(
             f"{self.base_url}/ohlc/pools/evm/{pool}", headers=self._headers, params=params, expected_type=OHLCResponse
         )
-        return response.data
+        return cast(OHLCResponse, response.data)
 
     async def get_ohlc_prices(
         self,
@@ -621,7 +623,7 @@ async def get_ohlc_prices(
         response = await self.manager.get(
             f"{self.base_url}/ohlc/prices/evm/{token}", headers=self._headers, params=params, expected_type=OHLCResponse
         )
-        return response.data
+        return cast(OHLCResponse, response.data)
 
     # ===== Historical Methods =====
 
@@ -666,4 +668,4 @@ async def get_historical_balances(
             params=params,
             expected_type=HistoricalBalancesResponse,
         )
-        return response.data
+        return cast(HistoricalBalancesResponse, response.data)

src/thegraph_token_api/simple.py

@@ -17,7 +17,7 @@
 
 import os
 from datetime import datetime, timedelta
-from typing import Any
+from typing import Any, cast
 
 from dotenv import load_dotenv
 
@@ -354,7 +354,7 @@ def __init__(
     def _extract_data(self, response) -> list[dict[Any, Any]]:
         """Extract clean data from API response."""
         if hasattr(response, "data") and isinstance(response.data, dict):
-            return response.data.get("data", [])
+            return cast(list[dict[Any, Any]], response.data.get("data", []))
         return []
 
     # ===== EVM Internal Methods =====

src/thegraph_token_api/svm.py

@@ -4,6 +4,8 @@
 Provides access to Solana blockchain data including SPL tokens, balances, transfers, and DEX swaps.
 """
 
+from typing import cast
+
 from .base import BaseTokenAPI
 from .types import (
     OrderBy,
@@ -103,7 +105,7 @@ async def get_balances(
         response = await self.manager.get(
             f"{self.base_url}/balances/svm", headers=self._headers, params=params, expected_type=SolanaBalancesResponse
         )
-        return response.data
+        return cast(SolanaBalancesResponse, response.data)
 
     # ===== Transfer Methods =====
 
@@ -174,7 +176,7 @@ async def get_transfers(
             params=params,
             expected_type=SolanaTransfersResponse,
         )
-        return response.data
+        return cast(SolanaTransfersResponse, response.data)
 
     # ===== Swap Methods =====
 
@@ -245,4 +247,4 @@ async def get_swaps(
         response = await self.manager.get(
             f"{self.base_url}/swaps/svm", headers=self._headers, params=params, expected_type=SolanaSwapsResponse
         )
-        return response.data
+        return cast(SolanaSwapsResponse, response.data)