----- Step 1 ------
Error in generating model output:
'text'
[Step 1: Duration 50.57 seconds]
---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
File ~/code/agents/smolagents/src/smolagents/agents.py:1637, in CodeAgent._step_stream(self, memory_step)
   1636 else:
-> 1637     chat_message: ChatMessage = self.model.generate(
   1638         input_messages,
   1639         stop_sequences=stop_sequences,
   1640         **additional_args,
   1641     )
   1642     memory_step.model_output_message = chat_message

File ~/code/agents/smolagents/src/smolagents/models.py:1902, in AmazonBedrockServerModel.generate(self, messages, stop_sequences, response_format, tools_to_call_from, **kwargs)
   1901 # Get first message
-> 1902 response["output"]["message"]["content"] = response["output"]["message"]["content"][0]["text"]
   1904 self._last_input_token_count = response["usage"]["inputTokens"]

KeyError: 'text'

The above exception was the direct cause of the following exception:

AgentGenerationError                      Traceback (most recent call last)
Cell In[1], line 19
     10 model = AmazonBedrockServerModel(model_id=model_id, client=boto_client, additionalModelRequestFields = {"thinking": {"type": "enabled", "budget_tokens": 4096}})
     11 agent = CodeAgent(
     12                 tools=[],
     13                 #additional_authorized_imports=["fractions", "sympy", "numpy", "functools", "numpy.random", "*"],
   (...)     17                 #logger=logger
     18             )
---> 19 result = agent.run(prompt)

File ~/code/agents/smolagents/src/smolagents/agents.py:468, in MultiStepAgent.run(self, task, stream, reset, images, additional_args, max_steps)
    465 run_start_time = time.time()
    466 # Outputs are returned only at the end. We only look at the last step.
--> 468 steps = list(self._run_stream(task=self.task, max_steps=max_steps, images=images))
    469 assert isinstance(steps[-1], FinalAnswerStep)
    470 output = steps[-1].output

File ~/code/agents/smolagents/src/smolagents/agents.py:562, in MultiStepAgent._run_stream(self, task, max_steps, images)
    558             action_step.is_final_answer = True
    560 except AgentGenerationError as e:
    561     # Agent generation errors are not caused by a Model error but an implementation error: so we should raise them and exit.
--> 562     raise e
    563 except AgentError as e:
    564     # Other AgentError types are caused by the Model, so we should log them and iterate.
    565     action_step.error = e

File ~/code/agents/smolagents/src/smolagents/agents.py:544, in MultiStepAgent._run_stream(self, task, max_steps, images)
    542 self.logger.log_rule(f"Step {self.step_number}", level=LogLevel.INFO)
    543 try:
--> 544     for output in self._step_stream(action_step):
    545         # Yield all
    546         yield output
    548         if isinstance(output, ActionOutput) and output.is_final_answer:

File ~/code/agents/smolagents/src/smolagents/agents.py:1659, in CodeAgent._step_stream(self, memory_step)
   1657     memory_step.model_output = output_text
   1658 except Exception as e:
-> 1659     raise AgentGenerationError(f"Error in generating model output:\n{e}", self.logger) from e
   1661 ### Parse output ###
   1662 try:

AgentGenerationError: Error in generating model output:
'text'