Skip to main content

A package for simplified interaction with OpenAI's GPT-3 and GPT-4 models.

Project description

🚀 GPTBase

Welcome to GPTBase! This Python library provides a straightforward and user-friendly interface to leverage OpenAI's powerful GPT-3.5-Turbo and GPT-4 models for text generation. 📚🔍

✨ Features

  • 🎛️ Configurable parameters for chat completion.
  • 💬 Support for both individual prompts and chat conversations.
  • ⚡ Stream output option for faster response times.
  • 📝 Built-in logging and rich printing for enhanced visualization.
  • 🔄 Customizable model choice, including GPT-3.5-Turbo, GPT-4, and other variants.
  • 📈 Easy model parameter adjustment.
  • 🤖 Support for function calls in chat completions.

📦 Installation

You can easily install the GPTBase library from PyPI with a single command line:

pip install gptbase

Once installed, import it in your Python script:

from gptbase.basev2 import ChatAssistant, CompletionParameters, FunctionCall

📚 Usage

The library includes two powerful classes: Assistant for individual prompts and ChatAssistant for engaging, multi-turn chat conversations.

Here's a snippet showcasing a simple usage of the ChatAssistant class:

from gptbase.basev2 import ChatAssistant, CompletionParameters

cm = CompletionParameters(stream=True)
assistant = ChatAssistant(memory_turns=3)
message = assistant.chat("I want to learn Python", cm)

And here's an example of how to use function calls:

from gptbase.basev2 import Assistant, CompletionParameters, FunctionCall

# Define the function
get_weather_function = FunctionCall("get_current_weather", {"location": "Boston"})

# Define the functions that the model can call
functions = [
    {
        "name": "get_current_weather",
        "description": "Gets the current weather for a given location",
        "parameters": {
            "type": "object",
            "properties": {
                "location": {
                    "type": "string",
                    "description": "The location to get the weather for"
                }
            },
            "required": ["location"]
        }
    }
]

# Define the parameters for the chat completion
params = CompletionParameters(
    model="gpt-3.5-turbo-0613",
    function_call=get_weather_function,
    functions=functions,
    # Add other parameters as needed
)

# Create an instance of the Assistant class
assistant = Assistant()

# Call the function
response = assistant.ask("What's the weather like in Boston?", params=params)

# Print the response
print(response)
extract_function_call(response)

def get_current_weather(location):
    """fake get current weather function."""
    print(f"Today's fake weather for {location} is _°")

func_dic = {get_current_weather.__name__: get_current_weather}
call_with_chat_completion(response, func_dic)

And you will get output like this:

Today's fake weather for Boston is _°

You can also use the command line interface for chat conversations:

gptbase-chat --memory-turns 3 --system-prompt "This is a system prompt."

For more examples and a deeper dive into usage, please refer to the detailed documentation. 📘

🙋 Support

If you have any questions, run into any issues, or just need a little help, don't hesitate to reach out. We're here to help! 🤝


Whether you're a veteran developer or just getting started, GPTBase is a powerful tool for any text generation task. Give it a try today! 🌟

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

gptbase-0.3.1.tar.gz (12.2 kB view hashes)

Uploaded Source

Built Distribution

gptbase-0.3.1-py3-none-any.whl (11.9 kB view hashes)

Uploaded Python 3

Supported by

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