fere-sdk 0.3.1.dev27__tar.gz → 0.4.0.dev29__tar.gz

{fere_sdk-0.3.1.dev27 → fere_sdk-0.4.0.dev29}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fere-sdk
-Version: 0.3.1.dev27
+Version: 0.4.0.dev29
 Summary: Python SDK for the FereAI Gateway API
 Author-email: Fere AI <info@fere.ai>
 License-Expression: MIT

{fere_sdk-0.3.1.dev27 → fere_sdk-0.4.0.dev29}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fere-sdk"
-version = "0.3.1.dev27"
+version = "0.4.0.dev29"
 description = "Python SDK for the FereAI Gateway API"
 readme = "README.md"
 requires-python = ">=3.10"

{fere_sdk-0.3.1.dev27 → fere_sdk-0.4.0.dev29}/src/fere_sdk/client.py

@@ -22,7 +22,9 @@ from .models import (
     DepositRequest,
     HooksRequest,
     LimitOrderRequest,
+    SecurityCheckResponse,
     SwapRequest,
+    TokenToCheck,
     WithdrawRequest,
 )
 
@@ -223,6 +225,29 @@ class FereClient:
         )
         return await self._api.set_hooks(req)
 
+    async def check_token_security(
+        self,
+        tokens: list[TokenToCheck] | list[dict],
+    ) -> SecurityCheckResponse:
+        """Pre-check tokens for honeypot / scam / rug-pull flags.
+
+        Pass a list of TokenToCheck or plain dicts with token_address
+        and chain_id. If a token is flagged and the user still wants
+        to proceed, call ``swap(..., bypass_security_check=True)``.
+        """
+        normalized: list[TokenToCheck] = []
+        for t in tokens:
+            if isinstance(t, TokenToCheck):
+                normalized.append(t)
+            else:
+                normalized.append(
+                    TokenToCheck(
+                        token_address=t["token_address"],
+                        chain_id=t["chain_id"],
+                    )
+                )
+        return await self._api.check_token_security(normalized)
+
     # ----------------------------------------------------------
     # Earn (blocks until complete)
     # ----------------------------------------------------------

{fere_sdk-0.3.1.dev27 → fere_sdk-0.4.0.dev29}/src/fere_sdk/low_level.py

@@ -16,10 +16,14 @@ from .models import (
     DepositRequest,
     HooksRequest,
     LimitOrderRequest,
+    SecurityCheckResponse,
+    SecurityCheckSummary,
     SwapRequest,
     TaskResponse,
     TaskStatusResponse,
     TokenInfo,
+    TokenSecurityResult,
+    TokenToCheck,
     WithdrawRequest,
 )
 
@@ -305,6 +309,49 @@ class FereAPI:
         """POST /v1/hooks"""
         return await self._authed_post("/v1/hooks", request.to_dict())
 
+    # ----------------------------------------------------------
+    # Security
+    # ----------------------------------------------------------
+
+    async def check_token_security(
+        self,
+        tokens: list[TokenToCheck],
+    ) -> SecurityCheckResponse:
+        """POST /v1/security/check.
+
+        Pre-check whether output tokens have been flagged by Honeypot.is
+        (EVM) or RugCheck.xyz (Solana) before submitting a swap. If a
+        token is flagged and the user still wants to proceed, pass
+        bypass_security_check=true on the swap request.
+        """
+        data = await self._authed_post(
+            "/v1/security/check",
+            {"tokens": [t.to_dict() for t in tokens]},
+        )
+        results = [
+            TokenSecurityResult(
+                token_address=r["token_address"],
+                chain_id=r["chain_id"],
+                allowed=r["allowed"],
+                status=r["status"],
+                provider=r["provider"],
+                reason=r.get("reason"),
+                risk_details=r.get("risk_details"),
+            )
+            for r in data.get("results", [])
+        ]
+        s = data.get("summary", {})
+        summary = SecurityCheckSummary(
+            total=s.get("total", 0),
+            passed=s.get("passed", 0),
+            failed=s.get("failed", 0),
+            warning=s.get("warning", 0),
+            api_unavailable=s.get("api_unavailable", 0),
+            unsupported_chain=s.get("unsupported_chain", 0),
+            skipped=s.get("skipped", 0),
+        )
+        return SecurityCheckResponse(results=results, summary=summary)
+
     # ----------------------------------------------------------
     # Wallets
     # ----------------------------------------------------------

{fere_sdk-0.3.1.dev27 → fere_sdk-0.4.0.dev29}/src/fere_sdk/models.py

@@ -18,6 +18,10 @@ __all__ = [
     "HooksRequest",
     "DepositRequest",
     "WithdrawRequest",
+    "TokenToCheck",
+    "TokenSecurityResult",
+    "SecurityCheckSummary",
+    "SecurityCheckResponse",
     "TaskTimeoutError",
 ]
 
@@ -78,6 +82,7 @@ class SwapRequest:
     stop_loss: dict | None = None
     take_profit: dict | None = None
     cancel_conditional_orders: bool | None = None
+    bypass_security_check: bool | None = None
 
     def to_dict(self) -> dict:
         d: dict = {
@@ -95,9 +100,51 @@ class SwapRequest:
             d["take_profit"] = self.take_profit
         if self.cancel_conditional_orders is not None:
             d["cancel_conditional_orders"] = self.cancel_conditional_orders
+        if self.bypass_security_check is not None:
+            d["bypass_security_check"] = self.bypass_security_check
         return d
 
 
+@dataclass
+class TokenToCheck:
+    token_address: str
+    chain_id: int | str
+
+    def to_dict(self) -> dict:
+        return {
+            "token_address": self.token_address,
+            "chain_id": self.chain_id,
+        }
+
+
+@dataclass
+class TokenSecurityResult:
+    token_address: str
+    chain_id: int
+    allowed: bool
+    status: str
+    provider: str
+    reason: str | None = None
+    risk_details: dict | None = None
+
+
+@dataclass
+class SecurityCheckSummary:
+    total: int
+    passed: int
+    failed: int
+    warning: int
+    api_unavailable: int
+    unsupported_chain: int = 0
+    skipped: int = 0
+
+
+@dataclass
+class SecurityCheckResponse:
+    results: list[TokenSecurityResult]
+    summary: SecurityCheckSummary
+
+
 @dataclass
 class LimitOrderRequest:
     chain_id_in: int