use the nightly model

caesarHQ · May 10, 2023 · 049b1fd · 049b1fd · vercel · May 10, 2023
1 parent 427f4a6
commit 049b1fd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/api/app/api/utils/sql_gen/text_to_sql.py b/api/app/api/utils/sql_gen/text_to_sql.py
@@ -39,9 +39,9 @@ def text_to_sql_with_retry(natural_language_query, table_names, k=3, messages=No
     Tries to take a natural language query and generate valid SQL to answer it K times
     """
     if scope == "SF":
-        model = "gpt-3.5-turbo-0301"
+        model = "gpt-3.5-turbo"
     else:
-        model = "gpt-3.5-turbo-0301"
+        model = "gpt-3.5-turbo"
 
     example_messages = []
     enums_message = [{'role': 'user', 'content': ''}]

diff --git a/api/app/api/utils/table_selection/table_selection.py b/api/app/api/utils/table_selection/table_selection.py
@@ -147,15 +147,15 @@ def get_relevant_tables(natural_language_query, scope="USA") -> List[str]:
     # temporary hack to always use LM for SF
     if scope == "SF":
         # model = "gpt-4"
-        model = "gpt-3.5-turbo-0301"
+        model = "gpt-3.5-turbo"
         return get_relevant_tables_from_lm(natural_language_query, scope, model)
 
     if PINECONE_KEY and PINECONE_ENV:
         return get_relevant_tables_from_pinecone(natural_language_query, scope=scope)
 
     if scope == "SF":
         # model = "gpt-4"
-        model = "gpt-3.5-turbo-0301"
+        model = "gpt-3.5-turbo"
     else:
         model = "gpt-3.5-turbo"