Просмотр исходного кода

feat: optimize message return (#822)

tags/0.3.13
takatost 2 лет назад
Родитель
Сommit
7898937eae
Аккаунт пользователя с таким Email не найден
1 измененных файлов: 6 добавлений и 3 удалений
  1. 6
    3
      api/core/model_providers/models/llm/base.py

+ 6
- 3
api/core/model_providers/models/llm/base.py Просмотреть файл

@@ -218,15 +218,18 @@ class BaseLLM(BaseProviderModel):

def _get_prompt_from_messages(self, messages: List[PromptMessage],
model_mode: Optional[ModelMode] = None) -> Union[str | List[BaseMessage]]:
if len(messages) == 0:
raise ValueError("prompt must not be empty.")

if not model_mode:
model_mode = self.model_mode

if model_mode == ModelMode.COMPLETION:
if len(messages) == 0:
return ''

return messages[0].content
else:
if len(messages) == 0:
return []

chat_messages = []
for message in messages:
if message.type == MessageType.HUMAN:

Загрузка…
Отмена
Сохранить