ipor-fusion 4.0.0.dev41__tar.gz → 4.0.0.dev43__tar.gz

{ipor_fusion-4.0.0.dev41 → ipor_fusion-4.0.0.dev43}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: ipor-fusion
-Version: 4.0.0.dev41
+Version: 4.0.0.dev43
 Summary: The IPOR Fusion SDK for Python
 Project-URL: homepage, https://ipor.io
 Project-URL: repository, https://github.com/IPOR-Labs/ipor-fusion.py

{ipor_fusion-4.0.0.dev41 → ipor_fusion-4.0.0.dev43}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ipor-fusion"
-version = "4.0.0.dev41"
+version = "4.0.0.dev43"
 description = "The IPOR Fusion SDK for Python"
 authors = [
     { name = "Adam Mydlarz", email = "amydlarz@gmail.com" },

{ipor_fusion-4.0.0.dev41 → ipor_fusion-4.0.0.dev43}/src/ipor_fusion/cli/vault_cmd.py

@@ -131,7 +131,7 @@ def add(address: ChecksumAddress, label: str | None, chain_id: ChainId | None) -
 
 @vault.command("remove")
 @click.argument("address", type=ADDRESS)
-def remove(address: str) -> None:
+def remove(address: ChecksumAddress) -> None:
     """Remove a saved vault."""
     cfg = load_config()
     before = len(cfg.vaults)

{ipor_fusion-4.0.0.dev41 → ipor_fusion-4.0.0.dev43}/src/ipor_fusion/mcp/server.py

@@ -437,6 +437,7 @@ def vault_add(
         label: Human-readable label (fetched from on-chain name() if empty).
         chain_id: Chain ID (auto-detected when only one provider configured).
     """
+    checksum = Web3.to_checksum_address(address)
     cfg = load_config()
 
     if not chain_id:
@@ -451,26 +452,21 @@ def vault_add(
     if not label:
         provider_url = _resolve_provider(cfg, chain_id)
         ctx = Web3Context.from_url(provider_url)
-        checksum = Web3.to_checksum_address(address)
         try:
             label = PlasmaVault(ctx, checksum).name().call()
         except Exception:
             label = checksum
 
     for vault_entry in cfg.vaults:
-        if vault_entry.address.lower() == address.lower():
+        if vault_entry.address.lower() == checksum.lower():
             vault_entry.label = label
             vault_entry.chain_id = chain_id
             save_config(cfg)
-            return ActionResult(message=f"Vault {label} ({address}) updated.")
+            return ActionResult(message=f"Vault {label} ({checksum}) updated.")
 
-    cfg.vaults.append(
-        VaultEntry(
-            address=Web3.to_checksum_address(address), label=label, chain_id=chain_id
-        )
-    )
+    cfg.vaults.append(VaultEntry(address=checksum, label=label, chain_id=chain_id))
     save_config(cfg)
-    return ActionResult(message=f"Vault {label} ({address}) added.")
+    return ActionResult(message=f"Vault {label} ({checksum}) added.")
 
 
 @mcp.tool()
@@ -635,7 +631,7 @@ def market_meta_morpho(
         chain_id: EVM chain ID.
     """
     try:
-        info = fetch_vault(vault_address, chain_id)
+        info = fetch_vault(Web3.to_checksum_address(vault_address), chain_id)
     except MorphoApiError as exc:
         raise ValueError(str(exc)) from exc
     payload = build_meta_morpho_json(info, chain_id)

{ipor_fusion-4.0.0.dev41 → ipor_fusion-4.0.0.dev43}/src/ipor_fusion/morpho_api.py

@@ -91,7 +91,7 @@ class VaultFlowCap:
     fee_wei: int
     max_in: int
     max_out: int
-    admin: str | None  # the address that can configure these flow caps
+    admin: ChecksumAddress | None  # the address that can configure these flow caps
 
 
 @dataclass(slots=True)
@@ -106,7 +106,7 @@ class VaultAllocation:
     total_assets: int
     supply_assets: int  # currently allocated to *this* market
     supply_cap: int  # cap configured by curator for *this* market
-    allocators: list[str]  # addresses with ALLOCATOR_ROLE on the vault
+    allocators: list[ChecksumAddress]  # addresses with ALLOCATOR_ROLE on the vault
     flow_cap: VaultFlowCap | None  # None if PublicAllocator not configured
 
 
@@ -122,8 +122,8 @@ class MorphoApiMarket:
     collateral_token: ChecksumAddress
     collateral_symbol: str
     collateral_decimals: int
-    oracle: str
-    irm: str
+    oracle: ChecksumAddress
+    irm: ChecksumAddress
     supply_assets: int
     borrow_assets: int
     liquidity_assets: int
@@ -190,8 +190,8 @@ def _parse_market(raw: dict) -> MorphoApiMarket:
         collateral_token=_api_address(raw["collateralAsset"]["address"]),
         collateral_symbol=raw["collateralAsset"]["symbol"],
         collateral_decimals=int(raw["collateralAsset"]["decimals"]),
-        oracle=raw["oracle"]["address"],
-        irm=raw["irmAddress"],
+        oracle=_api_address(raw["oracle"]["address"]),
+        irm=_api_address(raw["irmAddress"]),
         supply_assets=int(state.get("supplyAssets") or 0),
         borrow_assets=int(state.get("borrowAssets") or 0),
         liquidity_assets=int(state.get("liquidityAssets") or 0),
@@ -283,8 +283,8 @@ class VaultV2Cap:
     market_id: MorphoBlueMarketId | None
     loan_token: ChecksumAddress | None
     collateral_token: ChecksumAddress | None
-    oracle: str | None
-    irm: str | None
+    oracle: ChecksumAddress | None
+    irm: ChecksumAddress | None
     lltv: int | None
 
     @property
@@ -378,7 +378,7 @@ class VaultV1Info:
 
 
 def fetch_vault(
-    address: str, chain_id: ChainId, *, timeout: float = 10.0
+    address: ChecksumAddress, chain_id: ChainId, *, timeout: float = 10.0
 ) -> VaultV2Info | VaultV1Info:
     """Fetch a Morpho vault by address. Tries V2 first, falls back to V1.
 
@@ -625,7 +625,7 @@ def _parse_vault_v1(raw: dict) -> VaultV1Info:
             fee_wei=int(pac.get("fee") or 0),
             max_in=0,
             max_out=0,
-            admin=pac.get("admin"),
+            admin=_api_optional_address(pac.get("admin")),
         )
         for fc in pac.get("flowCaps") or []:
             if raw_mid := (fc.get("market") or {}).get("marketId"):
@@ -687,10 +687,12 @@ def _parse_vault(raw: dict, market_id: MorphoBlueMarketId) -> VaultAllocation:
                 fee_wei=int(pac.get("fee") or 0),
                 max_in=int(flow_for_market.get("maxIn") or 0),
                 max_out=int(flow_for_market.get("maxOut") or 0),
-                admin=pac.get("admin"),
+                admin=_api_optional_address(pac.get("admin")),
             )
     allocators = [
-        a["address"] for a in (raw.get("allocators") or []) if a.get("address")
+        _api_address(a["address"])
+        for a in (raw.get("allocators") or [])
+        if a.get("address")
     ]
     return VaultAllocation(
         vault_address=_api_address(raw["address"]),

{ipor_fusion-4.0.0.dev41 → ipor_fusion-4.0.0.dev43}/tests/test_cli_market_morpho_blue.py

@@ -4,6 +4,7 @@
 import json
 from unittest.mock import MagicMock, patch
 
+import pytest
 from click.testing import CliRunner
 from eth_abi import encode
 from web3 import Web3
@@ -18,6 +19,7 @@ from ipor_fusion.morpho_api import (
     MorphoApiMarket,
     VaultAllocation,
     VaultFlowCap,
+    _parse_market,
     fetch_market,
     fetch_vault,
 )
@@ -786,3 +788,31 @@ class TestMetaMorphoCommand:
         assert len(market_caps) == 1
         assert market_caps[0]["loan_token"] == LOAN_TOKEN
         assert market_caps[0]["lltv"] == str(915 * 10**15)
+
+
+class TestMorphoApiAddresses:
+    def _raw(self, oracle: str) -> dict:
+        return {
+            "marketId": "0x" + MARKET_ID,
+            "lltv": "915000000000000000",
+            "loanAsset": {"address": LOAN_TOKEN.lower(), "symbol": "U", "decimals": 6},
+            "collateralAsset": {
+                "address": COLLATERAL_TOKEN,
+                "symbol": "C",
+                "decimals": 18,
+            },
+            "oracle": {"address": oracle},
+            "irmAddress": IRM.lower(),
+            "state": {},
+            "supplyingVaults": [],
+        }
+
+    def test_addresses_are_checksummed(self):
+        market = _parse_market(self._raw(ORACLE.lower()))
+        assert market.loan_token == LOAN_TOKEN
+        assert market.oracle == ORACLE
+        assert market.irm == IRM
+
+    def test_a_malformed_address_is_an_api_error(self):
+        with pytest.raises(MorphoApiError):
+            _parse_market(self._raw("invalid oracle"))

{ipor_fusion-4.0.0.dev41 → ipor_fusion-4.0.0.dev43}/tests/test_mcp_server.py

@@ -89,12 +89,15 @@ def _config_with_keyed_provider():
     return FusionConfig(providers={"1": "https://rpc.example.com/v2/SECRETKEY"})
 
 
-def _config_with_vault():
+_SAVED_VAULT = "0x43ee0243ea8cf02f7087d8b16c8d2007cc9c7ca2"
+
+
+def _config_with_vault(address: str = _SAVED_VAULT):
     from ipor_fusion.cli.config_store import FusionConfig, VaultEntry
 
     return FusionConfig(
         providers={"1": "https://rpc.example.com"},
-        vaults=[VaultEntry(address="0xABC", label="Test", chain_id=1)],
+        vaults=[VaultEntry(address=address, label="Test", chain_id=1)],
     )
 
 
@@ -264,7 +267,7 @@ class TestConfigShow:
     def test_config_with_vault(self, _):
         result = config_show()
         assert len(result.vaults) == 1
-        assert result.vaults[0].address == "0xABC"
+        assert result.vaults[0].address == Web3.to_checksum_address(_SAVED_VAULT)
 
     @patch(
         "ipor_fusion.mcp.server.load_config",
@@ -319,7 +322,7 @@ class TestVaultList:
     def test_with_vault(self, _):
         result = vault_list()
         assert len(result) == 1
-        assert result[0].address == "0xABC"
+        assert result[0].address == Web3.to_checksum_address(_SAVED_VAULT)
         assert result[0].label == "Test"
         assert result[0].chain_id == 1
 
@@ -348,13 +351,23 @@ class TestVaultAdd:
         with pytest.raises(ValueError):
             vault_add(address="0xDEF", label="New Vault", chain_id=1)
 
+    @patch("ipor_fusion.mcp.server.save_config")
+    @patch(
+        "ipor_fusion.mcp.server.load_config",
+        return_value=_config_with_vault("0xABC"),
+    )
+    def test_a_malformed_saved_entry_is_not_updated(self, mock_load, mock_save):
+        with pytest.raises(ValueError):
+            vault_add(address="0xABC", label="Updated", chain_id=1)
+        mock_save.assert_not_called()
+
     @patch("ipor_fusion.mcp.server.save_config")
     @patch(
         "ipor_fusion.mcp.server.load_config",
         return_value=_config_with_vault(),
     )
     def test_update_existing(self, mock_load, mock_save):
-        result = vault_add(address="0xABC", label="Updated", chain_id=1)
+        result = vault_add(address=_SAVED_VAULT, label="Updated", chain_id=1)
         assert "updated" in result.message.lower()
         saved_cfg = mock_save.call_args[0][0]
         assert len(saved_cfg.vaults) == 1
@@ -368,7 +381,7 @@ class TestVaultRemove:
         return_value=_config_with_vault(),
     )
     def test_remove_existing(self, mock_load, mock_save):
-        result = vault_remove(address="0xABC")
+        result = vault_remove(address=_SAVED_VAULT)
         assert "removed" in result.message.lower()
         saved_cfg = mock_save.call_args[0][0]
         assert len(saved_cfg.vaults) == 0