File size: 478 Bytes
e3eb379
6ebbca3
 
e3eb379
 
6ebbca3
e3eb379
6ebbca3
 
 
 
e3eb379
 
6ebbca3
 
e3eb379
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
import json
from typing import List
from langchain_core.messages import BaseMessage


def format_messages(messages: List[BaseMessage]) -> str:
    formatted = []

    for m in messages:
        role = m.type  # Adjust if your object structure differs
        content = m.content
        content_string = content if isinstance(
            content, str) else json.dumps(content)
        formatted.append(f'<{role}>\n{content_string}\n</{role}>')

    return "\n".join(formatted)