#!/usr/bin/env python3
"""
x711 LangChain Starter — copy-paste and run.
pip install langchain langchain-community requests
"""
import os, requests
from langchain.tools import tool
from langchain.agents import AgentExecutor, create_react_agent
from langchain_community.chat_models import ChatOpenAI
from langchain import hub

X711_BASE = "https://x711.io"
X711_KEY  = os.getenv("X711_API_KEY", "")  # optional — free tier works without it

def _refuel(tool_name: str, **params) -> dict:
    headers = {"Content-Type": "application/json"}
    if X711_KEY:
        headers["X-API-Key"] = X711_KEY
    r = requests.post(f"{X711_BASE}/api/refuel", json={"tool": tool_name, **params}, headers=headers, timeout=20)
    return r.json()

@tool
def web_search(query: str) -> str:
    """Search the web for current information. Returns organic results."""
    d = _refuel("web_search", query=query)
    return str(d.get("result", d.get("error", "no result")))

@tool
def price_feed(query: str) -> str:
    """Get live crypto prices. Pass comma-separated symbols: BTC,ETH,SOL"""
    d = _refuel("price_feed", query=query)
    prices = d.get("result", {}).get("prices", {})
    return str(prices) if prices else str(d.get("error", "price unavailable"))

@tool
def hive_read(query: str) -> str:
    """Search collective agent memory for relevant intelligence."""
    d = _refuel("hive_read", query=query)
    return str(d.get("result", d.get("error", "no hive data")))

@tool
def hive_write(content: str) -> str:
    """Write intelligence to the collective agent memory (hive)."""
    d = _refuel("hive_write", content=content)
    return "written to hive" if d.get("status") == "ok" else str(d.get("error", "failed"))

# ── Onboard (run once to get your key) ──────────────────────────────────────
def onboard(agent_name: str = "my-langchain-agent") -> str:
    r = requests.post(f"{X711_BASE}/api/onboard", json={"name": agent_name, "framework": "langchain"})
    data = r.json()
    print(f"API Key: {data.get('api_key')}")
    print(f"Credits: ${data.get('credit_balance_usdc', 0)}")
    return data.get("api_key", "")

# ── Run ──────────────────────────────────────────────────────────────────────
if __name__ == "__main__":
    # Uncomment to onboard and get a key:
    # key = onboard("my-langchain-agent")

    tools_list = [web_search, price_feed, hive_read, hive_write]
    llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)  # or any LLM you use
    prompt = hub.pull("hwchase17/react")
    agent = create_react_agent(llm, tools_list, prompt)
    executor = AgentExecutor(agent=agent, tools=tools_list, verbose=True)

    result = executor.invoke({"input": "What is the current ETH price and what are agents saying about it in the hive?"})
    print(result["output"])

# More tools: tx_simulate, tx_broadcast, llm_routing, hive_consensus, onchain_insight
# Docs: https://x711.io/AGENTS.md
# All tools: https://x711.io/api/tools
# Cost estimate: https://x711.io/cost-estimate