mirror of
https://github.com/Aider-AI/aider
synced 2026-05-10 17:12:05 +02:00
be1cf5cc5d
# Context:
USER: can we stop the thread if it is still alive?
ASSISTANT: Yes, we can stop the thread if it's still alive using the `threading.Event` mechanism. We'll create an event object and pass it to the `run_coder` function. If the thread is still alive after the timeout, we'll set the event, and the `run_coder` function will check for the event being set and exit gracefully.
Here are the changes needed in the `scripts/benchmark.py` file:
```python
scripts/benchmark.py
<<<<<<< ORIGINAL
def run_coder():
coder.run(with_message=instructions)
=======
def run_coder(stop_event):
try:
coder.run(with_message=instructions)
except Exception as e:
if stop_event.is_set():
print("Thread stopped due to timeout.")
else:
raise e
>>>>>>> UPDATED
```
```python
scripts/benchmark.py
<<<<<<< ORIGINAL
coder_thread = threading.Thread(target=run_coder)
=======
stop_event = threading.Event()
coder_thread = threading.Thread(target=run_coder, args=(stop_event,))
>>>>>>> UPDATED
```
```python
scripts/benchmark.py
<<<<<<< ORIGINAL
if coder_thread.is_alive():
# Handle the case when the coder.run call takes longer than 60 seconds
# You can raise an exception or handle it accordingly
raise Exception("coder.run took longer than 60 seconds")
=======
if coder_thread.is_alive():
stop_event.set()
coder_thread.join() # Wait for the thread to exit gracefully
print("coder.run took longer than 60 seconds and was stopped.")
# Handle the case when the coder.run call takes longer than 60 seconds
# You can raise an exception or handle it accordingly
# raise Exception("coder.run took longer than 60 seconds")
>>>>>>> UPDATED
```
These changes create a `stop_event` object, pass it to the `run_coder` function, and set the event if the thread is still alive after the timeout. The `run_coder` function checks for the event being set and exits gracefully.