wayfinder-paths 0.1.23__py3-none-any.whl → 0.1.25__py3-none-any.whl

wayfinder_paths/adapters/token_adapter/manifest.yaml

@@ -0,0 +1,7 @@
+schema_version: "0.1"
+entrypoint: "adapters.token_adapter.adapter.TokenAdapter"
+capabilities:
+  - "token.read"
+  - "token.price"
+  - "token.gas"
+dependencies: []

wayfinder_paths/conftest.py

@@ -1,23 +1,30 @@
 import sys
 from pathlib import Path
 
-# This needs to be at index 0 to take precedence over repo root 'tests/' directory
-_wayfinder_path_dir = Path(__file__).parent
-_wayfinder_path_str = str(_wayfinder_path_dir)
+import pytest
+
+# Add repo root to path so tests.test_utils can be imported
+_repo_root = Path(__file__).parent.parent
+_repo_root_str = str(_repo_root)
 
 
 def pytest_configure(config):
-    if _wayfinder_path_str not in sys.path:
-        sys.path.insert(0, _wayfinder_path_str)
-    elif sys.path.index(_wayfinder_path_str) > 0:
-        # Move to front if it exists but isn't first
-        sys.path.remove(_wayfinder_path_str)
-        sys.path.insert(0, _wayfinder_path_str)
-
-
-# Also set it immediately (in case pytest_configure hasn't run yet)
-if _wayfinder_path_str not in sys.path:
-    sys.path.insert(0, _wayfinder_path_str)
-elif sys.path.index(_wayfinder_path_str) > 0:
-    sys.path.remove(_wayfinder_path_str)
-    sys.path.insert(0, _wayfinder_path_str)
+    config.addinivalue_line("markers", "smoke: mark test as a smoke test")
+    if _repo_root_str not in sys.path:
+        sys.path.insert(0, _repo_root_str)
+    elif sys.path.index(_repo_root_str) > 0:
+        sys.path.remove(_repo_root_str)
+        sys.path.insert(0, _repo_root_str)
+
+
+def pytest_collection_modifyitems(config, items):
+    for item in items:
+        if "smoke" in item.nodeid:
+            item.add_marker(pytest.mark.smoke)
+
+
+if _repo_root_str not in sys.path:
+    sys.path.insert(0, _repo_root_str)
+elif sys.path.index(_repo_root_str) > 0:
+    sys.path.remove(_repo_root_str)
+    sys.path.insert(0, _repo_root_str)

wayfinder_paths/core/__init__.py

@@ -1,5 +1,6 @@
 from wayfinder_paths.core.adapters.BaseAdapter import BaseAdapter
 from wayfinder_paths.core.strategies.Strategy import (
+    LiquidationResult,
     StatusDict,
     StatusTuple,
     Strategy,
@@ -7,6 +8,7 @@ from wayfinder_paths.core.strategies.Strategy import (
 
 __all__ = [
     "Strategy",
+    "LiquidationResult",
     "StatusDict",
     "StatusTuple",
     "BaseAdapter",

wayfinder_paths/core/adapters/BaseAdapter.py

@@ -14,30 +14,5 @@ class BaseAdapter(ABC):
         self.config = config or {}
         self.logger = logger.bind(adapter=self.__class__.__name__)
 
-    async def connect(self) -> bool:
-        return True
-
-    async def get_balance(self, asset: str) -> dict[str, Any]:
-        if not asset or not isinstance(asset, str) or not asset.strip():
-            raise ValueError("asset must be a non-empty string")
-        raise NotImplementedError(
-            f"get_balance not supported by {self.__class__.__name__}"
-        )
-
-    async def health_check(self) -> dict[str, Any]:
-        try:
-            connected = await self.connect()
-            return {
-                "status": "healthy" if connected else "unhealthy",
-                "connected": connected,
-                "adapter": self.adapter_type or self.__class__.__name__,
-            }
-        except Exception as e:
-            return {
-                "status": "error",
-                "error": str(e),
-                "adapter": self.adapter_type or self.__class__.__name__,
-            }
-
     async def close(self) -> None:
         pass

wayfinder_paths/core/adapters/models.py

@@ -4,9 +4,11 @@ from pydantic import BaseModel, Field
 
 
 class OperationBase(BaseModel):
-    adapter: str
-    transaction_hash: str | None
-    transaction_chain_id: int | None
+    # These are provided by adapters at execution time, but tests and callers may
+    # construct operations without them (e.g., for bookkeeping payloads).
+    adapter: str = "unknown"
+    transaction_hash: str | None = None
+    transaction_chain_id: int | None = None
 
 
 class SWAP(OperationBase):
@@ -23,14 +25,22 @@ class SWAP(OperationBase):
 
 class LEND(OperationBase):
     type: Literal["LEND"] = "LEND"
-    contract: str
-    amount: int
+    token_address: str
+    pool_address: str
+    amount: str
+    amount_usd: float
+    transaction_status: str | None = None
+    transaction_receipt: dict[str, Any] | None = None
 
 
 class UNLEND(OperationBase):
     type: Literal["UNLEND"] = "UNLEND"
-    contract: str
-    amount: int
+    token_address: str
+    pool_address: str
+    amount: str
+    amount_usd: float
+    transaction_status: str | None = None
+    transaction_receipt: dict[str, Any] | None = None
 
 
 # Type alias for operation types (currently only SWAP is used)

wayfinder_paths/core/clients/BRAPClient.py

@@ -72,7 +72,7 @@ class BRAPQuoteResponse(TypedDict):
 class BRAPClient(WayfinderClient):
     def __init__(self):
         super().__init__()
-        self.api_base_url = f"{get_api_base_url()}/v1/blockchain/braps"
+        self.api_base_url = f"{get_api_base_url()}/blockchain/braps"
 
     async def get_quote(
         self,

wayfinder_paths/core/clients/TokenClient.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import xml.etree.ElementTree as ET
 from typing import NotRequired, Required, TypedDict
 
 from wayfinder_paths.core.clients.WayfinderClient import WayfinderClient
@@ -76,10 +77,20 @@ class GasToken(TypedDict):
     chain: NotRequired[ChainInfo]
 
 
+class FuzzyTokenResult(TypedDict):
+    coingecko_id: NotRequired[str]
+    address: NotRequired[str]
+    chain: NotRequired[str]
+    name: NotRequired[str]
+    symbol: NotRequired[str]
+    price: NotRequired[float]
+    confidence: NotRequired[int]
+
+
 class TokenClient(WayfinderClient):
     def __init__(self):
         super().__init__()
-        self.api_base_url = f"{get_api_base_url()}/v1/blockchain/tokens"
+        self.api_base_url = f"{get_api_base_url()}/blockchain/tokens"
 
     async def get_token_details(
         self, query: str, market_data: bool = False, chain_id: int | None = None
@@ -103,3 +114,38 @@ class TokenClient(WayfinderClient):
         response.raise_for_status()
         data = response.json()
         return data.get("data", data)
+
+    async def fuzzy_search(
+        self, query: str, chain: str | None = None
+    ) -> dict[str, list[FuzzyTokenResult] | str | None]:
+        url = f"{self.api_base_url}/fuzzy/"
+        params: dict[str, str] = {"query": query}
+        if chain:
+            params["chain"] = chain
+        response = await self._authed_request("GET", url, params=params)
+        response.raise_for_status()
+        trace_id = response.headers.get("X-Token-Fuzzy-Trace")
+        tokens = self._parse_fuzzy_xml(response.text)
+        return {"tokens": tokens, "trace_id": trace_id}
+
+    def _parse_fuzzy_xml(self, xml_content: str) -> list[FuzzyTokenResult]:
+        root = ET.fromstring(xml_content)
+        tokens: list[FuzzyTokenResult] = []
+        for token_elem in root.findall("token"):
+            token: FuzzyTokenResult = {}
+            for field in ["coingecko_id", "address", "chain", "name", "symbol"]:
+                elem = token_elem.find(field)
+                if elem is not None and elem.text:
+                    token[field] = elem.text  # type: ignore[literal-required]
+            for num_field in ["price", "confidence"]:
+                elem = token_elem.find(num_field)
+                if elem is not None and elem.text:
+                    try:
+                        if num_field == "price":
+                            token["price"] = float(elem.text)
+                        else:
+                            token["confidence"] = int(elem.text)
+                    except ValueError:
+                        pass
+            tokens.append(token)
+        return tokens

wayfinder_paths/core/clients/WayfinderClient.py

@@ -65,8 +65,7 @@ class WayfinderClient:
         logger.debug(f"Making {method} request to {url}")
         start_time = time.time()
 
-        # Ensure API key is set in headers if available and not already set
-        # This ensures API keys are passed to all endpoints (including public ones) for rate limiting
+        # Pass API key to all endpoints (including public ones) for rate limiting
         if not self.headers.get("X-API-KEY"):
             creds = self._load_config_credentials()
             api_key = creds.get("api_key")

wayfinder_paths/core/clients/protocols.py

@@ -1,27 +1,26 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Protocol
-
-if TYPE_CHECKING:
-    from wayfinder_paths.core.clients.BRAPClient import BRAPQuoteResponse
-    from wayfinder_paths.core.clients.HyperlendClient import (
-        AssetsView,
-        LendRateHistory,
-        MarketEntry,
-        StableMarketsHeadroomResponse,
-    )
-    from wayfinder_paths.core.clients.LedgerClient import (
-        StrategyTransactionList,
-        TransactionRecord,
-    )
-    from wayfinder_paths.core.clients.PoolClient import (
-        LlamaMatchesResponse,
-        PoolList,
-    )
-    from wayfinder_paths.core.clients.TokenClient import (
-        GasToken,
-        TokenDetails,
-    )
+from typing import Any, Protocol
+
+from wayfinder_paths.core.clients.BRAPClient import BRAPQuoteResponse
+from wayfinder_paths.core.clients.HyperlendClient import (
+    AssetsView,
+    LendRateHistory,
+    MarketEntry,
+    StableMarketsHeadroomResponse,
+)
+from wayfinder_paths.core.clients.LedgerClient import (
+    StrategyTransactionList,
+    TransactionRecord,
+)
+from wayfinder_paths.core.clients.PoolClient import (
+    LlamaMatchesResponse,
+    PoolList,
+)
+from wayfinder_paths.core.clients.TokenClient import (
+    GasToken,
+    TokenDetails,
+)
 
 
 class TokenClientProtocol(Protocol):