Skip to content

Commit

Permalink
Explicitly load key for groq when needed (#486)
Browse files Browse the repository at this point in the history
  • Loading branch information
caufieldjh authored Dec 4, 2024
2 parents 0f83736 + 665edbc commit 96ccbb0
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion src/ontogpt/clients/llm_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,13 +42,17 @@ class LLMClient:

def __post_init__(self):
# Get appropriate API key for the model source
# and other details if needed
# and other details if needed.
# This will look at any env vars FIRST,
# then configs handled by oaklib.
if self.model.startswith("ollama"):
self.api_key = "" # Don't need an API key
if not self.api_key and not self.custom_llm_provider:
self.api_key = get_apikey_value("openai")
elif self.custom_llm_provider == "anthropic":
self.api_key = get_apikey_value("anthropic-key")
elif self.custom_llm_provider == "groq":
self.api_key = get_apikey_value("groq-key")
else:
for service in SERVICES:
if self.model.startswith(service):
Expand Down

0 comments on commit 96ccbb0

Please sign in to comment.