shortener_tool.py

import os
import requests
from crewai.tools import BaseTool

class ShortenerTool(BaseTool):
    name: str = "URL Shortener Tool"
    description: str = "Generates a short version of a given URL using an external API."

    def _run(self, original_url: str) -> str:
        api_url = "https://sales-mgmt-cb-bff-apigw.prd.naturacloud.com/cb-bff-cms/cms/shortener"
        headers = {
            "accept": "*/*",
            "accept-language": "pt-BR,pt;q=0.9,en-US;q=0.8,en;q=0.7",
            "authorization": f"Bearer {os.getenv('NATURA_API_TOKEN')}", # Get token from env
            "content-type": "application/json",
            "origin": "https://minhaloja.natura.com",
            "referer": "https://minhaloja.natura.com/",
            "sec-ch-ua": '"Not)A;Brand";v="8", "Chromium";v="138", "Google Chrome";v="138"' , # Escaped quotes
            "sec-ch-ua-mobile": "?0",
            "sec-ch-ua-platform": '"Windows"' , # Escaped quotes
            "sec-fetch-dest": "empty",
            "sec-fetch-mode": "cors",
            "sec-fetch-site": "cross-site",
            "siteid": "CBBrazil",
            "tenantid": "brazil-natura",
            "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36",
        }
        payload = {"url": original_url}

        try:
            response = requests.post(api_url, headers=headers, json=payload)
            response.raise_for_status()  # Raise an exception for HTTP errors
            short_url_data = response.json()
            return short_url_data.get("short", original_url) # Return original_url if short not found
        except requests.exceptions.RequestException as e:
            print(f"Warning: Error generating short URL for {original_url}: {e}. Returning original URL.")
            return original_url
        except ValueError:
            print(f"Warning: Invalid JSON response from shortener API for {original_url}. Returning original URL.")
            return original_url