瀏覽代碼

feat: optimize message return (#822)

takatost 1 年之前
父節點
當前提交
7898937eae
共有 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: