Skip to content

Commit 4a99598

Browse files
committed
Fixed bug where there was error on import when no OpenAIKey
1 parent 90f19e3 commit 4a99598

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

forecasting_tools/ai_models/model_archetypes/openai_text_model.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,11 @@
2222

2323
class OpenAiTextToTextModel(TraditionalOnlineLlm, ABC):
2424
_OPENAI_ASYNC_CLIENT = AsyncOpenAI(
25-
api_key=os.getenv("OPENAI_API_KEY"),
25+
api_key=(
26+
os.getenv("OPENAI_API_KEY")
27+
if os.getenv("OPENAI_API_KEY") is not None
28+
else "fake_key_so_it_doesn't_error_on_initialization"
29+
),
2630
max_retries=0, # Retry is implemented locally
2731
)
2832

forecasting_tools/ai_models/model_archetypes/perplexity_text_model.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,11 @@ class PerplexityTextModel(OpenAiTextToTextModel, PricedPerRequest, ABC):
2626
PRICE_PER_TOKEN: float
2727
PERPLEXITY_API_KEY = os.getenv("PERPLEXITY_API_KEY")
2828
_OPENAI_ASYNC_CLIENT = AsyncOpenAI(
29-
api_key=PERPLEXITY_API_KEY,
29+
api_key=(
30+
PERPLEXITY_API_KEY
31+
if PERPLEXITY_API_KEY is not None
32+
else "fake_key_so_it_doesn't_error_on_initialization"
33+
),
3034
base_url="https://api.perplexity.ai",
3135
max_retries=0, # Retry is implemented locally
3236
)

0 commit comments

Comments
 (0)