bugfix: retry doesn't work when Claude API errors out (such as when o…

…verloaded)

modules/models/Claude.py

@@ -74,16 +74,19 @@ def get_answer_stream_iter(self):
  system_prompt = self.system_prompt
  history = self._get_claude_style_history()
 
- with self.claude_client.messages.stream(
- model=self.model_name,
- max_tokens=self.max_generation_token,
- messages=history,
- system=system_prompt,
- ) as stream:
- partial_text = ""
- for text in stream.text_stream:
- partial_text += text
- yield partial_text
+ try:
+ with self.claude_client.messages.stream(
+ model=self.model_name,
+ max_tokens=self.max_generation_token,
+ messages=history,
+ system=system_prompt,
+ ) as stream:
+ partial_text = ""
+ for text in stream.text_stream:
+ partial_text += text
+ yield partial_text
+ except Exception as e:
+ yield i18n(GENERAL_ERROR_MSG) + ": " + str(e)
 
  def get_answer_at_once(self):
  system_prompt = self.system_prompt