future proofing

AbanteAI · Apr 18, 2024 · 109fddb · 109fddb
1 parent 1edafc3
commit 109fddb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mentat/llm_api_handler.py b/mentat/llm_api_handler.py
@@ -174,7 +174,7 @@ async def initialize_client(self):
  from mentat.session_input import collect_user_input
 
  match provider.name:
- case "open_ai":
+ case "open_ai" | "openai":
  env_variable = "OPENAI_API_KEY"
  case "anthropic":
  env_variable = "ANTHROPIC_API_KEY"