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"