fixed claude api key for claude case

CloudSecurityAlliance · May 8, 2024 · a01d66e · a01d66e
1 parent dd6c0f4
commit a01d66e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/csa_ai_toolkit/foundation_model_api_clients/src/foundation_model_api_client.py b/csa_ai_toolkit/foundation_model_api_clients/src/foundation_model_api_client.py
@@ -25,6 +25,7 @@ def get_model_mapping(self):
     def get_model_api_key(self):
         model_api_key = {
             'chatgpt': 'OPENAI_CHATGPT_API_KEY',
+            'claude': 'ANTHROPIC_CLAUDE_API_KEY',
             'claude-haiku': 'ANTHROPIC_CLAUDE_API_KEY',
             'claude-sonnet': 'ANTHROPIC_CLAUDE_API_KEY',
             'claude-opus': 'ANTHROPIC_CLAUDE_API_KEY',