@vegintech/langchain-react-agent 0.0.5 → 0.0.7

package/dist/index.d.mts

@@ -68,15 +68,13 @@ interface SenderSubmitParams {
   slotConfig?: SenderSlotConfig;
   skillData?: SkillType;
 }
-/** onPreSend 钩子返回值：messages + 其他任意字段作为 state */
+/** onPreSend 钩子返回值：只返回消息数组 */
 interface PreSendResult {
   /** 要发送的消息数组 */
   messages: Array<{
     type: string;
     content: string;
   }>;
-  /** 其他任意字段会作为额外参数一起 submit */
-  [x: string]: unknown;
 }
 /** Sender 组件的可定制参数 */
 interface SenderCustomizationProps {
@@ -222,6 +220,10 @@ interface AgentChatProps {
   inputConfig?: InputConfig;
   onError?: (error: Error) => void;
   interruptConfig?: InterruptConfig;
+  /** Agent 状态值 */
+  agentState?: Record<string, unknown>;
+  /** Agent 状态变化回调 */
+  onAgentStateChange?: (state: Record<string, unknown>) => void;
 }
 /** AgentChat 组件对外暴露的方法 */
 interface AgentChatRef {

package/dist/index.mjs

@@ -208,22 +208,6 @@ const CustomParagraph = (props) => {
 	const { node, ...rest } = props;
 	return /* @__PURE__ */ jsx("span", { ...rest });
 };
-const renderMarkdown = (content, customComponents, securityConfig) => {
-	return /* @__PURE__ */ jsx(Streamdown, {
-		components: {
-			p: CustomParagraph,
-			...customComponents
-		},
-		allowedTags: securityConfig?.allowedTags,
-		literalTagContent: securityConfig?.literalTagContent,
-		controls: { table: {
-			copy: false,
-			download: false,
-			fullscreen: false
-		} },
-		children: content ?? ""
-	});
-};
 const renderMessageContent = (message, isLastMessage, isLoading, tools, toolExecutions, components, securityConfig) => {
 	const hasToolCalls = message.toolCalls && message.toolCalls.length > 0;
 	const shouldBlink = isLastMessage && isLoading && !message.content && message.toolCalls?.length == 0;
@@ -267,19 +251,10 @@ const renderMessageContent = (message, isLastMessage, isLoading, tools, toolExec
 	}, message.id);
 };
 const toBubbleItem = (message, isLastMessage, isLoading, tools, toolExecutions, components, securityConfig) => {
-	const hasReasoning = !!message.reasoningContent;
-	const hasToolCalls = message.toolCalls && message.toolCalls.length > 0;
-	if (hasReasoning || hasToolCalls) return {
-		key: message.id,
-		role: message.type === "human" ? "user" : "ai",
-		content: renderMessageContent(message, isLastMessage, isLoading, tools, toolExecutions, components, securityConfig),
-		placement: message.type === "human" ? "end" : "start"
-	};
 	return {
 		key: message.id,
 		role: message.type === "human" ? "user" : "ai",
-		content: message.content,
-		contentRender: (content) => renderMarkdown(content, components, securityConfig),
+		content: renderMessageContent(message, isLastMessage, isLoading, tools, toolExecutions, components, securityConfig),
 		placement: message.type === "human" ? "end" : "start"
 	};
 };
@@ -740,7 +715,7 @@ function injectStyles() {
 //#endregion
 //#region src/components/AgentChat.tsx
 injectStyles();
-const AgentChat = forwardRef(({ apiUrl, assistantId, headers, threadId: externalThreadId, onThreadIdChange, className = "", tools, contexts, messageConfig, inputConfig, onError, interruptConfig }, ref) => {
+const AgentChat = forwardRef(({ apiUrl, assistantId, headers, threadId: externalThreadId, onThreadIdChange, className = "", tools, contexts, messageConfig, inputConfig, onError, interruptConfig, agentState, onAgentStateChange }, ref) => {
 	const [internalThreadId, setInternalThreadId] = useState(externalThreadId);
 	useEffect(() => {
 		setInternalThreadId(externalThreadId);
@@ -768,6 +743,9 @@ const AgentChat = forwardRef(({ apiUrl, assistantId, headers, threadId: external
 			onError?.(err);
 		}
 	});
+	useEffect(() => {
+		if (stream.values && onAgentStateChange) onAgentStateChange(stream.values);
+	}, [stream.values, onAgentStateChange]);
 	const { renderInterrupt: interruptRender } = useInterrupt({
 		interrupt: stream.interrupt,
 		config: interruptConfig,
@@ -787,10 +765,10 @@ const AgentChat = forwardRef(({ apiUrl, assistantId, headers, threadId: external
 			return next;
 		});
 	}, []);
-	const submitToStream = useCallback(async (submitMessages, extraState = {}) => {
+	const submitToStream = useCallback(async (submitMessages) => {
 		await stream.submit({
+			...agentState,
 			messages: submitMessages,
-			...extraState,
 			agentkit: {
 				actions: tools,
 				context: contexts
@@ -808,7 +786,8 @@ const AgentChat = forwardRef(({ apiUrl, assistantId, headers, threadId: external
 	}, [
 		stream,
 		tools,
-		contexts
+		contexts,
+		agentState
 	]);
 	const handleToolResult = useCallback((callId, name, result) => {
 		submitToStream([{
@@ -820,12 +799,8 @@ const AgentChat = forwardRef(({ apiUrl, assistantId, headers, threadId: external
 	}, [submitToStream]);
 	const handleSend = useCallback(async (params) => {
 		let messages = [];
-		let extraState = {};
-		if (onPreSend && params.message.trim()) {
-			const { messages: resultMessages, ...rest } = await onPreSend(params);
-			messages = resultMessages.filter((m) => m != null);
-			extraState = rest;
-		} else {
+		if (onPreSend && params.message.trim()) messages = (await onPreSend(params)).messages.filter((m) => m != null);
+		else {
 			if (!params.message.trim()) return;
 			messages = [{
 				type: "human",
@@ -833,7 +808,7 @@ const AgentChat = forwardRef(({ apiUrl, assistantId, headers, threadId: external
 			}];
 		}
 		if (messages.length === 0) return;
-		await submitToStream(messages, extraState);
+		await submitToStream(messages);
 	}, [onPreSend, submitToStream]);
 	const handleStop = useCallback(async () => {
 		await stream.stop();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vegintech/langchain-react-agent",
-  "version": "0.0.5",
+  "version": "0.0.7",
   "description": "LangChain Agent UI component library for React",
   "license": "MIT",
   "files": [