From c6d9f23c4bd6a6fc8873a90a0632c524c05f1809 Mon Sep 17 00:00:00 2001 From: Vedant Sahai Date: Tue, 29 Oct 2024 17:05:13 -0400 Subject: [PATCH] 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> --- agents-api/agents_api/activities/task_steps/prompt_step.py | 2 +- 1 file changed, 1 insertion(+), 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 index 0494980e2..f92c5b24c 100644 --- 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