AutoFNS

Introduction

AutoFNS is a utility for automatically calling functions when interacting with OpenAI Completions API.

Note: This is a work in progress and may not be ready for production use. If you find any bugs or have any suggestions, please open an issue or a pull request.

Requirements

• python >= 3.10
• openai
• regex

Installation

pip install autofns

Usage

from autofns import AutoFNS

FUNCTION_DEFINITIONS = [
    {
        "type": "function",
        "function": {
            "name": "get_temp_units",
            "description": "Get a list of temperature units",
        }
    },
    {
        "type": "function",
        "function": {
            "name": "get_current_temperature",
            "description": "Get the current temperature in a city",
        },
        "parameters": {
            "type": "object",
            "properties": {
                "city": {
                    "type": "string",
                    "description": "The city to get the temperature of",
                },
                "unit": {
                    "type": "string",
                    "description": "The unit to get the temperature in",
                    "enum": ["Fahrenheit", "Celsius", "Kelvin"],
                }
            },
            "required": ["city", "unit"],
        }
    },
]


def get_temp_units():
    return ["Fahrenheit", "Celsius", "Kelvin"]


def get_current_temperature(city: str, unit: str):
    return "The current temperature in {} is {} degrees {}".format(
        city, 72, unit
    )


fns = AutoFNS(
    "gpt-4-32k",
    fns_mapping={
        "get_temp_units": get_temp_units,
        "get_current_temperature": get_current_temperature
    },
    fns_definitions=FUNCTION_DEFINITIONS,
)

result = fns.create_completion(messages=[...])

You can also use the @fns.map_function decorator to add a function to AutoFNS internal mapping:

from autofns import AutoFNS

fns = AutoFNS(...)


@fns.map_function  # can also pass a name to map_function
def get_temp_units():
    return ["Fahrenheit", "Celsius", "Kelvin"]

You can also use the AutoFNSAsync class to use async functions:

from autofns import AutoFNSAsync

fns = AutoFNSAsync(...)

result = await fns.create_completion(messages=[...])