Bläddra i källkod

feat: optimize message return (#822)

tags/0.3.13
takatost 2 år sedan
förälder
incheckning
7898937eae
Inget konto är kopplat till bidragsgivarens mejladress
1 ändrade filer med 6 tillägg och 3 borttagningar
  1. 6
    3
      api/core/model_providers/models/llm/base.py

+ 6
- 3
api/core/model_providers/models/llm/base.py Visa fil

@@ -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:

Laddar…
Avbryt
Spara