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 AI to format text according to a specific structure or pattern.
 
    Args:
        text (str or list): The text to format (string or 2D list with a single cell)
        format_instruction (str): 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
        api_key (str, optional): API key for authentication (e.g. for Mistral AI)
        api_url (str, optional): OpenAI compatible URL. (e.g., https://api.mistral.ai/v1/chat/completions for Mistral AI)
 
    Returns:
        str: The formatted text according to the specified format
    """
    # 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):
        raise ValueError("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):
        raise ValueError("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": temperature,
        "model": model,
        "max_tokens": max_tokens
    }
 
    headers = {
        "Authorization": f"Bearer {api_key}",
        "Content-Type": "application/json",
        "Accept": "application/json"
    }
 
    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()