Browse Source

feat: optimize message return (#822)

tags/0.3.13
takatost 2 years ago
parent
commit
7898937eae
No account linked to committer's email address
1 changed files with 6 additions and 3 deletions
  1. 6
    3
      api/core/model_providers/models/llm/base.py

+ 6
- 3
api/core/model_providers/models/llm/base.py View File



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:

Loading…
Cancel
Save