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

View file

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