Perplexity-powered agents that research AND trade

import requests
import json

# API credentials
PPLX_API_KEY = "pplx-your-key-here"
PF_API_KEY   = "pf_live_your_key_here"

class PerplexityTradingAgent:
    def __init__(self):
        self.pplx_url = "https://api.perplexity.ai/chat/completions"
        self.pf_url   = "https://purpleflea.com/api/v1"

    def research_market(self, asset: str) -> dict:
        """Search for latest news and sentiment on an asset."""
        headers = {"Authorization": f"Bearer {PPLX_API_KEY}"}
        payload = {
            "model": "sonar-pro",
            "messages": [
                {
                    "role": "system",
                    "content": (
                        "You are a crypto market analyst. Analyze the latest news "
                        "for the given asset. Return JSON with fields: "
                        "sentiment (bullish/bearish/neutral), confidence (0-100), "
                        "key_catalysts (list), direction (long/short/hold)."
                    )
                },
                {
                    "role": "user",
                    "content": f"Latest news and sentiment for {asset} in the last 24 hours."
                }
            ],
            "temperature": 0.2,
            "return_citations": True
        }
        resp = requests.post(
            self.pplx_url,
            headers=headers,
            json=payload,
            timeout=30
        )
        resp.raise_for_status()
        content = resp.json()["choices"][0]["message"]["content"]
        try:
            return json.loads(content)
        except json.JSONDecodeError:
            return {"sentiment": "neutral", "confidence": 0, "direction": "hold"}

    def get_wallet_balance(self) -> float:
        """Fetch USDC balance from Purple Flea wallet."""
        resp = requests.get(
            f"{self.pf_url}/wallet/balance",
            headers={"X-API-Key": PF_API_KEY}
        )
        return resp.json()["usdc_balance"]

    def execute_trade(self, market: str, direction: str,
                        size_usd: float, leverage: int = 2) -> dict:
        """Open a perpetual position on Purple Flea."""
        payload = {
            "market":    market,      # e.g. "BTC-PERP"
            "side":      direction,  # "long" or "short"
            "size_usd":  size_usd,
            "leverage":  leverage,
            "order_type": "market"
        }
        resp = requests.post(
            f"{self.pf_url}/trading/perp/open",
            headers={"X-API-Key": PF_API_KEY},
            json=payload
        )
        return resp.json()

    def run_cycle(self, asset: str = "BTC"):
        """Full research-to-trade cycle."""
        print(f"[+] Researching {asset} via Perplexity...")
        signal = self.research_market(asset)
        print(f"    Sentiment: {signal['sentiment']} (confidence: {signal['confidence']}%)")

        if signal["confidence"] < 65 or signal["direction"] == "hold":
            print("    Low confidence — no trade.")
            return

        balance   = self.get_wallet_balance()
        trade_pct = 0.10 if signal["confidence"] < 80 else 0.20
        size      = round(balance * trade_pct, 2)

        print(f"[+] Opening {signal['direction']} {asset} @ ${size} (2x leverage)...")
        result = self.execute_trade(
            market    = f"{asset}-PERP",
            direction = signal["direction"],
            size_usd  = size
        )
        print(f"[+] Trade placed: {result}")


if __name__ == "__main__":
    agent = PerplexityTradingAgent()
    agent.run_cycle("BTC")
    agent.run_cycle("ETH")