FilesExpand file tree

main

/

structured_outputs_description.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

55 lines (45 loc) · 1.59 KB

main

/

structured_outputs_description.py

Copy path

Top

File metadata and controls

• Code
• Blame

55 lines (45 loc) · 1.59 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

import os

import azure.identity

import openai

import rich

from dotenv import load_dotenv

from pydantic import BaseModel, Field

# Setup the OpenAI client to use either Azure, OpenAI.com, or Ollama API

load_dotenv(override=True)

API_HOST = os.getenv("API_HOST", "azure")

if API_HOST == "azure":

token_provider = azure.identity.get_bearer_token_provider(

azure.identity.DefaultAzureCredential(), "https://cognitiveservices.azure.com/.default"

)

client = openai.OpenAI(

base_url=f"{os.environ['AZURE_OPENAI_ENDPOINT'].rstrip('/')}/openai/v1/",

api_key=token_provider,

)

MODEL_NAME = os.environ["AZURE_OPENAI_CHAT_DEPLOYMENT"]

elif API_HOST == "ollama":

client = openai.OpenAI(base_url=os.environ["OLLAMA_ENDPOINT"], api_key="nokeyneeded")

MODEL_NAME = os.environ["OLLAMA_MODEL"]

else:

client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])

MODEL_NAME = os.environ["OPENAI_MODEL"]

class CalendarEvent(BaseModel):

name: str

date: str = Field(..., description="A date in the format YYYY-MM-DD")

participants: list[str]

completion = client.responses.parse(

model=MODEL_NAME,

input=[

{

"role": "system",

"content": "Extract the event information. If no year is specified, assume the current year (2026).",

},

{"role": "user", "content": "Alice and Bob are going to a science fair on the 1st of april."},

],

text_format=CalendarEvent,

store=False,

)

if completion.output_parsed:

event = completion.output_parsed

rich.print(event)

else:

rich.print(completion.output_text)