From cb5907f731a7d2d8cc367f7ba976bf8ff0857b56 Mon Sep 17 00:00:00 2001 From: Kyle Kelley Date: Mon, 4 Mar 2024 14:21:04 -0800 Subject: [PATCH] default to 3.5 turbo by enum --- chatlab/chat.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/chatlab/chat.py b/chatlab/chat.py index 36e1b47..867def7 100644 --- a/chatlab/chat.py +++ b/chatlab/chat.py @@ -27,6 +27,8 @@ from .registry import FunctionRegistry, PythonHallucinationFunction from .views import ToolArguments, AssistantMessageView +from .models import GPT_3_5_TURBO + logger = logging.getLogger(__name__) @@ -65,7 +67,7 @@ def __init__( *initial_context: Union[ChatCompletionMessageParam, str], base_url=None, api_key=None, - model="gpt-3.5-turbo-0613", + model=GPT_3_5_TURBO, function_registry: Optional[FunctionRegistry] = None, chat_functions: Optional[List[Callable]] = None, allow_hallucinated_python: bool = False,