dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-31 01:35:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

feat: improve error handling for import loading

Browse source
This commit is contained in:
Paul Gauthier 2024-09-08 09:18:20 -07:00 • committed by Paul Gauthier (aider)
parent 1e9ff842f3
commit 9982f439e4
1 changed files with 12 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

16
aider/main.py
View file

@ -731,7 +731,14 @@ def check_and_load_imports(io, verbose=False):
io.tool_output( io.tool_output(
"First run for this version and executable, loading imports synchronously" "First run for this version and executable, loading imports synchronously"
) )
load_slow_imports() try:
load_slow_imports(swallow=False)
except Exception as err:
io.tool_error(str(err))
io.tool_output("Error loading required imports. Did you install aider properly?")
io.tool_output("https://aider.chat/docs/install.html")
sys.exit(1)
installs[str(key)] = True installs[str(key)] = True
installs_file.parent.mkdir(parents=True, exist_ok=True) installs_file.parent.mkdir(parents=True, exist_ok=True)
with open(installs_file, "w") as f: with open(installs_file, "w") as f:
@ -745,12 +752,12 @@ def check_and_load_imports(io, verbose=False):
thread.daemon = True thread.daemon = True
thread.start() thread.start()
except Exception as e: except Exception as e:
io.tool_warning(f"Error in check_and_load_imports: {e}", file=sys.stderr) io.tool_warning(f"Error in checking imports: {e}")
if verbose: if verbose:
io.tool_output(f"Full exception details: {traceback.format_exc()}") io.tool_output(f"Full exception details: {traceback.format_exc()}")
def load_slow_imports(): def load_slow_imports(swallow=True):
# These imports are deferred in various ways to # These imports are deferred in various ways to
# improve startup time. # improve startup time.
# This func is called either synchronously or in a thread # This func is called either synchronously or in a thread
@ -762,7 +769,8 @@ def load_slow_imports():
import networkx # noqa: F401 import networkx # noqa: F401
import numpy # noqa: F401 import numpy # noqa: F401
except Exception as e: except Exception as e:
print(f"Error in load_slow_imports: {e}", file=sys.stderr) if not swallow:
raise e
if __name__ == "__main__": if __name__ == "__main__":