瀏覽代碼

fix: image tokenizer (#2752)

tags/0.5.9
Yeuoly 1 年之前
父節點
當前提交
3231a8c51c
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 18 行新增1 行删除
  1. 18
    1
      api/core/model_runtime/model_providers/anthropic/llm/llm.py

+ 18
- 1
api/core/model_runtime/model_providers/anthropic/llm/llm.py 查看文件

@@ -424,8 +424,25 @@ class AnthropicLargeLanguageModel(LargeLanguageModel):

if isinstance(message, UserPromptMessage):
message_text = f"{human_prompt} {content}"
if not isinstance(message.content, list):
message_text = f"{ai_prompt} {content}"
else:
message_text = ""
for sub_message in message.content:
if sub_message.type == PromptMessageContentType.TEXT:
message_text += f"{human_prompt} {sub_message.data}"
elif sub_message.type == PromptMessageContentType.IMAGE:
message_text += f"{human_prompt} [IMAGE]"
elif isinstance(message, AssistantPromptMessage):
message_text = f"{ai_prompt} {content}"
if not isinstance(message.content, list):
message_text = f"{ai_prompt} {content}"
else:
message_text = ""
for sub_message in message.content:
if sub_message.type == PromptMessageContentType.TEXT:
message_text += f"{ai_prompt} {sub_message.data}"
elif sub_message.type == PromptMessageContentType.IMAGE:
message_text += f"{ai_prompt} [IMAGE]"
elif isinstance(message, SystemPromptMessage):
message_text = content
else:

Loading…
取消
儲存