Bladeren bron

fix: file type document is not supported (#9618)

tags/0.10.1
非法操作 1 jaar geleden
bovenliggende
commit
5f12616cb9
No account linked to committer's email address
1 gewijzigde bestanden met toevoegingen van 4 en 2 verwijderingen
  1. 4
    2
      api/core/memory/token_buffer_memory.py

+ 4
- 2
api/core/memory/token_buffer_memory.py Bestand weergeven

@@ -2,6 +2,7 @@ from typing import Optional

from core.app.app_config.features.file_upload.manager import FileUploadConfigManager
from core.file import file_manager
from core.file.models import FileType
from core.model_manager import ModelInstance
from core.model_runtime.entities import (
AssistantPromptMessage,
@@ -98,8 +99,9 @@ class TokenBufferMemory:
prompt_message_contents: list[PromptMessageContent] = []
prompt_message_contents.append(TextPromptMessageContent(data=message.query))
for file_obj in file_objs:
prompt_message = file_manager.to_prompt_message_content(file_obj)
prompt_message_contents.append(prompt_message)
if file_obj.type in {FileType.IMAGE, FileType.AUDIO}:
prompt_message = file_manager.to_prompt_message_content(file_obj)
prompt_message_contents.append(prompt_message)

prompt_messages.append(UserPromptMessage(content=prompt_message_contents))
else:

Laden…
Annuleren
Opslaan