Update agents-api/agents_api/activities/task_steps/prompt_step.py

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>
julep-ai · Oct 29, 2024 · c6d9f23 · c6d9f23
1 parent 98e3d3e
commit c6d9f23
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/agents-api/agents_api/activities/task_steps/prompt_step.py b/agents-api/agents_api/activities/task_steps/prompt_step.py
@@ -117,7 +117,7 @@ async def prompt_step(context: StepContext) -> StepOutcome:
     if "claude-3.5-sonnet-20241022" == agent_model.lower():
         # Retrieve the API key from the environment variable
         betas = [COMPUTER_USE_BETA_FLAG]
-        api_key = os.getenv("ANTHROPIC_API_KEY")
+        api_key = os.getenv("ANTHROPIC_API_KEY") or raise ValueError("API key is not set.")
         # Format all three Anthropic tools
         formatted_agent_tools = format_anthropic_tools(agent_tools)
         # Use Anthropic API directly