"""
AI endpoints interaction.
"""
from typing import Any, Dict
from httpx import AsyncClient

from ..utils import with_retry

class AIResource:
    """Resource defining AI generative operations."""
    def __init__(self, client: AsyncClient):
        self._client = client

    async def _handle_request(self, method: str, path: str, **kwargs) -> Any:
        async def _req():
             resp = await self._client.request(method, path, **kwargs)
             resp.raise_for_status()
             return resp.json()
        return await with_retry(_req)

    async def summarize_url(self, url: str, language: str = "français") -> Dict[str, Any]:
        return await self._handle_request("POST", "/api/v1/ai/summarize", json={"url": url, "language": language})
    
    async def draft_task(self, description: str, context: str = "") -> Dict[str, Any]:
        return await self._handle_request("POST", "/api/v1/ai/draft-task", json={"description": description, "context": context})