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

refactor: Optimize exception lookup using a set in LiteLLMExceptions

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-02-06 09:49:40 -08:00
parent 3714f9fdbd
commit 5e4852bd32
1 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
aider/exceptions.py
View file

@ -51,6 +51,7 @@ EXCEPTIONS = [
class LiteLLMExceptions:
exceptions = dict()
exception_names = {exi.name for exi in EXCEPTIONS}
def __init__(self):
self._load()
@ -63,11 +64,11 @@ class LiteLLMExceptions:
continue
ex_info = None
# collect these names into a set once, above ai!
for exi in EXCEPTIONS:
if var == exi.name:
ex_info = exi
break
if var in self.exception_names:
for exi in EXCEPTIONS:
if var == exi.name:
ex_info = exi
break
if strict and not ex_info:
raise ValueError(f"{var} is in litellm but not in aider's exceptions list")