Update utils.py

chtmp223 · Nov 21, 2023 · e0df264 · e0df264
1 parent 650d8de
commit e0df264
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/script/utils.py b/script/utils.py
@@ -28,7 +28,6 @@ def api_call(prompt, deployment_name, temperature, max_tokens, top_p):
     - top_p: top p parameter
     '''
     time.sleep(5)                           # Change to avoid rate limit
-    SystemExit
     if deployment_name in ["gpt-35-turbo", "gpt-4", "gpt-3.5-turbo"]:
         response = client.chat.completions.create(model=deployment_name, 
         temperature=float(temperature),  
@@ -332,4 +331,4 @@ def construct_sentences(p2_root, removed):
                 break
         if not removed_branches: 
             sentences.append(branch_str)
-    return sentences
+    return sentences