@surf-kit/agent 0.2.2 → 0.4.0

package/dist/{hooks-BGs8-4GK.d.ts → hooks-BLeiVk-x.d.ts}

@@ -1,5 +1,5 @@
-import { a as ChatMessage, C as ChatError, b as ConversationSummary } from './chat--OifhIRe.js';
-import { a as StreamState, S as StreamEvent } from './streaming-DbQxScpi.js';
+import { A as Attachment, a as ChatMessage, C as ChatError, b as ConversationSummary } from './chat-CcKc6OAR.js';
+import { a as StreamState, S as StreamEvent } from './streaming-C6mbU7My.js';
 import { A as AgentInfo } from './agent-BNSmiexZ.js';
 
 interface AgentChatConfig {
@@ -11,14 +11,29 @@ interface AgentChatConfig {
     feedbackPath?: string;
     /** Conversations endpoint path (appended to apiUrl) */
     conversationsPath?: string;
-    /** Request headers (e.g. Authorization) */
-    headers?: Record<string, string>;
+    /** Request headers (e.g. Authorization). Can be a static object or an async function that returns headers (useful for refreshing auth tokens). */
+    headers?: Record<string, string> | (() => Promise<Record<string, string>>);
     /** Request timeout in milliseconds */
     timeout?: number;
     /** Enable localStorage persistence for conversations */
     persistConversations?: boolean;
     /** Map of agent IDs to their display config */
     agentThemes?: Record<string, AgentInfo>;
+    /** Extra fields merged into every request body (e.g. `{ agent: "research" }`).
+     *  Keys here are shallow-merged after the default fields (message, conversation_id, attachments). */
+    bodyExtra?: Record<string, unknown>;
+    /** Custom stream reader for environments without ReadableStream (e.g. React Native).
+     *  When provided, this function handles sending the request and parsing SSE events
+     *  instead of the default fetch + getReader() approach. */
+    streamAdapter?: (url: string, options: {
+        method: string;
+        headers: Record<string, string>;
+        body: string;
+        signal: AbortSignal;
+    }, onEvent: (event: {
+        type: string;
+        [key: string]: unknown;
+    }) => void) => Promise<void>;
 }
 
 interface AgentChatState {
@@ -29,13 +44,15 @@ interface AgentChatState {
     inputValue: string;
     streamPhase: StreamState['phase'];
     streamingContent: string;
+    streamingAgent: string | null;
 }
 interface AgentChatActions {
-    sendMessage: (content: string) => Promise<void>;
+    sendMessage: (content: string, attachments?: Attachment[]) => Promise<void>;
     setInputValue: (value: string) => void;
     loadConversation: (conversationId: string, messages: ChatMessage[]) => void;
     submitFeedback: (messageId: string, rating: 'positive' | 'negative', comment?: string) => Promise<void>;
     retry: () => Promise<void>;
+    stop: () => void;
     reset: () => void;
 }
 declare function useAgentChat(config: AgentChatConfig): {

package/dist/{hooks-DLfF18IU.d.cts → hooks-CSGGLd7j.d.cts}

@@ -1,5 +1,5 @@
-import { a as ChatMessage, C as ChatError, b as ConversationSummary } from './chat-ChYl2XjV.cjs';
-import { a as StreamState, S as StreamEvent } from './streaming-DfT22A0z.cjs';
+import { A as Attachment, a as ChatMessage, C as ChatError, b as ConversationSummary } from './chat-BRY3xGg_.cjs';
+import { a as StreamState, S as StreamEvent } from './streaming-BHPXnwwo.cjs';
 import { A as AgentInfo } from './agent-BNSmiexZ.cjs';
 
 interface AgentChatConfig {
@@ -11,14 +11,29 @@ interface AgentChatConfig {
     feedbackPath?: string;
     /** Conversations endpoint path (appended to apiUrl) */
     conversationsPath?: string;
-    /** Request headers (e.g. Authorization) */
-    headers?: Record<string, string>;
+    /** Request headers (e.g. Authorization). Can be a static object or an async function that returns headers (useful for refreshing auth tokens). */
+    headers?: Record<string, string> | (() => Promise<Record<string, string>>);
     /** Request timeout in milliseconds */
     timeout?: number;
     /** Enable localStorage persistence for conversations */
     persistConversations?: boolean;
     /** Map of agent IDs to their display config */
     agentThemes?: Record<string, AgentInfo>;
+    /** Extra fields merged into every request body (e.g. `{ agent: "research" }`).
+     *  Keys here are shallow-merged after the default fields (message, conversation_id, attachments). */
+    bodyExtra?: Record<string, unknown>;
+    /** Custom stream reader for environments without ReadableStream (e.g. React Native).
+     *  When provided, this function handles sending the request and parsing SSE events
+     *  instead of the default fetch + getReader() approach. */
+    streamAdapter?: (url: string, options: {
+        method: string;
+        headers: Record<string, string>;
+        body: string;
+        signal: AbortSignal;
+    }, onEvent: (event: {
+        type: string;
+        [key: string]: unknown;
+    }) => void) => Promise<void>;
 }
 
 interface AgentChatState {
@@ -29,13 +44,15 @@ interface AgentChatState {
     inputValue: string;
     streamPhase: StreamState['phase'];
     streamingContent: string;
+    streamingAgent: string | null;
 }
 interface AgentChatActions {
-    sendMessage: (content: string) => Promise<void>;
+    sendMessage: (content: string, attachments?: Attachment[]) => Promise<void>;
     setInputValue: (value: string) => void;
     loadConversation: (conversationId: string, messages: ChatMessage[]) => void;
     submitFeedback: (messageId: string, rating: 'positive' | 'negative', comment?: string) => Promise<void>;
     retry: () => Promise<void>;
+    stop: () => void;
     reset: () => void;
 }
 declare function useAgentChat(config: AgentChatConfig): {

package/dist/hooks.cjs

@@ -39,7 +39,8 @@ var initialState = {
   error: null,
   inputValue: "",
   streamPhase: "idle",
-  streamingContent: ""
+  streamingContent: "",
+  streamingAgent: null
 };
 function reducer(state, action) {
   switch (action.type) {
@@ -53,12 +54,17 @@ function reducer(state, action) {
         error: null,
         inputValue: "",
         streamPhase: "thinking",
-        streamingContent: ""
+        streamingContent: "",
+        streamingAgent: null
       };
     case "STREAM_PHASE":
       return { ...state, streamPhase: action.phase };
     case "STREAM_CONTENT":
       return { ...state, streamingContent: state.streamingContent + action.content };
+    case "STREAM_CONTENT_RESET":
+      return { ...state, streamingContent: "" };
+    case "STREAM_AGENT":
+      return { ...state, streamingAgent: action.agent };
     case "SEND_SUCCESS":
       return {
         ...state,
@@ -74,7 +80,8 @@ function reducer(state, action) {
         isLoading: false,
         error: action.error,
         streamPhase: "idle",
-        streamingContent: ""
+        streamingContent: "",
+        streamingAgent: null
       };
     case "LOAD_CONVERSATION":
       return {
@@ -100,115 +107,172 @@ function useAgentChat(config) {
   const configRef = (0, import_react.useRef)(config);
   configRef.current = config;
   const lastUserMessageRef = (0, import_react.useRef)(null);
+  const lastUserAttachmentsRef = (0, import_react.useRef)(void 0);
+  const abortControllerRef = (0, import_react.useRef)(null);
   const sendMessage = (0, import_react.useCallback)(
-    async (content) => {
-      const { apiUrl, streamPath = "/chat/stream", headers = {}, timeout = 3e4 } = configRef.current;
+    async (content, attachments) => {
+      const { apiUrl, streamPath = "/chat/stream", headers: headersOrFn, timeout = 3e4, bodyExtra } = configRef.current;
+      const headers = typeof headersOrFn === "function" ? await headersOrFn() : headersOrFn ?? {};
       lastUserMessageRef.current = content;
+      lastUserAttachmentsRef.current = attachments;
       const userMessage = {
         id: generateMessageId(),
         role: "user",
         content,
+        attachments,
         timestamp: /* @__PURE__ */ new Date()
       };
       dispatch({ type: "SEND_START", message: userMessage });
       const controller = new AbortController();
+      abortControllerRef.current = controller;
       const timeoutId = setTimeout(() => controller.abort(), timeout);
+      const ctx = {
+        accumulatedContent: "",
+        agentResponse: null,
+        capturedAgent: null,
+        capturedConversationId: null,
+        hadStreamError: false
+      };
       try {
-        const response = await fetch(`${apiUrl}${streamPath}`, {
-          method: "POST",
-          headers: {
-            "Content-Type": "application/json",
-            Accept: "text/event-stream",
-            ...headers
-          },
-          body: JSON.stringify({
-            message: content,
-            conversation_id: state.conversationId
-          }),
-          signal: controller.signal
-        });
-        clearTimeout(timeoutId);
-        if (!response.ok) {
-          dispatch({
-            type: "SEND_ERROR",
-            error: {
-              code: "API_ERROR",
-              message: `HTTP ${response.status}: ${response.statusText}`,
-              retryable: response.status >= 500
-            }
-          });
-          return;
+        const url = `${apiUrl}${streamPath}`;
+        const mergedHeaders = {
+          "Content-Type": "application/json",
+          Accept: "text/event-stream",
+          ...headers
+        };
+        const requestBody = {
+          message: content,
+          conversation_id: state.conversationId,
+          ...bodyExtra
+        };
+        if (attachments && attachments.length > 0) {
+          requestBody.attachments = attachments.map((a) => ({
+            filename: a.filename,
+            content_type: a.content_type,
+            data: a.data
+          }));
         }
-        const reader = response.body?.getReader();
-        if (!reader) {
-          dispatch({
-            type: "SEND_ERROR",
-            error: { code: "STREAM_ERROR", message: "No response body", retryable: true }
+        const body = JSON.stringify(requestBody);
+        const handleEvent = (event) => {
+          switch (event.type) {
+            case "agent":
+              ctx.capturedAgent = event.agent;
+              dispatch({ type: "STREAM_AGENT", agent: ctx.capturedAgent });
+              break;
+            case "phase":
+              dispatch({ type: "STREAM_PHASE", phase: event.phase });
+              break;
+            case "delta":
+              ctx.accumulatedContent += event.content;
+              dispatch({ type: "STREAM_CONTENT", content: event.content });
+              break;
+            case "delta_reset":
+              ctx.accumulatedContent = "";
+              dispatch({ type: "STREAM_CONTENT_RESET" });
+              break;
+            case "done":
+              ctx.agentResponse = event.response;
+              ctx.capturedConversationId = event.conversation_id ?? null;
+              break;
+            case "error":
+              ctx.hadStreamError = true;
+              dispatch({ type: "SEND_ERROR", error: event.error });
+              break;
+          }
+        };
+        const { streamAdapter } = configRef.current;
+        if (streamAdapter) {
+          await streamAdapter(
+            url,
+            { method: "POST", headers: mergedHeaders, body, signal: controller.signal },
+            handleEvent
+          );
+          clearTimeout(timeoutId);
+        } else {
+          const response = await fetch(url, {
+            method: "POST",
+            headers: mergedHeaders,
+            body,
+            signal: controller.signal
           });
-          return;
-        }
-        const decoder = new TextDecoder();
-        let buffer = "";
-        let accumulatedContent = "";
-        let agentResponse = null;
-        let capturedAgent = null;
-        let capturedConversationId = null;
-        while (true) {
-          const { done, value } = await reader.read();
-          if (done) break;
-          buffer += decoder.decode(value, { stream: true });
-          const lines = buffer.split("\n");
-          buffer = lines.pop() ?? "";
-          for (const line of lines) {
-            if (!line.startsWith("data: ")) continue;
-            const data = line.slice(6).trim();
-            if (data === "[DONE]") continue;
-            try {
-              const event = JSON.parse(data);
-              switch (event.type) {
-                case "agent":
-                  capturedAgent = event.agent;
-                  break;
-                case "phase":
-                  dispatch({ type: "STREAM_PHASE", phase: event.phase });
-                  break;
-                case "delta":
-                  accumulatedContent += event.content;
-                  dispatch({ type: "STREAM_CONTENT", content: event.content });
-                  break;
-                case "done":
-                  agentResponse = event.response;
-                  capturedConversationId = event.conversation_id ?? null;
-                  break;
-                case "error":
-                  dispatch({ type: "SEND_ERROR", error: event.error });
-                  return;
+          clearTimeout(timeoutId);
+          if (!response.ok) {
+            dispatch({
+              type: "SEND_ERROR",
+              error: {
+                code: "API_ERROR",
+                message: `HTTP ${response.status}: ${response.statusText}`,
+                retryable: response.status >= 500
+              }
+            });
+            return;
+          }
+          const reader = response.body?.getReader();
+          if (!reader) {
+            dispatch({
+              type: "SEND_ERROR",
+              error: { code: "STREAM_ERROR", message: "No response body", retryable: true }
+            });
+            return;
+          }
+          const decoder = new TextDecoder();
+          let buffer = "";
+          while (true) {
+            const { done, value } = await reader.read();
+            if (done) break;
+            buffer += decoder.decode(value, { stream: true });
+            const lines = buffer.split("\n");
+            buffer = lines.pop() ?? "";
+            for (const line of lines) {
+              if (!line.startsWith("data: ")) continue;
+              const data = line.slice(6).trim();
+              if (data === "[DONE]") continue;
+              try {
+                const event = JSON.parse(data);
+                handleEvent(event);
+              } catch {
               }
-            } catch {
             }
           }
         }
+        if (ctx.hadStreamError) return;
         const assistantMessage = {
           id: generateMessageId(),
           role: "assistant",
-          content: agentResponse?.message ?? accumulatedContent,
-          response: agentResponse ?? void 0,
-          agent: capturedAgent ?? void 0,
+          content: ctx.agentResponse?.message ?? ctx.accumulatedContent,
+          response: ctx.agentResponse ?? void 0,
+          agent: ctx.capturedAgent ?? void 0,
           timestamp: /* @__PURE__ */ new Date()
         };
         dispatch({
           type: "SEND_SUCCESS",
           message: assistantMessage,
-          streamingContent: accumulatedContent,
-          conversationId: capturedConversationId
+          streamingContent: ctx.accumulatedContent,
+          conversationId: ctx.capturedConversationId
         });
       } catch (err) {
         clearTimeout(timeoutId);
         if (err.name === "AbortError") {
-          dispatch({
-            type: "SEND_ERROR",
-            error: { code: "TIMEOUT", message: "Request timed out", retryable: true }
-          });
+          if (ctx.accumulatedContent) {
+            const partialMessage = {
+              id: generateMessageId(),
+              role: "assistant",
+              content: ctx.accumulatedContent,
+              agent: ctx.capturedAgent ?? void 0,
+              timestamp: /* @__PURE__ */ new Date()
+            };
+            dispatch({
+              type: "SEND_SUCCESS",
+              message: partialMessage,
+              streamingContent: ctx.accumulatedContent,
+              conversationId: ctx.capturedConversationId
+            });
+          } else {
+            dispatch({
+              type: "SEND_ERROR",
+              error: { code: "ABORTED", message: "Request stopped", retryable: true }
+            });
+          }
         } else {
           dispatch({
             type: "SEND_ERROR",
@@ -219,6 +283,8 @@ function useAgentChat(config) {
             }
           });
         }
+      } finally {
+        abortControllerRef.current = null;
       }
     },
     [state.conversationId]
@@ -231,7 +297,8 @@ function useAgentChat(config) {
   }, []);
   const submitFeedback = (0, import_react.useCallback)(
     async (messageId, rating, comment) => {
-      const { apiUrl, feedbackPath = "/feedback", headers = {} } = configRef.current;
+      const { apiUrl, feedbackPath = "/feedback", headers: headersOrFn } = configRef.current;
+      const headers = typeof headersOrFn === "function" ? await headersOrFn() : headersOrFn ?? {};
       await fetch(`${apiUrl}${feedbackPath}`, {
         method: "POST",
         headers: { "Content-Type": "application/json", ...headers },
@@ -242,12 +309,16 @@ function useAgentChat(config) {
   );
   const retry = (0, import_react.useCallback)(async () => {
     if (lastUserMessageRef.current) {
-      await sendMessage(lastUserMessageRef.current);
+      await sendMessage(lastUserMessageRef.current, lastUserAttachmentsRef.current);
     }
   }, [sendMessage]);
+  const stop = (0, import_react.useCallback)(() => {
+    abortControllerRef.current?.abort();
+  }, []);
   const reset = (0, import_react.useCallback)(() => {
     dispatch({ type: "RESET" });
     lastUserMessageRef.current = null;
+    lastUserAttachmentsRef.current = void 0;
   }, []);
   const actions = {
     sendMessage,
@@ -255,6 +326,7 @@ function useAgentChat(config) {
     loadConversation,
     submitFeedback,
     retry,
+    stop,
     reset
   };
   return { state, actions };
@@ -607,7 +679,10 @@ function useCharacterDrain(target, msPerChar = 15) {
     const elapsed = now - lastTimeRef.current;
     const charsToAdvance = Math.floor(elapsed / msPerCharRef.current);
     if (charsToAdvance > 0 && indexRef.current < currentTarget.length) {
-      const nextIndex = Math.min(indexRef.current + charsToAdvance, currentTarget.length);
+      let nextIndex = Math.min(indexRef.current + charsToAdvance, currentTarget.length);
+      while (nextIndex < currentTarget.length && currentTarget[nextIndex - 1].trim() === "") {
+        nextIndex++;
+      }
       indexRef.current = nextIndex;
       lastTimeRef.current = now;
       setDisplayed(currentTarget.slice(0, nextIndex));