This commit is contained in:
Paul Gauthier 2024-04-18 09:29:26 -07:00
parent 15c1ad1f96
commit 93f4a46996
3 changed files with 7 additions and 10 deletions

View file

@ -582,14 +582,12 @@ def main(argv=None, input=None, output=None, force_git_root=None):
# Check in advance that we have model metadata # Check in advance that we have model metadata
try: try:
litellm.get_model_info(args.model) main_model = models.Model(args.model)
except Exception as err: except models.NoModelInfo as err:
io.tool_error(f"Unknown model {args.model}.") io.tool_error(f"Unknown model {args.model}.")
io.tool_error(str(err)) io.tool_error(str(err))
return 1 return 1
main_model = models.Model(args.model)
try: try:
coder = Coder.create( coder = Coder.create(
main_model=main_model, main_model=main_model,

View file

@ -1,17 +1,21 @@
import json import json
import math import math
from dataclasses import dataclass, fields from dataclasses import dataclass, fields
from typing import Optional
import litellm import litellm
from typing import Optional
from PIL import Image from PIL import Image
from aider.dump import dump from aider.dump import dump
DEFAULT_MODEL_NAME = "gpt-4-1106-preview"
class NoModelInfo(Exception): class NoModelInfo(Exception):
""" """
Exception raised when model information cannot be retrieved. Exception raised when model information cannot be retrieved.
""" """
def __init__(self, message: Optional[str] = None): def __init__(self, message: Optional[str] = None):
super().__init__(message or "No model information available.") super().__init__(message or "No model information available.")

View file

@ -1,5 +0,0 @@
from .model import Model
DEFAULT_MODEL_NAME = "gpt-4-1106-preview"
__all__ = [Model, DEFAULT_MODEL_NAME]