structured-outputs-anonLLM.py

from pydantic import BaseModel
from anonLLM.llm import OpenaiLanguageModel
from dotenv import load_dotenv

load_dotenv()

llm = OpenaiLanguageModel(anonymize=False, temperature=1)


class Person(BaseModel):
    name: str
    sex: str
    age: int
    email: str


response = llm.generate(
    prompt="Generate a person",
    output_format=Person
)

print(response)

# Returns: {'name': 'Alex', 'sex': 'Male', 'age': 32, 'email': '[email protected]'}