@librechat/agents 3.4.3 → 3.4.4

package/dist/cjs/stream.cjs

@@ -5,12 +5,13 @@ const require_truncation = require("./utils/truncation.cjs");
 const require_toolContent = require("./utils/toolContent.cjs");
 const require_ids = require("./messages/ids.cjs");
 const require_events = require("./utils/events.cjs");
+const require_assistantPhase = require("./messages/assistantPhase.cjs");
 require("./messages/index.cjs");
 const require_llm = require("./utils/llm.cjs");
 const require_streamedToolCallSeals = require("./tools/streamedToolCallSeals.cjs");
+const require_streamLimits = require("./llm/streamLimits.cjs");
 const require_eagerEventExecution = require("./tools/eagerEventExecution.cjs");
 const require_handlers = require("./tools/handlers.cjs");
-const require_streamLimits = require("./llm/streamLimits.cjs");
 const require_intentArg = require("./tools/intentArg.cjs");
 const require_toolOutputReferences = require("./tools/toolOutputReferences.cjs");
 //#region src/stream.ts
@@ -206,11 +207,14 @@ function shouldStartFreshMessageStepAfterGoogleServerSideTool({ graph, stepId, r
 	if (typeof content === "string") return true;
 	return content.every((c) => isTextContentPart(c)) || content.every((c) => isReasoningContentPart(c));
 }
-async function dispatchMessageCreationStep({ graph, stepKey, metadata }) {
+async function dispatchMessageCreationStep({ graph, stepKey, content, contentType, metadata }) {
 	const messageId = require_ids.getMessageId(stepKey, graph, true) ?? "";
 	return graph.dispatchRunStep(stepKey, {
 		type: "message_creation",
-		message_creation: { message_id: messageId }
+		message_creation: {
+			message_id: messageId,
+			...require_assistantPhase.getMessageCreationContentMetadata(content, contentType)
+		}
 	}, metadata);
 }
 async function dispatchMessageContentParts({ graph, stepKey, content, metadata }) {
@@ -218,6 +222,7 @@ async function dispatchMessageContentParts({ graph, stepKey, content, metadata }
 		const currentStepId = await dispatchMessageCreationStep({
 			graph,
 			stepKey,
+			content: [contentPart],
 			metadata
 		});
 		if (isGoogleServerSideToolContentPart(contentPart)) markGoogleServerSideToolMessageStep(graph, currentStepId);
@@ -229,6 +234,8 @@ async function dispatchReasoningContentParts({ graph, stepKey, content, metadata
 	const currentStepId = await dispatchMessageCreationStep({
 		graph,
 		stepKey,
+		content,
+		contentType: "think",
 		metadata
 	});
 	await graph.dispatchReasoningDelta(currentStepId, { content }, metadata);
@@ -804,11 +811,37 @@ var ChatModelStreamHandler = class {
 		}
 		if (isEmptyContent) return;
 		if (hasGoogleServerSideToolContent) return;
+		if (Array.isArray(content) && content.every(isTextContentPart)) {
+			const contentGroups = require_assistantPhase.splitAssistantTextContentByPhase(content);
+			const currentStepId = graph.stepKeyIds?.get(stepKey)?.at(-1);
+			const currentStep = currentStepId == null ? void 0 : graph.getRunStep(currentStepId);
+			const currentPhase = currentStep?.stepDetails.type === "message_creation" ? currentStep.stepDetails.message_creation.phase : void 0;
+			const nextPhase = require_assistantPhase.getMessageCreationContentMetadata(contentGroups[0]).phase;
+			const phaseChanged = currentPhase != null && nextPhase != null && currentPhase !== nextPhase;
+			if (contentGroups.length > 1 || phaseChanged) {
+				for (const contentGroup of contentGroups) {
+					const currentStepId = await dispatchMessageCreationStep({
+						graph,
+						stepKey,
+						content: contentGroup,
+						metadata
+					});
+					await graph.dispatchMessageDelta(currentStepId, { content: contentGroup }, metadata);
+				}
+				return;
+			}
+		}
 		const message_id = require_ids.getMessageId(stepKey, graph) ?? "";
-		if (message_id) await graph.dispatchRunStep(stepKey, {
-			type: "message_creation",
-			message_creation: { message_id }
-		}, metadata);
+		if (message_id) {
+			const fallbackContentType = agentContext.currentTokenType === "text" ? "text" : "think";
+			await graph.dispatchRunStep(stepKey, {
+				type: "message_creation",
+				message_creation: {
+					message_id,
+					...require_assistantPhase.getMessageCreationContentMetadata(content, fallbackContentType)
+				}
+			}, metadata);
+		}
 		let stepId = graph.getStepIdByKey(stepKey);
 		let runStep = graph.getRunStep(stepId);
 		if (shouldStartFreshMessageStepAfterGoogleServerSideTool({
@@ -820,6 +853,7 @@ var ChatModelStreamHandler = class {
 			stepId = await dispatchMessageCreationStep({
 				graph,
 				stepKey,
+				content,
 				metadata
 			});
 			runStep = graph.getRunStep(stepId);
@@ -861,7 +895,10 @@ hasToolCallChunks: ${hasToolCallChunks}
 				const message_id = require_ids.getMessageId(newStepKey, graph) ?? "";
 				await graph.dispatchRunStep(newStepKey, {
 					type: "message_creation",
-					message_creation: { message_id }
+					message_creation: {
+						message_id,
+						content_type: "text"
+					}
 				}, metadata);
 				const newStepId = graph.getStepIdByKey(newStepKey);
 				await graph.dispatchMessageDelta(newStepId, { content: [{