Agentic-Investor

"""Pydantic models for the Google Trends tool."""

from typing import List
from pydantic import BaseModel, Field, ConfigDict

from agentic_investor.interfaces.tool import BaseToolInput


class GoogleTrendsInput(BaseToolInput):
    """Input schema for Google Trends tool."""

    model_config = ConfigDict(
        json_schema_extra={
            "examples": [{"keywords": ["bitcoin", "ethereum"], "period_days": 7}]
        }
    )

    keywords: List[str] = Field(
        description="List of keywords to search for", examples=[["bitcoin", "ethereum"]]
    )
    period_days: int = Field(
        default=7, description="Number of days to look back", examples=[7, 30, 90]
    )


class GoogleTrendsOutput(BaseModel):
    """Output schema for Google Trends tool."""

    model_config = ConfigDict(
        json_schema_extra={
            "examples": [{"trends_data": "date,bitcoin,ethereum\n2024-01-01,50,45\n"}]
        }
    )

    trends_data: str = Field(
        description="CSV formatted trends data with dates and search interest values"
    )