Properly use argument timeout

plasma-umass · Feb 20, 2024 · afce64b · afce64b
1 parent 659ceeb
commit afce64b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/chatdbg/chatdbg_utils.py b/src/chatdbg/chatdbg_utils.py
@@ -102,7 +102,7 @@ def explain(
         return
 
     try:
-        client = openai.OpenAI(timeout=30)
+        client = openai.OpenAI(timeout=args.timeout)
     except openai.OpenAIError:
         print("You need an OpenAI key to use this tool.")
         print("You can get a key here: https://platform.openai.com/api-keys")