"""
title: WolframAlpha LLM API Integration
author: Morgan Blangeois
version: 1.0
license: MIT
description: A tool that integrates WolframAlpha LLM API.
requirements: requests
"""

import os
import requests
import urllib.parse
from typing import Optional, Dict, Any
from pydantic import BaseModel, Field


class Tools:
    class Valves(BaseModel):
        WOLFRAM_APP_ID: str = Field(
            default="", description="The App ID for WolframAlpha API"
        )
        API_BASE_URL: str = Field(
            default="https://www.wolframalpha.com/api/v1/llm-api",
            description="The base URL for WolframAlpha API",
        )
        MAX_CHARS: int = Field(
            default=6800,
            description="The default maximum number of characters in the response",
        )

    def __init__(self):
        self.valves = self.Valves()
        self.app_id = os.getenv("WOLFRAM_APP_ID", self.valves.WOLFRAM_APP_ID)
        self.api_base_url = self.valves.API_BASE_URL
        self.max_chars = self.valves.MAX_CHARS

    def perform_query(
        self, query: str, max_chars: Optional[int] = None, **kwargs
    ) -> Dict[str, Any]:
        """
        Perform a query to WolframAlpha LLM API.
        :param query: The query to send to WolframAlpha.
        :param max_chars: Maximum number of characters in the response (optional).
        :param kwargs: Additional parameters to include in the API request.
        :return: A dictionary containing the response data.
        """
        try:
            params = {
                "input": query,
                "appid": self.app_id,
                "maxchars": max_chars or self.max_chars,
                **kwargs,
            }

            encoded_params = urllib.parse.urlencode(params)
            url = f"{self.api_base_url}?{encoded_params}"

            headers = {
                "User-Agent": "WolframAlphaAPI/1.1",
                "Authorization": f"Bearer {self.app_id}",
            }

            response = requests.get(url, headers=headers, timeout=120)
            response.raise_for_status()

            return {"status": "success", "data": response.text, "url": response.url}
        except requests.exceptions.RequestException as e:
            return {
                "status": "error",
                "message": str(e),
                "url": url if "url" in locals() else None,
            }