AI_FORMAT

=AI_FORMAT(text, format_instruction, [temperature], [max_tokens], [model], [api_key], [api_url])

=AI_FORMAT("John Smith / Marketing Director - Acme Inc / [email protected] - 555.123.4567", "standard business contact card format")

=AI_FORMAT("Revenue: 2.4m; COGS: 1.1m; Gross Margin: 1.3m (54%); Opex: 950k; EBITDA: 350k", "professional financial statement format with proper currency notation")

=AI_FORMAT("Used the product for 3 weeks. Good quality but shipping took forever. Customer service was helpful though. Probably would buy again if they fix delivery issues.", "structured product review with ratings")

=AI_FORMAT("123 business park dr suite 12, austin tx 78701", "standard US business address format")

=AI_FORMAT("Tom said we need to finish the report by Friday. Maria will contact the client about budget concerns. Everyone should review the new marketing strategy before next meeting on 5/10.", "action items with responsible parties and deadlines")

import requests
import json
 
def ai_format(text, format_instruction, temperature=0.0, max_tokens=1500, model="mistral-small-latest", api_key=None, api_url="https://api.mistral.ai/v1/chat/completions"):
    """
    Uses an AI model to format text according to a specific structure or pattern.
 
    Args:
        text: str or list[list[str]], required. The text to format.
        format_instruction: str, required. Instructions describing the desired format.
        temperature: float, optional. Controls response creativity (0-2). Default is 0.0.
        max_tokens: int, optional. Maximum tokens for response generation. Default is 1500.
        model: str, optional. ID of the model to use. Default is "mistral-small-latest".
        api_key: str, optional. API key for authentication.
        api_url: str, optional. OpenAI-compatible URL. Default is "https://api.mistral.ai/v1/chat/completions".
 
    Returns:
        str: The formatted text according to the specified format, or an error message on failure.
 
    This example function is provided as-is without any representation of accuracy.
    """
    # Demo fallback: Boardflare endpoint if no api_key or api_url
    if api_key is None or api_url is None:
        if "idToken" in globals():
            api_key = globals()["idToken"]
            api_url = "https://llm.boardflare.com"
        else:
            return ("Login on the Functions tab for limited demo usage, or sign up for a free Mistral AI account at https://console.mistral.ai/ and add your own api_key.")
 
    # Handle 2D list input (flatten to a single string)
    if isinstance(text, list):
        if len(text) > 0 and len(text[0]) > 0:
            text = str(text[0][0])
        else:
            return "Error: Empty input text."
 
    # Validate temperature
    if not isinstance(temperature, (float, int)) or not (0 <= float(temperature) <= 2):
        return "Error: temperature must be a float between 0 and 2 (inclusive)."
    # Validate max_tokens
    if not isinstance(max_tokens, int) or not (5 <= max_tokens <= 5000):
        return "Error: max_tokens must be an integer between 5 and 5000 (inclusive)."
 
    # Construct a specific prompt for formatting
    format_prompt = f"Format the following text according to this format instruction: {format_instruction}\n\nText to format: {text}"
    format_prompt += "\n\nReturn ONLY the formatted text. Do not include any explanatory text, just the formatted text."
 
    # Prepare the API request payload
    payload = {
        "messages": [{"role": "user", "content": format_prompt}],
        "temperature": float(temperature),
        "model": model,
        "max_tokens": max_tokens
    }
 
    headers = {
        "Authorization": f"Bearer {api_key}",
        "Content-Type": "application/json",
        "Accept": "application/json"
    }
 
    try:
        response = requests.post(api_url, headers=headers, json=payload)
        if response.status_code == 429:
            return "You have hit the rate limit for the API. Please try again later."
        response.raise_for_status()
        response_data = response.json()
        content = response_data["choices"][0]["message"]["content"]
        return content.strip()
    except Exception as e:
        return f"Error: {str(e)}"