From 830f59e881b6d5e50b8b1450a52a2ba9241e3007 Mon Sep 17 00:00:00 2001 From: Liam Xu Date: Mon, 8 Apr 2024 17:23:05 -0700 Subject: [PATCH] removing the engine model map --- src/suql/prompt_continuation.py | 12 +----------- 1 file changed, 1 insertion(+), 11 deletions(-) diff --git a/src/suql/prompt_continuation.py b/src/suql/prompt_continuation.py index 2fa5031..58d5736 100644 --- a/src/suql/prompt_continuation.py +++ b/src/suql/prompt_continuation.py @@ -87,6 +87,7 @@ def _generate( no_line_break_start = "" no_line_break_length = 0 kwargs = { + "model": engine, "messages": [ {"role": "system", "content": filled_prompt + no_line_break_start} ], @@ -97,17 +98,6 @@ def _generate( "presence_penalty": presence_penalty, "stop": stop_tokens, } - engine_model_map = { - "gpt-4": "gpt-4", - "gpt-35-turbo": "gpt-3.5-turbo-1106", - "gpt-3.5-turbo": "gpt-3.5-turbo-1106", - "gpt-4-turbo": "gpt-4-1106-preview", - } - kwargs.update( - { - "model": engine_model_map[engine] if engine in engine_model_map else engine - } - ) generation_output = chat_completion_with_backoff(**kwargs) generation_output = no_line_break_start + generation_output