pydantic-openai

Pydantic models for OpenAI's API. These make it easier to construct requests and parse responses, as well as set up API-compatible servers.

Making requests is explicitly out of scope.

This is mostly translated from sashabaranov/go-openai, by ChatGPT. I welcome translating the rest over, I just don't really use the other API's.

Usage

See the package on PyPI.

pip install pydantic-openai

import openai
from pydantic_openai import ChatCompletionRequest, GPT3Models, ChatCompletionMessage

req = ChatCompletionRequest(
    model=GPT3Models.GPT3Dot5Turbo,
    messages=[
        ChatCompletionMessage(role="user", content="What's the capital of France?"),
    ],
    max_tokens=128,
)
resp = openai.ChatCompletion.create(**req.dict(exclude_none=True))
print(resp)