Skip to content

Commit

Permalink
OpenAI models Nov 20
Browse files Browse the repository at this point in the history
  • Loading branch information
MadcowD committed Nov 20, 2024
1 parent d66f4c7 commit 6b4e4ab
Showing 1 changed file with 10 additions and 2 deletions.
12 changes: 10 additions & 2 deletions src/ell/models/openai.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,6 +59,14 @@ def register(client: openai.Client):
'gpt-4o-mini-2024-07-18',
'gpt-4o-mini',
'gpt-4o-2024-08-06',
'gpt-4o-2024-11-20',
'chatgpt-4o-latest',
'gpt-4o-realtime-preview',
'gpt-4o-realtime-preview-2024-10-01',
'gpt-4o-audio-preview',
'gpt-4o-audio-preview-2024-10-01',
'o1-preview-2024-09-12',
'o1-mini-2024-09-12',
'gpt-3.5-turbo-0301',
'gpt-3.5-turbo-0613',
'tts-1',
Expand All @@ -82,8 +90,8 @@ def register(client: openai.Client):
config.register_model(model_id, client)

#XXX: Deprecation in 0.1.0
config.register_model('o1-preview', client, supports_streaming=False)
config.register_model('o1-mini', client, supports_streaming=False)
config.register_model('o1-preview', client, supports_streaming=True)
config.register_model('o1-mini', client, supports_streaming=True)

default_client = None
try:
Expand Down

0 comments on commit 6b4e4ab

Please sign in to comment.