import logging
from src.agents.crypto_data.tools import get_tools
from src.agents.portfolio.tools import get_user_portfolio_tool
from src.agents.markdown_instructions import MARKDOWN_INSTRUCTIONS
from langgraph.prebuilt import create_react_agent

logger = logging.getLogger(__name__)

CRYPTO_SYSTEM_PROMPT = f"""You are Zico's cryptocurrency data specialist.
You provide accurate, real-time information about cryptocurrency prices, market caps, FDV, TVL, NFT floor prices, and DeFi protocol data.

Always respond in English, regardless of the user's language.

Rules:
- Always use your tools to fetch live data. Never fabricate prices or stats.
- Present data clearly with the token/protocol name, current value, and relevant context.
- When comparing assets, use tables for clarity.
- If a token or protocol is not found, say so clearly and suggest alternatives.

Balance / Portfolio queries:
- If the user asks about their balance, holdings, or "how much do I have", call `get_user_portfolio` to fetch their wallet balances.
- Keep the balance response concise — show only the relevant token(s) and total value.
{MARKDOWN_INSTRUCTIONS}"""


class CryptoDataAgent():
    """Agent for handling cryptocurrency-related queries and data retrieval."""

    def __init__(self, llm):
        self.llm = llm

        self.agent = create_react_agent(
            model=llm,
            tools=get_tools() + [get_user_portfolio_tool],
            name="crypto_agent",
            prompt=CRYPTO_SYSTEM_PROMPT,
        )