Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2025-12-18 09:20:15 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

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); createdHandler(data, { ...submission, userMessage } as EventSubmission);
} else if (data.event) { } else if (data.event != null) {
stepHandler(data); stepHandler(data, { ...submission, userMessage } as EventSubmission);
} else if (data.sync != null) { } else if (data.sync != null) {
const runId = v4(); const runId = v4();
setActiveRunId(runId); setActiveRunId(runId);

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

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