"""
=========================================================
CV6 AI Trading OS
Claude AI Test API
=========================================================
"""

from fastapi import APIRouter

from app.ai.ai_models import AIProvider, AIRequest
from app.ai.providers.claude_provider import ClaudeProvider

router = APIRouter(
    prefix="/ai",
    tags=["AI Test"],
)


@router.post("/test/claude")
async def test_claude():

    provider = ClaudeProvider()

    request = AIRequest(
        provider=AIProvider.CLAUDE,
        symbol="NIFTY",
        timeframe="5m",
        indicators={
            "ema": "BUY",
            "rsi": "BUY",
            "macd": "BUY",
            "supertrend": "BUY",
            "vwap": "BUY",
        },
    )

    response = await provider.generate(request)

    return response