AI_LIST

=AI_LIST(prompt, [values], [temperature], [max_tokens], [model], [api_key], [api_url])

=AI_LIST("List essential marketing KPIs for quarterly performance reviews")

=AI_LIST("List effective risk mitigation strategies for enterprise software implementation")

=AI_LIST("List key compliance requirements for healthcare organizations")

=AI_LIST("List priority action items based on these quarterly business review notes:", A1:A3)

=AI_LIST("List 5 SMART goals for an HR department focused on improving employee retention")

import requests
import json
 
def ai_list(prompt, values=None, temperature=0.0, max_tokens=1000, model='mistral-small-latest', api_key=None, api_url="https://api.mistral.ai/v1/chat/completions"):
    """
    Uses AI to generate a list of items based on the prompt and optional values data.
 
    Args:
        prompt (str): Instruction for AI to create a list
        values (list, optional): 2D list containing additional data to append to prompt
        temperature (float, optional): Controls response creativity (0-2). Default is 0.0
        max_tokens (int, optional): Maximum tokens for response generation. Default is 1000
        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:
        list: 2D list representing the generated list data as a single column
    """
    # Demo fallback for Boardflare 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."]]
 
    # 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 list generation
    list_prompt = f"Generate a list based on this request: {prompt}"
    if values is not None:
        values_str = "\n".join([str(item[0]) for item in values]) if len(values) > 0 and len(values[0]) > 0 else ""
        if values_str:
            list_prompt += f"\n\nUse this information to help create the list:\n{values_str}"
    list_prompt += "\nReturn ONLY a JSON object with a key 'items' whose value is a JSON array of the items for the list. "
    list_prompt += "Each item should be a single value. "
    list_prompt += "Do not include any explanatory text, just the JSON object. "
    list_prompt += 'For example: {"items": ["item1", "item2", "item3"]}'
 
    payload = {
        "messages": [{"role": "user", "content": list_prompt}],
        "temperature": temperature,
        "model": model,
        "max_tokens": max_tokens,
        "response_format": {"type": "json_object"}
    }
    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"]
        try:
            list_data = json.loads(content)
            if isinstance(list_data, dict) and "items" in list_data:
                list_data = list_data["items"]
            elif isinstance(list_data, dict):
                for key, value in list_data.items():
                    if isinstance(value, list):
                        list_data = value
                        break
            if isinstance(list_data, list):
                result = []
                for item in list_data:
                    if isinstance(item, list):
                        if len(item) >= 1:
                            result.append([str(item[0])])
                        else:
                            result.append([""])
                    else:
                        result.append([str(item)])
                return result
            else:
                return [["Error: Unable to parse response. Expected a list."]]
        except (json.JSONDecodeError, ValueError):
            return [["Error: Unable to generate list. The AI response wasn't in the expected format."]]
    except requests.exceptions.RequestException as e:
        return [["Error: API request failed.", str(e)]]