jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add support for message conversion in ChatOutputResponse class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-26 17:08:10 -03:00
commit 7eac47f4c8
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
src/backend/langflow/utils/schemas.py
View file

@ -1,10 +1,18 @@
from typing import Dict, List, Optional, Union
from langchain_core.messages import BaseMessage
from pydantic import BaseModel
from typing import Optional
class ChatOutputResponse(BaseModel):
"""Chat output response schema."""
message: str
message: Union[str, List[Union[str, Dict]]]
sender: Optional[str] = "Machine"
sender_name: Optional[str] = "AI"
@classmethod
def from_message(cls, message: BaseMessage, sender: Optional[str] = "Machine", sender_name: Optional[str] = "AI"):
"""Build chat output response from message."""
content = message.content
return cls(message=content, sender=sender, sender_name=sender_name)