AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 06:25:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: optimize message return (#822)

Browse Source
This commit is contained in:
takatost 2023-08-13 13:51:12 +08:00 committed by GitHub
parent 1bd0a76a20
commit 7898937eae
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
api/core/model_providers/models/llm/base.py
View File

@ -218,15 +218,18 @@ class BaseLLM(BaseProviderModel):
def _get_prompt_from_messages(self, messages: List[PromptMessage], def _get_prompt_from_messages(self, messages: List[PromptMessage],
model_mode: Optional[ModelMode] = None) -> Union[str | List[BaseMessage]]: model_mode: Optional[ModelMode] = None) -> Union[str | List[BaseMessage]]:
if len(messages) == 0:
raise ValueError("prompt must not be empty.")
if not model_mode: if not model_mode:
model_mode = self.model_mode model_mode = self.model_mode
if model_mode == ModelMode.COMPLETION: if model_mode == ModelMode.COMPLETION:
if len(messages) == 0:
return ''
return messages[0].content return messages[0].content
else: else:
if len(messages) == 0:
return []
chat_messages = [] chat_messages = []
for message in messages: for message in messages:
if message.type == MessageType.HUMAN: if message.type == MessageType.HUMAN: