Skip to content

Commit

Permalink
Update aisuite/providers/fireworks_provider.py
Browse files Browse the repository at this point in the history
Co-authored-by: Kevin Solorio <[email protected]>
  • Loading branch information
rohitprasad15 and ksolo authored Oct 2, 2024
1 parent 3e8550c commit b4656a6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion aisuite/providers/fireworks_provider.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ def __init__(self, **config):
Initialize the Fireworks provider with the given configuration.
The API key is fetched from the config or environment variables.
"""
self.api_key = config.get("api_key") or os.getenv("FIREWORKS_API_KEY")
self.api_key = config.get("api_key", os.getenv("FIREWORKS_API_KEY"))
if not self.api_key:
raise ValueError(
"Fireworks API key is missing. Please provide it in the config or set the FIREWORKS_API_KEY environment variable."
Expand Down

0 comments on commit b4656a6

Please sign in to comment.