Skip to main content

A simple package for the purpose of providing a set of utilities that make it easier to invoke python functions and coroutines using OpenAI's GPT models.

Project description

GPT Function Calling Utility

The GPT Function Calling Utility is a Python package designed to streamline the process of calling Python methods using OpenAI's Function Calling API, without wrapping around the OpenAI library.

Please note that GPT Function Calling Utility does not directly make calls to OpenAI's API, but rather helps with function modeling and invocation when given a function_call field.

It also works with Ollama's Python Library.

Installation

python -m pip install -U gptfunctionutil

Key Features

  • Simplified Function Modeling Via Inheritance: This package utilizes subclasses decended from the GPTFunctionLibrary class that allows you to define sets of callable methods to be sent to OpenAI's Chat Completion endpoint. GPTFunctionLibrary contains methods to create a json schema describing your functions, and to invoke said functions using the Function Call field returned with a call to chat/completions.

  • Decorate Invokable Functions, and Coroutines: OpenAI's Function Calling Feature needs JSON schema to outline the name, description, and parameters of each invokable function. This utility uses two decorators, (@AILibFunction) and (@LibParam), as well as type annotation to create this schema for functions you want to use with the API.

    • set a display name and description with (@AILibFunction). You can also specify required parameters with this decorator, but it's not required.
    • apply small descriptions to arguments with (@LibParam), to inform the API on what it does.
    • (@LibParamSpec) can apply additional keywords depending on the type. (see https://json-schema.org/understanding-json-schema/index.html for details.)
    • You can decorate coroutines as well.
  • Parameter Typing and Descriptions: To ensure clarity and facilitate proper function formatting, GPT Function Calling Utility requires that all parameters intended to be passed into the AI have an applied type; strings, integers, floats, and bools. The library utilizes a collection of converter objects to generate schema for each parameter based on type annotations.

  • Convert into complex types: The utility is capable of converting some more complex data types into a json schema, such as datetimes and Literals.

    • Define Custom Converters to automatically use response arguments to initalize objects.
      • (see examples/custom_converters.py for an example.)
  • Schema Generation for API Calls: Before making a call to the OpenAI chat/completion endpoint, the utility has a get_schema() method to extract the formatted functions as a list of dictionaries. This schema is then passed as the functions field in the ChatCompletion call. If the AI determines that it should invoke a function call, you can pass the returned function_call field into the call_by_dict(function_call)(or call_by_dict_async if you're trying to call a decorated coroutine) method to call the corresponding function with the provided arguments.

    • The method also checks if there is a function by that name, falling back to a default response if something goes wrong.
    • Schema can also validate and convert responses returned from the chat/completions endpoint.
  • Integration with Discord.py: This utility was intended to be used with life as a discord.py utility, and can be easily integrated with discord.py bots.

    • Simply import gptfunctionutil into your Discord bot project, and decorate your commands with @LibParam and @AILibFunction. After passing the Commands into a GPTFunctionLibrary subclass with add_in_commands(your_bot_object_here), your bot commands will become invokable in the same way as a decorated GPTFunctionLibrary coroutine, provided you use the call_by_dict_ctx method.
      • (see examples/discord_bot.py for an example.)

Usage Example

Using GPT Function Calling Utility with OpenAI to get the current time:

from gptfunctionutil import GPTFunctionLibrary, AILibFunction, LibParam, LibParamSpec
from datetime import datetime
import openai
class MyLib(GPTFunctionLibrary):
    #Define methods here.
    @AILibFunction(name='get_time', description='Get the current time and day in UTC.')
    @LibParam(comment='An interesting, amusing remark.')
    def get_time(self, comment: str):
        # get the current time, with a small remark.
        return f"{comment}\n{str(datetime.now())}"

# Initialize the subclass somewhere in your code
mylib = MyLib()
client = openai.Client()

completion = client.chat.completions.create(
    model="gpt-4o-mini",
    messages=[
        {"role": "system", "content": "You are a helpful assistant."},
        #This message will invoke get_time
        {"role": "user", "content": "Hello, get me the current time in UTC."}
    ],
    "tools"=: mylib.get_tool_schema(),
    "tool_choice"= 'auto',
)
message=completion.choices[0]['message']
if message.tool_calls:
    for tool in message.tool_calls:
        output = mylib.call_by_tool(tool)
        print(tool.name,output)
else:
    print(completion.choices[0].message.content)

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

gptfunctionutil-0.3.8.0.tar.gz (30.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

gptfunctionutil-0.3.8.0-py3-none-any.whl (25.1 kB view details)

Uploaded Python 3

File details

Details for the file gptfunctionutil-0.3.8.0.tar.gz.

File metadata

  • Download URL: gptfunctionutil-0.3.8.0.tar.gz
  • Upload date:
  • Size: 30.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.9

File hashes

Hashes for gptfunctionutil-0.3.8.0.tar.gz
Algorithm Hash digest
SHA256 fad99e7f5ca8aff9adab34b22eae876e4527c4281d9f056040c4752f02701418
MD5 943316168c156fbb7e0f0f29c6fa60e9
BLAKE2b-256 9216b0e08dd16a4b9ded2b069358e80552c6bec878a0f03d65c6a84a225125dd

See more details on using hashes here.

File details

Details for the file gptfunctionutil-0.3.8.0-py3-none-any.whl.

File metadata

File hashes

Hashes for gptfunctionutil-0.3.8.0-py3-none-any.whl
Algorithm Hash digest
SHA256 90551b622c1f4c27c957e2d166d2098cb9f46348cd95eed84380d2b67b224e37
MD5 bd171f6f825a9b48347149d8cc3cacc4
BLAKE2b-256 a0e958c59c06311677ccbd56e66e8a2d577e04452cf3ee09e8002d1840b3a4a9

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page