weave_audio.py

import base64
from openai import OpenAI
import weave
import wave


client = OpenAI()
weave.init("audio-in-weave")

@weave.op()
def make_audio_file():
    completion = client.chat.completions.create(
        model="gpt-4o-audio-preview",
        modalities=["text", "audio"],
        audio={"voice": "alloy", "format": "wav"},
        messages=[
            {
                "role": "user",
                "content": "Is a golden retriever a good family dog?"
            }
        ]
    )

    print(completion.choices[0])

    wav_bytes = base64.b64decode(completion.choices[0].message.audio.data)
    with open("dog.wav", "wb") as f:
        f.write(wav_bytes)

    return wave.open("dog.wav", "rb")