Generates a text completion using the provider's API, with a given system and user prompt. This method should be decorated with retry

Categorizes the input data using the specified LLM provider. Args: input_data (str): The text data to be categorized.

Categorizes the input data into multiple categories using the specified LLM provider. Args: input_data (str): The text d

Apply a function to each batch of values in a column of a DataFrame or a normal Python list, starting from a specified index. Parameters:

Apply a function to each value in a column of a DataFrame or a normal Python list, starting from a specified index. Parameters: inpu

Categorizes the input data into multiple categories using the specified LLM provider. Args: input_data (str): The text d

Categorizes a batch of input data using the specified LLM provider. For less advanced LLMs, call this method on batches of 3-5 inputs (depend

Transforms the input data using the specified LLM provider. Args: input_data (str): The text data to be transformed.

Initializes the LLMProvider with an API key and retry parameters. Parameters: model (str): The default model to use for tex

Initializes the OllamaProvider with an optional Ollama client or host, and retry parameters. Parameters: model (str): The de

Initializes the ClaudeProvider with an API key and retry parameters. Parameters: api_key (str): Anthropic API key. m

Initializes the OpenAIProvider with an API key and retry parameters. Parameters: api_key (str): OpenAI API key. mult

Apply a function to the input column using adaptive batch processing. This function applies a batch processing function to the input column,

Generates a text completion using Ollama's API, with a given system and user prompt. This method is decorated with retry logic to han

Generates a text completion using Anthropic's Claude API, with a given system and user prompt. This method is decorated with retry lo

Generates a text completion using OpenAI's API, with a given system and user prompt. This method is decorated with retry logic to han

Decorator to apply retry logic with exponential backoff to an instance method. It captures the 'self' context to access instance attr

Decorator to apply retry logic with exponential backoff to an instance method. It captures the 'self' context to access instance attr

Test the apply_to_column function.

Test the apply_to_column_batch function.

Test the apply_to_column_batch function with a start index.

Test the BaseCategorizer class.

Test the BaseCategorizer class with a batch of examples.

Test the MultiCategorizer class.

Test the MultiCategorizer class with a batch of examples.

Transforms the input data into a list of dictionaries using the specified LLM provider. Args: input_data (str): The text

Transforms a batch of input data using the specified LLM provider. Args: input_data (List[str]): A list of text data to

Validates the categories dictionary. Args: v (Dict[str, str]): The categories dictionary to be validated. Raise

Validates the examples list. Args: v (List[Dict[str, str]]): The examples list to be validated. Raises:

Validates the examples list. Args: v (List[Dict[str, str]]): The examples list to be validated. Raises:

Validates the examples list. Args: v (List[Dict[str, str]]): The examples list to be validated. Raises:

Validates that the "response" values in the examples are within the categories keys. If there are multiple categories, they should be separat

Validates that the "response" values in the examples are within the categories keys.

Validates that the "response" values in the examples are valid JSON-formatted lists of dictionaries that match the output schema.

Validates the prompt. Args: v (str): The prompt to be validated. Raises: ValueError: If the prompt

Validates the output schema. Args: v (Dict[str, str]): The output schema to be validated. Raises: V