Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2025-12-17 00:40:14 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

fix(bedrock/agents): message regeneration

Browse source
This commit is contained in:
Danny Avila 2024-09-03 08:05:06 -04:00
parent b41e39e6e5
commit fac2acd4cf
No known key found for this signature in database
GPG key ID: 2DD9CC89B9B50364
2 changed files with 13 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

4
client/src/hooks/SSE/useSSE.ts
View file

@ -117,8 +117,8 @@ export default function useSSE(
};
createdHandler(data, { ...submission, userMessage } as EventSubmission);
} else if (data.event) {
stepHandler(data);
} else if (data.event != null) {
stepHandler(data, { ...submission, userMessage } as EventSubmission);
} else if (data.sync != null) {
const runId = v4();
setActiveRunId(runId);

16
client/src/hooks/SSE/useStepHandler.ts
View file

@ -1,6 +1,12 @@
import { useCallback, useRef } from 'react';
import { StepTypes, ContentTypes, ToolCallTypes } from 'librechat-data-provider';
import type { Agents, PartMetadata, TMessage, TMessageContentParts } from 'librechat-data-provider';
import type {
Agents,
PartMetadata,
TMessage,
TMessageContentParts,
EventSubmission,
} from 'librechat-data-provider';
import { getNonEmptyValue } from 'librechat-data-provider';
type TUseStepHandler = {
@ -96,8 +102,9 @@ export default function useStepHandler({ setMessages, getMessages }: TUseStepHan
};
return useCallback(
({ event, data }: TStepEvent) => {
({ event, data }: TStepEvent, submission: EventSubmission) => {
const messages = getMessages() || [];
const { userMessage } = submission;
if (event === 'on_run_step') {
const runStep = data as Agents.RunStep;
@ -112,12 +119,11 @@ export default function useStepHandler({ setMessages, getMessages }: TUseStepHan
if (!response) {
const responseMessage = messages[messages.length - 1] as TMessage;
const userMessage = messages[messages.length - 2] as TMessage | null;
response = {
...responseMessage,
parentMessageId: userMessage?.messageId ?? '',
conversationId: userMessage?.conversationId ?? '',
parentMessageId: userMessage.messageId,
conversationId: userMessage.conversationId,
messageId: responseMessageId,
content: [],
};