feat(format-history): update role determination
This commit is contained in:
parent
dd50d4927b
commit
37022fb11e
1 changed files with 5 additions and 2 deletions
|
|
@ -1,8 +1,11 @@
|
||||||
import { BaseMessage } from '@langchain/core/messages';
|
import { BaseMessage, isAIMessage } from '@langchain/core/messages';
|
||||||
|
|
||||||
const formatChatHistoryAsString = (history: BaseMessage[]) => {
|
const formatChatHistoryAsString = (history: BaseMessage[]) => {
|
||||||
return history
|
return history
|
||||||
.map((message) => `${message._getType()}: ${message.content}`)
|
.map(
|
||||||
|
(message) =>
|
||||||
|
`${isAIMessage(message) ? 'AI' : 'User:'}: ${message.content}`,
|
||||||
|
)
|
||||||
.join('\n');
|
.join('\n');
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue