12 lines
349 B
Python
12 lines
349 B
Python
from langchain_core.messages import HumanMessage, AIMessage
|
|
|
|
|
|
def dicts_to_messages(messages: list[dict]):
|
|
result = []
|
|
for m in messages:
|
|
if m["role"] == "user":
|
|
result.append(HumanMessage(content=m["content"]))
|
|
elif m["role"] == "ai":
|
|
result.append(AIMessage(content=m["content"]))
|
|
return result
|