@informedai/react 0.4.8 → 0.4.10

package/dist/index.js

@@ -536,13 +536,10 @@ function InformedAIProvider({ config, children }) {
     };
   }, [session?.id, startHeartbeat, stopHeartbeat, handleSessionDeleted, config]);
   const handleSSEEvent = (0, import_react.useCallback)((event) => {
-    console.log("[InformedAI] SSE event:", event.type, event);
     if (event.type === "content" && event.content) {
       setStreamingContent((prev) => prev + event.content);
     }
     if (event.type === "done" || event.type === "session_update") {
-      console.log("[InformedAI] Done event - session:", event.session);
-      console.log("[InformedAI] Done event - widgetMessages:", event.session?.widgetMessages);
       (0, import_react_dom.flushSync)(() => {
         if (event.session) {
           setSession(event.session);
@@ -564,6 +561,17 @@ function InformedAIProvider({ config, children }) {
   const sendMessage = (0, import_react.useCallback)(async (message) => {
     if (!clientRef.current || !session) return;
     try {
+      const optimisticMessage = {
+        id: `temp-${Date.now()}`,
+        role: "user",
+        content: message,
+        type: "user_input",
+        timestamp: (/* @__PURE__ */ new Date()).toISOString()
+      };
+      setSession((prev) => prev ? {
+        ...prev,
+        widgetMessages: [...prev.widgetMessages || [], optimisticMessage]
+      } : null);
       setIsStreaming(true);
       setStreamingContent("");
       setError(null);
@@ -740,9 +748,17 @@ function AssistantWidget({ className, theme, position = "inline", defaultCollaps
   const [isCollapsed, setIsCollapsed] = (0, import_react2.useState)(defaultCollapsed);
   const [inputValue, setInputValue] = (0, import_react2.useState)("");
   const messagesEndRef = (0, import_react2.useRef)(null);
+  const inputRef = (0, import_react2.useRef)(null);
+  const wasStreamingRef = (0, import_react2.useRef)(false);
   (0, import_react2.useEffect)(() => {
     messagesEndRef.current?.scrollIntoView({ behavior: "smooth" });
   }, [session?.widgetMessages, streamingContent]);
+  (0, import_react2.useEffect)(() => {
+    if (wasStreamingRef.current && !isStreaming) {
+      inputRef.current?.focus();
+    }
+    wasStreamingRef.current = isStreaming;
+  }, [isStreaming]);
   const handleSubmit = async (e) => {
     e.preventDefault();
     if (!inputValue.trim() || isStreaming) return;
@@ -1019,6 +1035,7 @@ function AssistantWidget({ className, theme, position = "inline", defaultCollaps
               /* @__PURE__ */ (0, import_jsx_runtime2.jsx)(
                 "input",
                 {
+                  ref: inputRef,
                   type: "text",
                   value: inputValue,
                   onChange: (e) => setInputValue(e.target.value),

package/dist/index.mjs

@@ -506,13 +506,10 @@ function InformedAIProvider({ config, children }) {
     };
   }, [session?.id, startHeartbeat, stopHeartbeat, handleSessionDeleted, config]);
   const handleSSEEvent = useCallback((event) => {
-    console.log("[InformedAI] SSE event:", event.type, event);
     if (event.type === "content" && event.content) {
       setStreamingContent((prev) => prev + event.content);
     }
     if (event.type === "done" || event.type === "session_update") {
-      console.log("[InformedAI] Done event - session:", event.session);
-      console.log("[InformedAI] Done event - widgetMessages:", event.session?.widgetMessages);
       flushSync(() => {
         if (event.session) {
           setSession(event.session);
@@ -534,6 +531,17 @@ function InformedAIProvider({ config, children }) {
   const sendMessage = useCallback(async (message) => {
     if (!clientRef.current || !session) return;
     try {
+      const optimisticMessage = {
+        id: `temp-${Date.now()}`,
+        role: "user",
+        content: message,
+        type: "user_input",
+        timestamp: (/* @__PURE__ */ new Date()).toISOString()
+      };
+      setSession((prev) => prev ? {
+        ...prev,
+        widgetMessages: [...prev.widgetMessages || [], optimisticMessage]
+      } : null);
       setIsStreaming(true);
       setStreamingContent("");
       setError(null);
@@ -710,9 +718,17 @@ function AssistantWidget({ className, theme, position = "inline", defaultCollaps
   const [isCollapsed, setIsCollapsed] = useState2(defaultCollapsed);
   const [inputValue, setInputValue] = useState2("");
   const messagesEndRef = useRef2(null);
+  const inputRef = useRef2(null);
+  const wasStreamingRef = useRef2(false);
   useEffect2(() => {
     messagesEndRef.current?.scrollIntoView({ behavior: "smooth" });
   }, [session?.widgetMessages, streamingContent]);
+  useEffect2(() => {
+    if (wasStreamingRef.current && !isStreaming) {
+      inputRef.current?.focus();
+    }
+    wasStreamingRef.current = isStreaming;
+  }, [isStreaming]);
   const handleSubmit = async (e) => {
     e.preventDefault();
     if (!inputValue.trim() || isStreaming) return;
@@ -989,6 +1005,7 @@ function AssistantWidget({ className, theme, position = "inline", defaultCollaps
               /* @__PURE__ */ jsx2(
                 "input",
                 {
+                  ref: inputRef,
                   type: "text",
                   value: inputValue,
                   onChange: (e) => setInputValue(e.target.value),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@informedai/react",
-  "version": "0.4.8",
+  "version": "0.4.10",
   "description": "React SDK for InformedAI Assistant - AI-powered content creation widget",
   "main": "dist/index.js",
   "module": "dist/index.mjs",