feat: Add Fireworks AI API key support to model configuration

This commit is contained in:
Paul Gauthier 2025-02-04 16:12:02 -08:00 committed by Paul Gauthier (aider)
parent b9f4f3f71c
commit 3e36f27987
2 changed files with 5 additions and 1 deletions

View file

@ -2,6 +2,8 @@ import importlib
import os
import warnings
from aider.dump import dump # noqa: F401
warnings.filterwarnings("ignore", category=UserWarning, module="pydantic")
AIDER_SITE_URL = "https://aider.chat"
@ -13,13 +15,14 @@ os.environ["LITELLM_MODE"] = "PRODUCTION"
# `import litellm` takes 1.5 seconds, defer it!
VERBOSE = False
VERBOSE = True
class LazyLiteLLM:
_lazy_module = None
def __getattr__(self, name):
dump(name)
if name == "_lazy_module":
return super()
self._load_litellm()

View file

@ -505,6 +505,7 @@ class Model(ModelSettings):
gemini="GEMINI_API_KEY",
anthropic="ANTHROPIC_API_KEY",
groq="GROQ_API_KEY",
fireworks_ai="FIREWORKS_API_KEY",
)
var = None
if model in OPENAI_MODELS: