From 15c1ad1f9673efb83409f90e880499531d9b1d79 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Thu, 18 Apr 2024 09:26:07 -0700 Subject: [PATCH] aider: Create the `NoModelInfo` class to handle exceptions when model information cannot be retrieved. # Aider chat conversation: USER: create the `NoModelInfo` class ASSISTANT: ```diff --- aider/models/model.py +++ aider/models/model.py @@ ... @@ from dataclasses import dataclass, fields import litellm +from typing import Optional from PIL import Image from aider.dump import dump +class NoModelInfo(Exception): + """ + Exception raised when model information cannot be retrieved. + """ + def __init__(self, message: Optional[str] = None): + super().__init__(message or "No model information available.") + @dataclass class ModelSettings: name: str ``` --- aider/models/model.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/aider/models/model.py b/aider/models/model.py index 65e6f3b0d..9ff629d05 100644 --- a/aider/models/model.py +++ b/aider/models/model.py @@ -3,10 +3,18 @@ import math from dataclasses import dataclass, fields import litellm +from typing import Optional from PIL import Image from aider.dump import dump +class NoModelInfo(Exception): + """ + Exception raised when model information cannot be retrieved. + """ + def __init__(self, message: Optional[str] = None): + super().__init__(message or "No model information available.") + @dataclass class ModelSettings: