From 4181d9e0b48884bab7a27e7e24f29ee0aa0db9ae Mon Sep 17 00:00:00 2001 From: Francis Beeson Date: Fri, 16 Feb 2024 16:53:21 +0000 Subject: [PATCH] Include new model names in _alias.py (#426) Co-authored-by: Jason Liu --- instructor/_types/_alias.py | 23 +++++++++++++++++++---- 1 file changed, 19 insertions(+), 4 deletions(-) diff --git a/instructor/_types/_alias.py b/instructor/_types/_alias.py index c4d52a6..2f8c9ac 100644 --- a/instructor/_types/_alias.py +++ b/instructor/_types/_alias.py @@ -3,10 +3,25 @@ from typing import Literal from typing_extensions import TypeAlias ModelNames: TypeAlias = Literal[ + "gpt-4-0125-preview", + "gpt-4-turbo-preview", + "gpt-4-1106-preview", + "gpt-4-vision-preview", + "gpt-4", + "gpt-4-0314", + "gpt-4-0613", + "gpt-4-32k", + "gpt-4-32k-0314", + "gpt-4-32k-0613", "gpt-3.5-turbo", "gpt-3.5-turbo-16k", - "gpt-4", - "gpt-4-32k", - "text-embedding-ada-002", - "text-embedding-ada-002-v2", + "gpt-3.5-turbo-0301", + "gpt-3.5-turbo-0613", + "gpt-3.5-turbo-1106", + "gpt-3.5-turbo-0125", + "gpt-3.5-turbo-16k-0613", + "gpt-3.5-turbo-instruct", + "text-embedding-ada-002", + "text-embedding-3-small", + "text-embedding-3-large" ]