"""
Configuration settings for the workflows package.

This module contains configuration settings and constants used across the workflows package,
including model configurations, workflow settings, and other package-wide constants.
"""

AVAILABLE_MODELS = {
    "OpenAI/gpt-4o": {
        "model": "gpt-4o-2024-11-20",
        "logprobs": True,
    },
    "OpenAI/gpt-4o-mini": {
        "model": "gpt-4o-mini-2024-07-18",
        "logprobs": True,
    },
    "OpenAI/gpt-3.5-turbo": {
        "model": "gpt-3.5-turbo-0125",
    },
    "Anthropic/claude-3-7-sonnet": {
        "model": "claude-3-7-sonnet-20250219",
    },
    "Anthropic/claude-3-5-sonnet": {
        "model": "claude-3-5-sonnet-20241022",
    },
    "Anthropic/claude-3-5-haiku": {
        "model": "claude-3-5-haiku-20241022",
    },
    "Cohere/command-r": {
        "model": "command-r-08-2024",
        "logprobs": True,
    },
    "Cohere/command-r-plus": {
        "model": "command-r-plus-08-2024",
        "logprobs": True,
    },
    "Cohere/command-r7b": {
        "model": "command-r7b-12-2024",
        "logprobs": False,
    },
}

# Function mapping for input/output transformations
TYPE_MAP = {
    "str": str,
    "int": int,
    "float": float,
    "bool": bool,
}

FUNCTION_MAP = {
    "upper": str.upper,
    "lower": str.lower,
    "len": len,
    "split": str.split,
}