bugfix: bulk response from Claude 3

GaiZhenbiao · Mar 16, 2024 · c24712b · c24712b
1 parent e9ac109
commit c24712b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/modules/models/Claude.py b/modules/models/Claude.py
@@ -91,8 +91,6 @@ def get_answer_stream_iter(self):
  def get_answer_at_once(self):
  system_prompt = self.system_prompt
  history = self._get_claude_style_history()
- if system_prompt is not None:
- history = [construct_system(system_prompt), *history]
 
  response = self.claude_client.messages.create(
  model=self.model_name,
@@ -101,6 +99,6 @@ def get_answer_at_once(self):
  system=system_prompt,
  )
  if response is not None:
- return response["content"][0]["text"], response["usage"]["output_tokens"]
+ return response.content[0].text, response.usage.output_tokens
  else:
  return i18n("获取资源错误"), 0
diff --git a/modules/presets.py b/modules/presets.py
@@ -63,6 +63,9 @@
  "GPT4 32K",
  "GPT4 Turbo",
  "GPT4 Vision",
+ "Claude 3 Haiku",
+ "Claude 3 Sonnet",
+ "Claude 3 Opus",
  "川虎助理",
  "川虎助理 Pro",
  "DALL-E 3",
@@ -82,9 +85,6 @@
  "讯飞星火大模型V3.0",
  "讯飞星火大模型V2.0",
  "讯飞星火大模型V1.5",
- "Claude 3 Haiku",
- "Claude 3 Sonnet",
- "Claude 3 Opus",
  "ERNIE-Bot-turbo",
  "ERNIE-Bot",
  "ERNIE-Bot-4",