AutoFNS is a utility for automatically calling functions when interacting with the OpenAI Completions.
Project description
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=[...])
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
autofns-0.1.6.tar.gz
(2.9 kB
view hashes)