1 files changed, 5 insertions, 2 deletions

diff --git a/etc/tool/improve_file.py b/etc/tool/improve_code.py
index cef41354..9a940b51 100644
--- a/etc/tool/improve_file.py
+++ b/etc/tool/improve_code.py
@@ -22,7 +22,10 @@ Improve the code in this file:
 ```py
 {code}
 ```
-Don't remove anything. Add type hints if possible.
+Don't remove anything.
+Add typehints if possible.
+Don't add any typehints to kwargs.
+Don't remove license comments.
 """
 
 print("Create code...")
@@ -30,7 +33,7 @@ response = []
 for chunk in g4f.ChatCompletion.create(
     model=g4f.models.gpt_35_long,
     messages=[{"role": "user", "content": prompt}],
-    timeout=0,
+    timeout=300,
     stream=True
 ):
     response.append(chunk)