default to 3.5 turbo by enum

rgbkrk · Mar 4, 2024 · fc5c759 · fc5c759
1 parent 8d3b49a
commit fc5c759
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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,